diff --git a/scripts/config/environment/__init__.py b/scripts/config/envs/__init__.py similarity index 100% rename from scripts/config/environment/__init__.py rename to scripts/config/envs/__init__.py diff --git a/scripts/config/envs/open_manipulator.py b/scripts/config/envs/open_manipulator.py new file mode 100755 index 0000000..2da7afb --- /dev/null +++ b/scripts/config/envs/open_manipulator.py @@ -0,0 +1,54 @@ +from math import pi + +from geometry_msgs.msg import Quaternion + + +config = { + "ENV_NAME": "OpenManipulatorReacher", + "MAX_EPISODE_STEPS": 100, + "TERM_COUNT": 5, + "SUCCESS_COUNT": 5, + "OVERHEAD_ORIENTATION": Quaternion( + x=-0.00142460053167, y=0.999994209902, z=-0.00177030764765, w=0.00253311793936 + ), + # box boundary + "POLAR_RADIAN_BOUNDARY": (0.134, 0.32), + "POLAR_THETA_BOUNDARY": (-pi * 0.7 / 4, pi * 0.7 / 4), + "Z_BOUNDARY": (0.05, 0.28), + "JOINT_LIMITS": { + "HIGH": { + "J1": pi * 0.9, + "J2": pi * 0.5, + "J3": pi * 0.44, + "J4": pi * 0.65, + "GRIP": 0.019, + }, + "LOW": { + "J1": -pi * 0.9, + "J2": -pi * 0.57, + "J3": -pi * 0.3, + "J4": -pi * 0.57, + "GRIP": -0.001, + }, + }, + # Global variables + "ACTION_DIM": 5, # Cartesian + "OBSERVATION_DIM": (25,), + # terminal condition + "INNER_RADIAN": 0.134, + "OUTER_RADIAN": 0.3, + "LOWER_RADIAN": 0.384, + "INNER_Z": 0.321, + "OUTER_Z": 0.250, + "LOWER_Z": 0.116, + "ENV_MODE": "sim", + "TRAIN_MODE": True, + "DISTANCE_THRESHOLD": 0.05, + "REWARD_RESCALE_RATIO": 1.0, + "REWARD_FUNC": "l2", + "CONTROL_MODE": "position", +} + + +def get(): + return config diff --git a/scripts/gazebo_test_open_manipulator.py b/scripts/gazebo_test_open_manipulator.py index 5fea141..cd8a4f9 100755 --- a/scripts/gazebo_test_open_manipulator.py +++ b/scripts/gazebo_test_open_manipulator.py @@ -3,9 +3,9 @@ from math import cos, pi, sin import numpy as np -from config.environment.open_manipulator import config as cfg import rospy +from config.envs.open_manipulator import config as cfg from envs.open_manipulator import OpenManipulatorReacherEnv from geometry_msgs.msg import Pose, Quaternion from open_manipulator_msgs.msg import JointPosition, KinematicsPose diff --git a/scripts/run_open_manipulator_reacher_v0.py b/scripts/run_open_manipulator_reacher_v0.py index 2547c01..7b3fea9 100755 --- a/scripts/run_open_manipulator_reacher_v0.py +++ b/scripts/run_open_manipulator_reacher_v0.py @@ -10,9 +10,8 @@ import argparse import importlib -from config.environment.open_manipulator import config as env_cfg - import algorithms.common.helper_functions as common_utils +from config.envs.open_manipulator import config as env_cfg from envs.open_manipulator.open_manipulator_reacher_env import OpenManipulatorReacherEnv # configurations