Spaces:
Sleeping
Sleeping
File size: 1,996 Bytes
66c9c8a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 | # Copyright (c) 2022 NVIDIA CORPORATION. All rights reserved.
# NVIDIA CORPORATION and its licensors retain all intellectual property
# and proprietary rights in and to this software, related documentation
# and any modifications thereto. Any use, reproduction, disclosure or
# distribution of this software and related documentation without an express
# license agreement from NVIDIA CORPORATION is strictly prohibited.
###########################################################################
# Cartpole environment
#
# Shows how to set up a simulation of a rigid-body cartpole articulation
# from a URDF using the Environment class.
# Note this example does not include a trained policy.
#
###########################################################################
import os
import math
import warp as wp
import warp.sim
from environment import Environment, run_env
class CartpoleEnvironment(Environment):
sim_name = "env_cartpole"
env_offset = (2.0, 0.0, 2.0)
opengl_render_settings = dict(scaling=3.0)
usd_render_settings = dict(scaling=100.0)
sim_substeps_euler = 32
sim_substeps_xpbd = 5
activate_ground_plane = False
show_joints = True
def create_articulation(self, builder):
wp.sim.parse_urdf(
os.path.join(os.path.dirname(__file__), "../assets/cartpole.urdf"),
builder,
xform=wp.transform((0.0, 0.0, 0.0), wp.quat_from_axis_angle(wp.vec3(1.0, 0.0, 0.0), -math.pi * 0.5)),
floating=False,
armature=0.1,
stiffness=0.0,
damping=0.0,
shape_ke=1.0e4,
shape_kd=1.0e2,
shape_kf=1.0e2,
shape_mu=1.0,
limit_ke=1.0e4,
limit_kd=1.0e1,
enable_self_collisions=False,
)
# joint initial positions
builder.joint_q[-3:] = [0.0, 0.3, 0.0]
builder.joint_target[:3] = [0.0, 0.0, 0.0]
if __name__ == "__main__":
run_env(CartpoleEnvironment)
|