RSI-PI/examples/advanced_motion/02_geometric_primitives.py
Adam cc19e102e8 Phase 4: Advanced Motion Control - Complete Implementation
Implements professional-grade trajectory planning and execution capabilities
for industrial robotics applications. Adds velocity profiling, geometric
motion primitives, path blending, and coordinate frame transformations.

Features Added:
- Velocity profiling (trapezoidal and S-curve profiles)
- Geometric motion primitives (arc, circle, spiral)
- Path blending with cubic Hermite spline interpolation
- Coordinate transformations (BASE/WORLD/TOOL/WORK frames)

New API Methods (MotionAPI):
- generate_velocity_profile(trajectory, max_velocity, max_acceleration, profile)
- generate_arc(center, radius, start_angle, end_angle, steps, plane)
- generate_circle(center, radius, steps, plane)
- generate_spiral(center, start_radius, end_radius, pitch, revolutions, steps, plane, axis)
- blend_trajectories(traj1, traj2, blend_radius, blend_steps)
- transform_coordinates(pose, from_frame, to_frame, frame_offset)

Helper Functions:
- _calculate_distance() - Euclidean distance between waypoints
- _trapezoidal_profile() - Bang-bang velocity control
- _s_curve_profile() - Jerk-limited smooth profiles
- _find_blend_point() - Locate blend zone boundaries
- _cubic_blend() - Cubic Hermite spline interpolation

Examples Created (examples/advanced_motion/):
- 01_velocity_profiles.py (234 lines) - Trapezoidal vs S-curve profiling
- 02_geometric_primitives.py (225 lines) - Arc, circle, spiral patterns
- 03_path_blending.py (253 lines) - Smooth trajectory transitions
- 04_coordinate_transforms.py (284 lines) - Frame transformations
- 05_combined_motion.py (336 lines) - Complete production application
- README.md (584 lines) - Comprehensive documentation

Documentation:
- PHASE_4_SUMMARY.md - Detailed implementation documentation
- Updated ROADMAP.md to mark Phase 4 complete
- Comprehensive API documentation in examples/advanced_motion/README.md

Files Modified:
- src/RSIPI/motion_api.py (~550 lines added)
- ROADMAP.md (updated Phase 4 status)

Files Created:
- PHASE_4_SUMMARY.md
- examples/advanced_motion/ (6 new files, 1,916 total lines)

Statistics:
- New API methods: 5 public methods + 4 helper functions
- Example code: ~1,332 lines
- Documentation: ~584 lines
- Total additions: ~2,466 lines

Production Applications:
- Drilling and milling (expanding/contracting spirals)
- Assembly (circular insertion, smooth approaches)
- Inspection (spiral scanning, circular features)
- Welding/coating (continuous beads, smooth transitions)
- Pick and place (optimized cycles, blended paths)

Phase 4 Status:  COMPLETE
Date: January 17, 2026
2026-01-17 01:38:48 +00:00

225 lines
7.6 KiB
Python

