blob: 2f5c3999a0acd019d0c9ae050e32fd50141c50d1 [file] [log] [blame]
milind-u086d7262022-01-19 20:44:18 -08001#ifndef Y2022_ACTORS_AUTO_SPLINES_H_
2#define Y2022_ACTORS_AUTO_SPLINES_H_
3
4#include "aos/events/event_loop.h"
Ravago Jones81e50632022-03-11 16:23:51 -08005#include "aos/flatbuffer_merge.h"
milind-u086d7262022-01-19 20:44:18 -08006#include "frc971/control_loops/control_loops_generated.h"
7#include "frc971/control_loops/drivetrain/drivetrain_goal_generated.h"
Ravago Jones81e50632022-03-11 16:23:51 -08008#include "frc971/input/joystick_state_generated.h"
milind-u086d7262022-01-19 20:44:18 -08009/*
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
16namespace y2022 {
17namespace actors {
18
19class AutonomousSplines {
20 public:
Ravago Jones81e50632022-03-11 16:23:51 -080021 AutonomousSplines()
22 : test_spline_(aos::JsonFileToFlatbuffer<frc971::MultiSpline>(
Ravago Jonese20f7f92022-03-11 19:54:38 -080023 "splines/test_spline.json")),
24 spline_1_(aos::JsonFileToFlatbuffer<frc971::MultiSpline>(
Henry Speiser09e8da92022-03-14 20:58:45 -070025 "splines/spline_5_ball_1.json")),
Ravago Jonese20f7f92022-03-11 19:54:38 -080026 spline_2_(aos::JsonFileToFlatbuffer<frc971::MultiSpline>(
Henry Speiser09e8da92022-03-14 20:58:45 -070027 "splines/spline_5_ball_2.json")),
Ravago Jonese20f7f92022-03-11 19:54:38 -080028 spline_3_(aos::JsonFileToFlatbuffer<frc971::MultiSpline>(
Henry Speiser5eed1de2022-04-07 21:52:10 -070029 "splines/spline_5_ball_3.json")),
Austin Schuhf3413b72022-04-16 19:09:33 -070030 spline_two1_(aos::JsonFileToFlatbuffer<frc971::MultiSpline>(
31 "splines/spline_2_ball.json")),
32 spline_two2_(aos::JsonFileToFlatbuffer<frc971::MultiSpline>(
33 "splines/spline_2_ball_2.json")) {}
Ravago Jones81e50632022-03-11 16:23:51 -080034
milind-u086d7262022-01-19 20:44:18 -080035 static flatbuffers::Offset<frc971::MultiSpline> BasicSSpline(
36 aos::Sender<frc971::control_loops::drivetrain::Goal>::Builder *builder);
37 static flatbuffers::Offset<frc971::MultiSpline> StraightLine(
38 aos::Sender<frc971::control_loops::drivetrain::Goal>::Builder *builder);
Ravago Jones81e50632022-03-11 16:23:51 -080039
40 flatbuffers::Offset<frc971::MultiSpline> TestSpline(
41 aos::Sender<frc971::control_loops::drivetrain::SplineGoal>::Builder
42 *builder,
43 aos::Alliance alliance);
44
Ravago Jonese20f7f92022-03-11 19:54:38 -080045 flatbuffers::Offset<frc971::MultiSpline> Spline1(
46 aos::Sender<frc971::control_loops::drivetrain::SplineGoal>::Builder
47 *builder,
48 aos::Alliance alliance);
49 flatbuffers::Offset<frc971::MultiSpline> Spline2(
50 aos::Sender<frc971::control_loops::drivetrain::SplineGoal>::Builder
51 *builder,
52 aos::Alliance alliance);
53 flatbuffers::Offset<frc971::MultiSpline> Spline3(
54 aos::Sender<frc971::control_loops::drivetrain::SplineGoal>::Builder
55 *builder,
56 aos::Alliance alliance);
Ravago Jonese20f7f92022-03-11 19:54:38 -080057
Austin Schuhf3413b72022-04-16 19:09:33 -070058 flatbuffers::Offset<frc971::MultiSpline> SplineTwoBall1(
59 aos::Sender<frc971::control_loops::drivetrain::SplineGoal>::Builder
60 *builder,
61 aos::Alliance alliance);
62 flatbuffers::Offset<frc971::MultiSpline> SplineTwoBall2(
Henry Speiser5eed1de2022-04-07 21:52:10 -070063 aos::Sender<frc971::control_loops::drivetrain::SplineGoal>::Builder
64 *builder,
65 aos::Alliance alliance);
66
Ravago Jones81e50632022-03-11 16:23:51 -080067 private:
68 aos::FlatbufferDetachedBuffer<frc971::MultiSpline> test_spline_;
Ravago Jonese20f7f92022-03-11 19:54:38 -080069 aos::FlatbufferDetachedBuffer<frc971::MultiSpline> spline_1_;
70 aos::FlatbufferDetachedBuffer<frc971::MultiSpline> spline_2_;
71 aos::FlatbufferDetachedBuffer<frc971::MultiSpline> spline_3_;
Austin Schuhf3413b72022-04-16 19:09:33 -070072 aos::FlatbufferDetachedBuffer<frc971::MultiSpline> spline_two1_;
73 aos::FlatbufferDetachedBuffer<frc971::MultiSpline> spline_two2_;
milind-u086d7262022-01-19 20:44:18 -080074};
75
76} // namespace actors
77} // namespace y2022
78
79#endif // Y2022_ACTORS_AUTO_SPLINES_H_