blob: 8e8768ebafb82a45561e7547cfc3255fdd1ff776 [file] [log] [blame]
Brian Silvermanb691f5e2015-08-02 11:37:55 -07001#ifndef Y2015_CONTROL_LOOPS_DRIVETRAIN_H_
2#define Y2015_CONTROL_LOOPS_DRIVETRAIN_H_
brians343bc112013-02-10 01:53:46 +00003
Austin Schuh2054f5f2013-10-27 14:54:10 -07004#include "Eigen/Dense"
5
Briana6553ed2014-04-02 21:26:46 -07006#include "aos/common/controls/polytope.h"
7#include "aos/common/controls/control_loop.h"
8#include "aos/common/controls/polytope.h"
Brian Silvermanb691f5e2015-08-02 11:37:55 -07009#include "y2015/control_loops/drivetrain/drivetrain.q.h"
Brian Silverman50a9d032014-02-16 17:20:57 -080010#include "aos/common/util/log_interval.h"
brians343bc112013-02-10 01:53:46 +000011
12namespace frc971 {
13namespace control_loops {
14
brians343bc112013-02-10 01:53:46 +000015class DrivetrainLoop
Brian Silvermanada5f2c2015-02-01 02:41:14 -050016 : public aos::controls::ControlLoop<control_loops::DrivetrainQueue> {
brians343bc112013-02-10 01:53:46 +000017 public:
18 // Constructs a control loop which can take a Drivetrain or defaults to the
19 // drivetrain at frc971::control_loops::drivetrain
Brian Silvermanada5f2c2015-02-01 02:41:14 -050020 explicit DrivetrainLoop(control_loops::DrivetrainQueue *my_drivetrain =
21 &control_loops::drivetrain_queue)
22 : aos::controls::ControlLoop<control_loops::DrivetrainQueue>(
23 my_drivetrain) {
Brian Silverman718b1d72013-10-28 16:22:45 -070024 ::aos::controls::HPolytope<0>::Init();
25 }
brians343bc112013-02-10 01:53:46 +000026
27 protected:
28 // Executes one cycle of the control loop.
29 virtual void RunIteration(
Brian Silvermanada5f2c2015-02-01 02:41:14 -050030 const control_loops::DrivetrainQueue::Goal *goal,
31 const control_loops::DrivetrainQueue::Position *position,
32 control_loops::DrivetrainQueue::Output *output,
33 control_loops::DrivetrainQueue::Status *status);
Brian Silverman50a9d032014-02-16 17:20:57 -080034
35 typedef ::aos::util::SimpleLogInterval SimpleLogInterval;
36 SimpleLogInterval no_position_ = SimpleLogInterval(
37 ::aos::time::Time::InSeconds(0.25), WARNING, "no position");
brians343bc112013-02-10 01:53:46 +000038};
39
40} // namespace control_loops
41} // namespace frc971
42
Brian Silvermanb691f5e2015-08-02 11:37:55 -070043#endif // Y2015_CONTROL_LOOPS_DRIVETRAIN_H_