pymomentum-cpu 0.0.0__cp312-cp312-manylinux_2_39_x86_64.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of pymomentum-cpu might be problematic. Click here for more details.

Files changed (555) hide show
  1. include/axel/BoundingBox.h +58 -0
  2. include/axel/Bvh.h +708 -0
  3. include/axel/BvhBase.h +75 -0
  4. include/axel/BvhCommon.h +43 -0
  5. include/axel/BvhEmbree.h +86 -0
  6. include/axel/BvhFactory.h +34 -0
  7. include/axel/Checks.h +21 -0
  8. include/axel/KdTree.h +199 -0
  9. include/axel/Log.h +22 -0
  10. include/axel/MeshToSdf.h +123 -0
  11. include/axel/Profile.h +64 -0
  12. include/axel/Ray.h +45 -0
  13. include/axel/SignedDistanceField.h +237 -0
  14. include/axel/SimdKdTree.h +515 -0
  15. include/axel/TriBvh.h +157 -0
  16. include/axel/TriBvhEmbree.h +57 -0
  17. include/axel/common/Constants.h +27 -0
  18. include/axel/common/Types.h +21 -0
  19. include/axel/common/VectorizationTypes.h +58 -0
  20. include/axel/math/BoundingBoxUtils.h +54 -0
  21. include/axel/math/ContinuousCollisionDetection.h +48 -0
  22. include/axel/math/CoplanarityCheck.h +30 -0
  23. include/axel/math/EdgeEdgeDistance.h +31 -0
  24. include/axel/math/MeshHoleFilling.h +98 -0
  25. include/axel/math/PointTriangleProjection.h +34 -0
  26. include/axel/math/PointTriangleProjectionDefinitions.h +209 -0
  27. include/axel/math/RayTriangleIntersection.h +36 -0
  28. include/momentum/character/blend_shape.h +91 -0
  29. include/momentum/character/blend_shape_base.h +70 -0
  30. include/momentum/character/blend_shape_skinning.h +96 -0
  31. include/momentum/character/character.h +272 -0
  32. include/momentum/character/character_state.h +108 -0
  33. include/momentum/character/character_utility.h +128 -0
  34. include/momentum/character/collision_geometry.h +80 -0
  35. include/momentum/character/collision_geometry_state.h +130 -0
  36. include/momentum/character/fwd.h +243 -0
  37. include/momentum/character/inverse_parameter_transform.h +58 -0
  38. include/momentum/character/joint.h +82 -0
  39. include/momentum/character/joint_state.h +241 -0
  40. include/momentum/character/linear_skinning.h +139 -0
  41. include/momentum/character/locator.h +82 -0
  42. include/momentum/character/locator_state.h +43 -0
  43. include/momentum/character/marker.h +48 -0
  44. include/momentum/character/parameter_limits.h +144 -0
  45. include/momentum/character/parameter_transform.h +250 -0
  46. include/momentum/character/pose_shape.h +65 -0
  47. include/momentum/character/skeleton.h +85 -0
  48. include/momentum/character/skeleton_state.h +181 -0
  49. include/momentum/character/skeleton_utility.h +38 -0
  50. include/momentum/character/skin_weights.h +67 -0
  51. include/momentum/character/skinned_locator.h +80 -0
  52. include/momentum/character/types.h +202 -0
  53. include/momentum/character_sequence_solver/fwd.h +200 -0
  54. include/momentum/character_sequence_solver/model_parameters_sequence_error_function.h +62 -0
  55. include/momentum/character_sequence_solver/multipose_solver.h +63 -0
  56. include/momentum/character_sequence_solver/multipose_solver_function.h +80 -0
  57. include/momentum/character_sequence_solver/sequence_error_function.h +95 -0
  58. include/momentum/character_sequence_solver/sequence_solver.h +144 -0
  59. include/momentum/character_sequence_solver/sequence_solver_function.h +126 -0
  60. include/momentum/character_sequence_solver/state_sequence_error_function.h +106 -0
  61. include/momentum/character_sequence_solver/vertex_sequence_error_function.h +125 -0
  62. include/momentum/character_solver/aim_error_function.h +112 -0
  63. include/momentum/character_solver/collision_error_function.h +88 -0
  64. include/momentum/character_solver/collision_error_function_stateless.h +71 -0
  65. include/momentum/character_solver/constraint_error_function-inl.h +321 -0
  66. include/momentum/character_solver/constraint_error_function.h +241 -0
  67. include/momentum/character_solver/distance_error_function.h +74 -0
  68. include/momentum/character_solver/error_function_utils.h +60 -0
  69. include/momentum/character_solver/fixed_axis_error_function.h +139 -0
  70. include/momentum/character_solver/fwd.h +924 -0
  71. include/momentum/character_solver/gauss_newton_solver_qr.h +62 -0
  72. include/momentum/character_solver/limit_error_function.h +53 -0
  73. include/momentum/character_solver/model_parameters_error_function.h +61 -0
  74. include/momentum/character_solver/normal_error_function.h +73 -0
  75. include/momentum/character_solver/orientation_error_function.h +74 -0
  76. include/momentum/character_solver/plane_error_function.h +102 -0
  77. include/momentum/character_solver/point_triangle_vertex_error_function.h +140 -0
  78. include/momentum/character_solver/pose_prior_error_function.h +76 -0
  79. include/momentum/character_solver/position_error_function.h +75 -0
  80. include/momentum/character_solver/projection_error_function.h +90 -0
  81. include/momentum/character_solver/simd_collision_error_function.h +95 -0
  82. include/momentum/character_solver/simd_normal_error_function.h +151 -0
  83. include/momentum/character_solver/simd_plane_error_function.h +154 -0
  84. include/momentum/character_solver/simd_position_error_function.h +155 -0
  85. include/momentum/character_solver/skeleton_error_function.h +136 -0
  86. include/momentum/character_solver/skeleton_solver_function.h +72 -0
  87. include/momentum/character_solver/skinned_locator_error_function.h +163 -0
  88. include/momentum/character_solver/skinned_locator_triangle_error_function.h +138 -0
  89. include/momentum/character_solver/skinning_weight_iterator.h +80 -0
  90. include/momentum/character_solver/state_error_function.h +90 -0
  91. include/momentum/character_solver/transform_pose.h +80 -0
  92. include/momentum/character_solver/trust_region_qr.h +78 -0
  93. include/momentum/character_solver/vertex_error_function.h +154 -0
  94. include/momentum/character_solver/vertex_projection_error_function.h +118 -0
  95. include/momentum/character_solver/vertex_vertex_distance_error_function.h +143 -0
  96. include/momentum/common/aligned.h +155 -0
  97. include/momentum/common/checks.h +27 -0
  98. include/momentum/common/exception.h +70 -0
  99. include/momentum/common/filesystem.h +20 -0
  100. include/momentum/common/fwd.h +27 -0
  101. include/momentum/common/log.h +173 -0
  102. include/momentum/common/log_channel.h +17 -0
  103. include/momentum/common/memory.h +71 -0
  104. include/momentum/common/profile.h +79 -0
  105. include/momentum/common/progress_bar.h +37 -0
  106. include/momentum/common/string.h +52 -0
  107. include/momentum/diff_ik/ceres_utility.h +73 -0
  108. include/momentum/diff_ik/fully_differentiable_body_ik.h +58 -0
  109. include/momentum/diff_ik/fully_differentiable_distance_error_function.h +69 -0
  110. include/momentum/diff_ik/fully_differentiable_motion_error_function.h +46 -0
  111. include/momentum/diff_ik/fully_differentiable_orientation_error_function.h +109 -0
  112. include/momentum/diff_ik/fully_differentiable_pose_prior_error_function.h +76 -0
  113. include/momentum/diff_ik/fully_differentiable_position_error_function.h +133 -0
  114. include/momentum/diff_ik/fully_differentiable_projection_error_function.h +65 -0
  115. include/momentum/diff_ik/fully_differentiable_skeleton_error_function.h +160 -0
  116. include/momentum/diff_ik/fully_differentiable_state_error_function.h +54 -0
  117. include/momentum/diff_ik/fwd.h +385 -0
  118. include/momentum/diff_ik/union_error_function.h +63 -0
  119. include/momentum/gui/rerun/eigen_adapters.h +70 -0
  120. include/momentum/gui/rerun/logger.h +102 -0
  121. include/momentum/gui/rerun/logging_redirect.h +27 -0
  122. include/momentum/io/character_io.h +56 -0
  123. include/momentum/io/common/gsl_utils.h +50 -0
  124. include/momentum/io/common/stream_utils.h +65 -0
  125. include/momentum/io/fbx/fbx_io.h +109 -0
  126. include/momentum/io/fbx/fbx_memory_stream.h +66 -0
  127. include/momentum/io/fbx/openfbx_loader.h +49 -0
  128. include/momentum/io/fbx/polygon_data.h +60 -0
  129. include/momentum/io/gltf/gltf_builder.h +135 -0
  130. include/momentum/io/gltf/gltf_file_format.h +19 -0
  131. include/momentum/io/gltf/gltf_io.h +136 -0
  132. include/momentum/io/gltf/utils/accessor_utils.h +299 -0
  133. include/momentum/io/gltf/utils/coordinate_utils.h +60 -0
  134. include/momentum/io/gltf/utils/json_utils.h +102 -0
  135. include/momentum/io/legacy_json/legacy_json_io.h +70 -0
  136. include/momentum/io/marker/c3d_io.h +29 -0
  137. include/momentum/io/marker/conversions.h +57 -0
  138. include/momentum/io/marker/coordinate_system.h +30 -0
  139. include/momentum/io/marker/marker_io.h +54 -0
  140. include/momentum/io/marker/trc_io.h +27 -0
  141. include/momentum/io/motion/mmo_io.h +97 -0
  142. include/momentum/io/shape/blend_shape_io.h +70 -0
  143. include/momentum/io/shape/pose_shape_io.h +21 -0
  144. include/momentum/io/skeleton/locator_io.h +41 -0
  145. include/momentum/io/skeleton/mppca_io.h +26 -0
  146. include/momentum/io/skeleton/parameter_limits_io.h +25 -0
  147. include/momentum/io/skeleton/parameter_transform_io.h +41 -0
  148. include/momentum/io/skeleton/parameters_io.h +20 -0
  149. include/momentum/io/urdf/urdf_io.h +26 -0
  150. include/momentum/io/usd/usd_io.h +36 -0
  151. include/momentum/marker_tracking/app_utils.h +62 -0
  152. include/momentum/marker_tracking/marker_tracker.h +213 -0
  153. include/momentum/marker_tracking/process_markers.h +58 -0
  154. include/momentum/marker_tracking/tracker_utils.h +90 -0
  155. include/momentum/math/constants.h +82 -0
  156. include/momentum/math/covariance_matrix.h +84 -0
  157. include/momentum/math/fmt_eigen.h +23 -0
  158. include/momentum/math/fwd.h +132 -0
  159. include/momentum/math/generalized_loss.h +61 -0
  160. include/momentum/math/intersection.h +32 -0
  161. include/momentum/math/mesh.h +84 -0
  162. include/momentum/math/mppca.h +67 -0
  163. include/momentum/math/online_householder_qr.h +516 -0
  164. include/momentum/math/random-inl.h +404 -0
  165. include/momentum/math/random.h +310 -0
  166. include/momentum/math/simd_generalized_loss.h +40 -0
  167. include/momentum/math/transform.h +229 -0
  168. include/momentum/math/types.h +461 -0
  169. include/momentum/math/utility.h +251 -0
  170. include/momentum/rasterizer/camera.h +453 -0
  171. include/momentum/rasterizer/fwd.h +102 -0
  172. include/momentum/rasterizer/geometry.h +83 -0
  173. include/momentum/rasterizer/image.h +18 -0
  174. include/momentum/rasterizer/rasterizer.h +583 -0
  175. include/momentum/rasterizer/tensor.h +140 -0
  176. include/momentum/rasterizer/utility.h +268 -0
  177. include/momentum/simd/simd.h +221 -0
  178. include/momentum/solver/fwd.h +131 -0
  179. include/momentum/solver/gauss_newton_solver.h +136 -0
  180. include/momentum/solver/gradient_descent_solver.h +65 -0
  181. include/momentum/solver/solver.h +155 -0
  182. include/momentum/solver/solver_function.h +126 -0
  183. include/momentum/solver/subset_gauss_newton_solver.h +109 -0
  184. include/rerun/archetypes/annotation_context.hpp +157 -0
  185. include/rerun/archetypes/arrows2d.hpp +271 -0
  186. include/rerun/archetypes/arrows3d.hpp +257 -0
  187. include/rerun/archetypes/asset3d.hpp +262 -0
  188. include/rerun/archetypes/asset_video.hpp +275 -0
  189. include/rerun/archetypes/bar_chart.hpp +261 -0
  190. include/rerun/archetypes/boxes2d.hpp +293 -0
  191. include/rerun/archetypes/boxes3d.hpp +369 -0
  192. include/rerun/archetypes/capsules3d.hpp +333 -0
  193. include/rerun/archetypes/clear.hpp +180 -0
  194. include/rerun/archetypes/depth_image.hpp +425 -0
  195. include/rerun/archetypes/ellipsoids3d.hpp +384 -0
  196. include/rerun/archetypes/encoded_image.hpp +250 -0
  197. include/rerun/archetypes/geo_line_strings.hpp +166 -0
  198. include/rerun/archetypes/geo_points.hpp +177 -0
  199. include/rerun/archetypes/graph_edges.hpp +152 -0
  200. include/rerun/archetypes/graph_nodes.hpp +206 -0
  201. include/rerun/archetypes/image.hpp +434 -0
  202. include/rerun/archetypes/instance_poses3d.hpp +221 -0
  203. include/rerun/archetypes/line_strips2d.hpp +289 -0
  204. include/rerun/archetypes/line_strips3d.hpp +270 -0
  205. include/rerun/archetypes/mesh3d.hpp +387 -0
  206. include/rerun/archetypes/pinhole.hpp +385 -0
  207. include/rerun/archetypes/points2d.hpp +333 -0
  208. include/rerun/archetypes/points3d.hpp +369 -0
  209. include/rerun/archetypes/recording_properties.hpp +132 -0
  210. include/rerun/archetypes/scalar.hpp +170 -0
  211. include/rerun/archetypes/scalars.hpp +153 -0
  212. include/rerun/archetypes/segmentation_image.hpp +305 -0
  213. include/rerun/archetypes/series_line.hpp +274 -0
  214. include/rerun/archetypes/series_lines.hpp +271 -0
  215. include/rerun/archetypes/series_point.hpp +265 -0
  216. include/rerun/archetypes/series_points.hpp +251 -0
  217. include/rerun/archetypes/tensor.hpp +213 -0
  218. include/rerun/archetypes/text_document.hpp +200 -0
  219. include/rerun/archetypes/text_log.hpp +211 -0
  220. include/rerun/archetypes/transform3d.hpp +925 -0
  221. include/rerun/archetypes/video_frame_reference.hpp +295 -0
  222. include/rerun/archetypes/view_coordinates.hpp +393 -0
  223. include/rerun/archetypes.hpp +43 -0
  224. include/rerun/arrow_utils.hpp +32 -0
  225. include/rerun/as_components.hpp +90 -0
  226. include/rerun/blueprint/archetypes/background.hpp +113 -0
  227. include/rerun/blueprint/archetypes/container_blueprint.hpp +259 -0
  228. include/rerun/blueprint/archetypes/dataframe_query.hpp +178 -0
  229. include/rerun/blueprint/archetypes/entity_behavior.hpp +130 -0
  230. include/rerun/blueprint/archetypes/force_center.hpp +115 -0
  231. include/rerun/blueprint/archetypes/force_collision_radius.hpp +141 -0
  232. include/rerun/blueprint/archetypes/force_link.hpp +136 -0
  233. include/rerun/blueprint/archetypes/force_many_body.hpp +124 -0
  234. include/rerun/blueprint/archetypes/force_position.hpp +132 -0
  235. include/rerun/blueprint/archetypes/line_grid3d.hpp +178 -0
  236. include/rerun/blueprint/archetypes/map_background.hpp +104 -0
  237. include/rerun/blueprint/archetypes/map_zoom.hpp +103 -0
  238. include/rerun/blueprint/archetypes/near_clip_plane.hpp +109 -0
  239. include/rerun/blueprint/archetypes/panel_blueprint.hpp +95 -0
  240. include/rerun/blueprint/archetypes/plot_legend.hpp +118 -0
  241. include/rerun/blueprint/archetypes/scalar_axis.hpp +116 -0
  242. include/rerun/blueprint/archetypes/tensor_scalar_mapping.hpp +146 -0
  243. include/rerun/blueprint/archetypes/tensor_slice_selection.hpp +167 -0
  244. include/rerun/blueprint/archetypes/tensor_view_fit.hpp +95 -0
  245. include/rerun/blueprint/archetypes/view_blueprint.hpp +170 -0
  246. include/rerun/blueprint/archetypes/view_contents.hpp +142 -0
  247. include/rerun/blueprint/archetypes/viewport_blueprint.hpp +200 -0
  248. include/rerun/blueprint/archetypes/visible_time_ranges.hpp +116 -0
  249. include/rerun/blueprint/archetypes/visual_bounds2d.hpp +109 -0
  250. include/rerun/blueprint/archetypes/visualizer_overrides.hpp +113 -0
  251. include/rerun/blueprint/archetypes.hpp +29 -0
  252. include/rerun/blueprint/components/active_tab.hpp +82 -0
  253. include/rerun/blueprint/components/apply_latest_at.hpp +79 -0
  254. include/rerun/blueprint/components/auto_layout.hpp +77 -0
  255. include/rerun/blueprint/components/auto_views.hpp +77 -0
  256. include/rerun/blueprint/components/background_kind.hpp +66 -0
  257. include/rerun/blueprint/components/column_share.hpp +78 -0
  258. include/rerun/blueprint/components/component_column_selector.hpp +81 -0
  259. include/rerun/blueprint/components/container_kind.hpp +65 -0
  260. include/rerun/blueprint/components/corner2d.hpp +64 -0
  261. include/rerun/blueprint/components/enabled.hpp +77 -0
  262. include/rerun/blueprint/components/filter_by_range.hpp +74 -0
  263. include/rerun/blueprint/components/filter_is_not_null.hpp +77 -0
  264. include/rerun/blueprint/components/force_distance.hpp +82 -0
  265. include/rerun/blueprint/components/force_iterations.hpp +82 -0
  266. include/rerun/blueprint/components/force_strength.hpp +82 -0
  267. include/rerun/blueprint/components/grid_columns.hpp +78 -0
  268. include/rerun/blueprint/components/grid_spacing.hpp +78 -0
  269. include/rerun/blueprint/components/included_content.hpp +86 -0
  270. include/rerun/blueprint/components/lock_range_during_zoom.hpp +82 -0
  271. include/rerun/blueprint/components/map_provider.hpp +64 -0
  272. include/rerun/blueprint/components/near_clip_plane.hpp +82 -0
  273. include/rerun/blueprint/components/panel_state.hpp +61 -0
  274. include/rerun/blueprint/components/query_expression.hpp +89 -0
  275. include/rerun/blueprint/components/root_container.hpp +77 -0
  276. include/rerun/blueprint/components/row_share.hpp +78 -0
  277. include/rerun/blueprint/components/selected_columns.hpp +76 -0
  278. include/rerun/blueprint/components/tensor_dimension_index_slider.hpp +90 -0
  279. include/rerun/blueprint/components/timeline_name.hpp +76 -0
  280. include/rerun/blueprint/components/view_class.hpp +76 -0
  281. include/rerun/blueprint/components/view_fit.hpp +61 -0
  282. include/rerun/blueprint/components/view_maximized.hpp +79 -0
  283. include/rerun/blueprint/components/view_origin.hpp +81 -0
  284. include/rerun/blueprint/components/viewer_recommendation_hash.hpp +82 -0
  285. include/rerun/blueprint/components/visible_time_range.hpp +77 -0
  286. include/rerun/blueprint/components/visual_bounds2d.hpp +74 -0
  287. include/rerun/blueprint/components/visualizer_override.hpp +86 -0
  288. include/rerun/blueprint/components/zoom_level.hpp +78 -0
  289. include/rerun/blueprint/components.hpp +41 -0
  290. include/rerun/blueprint/datatypes/component_column_selector.hpp +61 -0
  291. include/rerun/blueprint/datatypes/filter_by_range.hpp +59 -0
  292. include/rerun/blueprint/datatypes/filter_is_not_null.hpp +61 -0
  293. include/rerun/blueprint/datatypes/selected_columns.hpp +62 -0
  294. include/rerun/blueprint/datatypes/tensor_dimension_index_slider.hpp +63 -0
  295. include/rerun/blueprint/datatypes.hpp +9 -0
  296. include/rerun/c/arrow_c_data_interface.h +111 -0
  297. include/rerun/c/compiler_utils.h +10 -0
  298. include/rerun/c/rerun.h +627 -0
  299. include/rerun/c/sdk_info.h +28 -0
  300. include/rerun/collection.hpp +496 -0
  301. include/rerun/collection_adapter.hpp +43 -0
  302. include/rerun/collection_adapter_builtins.hpp +138 -0
  303. include/rerun/compiler_utils.hpp +61 -0
  304. include/rerun/component_batch.hpp +163 -0
  305. include/rerun/component_column.hpp +111 -0
  306. include/rerun/component_descriptor.hpp +142 -0
  307. include/rerun/component_type.hpp +35 -0
  308. include/rerun/components/aggregation_policy.hpp +76 -0
  309. include/rerun/components/albedo_factor.hpp +74 -0
  310. include/rerun/components/annotation_context.hpp +102 -0
  311. include/rerun/components/axis_length.hpp +74 -0
  312. include/rerun/components/blob.hpp +73 -0
  313. include/rerun/components/class_id.hpp +71 -0
  314. include/rerun/components/clear_is_recursive.hpp +75 -0
  315. include/rerun/components/color.hpp +99 -0
  316. include/rerun/components/colormap.hpp +99 -0
  317. include/rerun/components/depth_meter.hpp +84 -0
  318. include/rerun/components/draw_order.hpp +79 -0
  319. include/rerun/components/entity_path.hpp +83 -0
  320. include/rerun/components/fill_mode.hpp +72 -0
  321. include/rerun/components/fill_ratio.hpp +79 -0
  322. include/rerun/components/gamma_correction.hpp +80 -0
  323. include/rerun/components/geo_line_string.hpp +63 -0
  324. include/rerun/components/graph_edge.hpp +75 -0
  325. include/rerun/components/graph_node.hpp +79 -0
  326. include/rerun/components/graph_type.hpp +57 -0
  327. include/rerun/components/half_size2d.hpp +91 -0
  328. include/rerun/components/half_size3d.hpp +95 -0
  329. include/rerun/components/image_buffer.hpp +86 -0
  330. include/rerun/components/image_format.hpp +84 -0
  331. include/rerun/components/image_plane_distance.hpp +77 -0
  332. include/rerun/components/interactive.hpp +76 -0
  333. include/rerun/components/keypoint_id.hpp +74 -0
  334. include/rerun/components/lat_lon.hpp +89 -0
  335. include/rerun/components/length.hpp +77 -0
  336. include/rerun/components/line_strip2d.hpp +73 -0
  337. include/rerun/components/line_strip3d.hpp +73 -0
  338. include/rerun/components/magnification_filter.hpp +63 -0
  339. include/rerun/components/marker_shape.hpp +82 -0
  340. include/rerun/components/marker_size.hpp +74 -0
  341. include/rerun/components/media_type.hpp +157 -0
  342. include/rerun/components/name.hpp +83 -0
  343. include/rerun/components/opacity.hpp +77 -0
  344. include/rerun/components/pinhole_projection.hpp +94 -0
  345. include/rerun/components/plane3d.hpp +75 -0
  346. include/rerun/components/pose_rotation_axis_angle.hpp +73 -0
  347. include/rerun/components/pose_rotation_quat.hpp +71 -0
  348. include/rerun/components/pose_scale3d.hpp +102 -0
  349. include/rerun/components/pose_transform_mat3x3.hpp +87 -0
  350. include/rerun/components/pose_translation3d.hpp +96 -0
  351. include/rerun/components/position2d.hpp +86 -0
  352. include/rerun/components/position3d.hpp +90 -0
  353. include/rerun/components/radius.hpp +98 -0
  354. include/rerun/components/range1d.hpp +75 -0
  355. include/rerun/components/resolution.hpp +88 -0
  356. include/rerun/components/rotation_axis_angle.hpp +72 -0
  357. include/rerun/components/rotation_quat.hpp +71 -0
  358. include/rerun/components/scalar.hpp +76 -0
  359. include/rerun/components/scale3d.hpp +102 -0
  360. include/rerun/components/series_visible.hpp +76 -0
  361. include/rerun/components/show_labels.hpp +79 -0
  362. include/rerun/components/stroke_width.hpp +74 -0
  363. include/rerun/components/tensor_data.hpp +94 -0
  364. include/rerun/components/tensor_dimension_index_selection.hpp +77 -0
  365. include/rerun/components/tensor_height_dimension.hpp +71 -0
  366. include/rerun/components/tensor_width_dimension.hpp +71 -0
  367. include/rerun/components/texcoord2d.hpp +101 -0
  368. include/rerun/components/text.hpp +83 -0
  369. include/rerun/components/text_log_level.hpp +110 -0
  370. include/rerun/components/timestamp.hpp +76 -0
  371. include/rerun/components/transform_mat3x3.hpp +92 -0
  372. include/rerun/components/transform_relation.hpp +66 -0
  373. include/rerun/components/translation3d.hpp +96 -0
  374. include/rerun/components/triangle_indices.hpp +85 -0
  375. include/rerun/components/value_range.hpp +78 -0
  376. include/rerun/components/vector2d.hpp +92 -0
  377. include/rerun/components/vector3d.hpp +96 -0
  378. include/rerun/components/video_timestamp.hpp +120 -0
  379. include/rerun/components/view_coordinates.hpp +346 -0
  380. include/rerun/components/visible.hpp +74 -0
  381. include/rerun/components.hpp +77 -0
  382. include/rerun/config.hpp +52 -0
  383. include/rerun/datatypes/angle.hpp +76 -0
  384. include/rerun/datatypes/annotation_info.hpp +76 -0
  385. include/rerun/datatypes/blob.hpp +67 -0
  386. include/rerun/datatypes/bool.hpp +57 -0
  387. include/rerun/datatypes/channel_datatype.hpp +87 -0
  388. include/rerun/datatypes/class_description.hpp +92 -0
  389. include/rerun/datatypes/class_description_map_elem.hpp +69 -0
  390. include/rerun/datatypes/class_id.hpp +62 -0
  391. include/rerun/datatypes/color_model.hpp +68 -0
  392. include/rerun/datatypes/dvec2d.hpp +76 -0
  393. include/rerun/datatypes/entity_path.hpp +60 -0
  394. include/rerun/datatypes/float32.hpp +62 -0
  395. include/rerun/datatypes/float64.hpp +62 -0
  396. include/rerun/datatypes/image_format.hpp +107 -0
  397. include/rerun/datatypes/keypoint_id.hpp +63 -0
  398. include/rerun/datatypes/keypoint_pair.hpp +65 -0
  399. include/rerun/datatypes/mat3x3.hpp +105 -0
  400. include/rerun/datatypes/mat4x4.hpp +119 -0
  401. include/rerun/datatypes/pixel_format.hpp +142 -0
  402. include/rerun/datatypes/plane3d.hpp +60 -0
  403. include/rerun/datatypes/quaternion.hpp +110 -0
  404. include/rerun/datatypes/range1d.hpp +59 -0
  405. include/rerun/datatypes/range2d.hpp +55 -0
  406. include/rerun/datatypes/rgba32.hpp +94 -0
  407. include/rerun/datatypes/rotation_axis_angle.hpp +67 -0
  408. include/rerun/datatypes/tensor_buffer.hpp +529 -0
  409. include/rerun/datatypes/tensor_data.hpp +100 -0
  410. include/rerun/datatypes/tensor_dimension_index_selection.hpp +58 -0
  411. include/rerun/datatypes/tensor_dimension_selection.hpp +56 -0
  412. include/rerun/datatypes/time_int.hpp +62 -0
  413. include/rerun/datatypes/time_range.hpp +55 -0
  414. include/rerun/datatypes/time_range_boundary.hpp +175 -0
  415. include/rerun/datatypes/uint16.hpp +62 -0
  416. include/rerun/datatypes/uint32.hpp +62 -0
  417. include/rerun/datatypes/uint64.hpp +62 -0
  418. include/rerun/datatypes/utf8.hpp +76 -0
  419. include/rerun/datatypes/utf8pair.hpp +62 -0
  420. include/rerun/datatypes/uuid.hpp +60 -0
  421. include/rerun/datatypes/uvec2d.hpp +76 -0
  422. include/rerun/datatypes/uvec3d.hpp +80 -0
  423. include/rerun/datatypes/uvec4d.hpp +59 -0
  424. include/rerun/datatypes/vec2d.hpp +76 -0
  425. include/rerun/datatypes/vec3d.hpp +80 -0
  426. include/rerun/datatypes/vec4d.hpp +84 -0
  427. include/rerun/datatypes/video_timestamp.hpp +67 -0
  428. include/rerun/datatypes/view_coordinates.hpp +87 -0
  429. include/rerun/datatypes/visible_time_range.hpp +57 -0
  430. include/rerun/datatypes.hpp +51 -0
  431. include/rerun/demo_utils.hpp +75 -0
  432. include/rerun/entity_path.hpp +20 -0
  433. include/rerun/error.hpp +180 -0
  434. include/rerun/half.hpp +10 -0
  435. include/rerun/image_utils.hpp +187 -0
  436. include/rerun/indicator_component.hpp +59 -0
  437. include/rerun/loggable.hpp +54 -0
  438. include/rerun/recording_stream.hpp +960 -0
  439. include/rerun/rerun_sdk_export.hpp +25 -0
  440. include/rerun/result.hpp +86 -0
  441. include/rerun/rotation3d.hpp +33 -0
  442. include/rerun/sdk_info.hpp +20 -0
  443. include/rerun/spawn.hpp +21 -0
  444. include/rerun/spawn_options.hpp +57 -0
  445. include/rerun/string_utils.hpp +16 -0
  446. include/rerun/third_party/cxxopts.hpp +2198 -0
  447. include/rerun/time_column.hpp +288 -0
  448. include/rerun/timeline.hpp +38 -0
  449. include/rerun/type_traits.hpp +40 -0
  450. include/rerun.hpp +86 -0
  451. lib/cmake/rerun_sdk/rerun_sdkConfig.cmake +70 -0
  452. lib/cmake/rerun_sdk/rerun_sdkConfigVersion.cmake +83 -0
  453. lib/cmake/rerun_sdk/rerun_sdkTargets-release.cmake +19 -0
  454. lib/cmake/rerun_sdk/rerun_sdkTargets.cmake +108 -0
  455. lib/libarrow.a +0 -0
  456. lib/libarrow_bundled_dependencies.a +0 -0
  457. lib/librerun_c__linux_x64.a +0 -0
  458. lib/librerun_sdk.a +0 -0
  459. lib64/cmake/axel/axel-config.cmake +45 -0
  460. lib64/cmake/axel/axelTargets-release.cmake +19 -0
  461. lib64/cmake/axel/axelTargets.cmake +108 -0
  462. lib64/cmake/momentum/Findre2.cmake +52 -0
  463. lib64/cmake/momentum/momentum-config.cmake +67 -0
  464. lib64/cmake/momentum/momentumTargets-release.cmake +269 -0
  465. lib64/cmake/momentum/momentumTargets.cmake +445 -0
  466. lib64/libaxel.a +0 -0
  467. lib64/libmomentum_app_utils.a +0 -0
  468. lib64/libmomentum_character.a +0 -0
  469. lib64/libmomentum_character_sequence_solver.a +0 -0
  470. lib64/libmomentum_character_solver.a +0 -0
  471. lib64/libmomentum_common.a +0 -0
  472. lib64/libmomentum_diff_ik.a +0 -0
  473. lib64/libmomentum_io.a +0 -0
  474. lib64/libmomentum_io_common.a +0 -0
  475. lib64/libmomentum_io_fbx.a +0 -0
  476. lib64/libmomentum_io_gltf.a +0 -0
  477. lib64/libmomentum_io_legacy_json.a +0 -0
  478. lib64/libmomentum_io_marker.a +0 -0
  479. lib64/libmomentum_io_motion.a +0 -0
  480. lib64/libmomentum_io_shape.a +0 -0
  481. lib64/libmomentum_io_skeleton.a +0 -0
  482. lib64/libmomentum_io_urdf.a +0 -0
  483. lib64/libmomentum_marker_tracker.a +0 -0
  484. lib64/libmomentum_math.a +0 -0
  485. lib64/libmomentum_online_qr.a +0 -0
  486. lib64/libmomentum_process_markers.a +0 -0
  487. lib64/libmomentum_rasterizer.a +0 -0
  488. lib64/libmomentum_rerun.a +0 -0
  489. lib64/libmomentum_simd_constraints.a +0 -0
  490. lib64/libmomentum_simd_generalized_loss.a +0 -0
  491. lib64/libmomentum_skeleton.a +0 -0
  492. lib64/libmomentum_solver.a +0 -0
  493. pymomentum/axel.cpython-312-x86_64-linux-gnu.so +0 -0
  494. pymomentum/backend/__init__.py +16 -0
  495. pymomentum/backend/skel_state_backend.py +614 -0
  496. pymomentum/backend/trs_backend.py +871 -0
  497. pymomentum/backend/utils.py +224 -0
  498. pymomentum/geometry.cpython-312-x86_64-linux-gnu.so +0 -0
  499. pymomentum/marker_tracking.cpython-312-x86_64-linux-gnu.so +0 -0
  500. pymomentum/quaternion.py +740 -0
  501. pymomentum/renderer.cpython-312-x86_64-linux-gnu.so +0 -0
  502. pymomentum/skel_state.py +514 -0
  503. pymomentum/solver.cpython-312-x86_64-linux-gnu.so +0 -0
  504. pymomentum/solver2.cpython-312-x86_64-linux-gnu.so +0 -0
  505. pymomentum/torch/character.py +809 -0
  506. pymomentum/torch/parameter_limits.py +494 -0
  507. pymomentum/torch/utility.py +20 -0
  508. pymomentum/trs.py +535 -0
  509. pymomentum_cpu-0.0.0.dist-info/METADATA +209 -0
  510. pymomentum_cpu-0.0.0.dist-info/RECORD +555 -0
  511. pymomentum_cpu-0.0.0.dist-info/WHEEL +5 -0
  512. pymomentum_cpu-0.0.0.dist-info/licenses/LICENSE +21 -0
  513. pymomentum_cpu.libs/libabsl_base-86f3b38c.so.2505.0.0 +0 -0
  514. pymomentum_cpu.libs/libabsl_city-31b65ca2.so.2505.0.0 +0 -0
  515. pymomentum_cpu.libs/libabsl_debugging_internal-38680253.so.2505.0.0 +0 -0
  516. pymomentum_cpu.libs/libabsl_decode_rust_punycode-750652c3.so.2505.0.0 +0 -0
  517. pymomentum_cpu.libs/libabsl_demangle_internal-9a0351a3.so.2505.0.0 +0 -0
  518. pymomentum_cpu.libs/libabsl_demangle_rust-71629506.so.2505.0.0 +0 -0
  519. pymomentum_cpu.libs/libabsl_examine_stack-57661ecd.so.2505.0.0 +0 -0
  520. pymomentum_cpu.libs/libabsl_hash-8c523b7e.so.2505.0.0 +0 -0
  521. pymomentum_cpu.libs/libabsl_hashtablez_sampler-b5c3e343.so.2505.0.0 +0 -0
  522. pymomentum_cpu.libs/libabsl_int128-295bfed5.so.2505.0.0 +0 -0
  523. pymomentum_cpu.libs/libabsl_kernel_timeout_internal-29296ac1.so.2505.0.0 +0 -0
  524. pymomentum_cpu.libs/libabsl_log_globals-6cfa8af5.so.2505.0.0 +0 -0
  525. pymomentum_cpu.libs/libabsl_log_internal_format-a5c79460.so.2505.0.0 +0 -0
  526. pymomentum_cpu.libs/libabsl_log_internal_globals-481e9a7c.so.2505.0.0 +0 -0
  527. pymomentum_cpu.libs/libabsl_log_internal_log_sink_set-ac08f942.so.2505.0.0 +0 -0
  528. pymomentum_cpu.libs/libabsl_log_internal_message-7dfe150a.so.2505.0.0 +0 -0
  529. pymomentum_cpu.libs/libabsl_log_internal_nullguard-883adc72.so.2505.0.0 +0 -0
  530. pymomentum_cpu.libs/libabsl_log_internal_proto-a5da8c75.so.2505.0.0 +0 -0
  531. pymomentum_cpu.libs/libabsl_log_internal_structured_proto-e601fd9b.so.2505.0.0 +0 -0
  532. pymomentum_cpu.libs/libabsl_log_sink-894261b2.so.2505.0.0 +0 -0
  533. pymomentum_cpu.libs/libabsl_low_level_hash-a3284638.so.2505.0.0 +0 -0
  534. pymomentum_cpu.libs/libabsl_malloc_internal-814569de.so.2505.0.0 +0 -0
  535. pymomentum_cpu.libs/libabsl_raw_hash_set-922d64ad.so.2505.0.0 +0 -0
  536. pymomentum_cpu.libs/libabsl_raw_logging_internal-477f78ec.so.2505.0.0 +0 -0
  537. pymomentum_cpu.libs/libabsl_spinlock_wait-8b85a473.so.2505.0.0 +0 -0
  538. pymomentum_cpu.libs/libabsl_stacktrace-7369e71d.so.2505.0.0 +0 -0
  539. pymomentum_cpu.libs/libabsl_str_format_internal-98de729d.so.2505.0.0 +0 -0
  540. pymomentum_cpu.libs/libabsl_strerror-39a52998.so.2505.0.0 +0 -0
  541. pymomentum_cpu.libs/libabsl_strings-a57d5127.so.2505.0.0 +0 -0
  542. pymomentum_cpu.libs/libabsl_strings_internal-ed8c7c0d.so.2505.0.0 +0 -0
  543. pymomentum_cpu.libs/libabsl_symbolize-eba17dd1.so.2505.0.0 +0 -0
  544. pymomentum_cpu.libs/libabsl_synchronization-2f8cf326.so.2505.0.0 +0 -0
  545. pymomentum_cpu.libs/libabsl_time-066c0dde.so.2505.0.0 +0 -0
  546. pymomentum_cpu.libs/libabsl_time_zone-72867365.so.2505.0.0 +0 -0
  547. pymomentum_cpu.libs/libabsl_tracing_internal-021e37ee.so.2505.0.0 +0 -0
  548. pymomentum_cpu.libs/libabsl_utf8_for_code_point-de2a4d4a.so.2505.0.0 +0 -0
  549. pymomentum_cpu.libs/libconsole_bridge-f26e11cc.so.1.0 +0 -0
  550. pymomentum_cpu.libs/libdeflate-577b71e3.so.0 +0 -0
  551. pymomentum_cpu.libs/libdispenso-67ac1721.so.1.4.0 +0 -0
  552. pymomentum_cpu.libs/libezc3d-4a95ab2c.so +0 -0
  553. pymomentum_cpu.libs/libre2-985fb83c.so.11 +0 -0
  554. pymomentum_cpu.libs/libtinyxml2-8d10763c.so.11.0.0 +0 -0
  555. pymomentum_cpu.libs/liburdfdom_model-7b26ae88.so.4.0 +0 -0
