antioch-py 2.0.6__py3-none-any.whl → 2.1.0__py3-none-any.whl

common/session/views/camera.py

@@ -1,147 +0,0 @@
-from enum import Enum
-
-from pydantic import Field
-
-from common.message import CameraInfo, Message, Pose
-
-
-class CameraMode(str, Enum):
-    """
-    Camera capture modes.
-    """
-
-    RGB = "rgb"
-    DEPTH = "depth"
-
-
-class DistortionModel(str, Enum):
-    """
-    Camera lens distortion model types.
-    """
-
-    PINHOLE = "pinhole"
-    OPENCV_PINHOLE = "opencv_pinhole"
-    OPENCV_FISHEYE = "opencv_fisheye"
-    FTHETA = "ftheta"
-    KANNALA_BRANDT_K3 = "kannala_brandt_k3"
-    RAD_TAN_THIN_PRISM = "rad_tan_thin_prism"
-
-
-class CameraConfig(Message):
-    """
-    Configuration for camera sensor.
-    """
-
-    mode: CameraMode = Field(default=CameraMode.RGB, description="Camera capture mode (RGB or depth)")
-    frequency: int = Field(default=30, description="Camera update frequency in Hz")
-    width: int = Field(default=640, description="Image width in pixels")
-    height: int = Field(default=480, description="Image height in pixels")
-    focal_length: float = Field(default=50.0, description="Focal length in mm")
-    sensor_width: float = Field(default=20.4, description="Physical sensor width in mm")
-    sensor_height: float = Field(default=15.3, description="Physical sensor height in mm")
-    near_clip: float = Field(default=0.1, description="Near clipping plane in meters")
-    far_clip: float = Field(default=1000.0, description="Far clipping plane in meters")
-    f_stop: float = Field(default=0.0, description="F-stop for depth of field")
-    focus_distance: float = Field(default=10.0, description="Focus distance in meters")
-    principal_point_x: float = Field(default=0.0, description="Principal point X offset in pixels")
-    principal_point_y: float = Field(default=0.0, description="Principal point Y offset in pixels")
-    distortion_model: DistortionModel = Field(default=DistortionModel.PINHOLE, description="Lens distortion model")
-    distortion_coefficients: list[float] | None = Field(default=None, description="Distortion coefficients")
-
-    def to_camera_info(self, frame_id: str = "camera_optical_frame") -> CameraInfo:
-        """
-        Convert camera configuration to CameraInfo with calculated intrinsics.
-
-        :param frame_id: The coordinate frame ID for the camera.
-        :return: CameraInfo with full intrinsics and distortion parameters.
-        """
-
-        # Convert mm to pixels for focal length
-        fx = self.width * self.focal_length / self.sensor_width
-        fy = self.height * self.focal_length / self.sensor_height
-
-        # Principal point (image center + offset)
-        cx = self.width / 2.0 + self.principal_point_x
-        cy = self.height / 2.0 + self.principal_point_y
-
-        return CameraInfo(
-            width=self.width,
-            height=self.height,
-            fx=fx,
-            fy=fy,
-            cx=cx,
-            cy=cy,
-            distortion_model=self.distortion_model.value,
-            distortion_coefficients=self.distortion_coefficients or [],
-            frame_id=frame_id,
-        )
-
-
-class GetCamera(Message):
-    """
-    Get an existing camera sensor.
-    """
-
-    path: str | None = Field(default=None, description="USD path of the camera prim")
-    resolution: tuple[int, int] | None = Field(default=None, description="Image resolution in pixels (width, height)")
-
-
-class GetCameraResponse(Message):
-    """
-    Response from getting a camera.
-    """
-
-    path: str
-
-
-class AddCamera(Message):
-    """
-    Add a camera sensor.
-    """
-
-    path: str = Field(description="USD path for the camera")
-    config: CameraConfig
-    world_pose: Pose | None = Field(default=None, description="World pose (position and orientation)")
-    local_pose: Pose | None = Field(default=None, description="Local pose (translation and orientation)")
-
-
-class GetCameraFrame(Message):
-    """
-    Get camera frame with image data.
-    """
-
-    path: str
-
-
-class BufferCameraRgbRead(Message):
-    """
-    Request to buffer current camera RGB frame at simulation time.
-    """
-
-    path: str
-
-
-class GetBufferedCameraRgbRead(Message):
-    """
-    Request to get buffered camera RGB frame at or before simulation time.
-    """
-
-    path: str
-    read_sim_time: float
-
-
-class BufferCameraDepthRead(Message):
-    """
-    Request to buffer current camera depth frame at simulation time.
-    """
-
-    path: str
-
-
-class GetBufferedCameraDepthRead(Message):
-    """
-    Request to get buffered camera depth frame at or before simulation time.
-    """
-
-    path: str
-    read_sim_time: float

