src/ATC/trafficcontrol.hxx

   1 // trafficcontrol.hxx - classes to manage AIModels based air traffic control
   2 // Written by Durk Talsma, started September 2006.
   3 //
   4 // This program is free software; you can redistribute it and/or
   5 // modify it under the terms of the GNU General Public License as
   6 // published by the Free Software Foundation; either version 2 of the
   7 // License, or (at your option) any later version.
   8 //
   9 // This program is distributed in the hope that it will be useful, but
  10 // WITHOUT ANY WARRANTY; without even the implied warranty of
  11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12 // General Public License for more details.
  13 //
  14 // You should have received a copy of the GNU General Public License
  15 // along with this program; if not, write to the Free Software
  16 // Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  17 //
  18 // $Id$
  19
  20
  21 #ifndef _TRAFFIC_CONTROL_HXX_
  22 #define _TRAFFIC_CONTROL_HXX_
  23
  24
  25 #ifndef __cplusplus
  26 # error This library requires C++
  27 #endif
  28
  29 #include <osg/Geode>
  30 #include <osg/Geometry>
  31 #include <osg/MatrixTransform>
  32 #include <osg/Shape>
  33
  34 #include <simgear/compiler.h>
  35 // There is probably a better include than sg_geodesy to get the SG_NM_TO_METER...
  36 #include <simgear/math/sg_geodesy.hxx>
  37 #include <simgear/debug/logstream.hxx>
  38 #include <simgear/structure/SGReferenced.hxx>
  39 #include <simgear/structure/SGSharedPtr.hxx>
  40
  41 #include <string>
  42 #include <vector>
  43
  44 using std::string;
  45 using std::vector;
  46
  47
  48 typedef vector<int> intVec;
  49 typedef vector<int>::iterator intVecIterator;
  50
  51
  52 class FGAIFlightPlan;  // forward reference
  53 class FGGroundNetwork; // forward reference
  54 class FGAIAircraft;    // forward reference
  55 class FGAirportDynamics;
  56
  57 /**************************************************************************************
  58  * class FGATCInstruction
  59  * like class FGATC Controller, this class definition should go into its own file
  60  * and or directory... For now, just testing this stuff out though...
  61  *************************************************************************************/
  62 class FGATCInstruction
  63 {
  64 private:
  65   bool holdPattern;
  66   bool holdPosition;
  67   bool changeSpeed;
  68   bool changeHeading;
  69   bool changeAltitude;
  70   bool resolveCircularWait;
  71
  72   double speed;
  73   double heading;
  74   double alt;
  75 public:
  76
  77   FGATCInstruction();
  78   bool hasInstruction   ();
  79   bool getHoldPattern   () { return holdPattern;    };
  80   bool getHoldPosition  () { return holdPosition;   };
  81   bool getChangeSpeed   () { return changeSpeed;    };
  82   bool getChangeHeading () { return changeHeading;  };
  83   bool getChangeAltitude() { return changeAltitude; };
  84
  85   double getSpeed       () { return speed; };
  86   double getHeading     () { return heading; };
  87   double getAlt         () { return alt; };
  88
  89   bool getCheckForCircularWait() { return resolveCircularWait; };
  90
  91   void setHoldPattern   (bool val) { holdPattern    = val; };
  92   void setHoldPosition  (bool val) { holdPosition   = val; };
  93   void setChangeSpeed   (bool val) { changeSpeed    = val; };
  94   void setChangeHeading (bool val) { changeHeading  = val; };
  95   void setChangeAltitude(bool val) { changeAltitude = val; };
  96
  97   void setResolveCircularWait (bool val) { resolveCircularWait = val; };
  98
  99   void setSpeed       (double val) { speed   = val; };
 100   void setHeading     (double val) { heading = val; };
 101   void setAlt         (double val) { alt     = val; };
 102 };
 103
 104
 105
 106
 107
 108 /**************************************************************************************
 109  * class FGTrafficRecord
 110  *************************************************************************************/
 111 class FGTrafficRecord
 112 {
 113 private:
 114   int id, waitsForId;
 115   int currentPos;
 116   int leg;
 117   int frequencyId;
 118   int state;
 119   bool allowTransmission;
 120   time_t timer;
 121   intVec intentions;
 122   FGATCInstruction instruction;
 123   double latitude, longitude, heading, speed, altitude, radius;
 124   string runway;
 125   //FGAISchedule *trafficRef;
 126   FGAIAircraft *aircraft;
 127
 128
 129 public:
 130   FGTrafficRecord();
 131
 132   void setId(int val)  { id = val; };
 133   void setRadius(double rad) { radius = rad;};
 134   void setPositionAndIntentions(int pos, FGAIFlightPlan *route);
 135   void setRunway(string rwy) { runway = rwy;};
 136   void setLeg(int lg) { leg = lg;};
 137   int getId() { return id;};
 138   int getState() { return state;};
 139   void setState(int s) { state = s;}
 140   FGATCInstruction getInstruction() { return instruction;};
 141   bool hasInstruction() { return instruction.hasInstruction(); };
 142   void setPositionAndHeading(double lat, double lon, double hdg, double spd, double alt);
 143   bool checkPositionAndIntentions(FGTrafficRecord &other);
 144   int  crosses                   (FGGroundNetwork *, FGTrafficRecord &other);
 145   bool isOpposing                (FGGroundNetwork *, FGTrafficRecord &other, int node);
 146
 147   bool onRoute(FGGroundNetwork *, FGTrafficRecord &other);
 148
 149   bool getSpeedAdjustment() { return instruction.getChangeSpeed(); };
 150
 151   double getLatitude () { return latitude ; };
 152   double getLongitude() { return longitude; };
 153   double getHeading  () { return heading  ; };
 154   double getSpeed    () { return speed    ; };
 155   double getAltitude () { return altitude ; };
 156   double getRadius   () { return radius   ; };
 157
 158   int getWaitsForId  () { return waitsForId; };
 159
 160   void setSpeedAdjustment(double spd);
 161   void setHeadingAdjustment(double heading);
 162   void clearSpeedAdjustment  () { instruction.setChangeSpeed  (false); };
 163   void clearHeadingAdjustment() { instruction.setChangeHeading(false); };
 164
 165   bool hasHeadingAdjustment() { return instruction.getChangeHeading(); };
 166   bool hasHoldPosition() { return instruction.getHoldPosition(); };
 167   void setHoldPosition (bool inst) { instruction.setHoldPosition(inst); };
 168
 169   void setWaitsForId(int id) { waitsForId = id; };
 170
 171   void setResolveCircularWait()   { instruction.setResolveCircularWait(true);  };
 172   void clearResolveCircularWait() { instruction.setResolveCircularWait(false); };
 173
 174   string getRunway() { return runway; };
 175   //void setCallSign(string clsgn) { callsign = clsgn; };
 176   void setAircraft(FGAIAircraft *ref) { aircraft = ref;};
 177   void updateState() { state++; allowTransmission=true; };
 178   //string getCallSign() { return callsign; };
 179   FGAIAircraft *getAircraft() { return aircraft;};
 180   int getTime() { return timer; };
 181   int getLeg() { return leg; };
 182   void setTime(time_t time) { timer = time; };
 183
 184   bool pushBackAllowed();
 185   bool allowTransmissions() { return allowTransmission; };
 186   void suppressRepeatedTransmissions () { allowTransmission=false; };
 187   void allowRepeatedTransmissions () { allowTransmission=true; };
 188   void nextFrequency() { frequencyId++; };
 189   int  getNextFrequency() { return frequencyId; };
 190   intVec& getIntentions() { return intentions; };
 191   int getCurrentPosition() { return currentPos; };
 192 };
 193
 194 typedef vector<FGTrafficRecord> TrafficVector;
 195 typedef vector<FGTrafficRecord>::iterator TrafficVectorIterator;
 196
 197 typedef vector<time_t> TimeVector;
 198 typedef vector<time_t>::iterator TimeVectorIterator;
 199
 200 typedef vector<FGAIAircraft*> AircraftVec;
 201 typedef vector<FGAIAircraft*>::iterator AircraftVecIterator;
 202
 203 /***********************************************************************
 204  * Active runway, a utility class to keep track of which aircraft has
 205  * clearance for a given runway.
 206  **********************************************************************/
 207 class ActiveRunway
 208 {
 209 private:
 210   string rwy;
 211   int currentlyCleared;
 212   double distanceToFinal;
 213   TimeVector estimatedArrivalTimes;
 214   AircraftVec departureCue;
 215
 216 public:
 217   ActiveRunway(string r, int cc) { rwy = r; currentlyCleared = cc; distanceToFinal = 6.0 * SG_NM_TO_METER; };
 218
 219   string getRunwayName() { return rwy; };
 220   int    getCleared   () { return currentlyCleared; };
 221   double getApproachDistance() { return distanceToFinal; };
 222   //time_t getEstApproachTime() { return estimatedArrival; };
 223
 224   //void setEstApproachTime(time_t time) { estimatedArrival = time; };
 225   void addToDepartureCue(FGAIAircraft *ac) { departureCue.push_back(ac); };
 226   void setCleared(int number) { currentlyCleared = number; };
 227   time_t requestTimeSlot(time_t eta);
 228
 229    int getDepartureCueSize() { return departureCue.size(); };
 230    FGAIAircraft* getFirstAircraftInDepartureCue() { return departureCue.size() ? *(departureCue.begin()) : NULL; };
 231    void updateDepartureCue() { departureCue.erase(departureCue.begin()); }
 232 };
 233
 234 typedef vector<ActiveRunway> ActiveRunwayVec;
 235 typedef vector<ActiveRunway>::iterator ActiveRunwayVecIterator;
 236
 237 /**
 238  * class FGATCController
 239  * NOTE: this class serves as an abstraction layer for all sorts of ATC controllers.
 240  *************************************************************************************/
 241 class FGATCController
 242 {
 243 private:
 244      bool initialized;
 245
 246 protected:
 247   bool available;
 248   time_t lastTransmission;
 249
 250   double dt_count;
 251   osg::Group* group;
 252
 253   string formatATCFrequency3_2(int );
 254   string genTransponderCode(string fltRules);
 255   bool isUserAircraft(FGAIAircraft*);
 256
 257 public:
 258   typedef enum {
 259       MSG_ANNOUNCE_ENGINE_START,
 260       MSG_REQUEST_ENGINE_START,
 261       MSG_PERMIT_ENGINE_START,
 262       MSG_DENY_ENGINE_START,
 263       MSG_ACKNOWLEDGE_ENGINE_START,
 264       MSG_REQUEST_PUSHBACK_CLEARANCE,
 265       MSG_PERMIT_PUSHBACK_CLEARANCE,
 266       MSG_HOLD_PUSHBACK_CLEARANCE,
 267       MSG_ACKNOWLEDGE_SWITCH_GROUND_FREQUENCY,
 268       MSG_INITIATE_CONTACT,
 269       MSG_ACKNOWLEDGE_INITIATE_CONTACT,
 270       MSG_REQUEST_TAXI_CLEARANCE,
 271       MSG_ISSUE_TAXI_CLEARANCE,
 272       MSG_ACKNOWLEDGE_TAXI_CLEARANCE,
 273       MSG_HOLD_POSITION,
 274       MSG_ACKNOWLEDGE_HOLD_POSITION,
 275       MSG_RESUME_TAXI,
 276       MSG_ACKNOWLEDGE_RESUME_TAXI } AtcMsgId;
 277   typedef enum {
 278       ATC_AIR_TO_GROUND,
 279       ATC_GROUND_TO_AIR } AtcMsgDir;
 280   FGATCController();
 281   virtual ~FGATCController();
 282   void init();
 283
 284   virtual void announcePosition(int id, FGAIFlightPlan *intendedRoute, int currentRoute,
 285                                 double lat, double lon,
 286                                 double hdg, double spd, double alt, double radius, int leg,
 287                                 FGAIAircraft *aircraft) = 0;
 288   virtual void             signOff(int id) = 0;
 289   virtual void             updateAircraftInformation(int id, double lat, double lon,
 290                                                      double heading, double speed, double alt, double dt) = 0;
 291   virtual bool             hasInstruction(int id) = 0;
 292   virtual FGATCInstruction getInstruction(int id) = 0;
 293
 294   double getDt() { return dt_count; };
 295   void   setDt(double dt) { dt_count = dt;};
 296   void transmit(FGTrafficRecord *rec, AtcMsgId msgId, AtcMsgDir msgDir, bool audible);
 297   string getGateName(FGAIAircraft *aircraft);
 298   virtual void render(bool) = 0;
 299   virtual string getName()  = 0;
 300
 301
 302 private:
 303
 304  AtcMsgDir lastTransmissionDirection;
 305 };
 306
 307 /******************************************************************************
 308  * class FGTowerControl
 309  *****************************************************************************/
 310 class FGTowerController : public FGATCController
 311 {
 312 private:
 313   TrafficVector activeTraffic;
 314   ActiveRunwayVec activeRunways;
 315   FGAirportDynamics *parent;
 316
 317 public:
 318   FGTowerController(FGAirportDynamics *parent);
 319   virtual ~FGTowerController() {};
 320   virtual void announcePosition(int id, FGAIFlightPlan *intendedRoute, int currentRoute,
 321                                 double lat, double lon,
 322                                 double hdg, double spd, double alt, double radius, int leg,
 323                                 FGAIAircraft *aircraft);
 324   virtual void             signOff(int id);
 325   virtual void             updateAircraftInformation(int id, double lat, double lon,
 326                                   double heading, double speed, double alt, double dt);
 327   virtual bool             hasInstruction(int id);
 328   virtual FGATCInstruction getInstruction(int id);
 329
 330   virtual void render(bool);
 331   virtual string getName();
 332   bool hasActiveTraffic() { return activeTraffic.size() != 0; };
 333   TrafficVector &getActiveTraffic() { return activeTraffic; };
 334 };
 335
 336 /******************************************************************************
 337  * class FGStartupController
 338  * handle
 339  *****************************************************************************/
 340
 341 class FGStartupController : public FGATCController
 342 {
 343 private:
 344   TrafficVector activeTraffic;
 345   //ActiveRunwayVec activeRunways;
 346   FGAirportDynamics *parent;
 347
 348 public:
 349   FGStartupController(FGAirportDynamics *parent);
 350   virtual ~FGStartupController() {};
 351   virtual void announcePosition(int id, FGAIFlightPlan *intendedRoute, int currentRoute,
 352                                 double lat, double lon,
 353                                 double hdg, double spd, double alt, double radius, int leg,
 354                                 FGAIAircraft *aircraft);
 355   virtual void             signOff(int id);
 356   virtual void             updateAircraftInformation(int id, double lat, double lon,
 357                                   double heading, double speed, double alt, double dt);
 358   virtual bool             hasInstruction(int id);
 359   virtual FGATCInstruction getInstruction(int id);
 360
 361   virtual void render(bool);
 362   virtual string getName();
 363
 364   bool hasActiveTraffic() { return activeTraffic.size() != 0; };
 365   TrafficVector &getActiveTraffic() { return activeTraffic; };
 366
 367   // Hpoefully, we can move this function to the base class, but I need to verify what is needed for the other controllers before doing so.
 368   bool checkTransmissionState(int st, time_t now, time_t startTime, TrafficVectorIterator i, AtcMsgId msgId,
 369                                AtcMsgDir msgDir);
 370
 371 };
 372
 373 /******************************************************************************
 374  * class FGTowerControl
 375  *****************************************************************************/
 376 class FGApproachController : public FGATCController
 377 {
 378 private:
 379   TrafficVector activeTraffic;
 380   ActiveRunwayVec activeRunways;
 381   FGAirportDynamics *parent;
 382
 383 public:
 384   FGApproachController(FGAirportDynamics * parent);
 385   virtual ~FGApproachController() { };
 386   virtual void announcePosition(int id, FGAIFlightPlan *intendedRoute, int currentRoute,
 387                                 double lat, double lon,
 388                                 double hdg, double spd, double alt, double radius, int leg,
 389                                 FGAIAircraft *aircraft);
 390   virtual void             signOff(int id);
 391   virtual void             updateAircraftInformation(int id, double lat, double lon,
 392                                   double heading, double speed, double alt, double dt);
 393   virtual bool             hasInstruction(int id);
 394   virtual FGATCInstruction getInstruction(int id);
 395
 396   virtual void render(bool);
 397   virtual string getName();
 398
 399   ActiveRunway* getRunway(string name);
 400
 401   bool hasActiveTraffic() { return activeTraffic.size() != 0; };
 402   TrafficVector &getActiveTraffic() { return activeTraffic; };
 403 };
 404
 405
 406 #endif // _TRAFFIC_CONTROL_HXX