Flair
Framework Libre Air
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties
UavStateMachine.h
1 // %flair:license{
2 // This file is part of the Flair framework distributed under the
3 // CECILL-C License, Version 1.0.
4 // %flair:license}
5 // created: 2014/04/29
6 // filename: UavStateMachine.h
7 //
8 // author: Gildas Bayard, Guillaume Sanahuja
9 // Copyright Heudiasyc UMR UTC/CNRS 7253
10 //
11 // version: $Id: $
12 //
13 // purpose: state machine for UAV
14 //
15 //
16 /*********************************************************************/
17 
18 #ifndef UAVSTATEMACHINE_H
19 #define UAVSTATEMACHINE_H
20 
21 #include <Thread.h>
22 #include <Vector2D.h>
23 #include <Vector3D.h>
24 #include <Euler.h>
25 #include <Quaternion.h>
26 
27 namespace flair {
28 namespace core {
29 class FrameworkManager;
30 class AhrsData;
31 class io_data;
32 }
33 namespace gui {
34 class PushButton;
35 class GridLayout;
36 class Tab;
37 class DoubleSpinBox;
38 }
39 namespace filter {
40 class ControlLaw;
41 class NestedSat;
42 class Pid;
43 class PidThrust;
44 class TrajectoryGenerator1D;
45 }
46 namespace sensor {
47 class TargetController;
48 }
49 namespace meta {
50 class MetaDualShock3;
51 class Uav;
52 }
53 }
54 
55 namespace flair {
56 namespace meta {
57 
66 class UavStateMachine : public core::Thread {
67 protected:
68  enum class AltitudeMode_t { Manual, Custom };
69  const AltitudeMode_t &GetAltitudeMode(void) const { return altitudeMode; }
70  bool SetAltitudeMode(const AltitudeMode_t &altitudeMode);
71 
72  // uses TrajectoryGenerator1D *altitudeTrajectory to go to desiredAltitude
73  // available in mode AltitudeMode_t::Manual
74  // return true if goto is possible
75  bool GotoAltitude(float desiredAltitude);
76 
77  enum class OrientationMode_t { Manual, Custom };
78  const OrientationMode_t &GetOrientationMode(void) const {
79  return orientationMode;
80  }
81  bool SetOrientationMode(const OrientationMode_t &orientationMode);
82 
83  enum class ThrustMode_t { Default, Custom };
84  const ThrustMode_t &GetThrustMode() const { return thrustMode; }
85  bool SetThrustMode(const ThrustMode_t &thrustMode);
86 
87  enum class TorqueMode_t { Default, Custom };
88  const TorqueMode_t &GetTorqueMode(void) const { return torqueMode; }
89  bool SetTorqueMode(const TorqueMode_t &torqueMode);
90 
91  enum class Event_t {
92  EnteringFailSafeMode,
93  EnteringControlLoop,
94  StartLanding,
95  FinishLanding,
96  Stopped,
97  TakingOff,
98  EmergencyStop,
99  Stabilized, // as soon as uav is 3cm far from the ground
100  ZTrajectoryFinished,
101  };
102 
103  UavStateMachine(meta::Uav *uav, uint16_t ds3Port = 20000);
104  ~UavStateMachine();
105 
106  const core::Quaternion &GetCurrentQuaternion(void) const;
107 
108  const core::Vector3D &GetCurrentAngularSpeed(void) const;
109 
110  const meta::Uav *GetUav(void) const;
111 
112  void Land(void);
113  void TakeOff(void);
114  void EmergencyStop(void);
116 
119  virtual void SignalEvent(Event_t event);
120 
121  virtual const core::AhrsData *GetOrientation(void) const;
122  const core::AhrsData *GetDefaultOrientation(void) const;
123 
124  virtual void AltitudeValues(float &z, float &dz) const; // in uav coordinate!
125  void EnterFailSafeMode(void);
126  bool ExitFailSafeMode(void);
127  void FailSafeAltitudeValues(float &z, float &dz) const; // in uav coordinate!
128 
129  gui::GridLayout *GetButtonsLayout(void) const;
130  virtual void ExtraSecurityCheck(void){};
131  virtual void ExtraCheckJoystick(void){};
132  virtual void ExtraCheckPushButton(void){};
133 
134  void GetDefaultReferenceAltitude(float &refAltitude,
135  float &refVerticalVelocity);
136  virtual void GetReferenceAltitude(float &refAltitude,
137  float &refVerticalVelocity);
138  // float GetDefaultThrustOffset(void);
139  const core::AhrsData *GetDefaultReferenceOrientation(void) const;
140  virtual const core::AhrsData *GetReferenceOrientation(void);
141 
152  virtual void ComputeCustomTorques(core::Euler &torques);
153 
164  void ComputeDefaultTorques(core::Euler &torques);
165 
171  // const core::Euler &GetTorques() const;
172 
183  virtual float ComputeCustomThrust(void);
184 
195  float ComputeDefaultThrust(void);
196 
202  // float GetThrust() const;
203 
214  void AddDeviceToControlLawLog(const core::IODevice *device);
215 
226  void AddDataToControlLawLog(const core::io_data *data);
227 
228  const sensor::TargetController *GetJoystick(void) const;
229 
230  gui::Tab *setupLawTab, *graphLawTab;
231 
232 private:
237  enum class AltitudeState_t {
238  Stopped,
239  TakingOff,
241  Stabilized,
242  StartLanding,
245  };
246  AltitudeState_t altitudeState;
247  void ProcessAltitudeFiniteStateMachine();
248  void ComputeReferenceAltitude(float &refAltitude, float &refVerticalVelocity);
249 
250  float groundAltitude; // effective altitude when the uav leaves the ground
251  float currentAltitude, currentVerticalSpeed;
252 
253  bool failSafeMode;
254  void SecurityCheck(void);
255  void MandatorySecurityCheck(void);
256  void CheckJoystick();
257  void GenericCheckJoystick();
258  void CheckPushButton(void);
259  void GenericCheckPushButton(void);
260  void Run(void);
261  void StopMotors(void);
262 
263  meta::Uav *uav;
264 
265  core::Quaternion currentQuaternion;
266  core::Vector3D currentAngularSpeed;
267 
268  const core::AhrsData *ComputeReferenceOrientation(void);
269 
270  void ComputeOrientation(void);
271  void ComputeAltitude(void);
272 
273  void ComputeTorques(void);
274  core::Euler currentTorques, savedDefaultTorques;
275  bool needToComputeDefaultTorques;
276 
277  void ComputeThrust(void);
278  float currentThrust, savedDefaultThrust;
279  bool needToComputeDefaultThrust;
280 
281  gui::PushButton *button_kill, *button_take_off, *button_land,
282  *button_start_log, *button_stop_log;
283  gui::GridLayout *buttonslayout;
284  gui::DoubleSpinBox *desiredTakeoffAltitude, *desiredLandingAltitude;
285  AltitudeMode_t altitudeMode;
286  OrientationMode_t orientationMode;
287  ThrustMode_t thrustMode;
288  TorqueMode_t torqueMode;
289  bool flagBatteryLow;
290  bool flagConnectionLost;
291  bool flagZTrajectoryFinished;
292  filter::NestedSat *uRoll, *uPitch;
293  filter::Pid *uYaw;
294  filter::PidThrust *uZ;
295 
296  MetaDualShock3 *joy;
297  filter::TrajectoryGenerator1D *altitudeTrajectory;
298 };
299 };
300 };
301 #endif // UAVSTATEMACHINE_H
Abstract class for data types.
Definition: io_data.h:77
virtual void SignalEvent(Event_t event)
Used to signal an event.
Classe intégrant la manette MetaDualShock3.
Definition: MetaDualShock3.h:37
Abstract class for input/ouput system.
Definition: IODevice.h:44
Class defining a 3D vector.
Definition: Vector3D.h:28
Class defining euler angles.
Definition: Euler.h:27
float ComputeDefaultThrust(void)
Compute Default Thrust.
Class defining a 2D vector.
Class displaying a QGridLayout on the ground station.
Definition: GridLayout.h:27
Class defining a Pid for Thrust. This Pid as an extra offset for compensating gravity.
Definition: PidThrust.h:33
Class defining AHRS datas.
Definition: AhrsData.h:30
Class defining a 3D vector.
State machine for UAV.
Definition: UavStateMachine.h:66
Base Class for target side remote controls.
Definition: TargetController.h:49
Class generating a trajectory in 1D.
Definition: TrajectoryGenerator1D.h:47
virtual float ComputeCustomThrust(void)
Get Torques.
void AddDeviceToControlLawLog(const core::IODevice *device)
Get Thrust.
Class displaying a QPushButton on the ground station.
Definition: PushButton.h:28
virtual void ComputeCustomTorques(core::Euler &torques)
Compute Custom Torques.
Class defining a PID.
Definition: Pid.h:32
Abstract class for a thread.
Class defining a quaternion.
Class defining a PID with saturations.
Definition: NestedSat.h:43
Abstract class for a thread.
Definition: Thread.h:38
Class displaying a QDoubleSpinBox on the ground station.
Definition: DoubleSpinBox.h:28
Base class to construct sensors/actuators depending on uav type.
Definition: Uav.h:47
void AddDataToControlLawLog(const core::io_data *data)
Add an io_data to the control law logs.
void ComputeDefaultTorques(core::Euler &torques)
Compute Default Torques.
Class defining a quaternion.
Definition: Quaternion.h:25
Class displaying a QTab on the ground station.
Definition: Tab.h:29
Class defining euler angles.