common/session/views/collision.py

@@ -1,59 +0,0 @@
-from pydantic import Field
-
-from common.message import Message
-from common.session.views.geometry import MeshApproximation
-
-
-class SetCollision(Message):
-    """
-    Apply collision API to a prim, optionally with mesh approximation.
-    """
-
-    path: str = Field(description="USD path to the prim")
-    mesh_approximation: MeshApproximation | None = Field(
-        default=None,
-        description="Optional mesh approximation method for collision geometry",
-    )
-
-
-class RemoveCollision(Message):
-    """
-    Remove collision API from a prim.
-    """
-
-    path: str = Field(description="USD path to the prim")
-
-
-class HasCollision(Message):
-    """
-    Check if a prim has collision API applied.
-    """
-
-    path: str = Field(description="USD path to the prim")
-
-
-class HasCollisionResponse(Message):
-    """
-    Response for collision check.
-    """
-
-    has_collision: bool = Field(description="Whether the prim has collision API")
-
-
-class GetMeshApproximation(Message):
-    """
-    Get mesh collision approximation from a prim.
-    """
-
-    path: str = Field(description="USD path to the prim")
-
-
-class GetMeshApproximationResponse(Message):
-    """
-    Response for mesh approximation query.
-    """
-
-    approximation: MeshApproximation | None = Field(
-        default=None,
-        description="Mesh approximation method, or None if not set",
-    )

common/session/views/geometry.py

@@ -1,102 +0,0 @@
-from enum import Enum
-
-from pydantic import Field
-from pydantic.alias_generators import to_camel
-
-from common.message import Message, Pose, Vector3
-
-
-class GeometryType(str, Enum):
-    """
-    Supported geometry types.
-    """
-
-    SPHERE = "sphere"
-    CUBE = "cube"
-    CYLINDER = "cylinder"
-    CONE = "cone"
-    CAPSULE = "capsule"
-    MESH = "mesh"
-
-
-class MeshApproximation(str, Enum):
-    """
-    Collision mesh approximation type.
-
-    Values are stored in snake_case for consistency with Rust. Use to_usd()
-    to get the camelCase format required by USD/Isaac Sim.
-    """
-
-    NONE = "none"
-    CONVEX_HULL = "convex_hull"
-    CONVEX_DECOMPOSITION = "convex_decomposition"
-    BOUNDING_SPHERE = "bounding_sphere"
-    BOUNDING_CUBE = "bounding_cube"
-    MESH_SIMPLIFICATION = "mesh_simplification"
-    SDF = "sdf"
-    SPHERE_FILL = "sphere_fill"
-
-    def to_usd(self) -> str:
-        """
-        Convert to USD camelCase format.
-
-        :return: The USD camelCase format.
-        """
-
-        return to_camel(self.value)
-
-
-class GeometryConfig(Message):
-    """
-    Configuration for creating geometry.
-    """
-
-    geometry_type: GeometryType = Field(description="Geometry type")
-    radius: float | None = Field(default=None, description="Radius for sphere/cylinder/cone/capsule")
-    height: float | None = Field(default=None, description="Height for cylinder/cone/capsule")
-    size: float | None = Field(default=None, description="Size for cube (uniform)")
-    color: Vector3 | None = Field(default=None, description="RGB color (0-1)")
-    opacity: float = Field(default=1.0, description="Opacity (0=transparent, 1=opaque)")
-    enable_collision: bool = Field(default=True, description="Whether to enable collision on this geometry")
-    static_friction: float = Field(default=0.5, description="Static friction coefficient")
-    dynamic_friction: float = Field(default=0.5, description="Dynamic friction coefficient")
-    restitution: float = Field(default=0.2, description="Restitution (bounciness)")
-    mesh_file_path: str | None = Field(
-        default=None,
-        description="Path to mesh file (FBX, OBJ, glTF, STL, etc.) - required for MESH type",
-    )
-    mesh_approximation: MeshApproximation = Field(
-        default=MeshApproximation.CONVEX_DECOMPOSITION,
-        description="Mesh approximation method",
-    )
-    contact_offset: float | None = Field(default=None, description="Distance at which collision detection begins")
-    rest_offset: float | None = Field(default=None, description="Minimum separation distance between objects")
-    torsional_patch_radius: float | None = Field(default=None, description="Radius for torsional friction calculations")
-    min_torsional_patch_radius: float | None = Field(default=None, description="Minimum radius for torsional friction")
-
-
-class GetGeometry(Message):
-    """
-    Get an existing geometry view from a prim.
-    """
-
-    path: str | None = Field(default=None, description="USD path of the geometry prim")
-
-
-class GetGeometryResponse(Message):
-    """
-    Response from getting a geometry.
-    """
-
-    path: str
-
-
-class AddGeometry(Message):
-    """
-    Add a geometry prim with optional pose.
-    """
-
-    path: str = Field(description="USD path for the geometry")
-    config: GeometryConfig
-    world_pose: "Pose | None" = Field(default=None, description="Optional world pose")
-    local_pose: "Pose | None" = Field(default=None, description="Optional local pose")

