mirror of
https://github.com/wassname/pyrobolearn.git
synced 2026-10-02 12:40:58 +08:00
39 lines
854 B
Python
39 lines
854 B
Python
#!/usr/bin/env python
|
|
"""Demonstrate how the distance cost decreases as you move the manipulator (with your mouse) closer to the target
|
|
object in the world.
|
|
"""
|
|
|
|
from itertools import count
|
|
import pyrobolearn as prl
|
|
|
|
|
|
# Create simulator
|
|
sim = prl.simulators.Bullet()
|
|
|
|
# create world
|
|
world = prl.worlds.BasicWorld(sim)
|
|
|
|
# create robot
|
|
robot = world.load_robot('kuka_iiwa')
|
|
print(robot)
|
|
|
|
# desired position
|
|
sphere = world.load_visual_sphere([0.5, 0., 0.5], radius=0.05, color=(1, 0, 0, 0.5), return_body=True)
|
|
|
|
# create state
|
|
state = prl.states.LinkPositionState(robot, link_ids=robot.end_effectors)
|
|
|
|
# create reward
|
|
# reward = prl.rewards.DistanceCost()
|
|
|
|
# run simulation
|
|
for t in count():
|
|
# update state
|
|
state()
|
|
|
|
# compute reward
|
|
# print("Reward value = {}".format(reward()))
|
|
|
|
# perform a step in the simulator
|
|
world.step(sim.dt)
|