o
wi @ s^ d Z ddlZddlmZ ddlmZmZ dZdZ e dZ
eG dd d Z G d
d d Z dS ) z)Base exoskeleton configuration framework. N) dataclass)DictTuplez ../../so100_blender_testingszrobot_models/so100_modelz/../board_imgsc @ sb e Zd ZU dZeed< eed< eed< eed< ejed< ejed< eed< eed < d
Z e
ed < d
S )
LinkConfigz8Configuration for a single robot link with ArUco marker. mujoco_name
pybullet_namerobot_mesh_path
exo_mesh_patharuco_offset_posaruco_offset_rotaruco_board_name board_lengthN plane_scale) __name__
__module__ __qualname____doc__str__annotations__npndarrayfloatr tuple r r N/data/cameron/para/panda_streaming/hand_eye_calib/../ExoConfigs/exoskeleton.pyr s
r c @ s e Zd ZU dZdZeed< dZeed< dZeed< dZ eed < i Z
e eef ed
< d Z e
ed < d
Ze
ed< i Ze eef ed< dZe
ed< dZe
ed< dZe
ed< dd ZdefddZdS ) ExoskeletonConfigzCBase configuration class for robots with ArUco marker exoskeletons.ZRobotname
base_xml_pathz'robot_models/so100_model/background.xmlbackground_xml_pathz./compiler_meshdir aruco_boardsg
ףp=
?board_length_smallgR Q?board_length_largelinksg333333? exo_alphag?exo_link_alphag ? aruco_alphac C s | | _dS )z Initialize and generate XML.N)
_generate_xmlxml)selfr r r __init__0 s zExoskeletonConfig.__init__returnc C s d| j d| j d| j d}| j D ]$\}}|jdr dnd}|d| d|j d | d
|j d | d 7 }q|d
t dt dt dt dt d 7 }| j D ]\}}|d| d| d| d| d 7 }qR|d7 }| j D ]{\}}|j
dur|j
\}}| d} | dd | d t d |d d |d d! d"}n'|j d#krd$}
n|j d%krd&}
n|j d'krd(}
n
|j d)krd*}
nd+}
d,|
} |d-| d.| d/| d0| j
d1| d2| d3| j d1| d4| d5|j d6| j d77 }qq|d87 }|S )9z@Generate complete MuJoCo XML with exoskeletons and ArUco planes.z?
z.stlz scale=".001 .001 .001"r z!
z%
z
z
NZ _aruco_planezz
gR Q?Z
6x6_largergףp=
?Z
5x5_largerg)\(?Z even_largerg?largesmallZ aruco_plane_z
z )r r r r$ itemsr endswithr BLENDER_STL_DIRr! r replacer
r& r% r r' ) r* r) r cfgZ exo_scaleZ
board_nameZimg_pathsxsyZplane_mesh_nameZ
plane_sizer r r r( 4 s
&
z ExoskeletonConfig._generate_xmlN)r r r r r r r r r r r! r r" r r# r$ r r% r& r' r+ r( r r r r r s
r )
r numpyr dataclassesr typingr r r4 SO100_MODEL_DIR
BOARD_IMG_DIRr r r r r r s