"""
Geometric Motion Primitives Example
Demonstrates arc, circle, and spiral trajectory generation for
complex motion patterns.
Usage:
python 02_geometric_primitives.py --config RSI_EthernetConfig.xml
"""
import argparse
import logging
from RSIPI import RSIAPI
logging.basicConfig(
level=logging.INFO,
format='%(asctime)s - %(levelname)s - %(message)s'
)
def geometric_primitives_example(config_file: str) -> None:
"""
Demonstrate geometric motion primitives.
Args:
config_file: Path to RSI configuration XML file
"""
api = RSIAPI(config_file)
try:
logging.info("Starting RSI communication...")
api.start()
logging.info("✅ RSI started successfully")
center = {"X": 100, "Y": 0, "Z": 500}
# ==================================================
# Example 1: Circular Arc
# ==================================================
logging.info("\n" + "=" * 60)
logging.info("Example 1: Circular Arc (90 degrees)")
logging.info("=" * 60)
arc = api.motion.generate_arc(
center=center,
radius=50.0, # mm
start_angle=0, # degrees
end_angle=90, # degrees
steps=50,
plane='XY'
)
logging.info(f"Generated arc with {len(arc)} waypoints")
logging.info(f"Start point: {arc[0]}")
logging.info(f"End point: {arc[-1]}")
logging.info("Executing arc motion...")
api.motion.execute_trajectory(arc, space='cartesian', rate=0.02)
logging.info("✅ Arc complete")
# ==================================================
# Example 2: Full Circle
# ==================================================
logging.info("\n" + "=" * 60)
logging.info("Example 2: Full Circle (360 degrees)")
logging.info("=" * 60)
circle = api.motion.generate_circle(
center=center,
radius=30.0, # mm
steps=100,
plane='XY'
)
logging.info(f"Generated circle with {len(circle)} waypoints")
logging.info(f"Radius: 30.0 mm")
logging.info(f"Circumference: ~{2 * 3.14159 * 30.0:.2f} mm")
logging.info("Executing circular motion...")
api.motion.execute_trajectory(circle, space='cartesian', rate=0.02)
logging.info("✅ Circle complete")
# ==================================================
# Example 3: Expanding Spiral (Drilling Pattern)
# ==================================================
logging.info("\n" + "=" * 60)
logging.info("Example 3: Expanding Spiral (Drilling Pattern)")
logging.info("=" * 60)
spiral_expand = api.motion.generate_spiral(
center={"X": 100, "Y": 0, "Z": 500},
start_radius=5.0, # Start at 5mm
end_radius=40.0, # End at 40mm
pitch=10.0, # Descend 10mm per revolution
revolutions=3.0, # 3 complete turns
steps=150,
plane='XY',
axis='Z'
)
logging.info(f"Generated expanding spiral:")
logging.info(f" Waypoints: {len(spiral_expand)}")
logging.info(f" Start radius: 5.0 mm")
logging.info(f" End radius: 40.0 mm")
logging.info(f" Pitch: 10.0 mm/rev (descending)")
logging.info(f" Revolutions: 3.0")
logging.info(f" Total Z travel: 30.0 mm")
logging.info("Executing expanding spiral...")
api.motion.execute_trajectory(spiral_expand, space='cartesian', rate=0.02)
logging.info("✅ Expanding spiral complete")
# ==================================================
# Example 4: Contracting Spiral (Retraction Pattern)
# ==================================================
logging.info("\n" + "=" * 60)
logging.info("Example 4: Contracting Spiral (Retraction Pattern)")
logging.info("=" * 60)
spiral_contract = api.motion.generate_spiral(
center={"X": 100, "Y": 0, "Z": 470}, # Start at bottom
start_radius=40.0, # Start wide
end_radius=5.0, # End narrow
pitch=-10.0, # Ascend 10mm per revolution (negative pitch)
revolutions=3.0,
steps=150,
plane='XY',
axis='Z'
)
logging.info(f"Generated contracting spiral:")
logging.info(f" Waypoints: {len(spiral_contract)}")
logging.info(f" Start radius: 40.0 mm")
logging.info(f" End radius: 5.0 mm")
logging.info(f" Pitch: -10.0 mm/rev (ascending)")
logging.info(f" Revolutions: 3.0")
logging.info(f" Total Z travel: -30.0 mm (upward)")
logging.info("Executing contracting spiral...")
api.motion.execute_trajectory(spiral_contract, space='cartesian', rate=0.02)
logging.info("✅ Contracting spiral complete")
# ==================================================
# Example 5: Different Planes
# ==================================================
logging.info("\n" + "=" * 60)
logging.info("Example 5: Circles in Different Planes")
logging.info("=" * 60)
# Circle in XZ plane (vertical)
circle_xz = api.motion.generate_circle(
center={"X": 100, "Y": 0, "Z": 500},
radius=25.0,
steps=80,
plane='XZ'
)
logging.info("Circle in XZ plane (vertical):")
logging.info(f" Waypoints: {len(circle_xz)}")
logging.info(f" Plane: XZ (vertical circle)")
logging.info("Executing XZ circle...")
api.motion.execute_trajectory(circle_xz, space='cartesian', rate=0.02)
logging.info("✅ XZ circle complete")
# ==================================================
# Application Examples
# ==================================================
logging.info("\n" + "=" * 60)
logging.info("Application Examples")
logging.info("=" * 60)
logging.info("\nDrilling/Milling Applications:")
logging.info(" - Expanding spiral: Drill out large holes, pocket milling")
logging.info(" - Contracting spiral: Retracting from deep holes")
logging.info(" - Circular: Bore existing holes, circular pockets")
logging.info("\nAssembly Applications:")
logging.info(" - Circular: Screw driving, peg insertion with clearance")
logging.info(" - Arc: Curved insertion paths, avoiding obstacles")
logging.info("\nInspection Applications:")
logging.info(" - Circle: Scanning circular features")
logging.info(" - Spiral: Scanning large areas with overlap")
except KeyboardInterrupt:
logging.warning("\n⚠️ Interrupted by user")
except Exception as e:
logging.error(f"❌ Error during geometric primitives: {e}")
finally:
logging.info("Stopping RSI communication...")
api.stop()
logging.info("✅ API stopped successfully")
def main():
"""Main entry point."""
parser = argparse.ArgumentParser(description='Geometric Primitives Example')
parser.add_argument(
'--config',
type=str,
default='RSI_EthernetConfig.xml',
help='Path to RSI configuration file'
)
args = parser.parse_args()
logging.info("=" * 60)
logging.info("RSIPI - Geometric Motion Primitives Example")
logging.info("=" * 60)
logging.info(f"Config: {args.config}")
logging.info("=" * 60)
geometric_primitives_example(args.config)
logging.info("=" * 60)
logging.info("Example complete!")
logging.info("=" * 60)
if __name__ == '__main__':
main()