milind-u | 086d726 | 2022-01-19 20:44:18 -0800 | [diff] [blame] | 1 | #ifndef Y2022_ACTORS_AUTO_SPLINES_H_ |
| 2 | #define Y2022_ACTORS_AUTO_SPLINES_H_ |
| 3 | |
| 4 | #include "aos/events/event_loop.h" |
Ravago Jones | 81e5063 | 2022-03-11 16:23:51 -0800 | [diff] [blame] | 5 | #include "aos/flatbuffer_merge.h" |
milind-u | 086d726 | 2022-01-19 20:44:18 -0800 | [diff] [blame] | 6 | #include "frc971/control_loops/control_loops_generated.h" |
| 7 | #include "frc971/control_loops/drivetrain/drivetrain_goal_generated.h" |
Ravago Jones | 81e5063 | 2022-03-11 16:23:51 -0800 | [diff] [blame] | 8 | #include "frc971/input/joystick_state_generated.h" |
milind-u | 086d726 | 2022-01-19 20:44:18 -0800 | [diff] [blame] | 9 | /* |
| 10 | |
| 11 | The cooridinate system for the autonomous splines is the same as the spline |
| 12 | python generator and drivetrain spline systems. |
| 13 | |
| 14 | */ |
| 15 | |
Stephan Pleines | d99b1ee | 2024-02-02 20:56:44 -0800 | [diff] [blame] | 16 | namespace y2022::actors { |
milind-u | 086d726 | 2022-01-19 20:44:18 -0800 | [diff] [blame] | 17 | |
| 18 | class AutonomousSplines { |
| 19 | public: |
Ravago Jones | 81e5063 | 2022-03-11 16:23:51 -0800 | [diff] [blame] | 20 | AutonomousSplines() |
| 21 | : test_spline_(aos::JsonFileToFlatbuffer<frc971::MultiSpline>( |
Ravago Jones | e20f7f9 | 2022-03-11 19:54:38 -0800 | [diff] [blame] | 22 | "splines/test_spline.json")), |
| 23 | spline_1_(aos::JsonFileToFlatbuffer<frc971::MultiSpline>( |
Henry Speiser | 09e8da9 | 2022-03-14 20:58:45 -0700 | [diff] [blame] | 24 | "splines/spline_5_ball_1.json")), |
Ravago Jones | e20f7f9 | 2022-03-11 19:54:38 -0800 | [diff] [blame] | 25 | spline_2_(aos::JsonFileToFlatbuffer<frc971::MultiSpline>( |
Henry Speiser | 09e8da9 | 2022-03-14 20:58:45 -0700 | [diff] [blame] | 26 | "splines/spline_5_ball_2.json")), |
Ravago Jones | e20f7f9 | 2022-03-11 19:54:38 -0800 | [diff] [blame] | 27 | spline_3_(aos::JsonFileToFlatbuffer<frc971::MultiSpline>( |
Henry Speiser | 5eed1de | 2022-04-07 21:52:10 -0700 | [diff] [blame] | 28 | "splines/spline_5_ball_3.json")), |
Austin Schuh | f3413b7 | 2022-04-16 19:09:33 -0700 | [diff] [blame] | 29 | spline_two1_(aos::JsonFileToFlatbuffer<frc971::MultiSpline>( |
| 30 | "splines/spline_2_ball.json")), |
| 31 | spline_two2_(aos::JsonFileToFlatbuffer<frc971::MultiSpline>( |
| 32 | "splines/spline_2_ball_2.json")) {} |
Ravago Jones | 81e5063 | 2022-03-11 16:23:51 -0800 | [diff] [blame] | 33 | |
milind-u | 086d726 | 2022-01-19 20:44:18 -0800 | [diff] [blame] | 34 | static flatbuffers::Offset<frc971::MultiSpline> BasicSSpline( |
| 35 | aos::Sender<frc971::control_loops::drivetrain::Goal>::Builder *builder); |
| 36 | static flatbuffers::Offset<frc971::MultiSpline> StraightLine( |
| 37 | aos::Sender<frc971::control_loops::drivetrain::Goal>::Builder *builder); |
Ravago Jones | 81e5063 | 2022-03-11 16:23:51 -0800 | [diff] [blame] | 38 | |
| 39 | flatbuffers::Offset<frc971::MultiSpline> TestSpline( |
| 40 | aos::Sender<frc971::control_loops::drivetrain::SplineGoal>::Builder |
| 41 | *builder, |
| 42 | aos::Alliance alliance); |
| 43 | |
Ravago Jones | e20f7f9 | 2022-03-11 19:54:38 -0800 | [diff] [blame] | 44 | flatbuffers::Offset<frc971::MultiSpline> Spline1( |
| 45 | aos::Sender<frc971::control_loops::drivetrain::SplineGoal>::Builder |
| 46 | *builder, |
| 47 | aos::Alliance alliance); |
| 48 | flatbuffers::Offset<frc971::MultiSpline> Spline2( |
| 49 | aos::Sender<frc971::control_loops::drivetrain::SplineGoal>::Builder |
| 50 | *builder, |
| 51 | aos::Alliance alliance); |
| 52 | flatbuffers::Offset<frc971::MultiSpline> Spline3( |
| 53 | aos::Sender<frc971::control_loops::drivetrain::SplineGoal>::Builder |
| 54 | *builder, |
| 55 | aos::Alliance alliance); |
Ravago Jones | e20f7f9 | 2022-03-11 19:54:38 -0800 | [diff] [blame] | 56 | |
Austin Schuh | f3413b7 | 2022-04-16 19:09:33 -0700 | [diff] [blame] | 57 | flatbuffers::Offset<frc971::MultiSpline> SplineTwoBall1( |
| 58 | aos::Sender<frc971::control_loops::drivetrain::SplineGoal>::Builder |
| 59 | *builder, |
| 60 | aos::Alliance alliance); |
| 61 | flatbuffers::Offset<frc971::MultiSpline> SplineTwoBall2( |
Henry Speiser | 5eed1de | 2022-04-07 21:52:10 -0700 | [diff] [blame] | 62 | aos::Sender<frc971::control_loops::drivetrain::SplineGoal>::Builder |
| 63 | *builder, |
| 64 | aos::Alliance alliance); |
| 65 | |
Ravago Jones | 81e5063 | 2022-03-11 16:23:51 -0800 | [diff] [blame] | 66 | private: |
| 67 | aos::FlatbufferDetachedBuffer<frc971::MultiSpline> test_spline_; |
Ravago Jones | e20f7f9 | 2022-03-11 19:54:38 -0800 | [diff] [blame] | 68 | aos::FlatbufferDetachedBuffer<frc971::MultiSpline> spline_1_; |
| 69 | aos::FlatbufferDetachedBuffer<frc971::MultiSpline> spline_2_; |
| 70 | aos::FlatbufferDetachedBuffer<frc971::MultiSpline> spline_3_; |
Austin Schuh | f3413b7 | 2022-04-16 19:09:33 -0700 | [diff] [blame] | 71 | aos::FlatbufferDetachedBuffer<frc971::MultiSpline> spline_two1_; |
| 72 | aos::FlatbufferDetachedBuffer<frc971::MultiSpline> spline_two2_; |
milind-u | 086d726 | 2022-01-19 20:44:18 -0800 | [diff] [blame] | 73 | }; |
| 74 | |
Stephan Pleines | d99b1ee | 2024-02-02 20:56:44 -0800 | [diff] [blame] | 75 | } // namespace y2022::actors |
milind-u | 086d726 | 2022-01-19 20:44:18 -0800 | [diff] [blame] | 76 | |
| 77 | #endif // Y2022_ACTORS_AUTO_SPLINES_H_ |