common/session/views/ground_plane.py

@@ -1,41 +0,0 @@
-from pydantic import Field
-
-from common.message import Message, Vector3
-
-
-class GroundPlaneConfig(Message):
-    """
-    Configuration for creating a ground plane.
-    """
-
-    size: float = Field(default=5000.0, description="Size of the ground plane in meters")
-    z_position: float = Field(default=0.0, description="Z position of the ground plane")
-    color: Vector3 = Field(default=Vector3.new(0.5, 0.5, 0.5), description="RGB color (0-1)")
-    static_friction: float = Field(default=0.5, description="Static friction coefficient")
-    dynamic_friction: float = Field(default=0.5, description="Dynamic friction coefficient")
-    restitution: float = Field(default=0.0, description="Restitution (bounciness)")
-
-
-class GetGroundPlane(Message):
-    """
-    Get an existing ground plane.
-    """
-
-    path: str | None = Field(default=None, description="USD path of the ground plane prim")
-
-
-class GetGroundPlaneResponse(Message):
-    """
-    Response from getting a ground plane.
-    """
-
-    path: str
-
-
-class AddGroundPlane(Message):
-    """
-    Add a ground plane.
-    """
-
-    path: str = Field(description="USD path for the ground plane")
-    config: GroundPlaneConfig

common/session/views/imu.py

@@ -1,66 +0,0 @@
-from pydantic import Field
-
-from common.message import Message, Pose
-
-
-class ImuConfig(Message):
-    """
-    Configuration for IMU sensor.
-    """
-
-    frequency: int | None = Field(default=None, description="Sensor update frequency in Hz (optional, defaults to physics rate)")
-    linear_acceleration_filter_size: int = Field(default=10, description="Filter window size for linear acceleration")
-    angular_velocity_filter_size: int = Field(default=10, description="Filter window size for angular velocity")
-    orientation_filter_size: int = Field(default=10, description="Filter window size for orientation")
-
-
-class GetImu(Message):
-    """
-    Get an existing IMU sensor.
-    """
-
-    path: str | None = Field(default=None, description="USD path of the IMU prim")
-
-
-class GetImuResponse(Message):
-    """
-    Response from getting an IMU.
-    """
-
-    path: str
-
-
-class AddImu(Message):
-    """
-    Add an IMU sensor.
-    """
-
-    path: str = Field(description="USD path for the IMU")
-    config: ImuConfig
-    world_pose: Pose | None = Field(default=None, description="World pose (position and orientation)")
-    local_pose: Pose | None = Field(default=None, description="Local pose (translation and orientation)")
-
-
-class GetImuSample(Message):
-    """
-    Get IMU sensor sample.
-    """
-
-    path: str
-
-
-class BufferImuRead(Message):
-    """
-    Request to buffer current IMU data at simulation time.
-    """
-
-    path: str
-
-
-class GetBufferedImuRead(Message):
-    """
-    Request to get buffered IMU data at or before simulation time.
-    """
-
-    path: str
-    read_sim_time: float

common/session/views/joint.py