@@ -0,0 +1,131 @@
1
+ /*
2
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
3
+ *
4
+ * This source code is licensed under the MIT license found in the
5
+ * LICENSE file in the root directory of this source tree.
6
+ */
7
+
8
+ // This file is @generated by gen_fwd.py
9
+ // Do not edit this file directly, instead update gen_fwd_input.toml and run:
10
+ // buck run @arvr/mode/platform010/opt //arvr/libraries/momentum:gen_fwd
11
+
12
+ #pragma once
13
+
14
+ #include <memory>
15
+
16
+ namespace momentum {
17
+
18
+ struct SolverOptions;
19
+
20
+ using SolverOptions_p = ::std::shared_ptr<SolverOptions>;
21
+ using SolverOptions_u = ::std::unique_ptr<SolverOptions>;
22
+ using SolverOptions_w = ::std::weak_ptr<SolverOptions>;
23
+ using SolverOptions_const_p = ::std::shared_ptr<const SolverOptions>;
24
+ using SolverOptions_const_u = ::std::unique_ptr<const SolverOptions>;
25
+ using SolverOptions_const_w = ::std::weak_ptr<const SolverOptions>;
26
+
27
+ struct GaussNewtonSolverOptions;
28
+
29
+ using GaussNewtonSolverOptions_p = ::std::shared_ptr<GaussNewtonSolverOptions>;
30
+ using GaussNewtonSolverOptions_u = ::std::unique_ptr<GaussNewtonSolverOptions>;
31
+ using GaussNewtonSolverOptions_w = ::std::weak_ptr<GaussNewtonSolverOptions>;
32
+ using GaussNewtonSolverOptions_const_p = ::std::shared_ptr<const GaussNewtonSolverOptions>;
33
+ using GaussNewtonSolverOptions_const_u = ::std::unique_ptr<const GaussNewtonSolverOptions>;
34
+ using GaussNewtonSolverOptions_const_w = ::std::weak_ptr<const GaussNewtonSolverOptions>;
35
+
36
+ template <typename T>
37
+ class SolverT;
38
+ using Solver = SolverT<float>;
39
+ using Solverd = SolverT<double>;
40
+
41
+ using Solver_p = ::std::shared_ptr<Solver>;
42
+ using Solver_u = ::std::unique_ptr<Solver>;
43
+ using Solver_w = ::std::weak_ptr<Solver>;
44
+ using Solver_const_p = ::std::shared_ptr<const Solver>;
45
+ using Solver_const_u = ::std::unique_ptr<const Solver>;
46
+ using Solver_const_w = ::std::weak_ptr<const Solver>;
47
+
48
+ using Solverd_p = ::std::shared_ptr<Solverd>;
49
+ using Solverd_u = ::std::unique_ptr<Solverd>;
50
+ using Solverd_w = ::std::weak_ptr<Solverd>;
51
+ using Solverd_const_p = ::std::shared_ptr<const Solverd>;
52
+ using Solverd_const_u = ::std::unique_ptr<const Solverd>;
53
+ using Solverd_const_w = ::std::weak_ptr<const Solverd>;
54
+
55
+ template <typename T>
56
+ class SolverFunctionT;
57
+ using SolverFunction = SolverFunctionT<float>;
58
+ using SolverFunctiond = SolverFunctionT<double>;
59
+
60
+ using SolverFunction_p = ::std::shared_ptr<SolverFunction>;
61
+ using SolverFunction_u = ::std::unique_ptr<SolverFunction>;
62
+ using SolverFunction_w = ::std::weak_ptr<SolverFunction>;
63
+ using SolverFunction_const_p = ::std::shared_ptr<const SolverFunction>;
64
+ using SolverFunction_const_u = ::std::unique_ptr<const SolverFunction>;
65
+ using SolverFunction_const_w = ::std::weak_ptr<const SolverFunction>;
66
+
67
+ using SolverFunctiond_p = ::std::shared_ptr<SolverFunctiond>;
68
+ using SolverFunctiond_u = ::std::unique_ptr<SolverFunctiond>;
69
+ using SolverFunctiond_w = ::std::weak_ptr<SolverFunctiond>;
70
+ using SolverFunctiond_const_p = ::std::shared_ptr<const SolverFunctiond>;
71
+ using SolverFunctiond_const_u = ::std::unique_ptr<const SolverFunctiond>;
72
+ using SolverFunctiond_const_w = ::std::weak_ptr<const SolverFunctiond>;
73
+
74
+ template <typename T>
75
+ class GaussNewtonSolverT;
76
+ using GaussNewtonSolver = GaussNewtonSolverT<float>;
77
+ using GaussNewtonSolverd = GaussNewtonSolverT<double>;
78
+
79
+ using GaussNewtonSolver_p = ::std::shared_ptr<GaussNewtonSolver>;
80
+ using GaussNewtonSolver_u = ::std::unique_ptr<GaussNewtonSolver>;
81
+ using GaussNewtonSolver_w = ::std::weak_ptr<GaussNewtonSolver>;
82
+ using GaussNewtonSolver_const_p = ::std::shared_ptr<const GaussNewtonSolver>;
83
+ using GaussNewtonSolver_const_u = ::std::unique_ptr<const GaussNewtonSolver>;
84
+ using GaussNewtonSolver_const_w = ::std::weak_ptr<const GaussNewtonSolver>;
85
+
86
+ using GaussNewtonSolverd_p = ::std::shared_ptr<GaussNewtonSolverd>;
87
+ using GaussNewtonSolverd_u = ::std::unique_ptr<GaussNewtonSolverd>;
88
+ using GaussNewtonSolverd_w = ::std::weak_ptr<GaussNewtonSolverd>;
89
+ using GaussNewtonSolverd_const_p = ::std::shared_ptr<const GaussNewtonSolverd>;
90
+ using GaussNewtonSolverd_const_u = ::std::unique_ptr<const GaussNewtonSolverd>;
91
+ using GaussNewtonSolverd_const_w = ::std::weak_ptr<const GaussNewtonSolverd>;
92
+
93
+ template <typename T>
94
+ class GradientDescentSolverT;
95
+ using GradientDescentSolver = GradientDescentSolverT<float>;
96
+ using GradientDescentSolverd = GradientDescentSolverT<double>;
97
+
98
+ using GradientDescentSolver_p = ::std::shared_ptr<GradientDescentSolver>;
99
+ using GradientDescentSolver_u = ::std::unique_ptr<GradientDescentSolver>;
100
+ using GradientDescentSolver_w = ::std::weak_ptr<GradientDescentSolver>;
101
+ using GradientDescentSolver_const_p = ::std::shared_ptr<const GradientDescentSolver>;
102
+ using GradientDescentSolver_const_u = ::std::unique_ptr<const GradientDescentSolver>;
103
+ using GradientDescentSolver_const_w = ::std::weak_ptr<const GradientDescentSolver>;
104
+
105
+ using GradientDescentSolverd_p = ::std::shared_ptr<GradientDescentSolverd>;
106
+ using GradientDescentSolverd_u = ::std::unique_ptr<GradientDescentSolverd>;
107
+ using GradientDescentSolverd_w = ::std::weak_ptr<GradientDescentSolverd>;
108
+ using GradientDescentSolverd_const_p = ::std::shared_ptr<const GradientDescentSolverd>;
109
+ using GradientDescentSolverd_const_u = ::std::unique_ptr<const GradientDescentSolverd>;
110
+ using GradientDescentSolverd_const_w = ::std::weak_ptr<const GradientDescentSolverd>;
111
+
112
+ template <typename T>
113
+ class SubsetGaussNewtonSolverT;
114
+ using SubsetGaussNewtonSolver = SubsetGaussNewtonSolverT<float>;
115
+ using SubsetGaussNewtonSolverd = SubsetGaussNewtonSolverT<double>;
116
+
117
+ using SubsetGaussNewtonSolver_p = ::std::shared_ptr<SubsetGaussNewtonSolver>;
118
+ using SubsetGaussNewtonSolver_u = ::std::unique_ptr<SubsetGaussNewtonSolver>;
119
+ using SubsetGaussNewtonSolver_w = ::std::weak_ptr<SubsetGaussNewtonSolver>;
120
+ using SubsetGaussNewtonSolver_const_p = ::std::shared_ptr<const SubsetGaussNewtonSolver>;
121
+ using SubsetGaussNewtonSolver_const_u = ::std::unique_ptr<const SubsetGaussNewtonSolver>;
122
+ using SubsetGaussNewtonSolver_const_w = ::std::weak_ptr<const SubsetGaussNewtonSolver>;
123
+
124
+ using SubsetGaussNewtonSolverd_p = ::std::shared_ptr<SubsetGaussNewtonSolverd>;
125
+ using SubsetGaussNewtonSolverd_u = ::std::unique_ptr<SubsetGaussNewtonSolverd>;
126
+ using SubsetGaussNewtonSolverd_w = ::std::weak_ptr<SubsetGaussNewtonSolverd>;
127
+ using SubsetGaussNewtonSolverd_const_p = ::std::shared_ptr<const SubsetGaussNewtonSolverd>;
128
+ using SubsetGaussNewtonSolverd_const_u = ::std::unique_ptr<const SubsetGaussNewtonSolverd>;
129
+ using SubsetGaussNewtonSolverd_const_w = ::std::weak_ptr<const SubsetGaussNewtonSolverd>;
130
+
131
+ } // namespace momentum
@@ -0,0 +1,136 @@
1
+ /*
2
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
3
+ *
4
+ * This source code is licensed under the MIT license found in the
5
+ * LICENSE file in the root directory of this source tree.
6
+ */
7
+
8
+ #pragma once
9
+
10
+ #include <momentum/solver/fwd.h>
11
+ #include <momentum/solver/solver.h>
12
+
13
+ namespace momentum {
14
+
15
+ /// Extended options specific to the Gauss-Newton optimization algorithm
16
+ struct GaussNewtonSolverOptions : SolverOptions {
17
+ /// Damping parameter added to Hessian diagonal for numerical stability; see
18
+ /// https://en.wikipedia.org/wiki/Levenberg%E2%80%93Marquardt_algorithm
19
+ ///
20
+ /// Higher values improve stability but may slow convergence
21
+ float regularization = 0.05f;
22
+
23
+ /// Enables backtracking line search to ensure error reduction at each step
24
+ bool doLineSearch = false;
25
+
26
+ /// Uses pre-computed JᵀJ and JᵀR from the solver function
27
+ ///
28
+ /// Can improve performance for problems with specialized structure
29
+ bool useBlockJtJ = false;
30
+
31
+ /// Directly computes sparse JᵀJ without dense intermediate representation
32
+ ///
33
+ /// Only effective when useBlockJtJ is true
34
+ bool directSparseJtJ = false;
35
+
36
+ /// Parameter count threshold for switching to sparse matrix operations
37
+ ///
38
+ /// Problems with more parameters than this threshold will use sparse solvers
39
+ size_t sparseMatrixThreshold = 200;
40
+
41
+ /// Default constructor
42
+ GaussNewtonSolverOptions() = default;
43
+
44
+ /// Construct from base solver options while preserving Gauss-Newton defaults
45
+ /* implicit */ GaussNewtonSolverOptions(const SolverOptions& baseOptions)
46
+ : SolverOptions(baseOptions) {}
47
+ };
48
+
49
+ /// Implementation of the Gauss-Newton optimization algorithm
50
+ ///
51
+ /// Minimizes non-linear least squares problems by iteratively approximating
52
+ /// the objective function with a quadratic model based on first derivatives.
53
+ /// Supports both dense and sparse matrix operations depending on problem size.
54
+ template <typename T>
55
+ class GaussNewtonSolverT : public SolverT<T> {
56
+ public:
57
+ /// Creates a solver with the specified options and function to optimize
58
+ GaussNewtonSolverT(const SolverOptions& options, SolverFunctionT<T>* solver);
59
+
60
+ /// Returns "GaussNewton" as the solver name
61
+ [[nodiscard]] std::string_view getName() const override;
62
+
63
+ /// Updates solver configuration, handling both base and Gauss-Newton specific options
64
+ void setOptions(const SolverOptions& options) final;
65
+
66
+ protected:
67
+ /// Performs one iteration of the Gauss-Newton algorithm
68
+ ///
69
+ /// Delegates to either doIterationDense or doIterationSparse based on problem size
70
+ void doIteration() final;
71
+
72
+ /// Initializes solver state before optimization begins
73
+ void initializeSolver() final;
74
+
75
+ private:
76
+ /// Performs one iteration using dense matrix operations
77
+ void doIterationDense();
78
+
79
+ /// Performs one iteration using sparse matrix operations
80
+ void doIterationSparse();
81
+
82
+ /// Updates parameters using the computed step direction
83
+ ///
84
+ /// Optionally performs line search if enabled
85
+ void updateParameters(Eigen::VectorX<T>& delta);
86
+
87
+ /// Whether to use pre-computed JᵀJ and JᵀR from solver function
88
+ bool useBlockJtJ_{};
89
+
90
+ /// Whether to directly compute sparse JᵀJ without dense intermediate
91
+ bool directSparseJtJ_{};
92
+
93
+ /// Parameter count threshold for switching to sparse operations
94
+ size_t sparseMatrixThreshold_{200};
95
+
96
+ /// Whether the solver has been initialized
97
+ bool initialized_;
98
+
99
+ /// Whether to perform line search during parameter updates
100
+ bool doLineSearch_;
101
+
102
+ /// Sparse Cholesky factorization solver
103
+ Eigen::SimplicialLLT<Eigen::SparseMatrix<T>, Eigen::Lower> lltSolver_;
104
+
105
+ /// Sparse approximation of Hessian matrix (JᵀJ)
106
+ Eigen::SparseMatrix<T> JtJ_;
107
+
108
+ /// Sparse identity matrix for regularization
109
+ Eigen::SparseMatrix<T> D_;
110
+
111
+ /// Jacobian matrix for dense operations
112
+ Eigen::MatrixX<T> jacobian_;
113
+
114
+ /// Dense approximation of Hessian matrix (JᵀJ)
115
+ Eigen::MatrixX<T> hessianApprox_;
116
+
117
+ /// Gradient vector (JᵀR)
118
+ Eigen::VectorX<T> JtR_;
119
+
120
+ /// Residual vector
121
+ Eigen::VectorX<T> residual_;
122
+
123
+ /// Dense Cholesky factorization solver
124
+ Eigen::LLT<Eigen::MatrixX<T>> llt_;
125
+
126
+ /// Base regularization parameter
127
+ T regularization_;
128
+
129
+ /// Current regularization value (may be adjusted during iterations)
130
+ T alpha_;
131
+
132
+ /// Whether the current iteration uses dense operations
133
+ bool denseIteration_{};
134
+ };
135
+
136
+ } // namespace momentum
@@ -0,0 +1,65 @@
1
+ /*
2
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
3
+ *
4
+ * This source code is licensed under the MIT license found in the
5
+ * LICENSE file in the root directory of this source tree.
6
+ */
7
+
8
+ #pragma once
9
+
10
+ #include <momentum/solver/fwd.h>
11
+ #include <momentum/solver/solver.h>
12
+
13
+ namespace momentum {
14
+
15
+ /// Extended options for the Gradient Descent optimization algorithm
16
+ ///
17
+ /// Provides configuration specific to the first-order gradient descent method
18
+ struct GradientDescentSolverOptions : SolverOptions {
19
+ /// Step size for parameter updates during optimization
20
+ ///
21
+ /// Controls how far to move in the direction of the negative gradient
22
+ float learningRate = 0.01f;
23
+
24
+ /// Default constructor
25
+ GradientDescentSolverOptions() = default;
26
+
27
+ /// Construct from base solver options while preserving Gradient Descent defaults
28
+ /* implicit */ GradientDescentSolverOptions(const SolverOptions& baseOptions)
29
+ : SolverOptions(baseOptions) {}
30
+ };
31
+
32
+ /// First-order optimization algorithm that follows the negative gradient
33
+ ///
34
+ /// Implements the standard gradient descent method which iteratively
35
+ /// updates parameters by moving in the direction of steepest descent
36
+ template <typename T>
37
+ class GradientDescentSolverT : public SolverT<T> {
38
+ public:
39
+ /// Creates a solver with the specified options and function to optimize
40
+ GradientDescentSolverT(const SolverOptions& options, SolverFunctionT<T>* solver);
41
+
42
+ /// Returns "GradientDescent" as the solver name
43
+ [[nodiscard]] std::string_view getName() const override;
44
+
45
+ /// Updates solver configuration, handling both base and Gradient Descent specific options
46
+ void setOptions(const SolverOptions& options) final;
47
+
48
+ protected:
49
+ /// Performs one iteration of the gradient descent algorithm
50
+ ///
51
+ /// Computes the gradient and updates parameters by moving in the negative gradient direction
52
+ void doIteration() final;
53
+
54
+ /// Initializes solver state before optimization begins
55
+ void initializeSolver() final;
56
+
57
+ private:
58
+ /// Gradient vector at current parameter values
59
+ Eigen::VectorX<T> gradient_;
60
+
61
+ /// Step size for parameter updates
62
+ float learningRate_;
63
+ };
64
+
65
+ } // namespace momentum
@@ -0,0 +1,155 @@
1
+ /*
2
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
3
+ *
4
+ * This source code is licensed under the MIT license found in the
5
+ * LICENSE file in the root directory of this source tree.
6
+ */
7
+
8
+ #pragma once
9
+
10
+ #include <momentum/math/types.h>
11
+ #include <momentum/solver/fwd.h>
12
+
13
+ #include <string>
14
+ #include <unordered_map>
15
+
16
+ namespace momentum {
17
+
18
+ /// Common options for numerical optimization solvers.
19
+ struct SolverOptions {
20
+ /// Minimum number of iterations before checking convergence criteria
21
+ size_t minIterations = 1;
22
+
23
+ /// Maximum number of iterations before terminating
24
+ size_t maxIterations = 2;
25
+
26
+ /// Convergence threshold for relative error change
27
+ float threshold = 1.0f;
28
+
29
+ /// Enable detailed logging during optimization
30
+ bool verbose = false;
31
+
32
+ /// Virtual destructor for polymorphic behavior
33
+ virtual ~SolverOptions() = default;
34
+ };
35
+
36
+ /// Base class for numerical optimization solvers.
37
+ ///
38
+ /// Provides common functionality for iterative optimization algorithms
39
+ /// including parameter management, convergence checking, and history tracking.
40
+ template <typename T>
41
+ class SolverT {
42
+ public:
43
+ /// Constructor with solver options and function to optimize
44
+ SolverT(const SolverOptions& options, SolverFunctionT<T>* solver);
45
+
46
+ virtual ~SolverT() = default;
47
+
48
+ /// Returns the name of the solver implementation
49
+ [[nodiscard]] virtual std::string_view getName() const = 0;
50
+
51
+ /// Updates solver configuration with new options
52
+ virtual void setOptions(const SolverOptions& options);
53
+
54
+ /// Solves the optimization problem
55
+ ///
56
+ /// @param[in,out] params Vector of initial guess, updated with the optimized solution
57
+ /// @return Final objective function value after optimization
58
+ double solve(Eigen::VectorX<T>& params);
59
+
60
+ /// Specifies which parameters should be optimized
61
+ ///
62
+ /// @param parameters Bitset where each bit indicates if the corresponding parameter is enabled
63
+ virtual void setEnabledParameters(const ParameterSet& parameters);
64
+
65
+ /// Returns the current set of enabled parameters
66
+ [[nodiscard]] const ParameterSet& getActiveParameters() const;
67
+
68
+ /// Sets the current parameter values without solving
69
+ void setParameters(const Eigen::VectorX<T>& params);
70
+
71
+ /// Controls whether to store iteration history for debugging and analysis
72
+ void setStoreHistory(bool b);
73
+
74
+ /// Returns the history of the solver's iterations
75
+ ///
76
+ /// The history contains matrices for parameters, errors, and other solver-specific data
77
+ [[nodiscard]] const std::unordered_map<std::string, Eigen::MatrixX<T>>& getHistory() const;
78
+
79
+ /// Returns the minimum number of iterations before checking convergence
80
+ [[nodiscard]] size_t getMinIterations() const;
81
+
82
+ /// Returns the maximum number of iterations before terminating
83
+ [[nodiscard]] size_t getMaxIterations() const;
84
+
85
+ [[nodiscard]] size_t getNumParameters() const {
86
+ return numParameters_;
87
+ }
88
+
89
+ /// Returns the history of objective function values over the iterations of the solve
90
+ [[nodiscard]] const std::vector<double>& getErrorHistory() const {
91
+ return errorHistory_;
92
+ }
93
+
94
+ protected:
95
+ /// Initializes solver state before optimization begins
96
+ virtual void initializeSolver() = 0;
97
+
98
+ /// Performs a single iteration of the optimization algorithm
99
+ virtual void doIteration() = 0;
100
+
101
+ protected:
102
+ /// Total number of parameters in the optimization problem
103
+ size_t numParameters_;
104
+
105
+ /// Function to be optimized
106
+ SolverFunctionT<T>* solverFunction_;
107
+
108
+ /// Current parameter values
109
+ Eigen::VectorX<T> parameters_;
110
+
111
+ /// Bitset indicating which parameters are enabled for optimization
112
+ ParameterSet activeParameters_;
113
+
114
+ /// Number of parameters currently enabled for optimization
115
+ int actualParameters_;
116
+
117
+ /// Flag indicating parameter structure has changed
118
+ ///
119
+ /// Used to trigger reinitialization of solver data structures
120
+ bool newParameterPattern_;
121
+
122
+ /// Current iteration count
123
+ size_t iteration_{};
124
+
125
+ /// Current objective function value
126
+ double error_{};
127
+
128
+ /// Previous iteration's objective function value
129
+ double lastError_{};
130
+
131
+ /// Whether to record optimization progress for analysis
132
+ bool storeHistory = false;
133
+
134
+ /// Recorded data from optimization process
135
+ ///
136
+ /// Contains matrices for parameters, errors, and other solver-specific data
137
+ std::unordered_map<std::string, Eigen::MatrixX<T>> iterationHistory_;
138
+ /// History of objective function values over the iterations
139
+ std::vector<double> errorHistory_;
140
+
141
+ /// Whether to output detailed progress information
142
+ bool verbose_ = false;
143
+
144
+ private:
145
+ /// Minimum iterations before checking convergence criteria
146
+ size_t minIterations_{};
147
+
148
+ /// Maximum iterations before terminating optimization
149
+ size_t maxIterations_{};
150
+
151
+ /// Relative error change threshold for convergence
152
+ float threshold_{};
153
+ };
154
+
155
+ } // namespace momentum
@@ -0,0 +1,126 @@
1
+ /*
2
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
3
+ *
4
+ * This source code is licensed under the MIT license found in the
5
+ * LICENSE file in the root directory of this source tree.
6
+ */
7
+
8
+ #pragma once
9
+
10
+ #include <momentum/math/types.h>
11
+ #include <momentum/solver/fwd.h>
12
+
13
+ #include <unordered_map>
14
+
15
+ namespace momentum {
16
+
17
+ /// Abstract base class for optimization objective functions
18
+ ///
19
+ /// Provides the interface for computing objective function values, gradients,
20
+ /// and other derivatives needed by numerical optimization algorithms.
21
+ template <typename T>
22
+ class SolverFunctionT {
23
+ public:
24
+ virtual ~SolverFunctionT() = default;
25
+
26
+ /// Evaluates the objective function at the given parameter values
27
+ ///
28
+ /// @param parameters Current parameter values
29
+ /// @return Objective function value (typically sum of squared errors)
30
+ virtual double getError(const VectorX<T>& parameters) = 0;
31
+
32
+ /// Computes the gradient of the objective function
33
+ ///
34
+ /// @param parameters Current parameter values
35
+ /// @param[out] gradient Computed gradient vector
36
+ /// @return Objective function value
37
+ virtual double getGradient(const VectorX<T>& parameters, VectorX<T>& gradient) = 0;
38
+
39
+ /// Computes the Jacobian matrix for least squares problems
40
+ ///
41
+ /// @param parameters Current parameter values
42
+ /// @param[out] jacobian Jacobian matrix (m×n for m residuals and n parameters)
43
+ /// @param[out] residual Vector of residual values
44
+ /// @param[out] actualRows Number of active residual rows
45
+ /// @return Objective function value
46
+ virtual double getJacobian(
47
+ const VectorX<T>& parameters,
48
+ MatrixX<T>& jacobian,
49
+ VectorX<T>& residual,
50
+ size_t& actualRows) = 0;
51
+
52
+ /// Computes the Hessian matrix of second derivatives
53
+ ///
54
+ /// Default implementation throws an exception as this is rarely needed
55
+ /// @param parameters Current parameter values
56
+ /// @param[out] hessian Computed Hessian matrix
57
+ virtual void getHessian(const VectorX<T>& parameters, MatrixX<T>& hessian);
58
+
59
+ /// Computes JᵀJ and JᵀR for Gauss-Newton optimization
60
+ ///
61
+ /// Default implementation computes these from the Jacobian
62
+ /// @param parameters Current parameter values
63
+ /// @param[out] jtj Approximated Hessian matrix (JᵀJ)
64
+ /// @param[out] jtr Gradient vector (JᵀR)
65
+ /// @return Objective function value
66
+ virtual double getJtJR(const VectorX<T>& parameters, MatrixX<T>& jtj, VectorX<T>& jtr);
67
+
68
+ /// Computes sparse JᵀJ and JᵀR for large-scale problems
69
+ ///
70
+ /// Default implementation returns 0.0 and must be overridden for sparse optimization
71
+ /// @param parameters Current parameter values
72
+ /// @param[out] jtj Sparse approximated Hessian matrix
73
+ /// @param[out] jtr Gradient vector
74
+ /// @return Objective function value
75
+ virtual double
76
+ getJtJR_Sparse(const VectorX<T>& parameters, SparseMatrix<T>& jtj, VectorX<T>& jtr);
77
+
78
+ /// Computes derivatives needed by the solver
79
+ ///
80
+ /// Default implementation calls getJtJR
81
+ /// @param parameters Current parameter values
82
+ /// @param[out] hess Hessian matrix or approximation
83
+ /// @param[out] grad Gradient vector
84
+ /// @return Objective function value
85
+ virtual double
86
+ getSolverDerivatives(const VectorX<T>& parameters, MatrixX<T>& hess, VectorX<T>& grad);
87
+
88
+ /// Updates parameters using the computed step direction
89
+ ///
90
+ /// @param[in,out] parameters Current parameters, updated in-place
91
+ /// @param gradient Step direction (typically the negative gradient)
92
+ virtual void updateParameters(VectorX<T>& parameters, const VectorX<T>& gradient) = 0;
93
+
94
+ /// Specifies which parameters should be optimized
95
+ ///
96
+ /// Default implementation does nothing
97
+ /// @param parameterSet Bitset where each bit indicates if the corresponding parameter is enabled
98
+ virtual void setEnabledParameters(const ParameterSet& parameterSet);
99
+
100
+ /// Returns the total number of parameters in the optimization problem
101
+ [[nodiscard]] size_t getNumParameters() const;
102
+
103
+ /// Returns the number of parameters currently enabled for optimization
104
+ [[nodiscard]] size_t getActualParameters() const;
105
+
106
+ /// Records solver state for debugging and analysis
107
+ ///
108
+ /// @param[in,out] history Map to store iteration data
109
+ /// @param iteration Current iteration number
110
+ /// @param maxIterations Maximum number of iterations
111
+ virtual void storeHistory(
112
+ std::unordered_map<std::string, MatrixX<T>>& history,
113
+ size_t iteration,
114
+ size_t maxIterations_);
115
+
116
+ protected:
117
+ /// Total number of parameters in the optimization problem
118
+ size_t numParameters_{};
119
+
120
+ /// Number of parameters currently enabled for optimization
121
+ ///
122
+ /// Always less than or equal to numParameters_
123
+ size_t actualParameters_{};
124
+ };
125
+
126
+ } // namespace momentum