@@ -1,81 +0,0 @@
-from enum import Enum
-
-from pydantic import Field
-
-from common.message import Message, Pose
-
-
-class JointType(str, Enum):
-    """
-    Types of joints supported by the simulator.
-    """
-
-    REVOLUTE = "revolute"
-    PRISMATIC = "prismatic"
-    FIXED = "fixed"
-    SPHERICAL = "spherical"
-    DISTANCE = "distance"
-    GENERIC = "generic"
-
-
-class JointAxis(str, Enum):
-    """
-    Axis of motion for joints.
-    """
-
-    X = "x"
-    Y = "y"
-    Z = "z"
-
-
-class JointConfig(Message):
-    """
-    Configuration for a joint object that connects two bodies.
-    """
-
-    # Joint relationships
-    parent_path: str = Field(description="USD path to parent body")
-    child_path: str = Field(description="USD path to child body")
-
-    # Transform
-    pose: Pose = Field(default_factory=Pose.identity, description="Joint pose relative to parent")
-
-    # Joint properties
-    joint_type: JointType = Field(default=JointType.FIXED, description="Type of joint motion allowed")
-    axis: JointAxis = Field(default=JointAxis.X, description="Axis of motion for non-fixed joints")
-
-    # Motion limits (for revolute: degrees, for prismatic: meters)
-    lower_limit: float | None = Field(default=None, description="Lower motion limit")
-    upper_limit: float | None = Field(default=None, description="Upper motion limit")
-
-    # Physics properties
-    friction: float = Field(default=0.01, description="Joint friction (unitless)")
-    armature: float = Field(default=0.1, description="Joint armature (kg for prismatic, kg-m^2 for revolute)")
-
-    # Special properties
-    exclude_from_articulation: bool = Field(default=False, description="Whether to exclude this joint from articulation")
-
-
-class GetJoint(Message):
-    """
-    Get an existing joint.
-    """
-
-    path: str | None = Field(default=None, description="USD path of the joint prim")
-
-
-class GetJointResponse(Message):
-    """
-    Response from getting a joint.
-    """
-
-    path: str
-
-
-class AddJoint(Message):
-    """
-    Add a joint connecting two bodies.
-    """
-
-    path: str = Field(description="USD path for the joint")
-    config: JointConfig

common/session/views/light.py

@@ -1,96 +0,0 @@
-from enum import Enum
-
-from pydantic import Field
-
-from common.message import Message, Pose, Vector3
-
-
-class LightType(str, Enum):
-    """
-    Supported light types.
-    """
-
-    SPHERE = "sphere"
-    RECT = "rect"
-    DISK = "disk"
-    CYLINDER = "cylinder"
-    DISTANT = "distant"
-    DOME = "dome"
-
-
-class LightConfig(Message):
-    """
-    Configuration for creating a light.
-    """
-
-    light_type: LightType = Field(default=LightType.SPHERE, description="Light type")
-    intensity: float = Field(default=30000.0, description="Light intensity")
-    exposure: float = Field(default=10.0, description="Light exposure")
-    color: Vector3 = Field(default_factory=Vector3.ones, description="RGB color (0-1)")
-    radius: float = Field(default=0.1, description="Radius for sphere lights (meters)")
-    width: float | None = Field(default=None, description="Width for rect lights (meters)")
-    height: float | None = Field(default=None, description="Height for rect/cylinder lights (meters)")
-    length: float | None = Field(default=None, description="Length for cylinder lights (meters)")
-    angle: float | None = Field(default=None, description="Angle for distant lights (degrees)")
-    texture_file: str | None = Field(default=None, description="Texture file for dome lights")
-
-
-class GetLight(Message):
-    """
-    Get an existing light source.
-    """
-
-    path: str | None = Field(default=None, description="USD path of the light prim")
-
-
-class GetLightResponse(Message):
-    """
-    Response from getting a light.
-    """
-
-    path: str
-
-
-class AddLight(Message):
-    """
-    Add a light source.
-    """
-
-    path: str = Field(description="USD path for the light")
-    config: LightConfig
-    world_pose: Pose | None = Field(default=None, description="World pose (position and orientation)")
-    local_pose: Pose | None = Field(default=None, description="Local pose (translation and orientation)")
-
-
-class SetLightIntensity(Message):
-    """
-    Set light intensity.
-    """
-
-    path: str
-    intensity: float
-
-
-class SetLightColor(Message):
-    """
-    Set light color.
-    """
-
-    path: str
-    color: Vector3
-
-
-class EnableLight(Message):
-    """
-    Enable light.
-    """
-
-    path: str
-
-
-class DisableLight(Message):
-    """
-    Disable light.
-    """
-
-    path: str