src/Autopilot/autopilot.cxx

   1 // autopilot.cxx - an even more flexible, generic way to build autopilots
   2 //
   3 // Written by Torsten Dreyer
   4 // Based heavily on work created by Curtis Olson, started January 2004.
   5 //
   6 // Copyright (C) 2004  Curtis L. Olson  - http://www.flightgear.org/~curt
   7 // Copyright (C) 2010  Torsten Dreyer - Torsten (at) t3r (dot) de
   8 //
   9 // This program is free software; you can redistribute it and/or
  10 // modify it under the terms of the GNU General Public License as
  11 // published by the Free Software Foundation; either version 2 of the
  12 // License, or (at your option) any later version.
  13 //
  14 // This program is distributed in the hope that it will be useful, but
  15 // WITHOUT ANY WARRANTY; without even the implied warranty of
  16 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  17 // General Public License for more details.
  18 //
  19 // You should have received a copy of the GNU General Public License
  20 // along with this program; if not, write to the Free Software
  21 // Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  22 //
  23
  24 #ifdef HAVE_CONFIG_H
  25 #  include <config.h>
  26 #endif
  27
  28 #include "functor.hxx"
  29 #include "predictor.hxx"
  30 #include "digitalfilter.hxx"
  31 #include "pisimplecontroller.hxx"
  32 #include "pidcontroller.hxx"
  33 #include "autopilot.hxx"
  34 #include "logic.hxx"
  35 #include "flipflop.hxx"
  36
  37 #include "Main/fg_props.hxx"
  38
  39 using std::map;
  40 using std::string;
  41
  42 using namespace FGXMLAutopilot;
  43
  44 class ComponentForge : public map<string,FunctorBase<Component> *> {
  45 public:
  46     virtual ~ ComponentForge();
  47 };
  48
  49 ComponentForge::~ComponentForge()
  50 {
  51     for( iterator it = begin(); it != end(); ++it )
  52         delete it->second;
  53 }
  54
  55 static ComponentForge componentForge;
  56
  57 Autopilot::Autopilot( SGPropertyNode_ptr rootNode, SGPropertyNode_ptr configNode ) :
  58   _name("unnamed autopilot"),
  59   _serviceable(true),
  60   _rootNode(rootNode)
  61 {
  62
  63   componentForge["pid-controller"]       = new CreateAndConfigureFunctor<PIDController,Component>();
  64   componentForge["pi-simple-controller"] = new CreateAndConfigureFunctor<PISimpleController,Component>();
  65   componentForge["predict-simple"]       = new CreateAndConfigureFunctor<Predictor,Component>();
  66   componentForge["filter"]               = new CreateAndConfigureFunctor<DigitalFilter,Component>();
  67   componentForge["logic"]                = new CreateAndConfigureFunctor<Logic,Component>();
  68   componentForge["flipflop"]             = new CreateAndConfigureFunctor<FlipFlop,Component>();
  69
  70   if( configNode == NULL ) configNode = rootNode;
  71
  72   int count = configNode->nChildren();
  73   for ( int i = 0; i < count; ++i ) {
  74     SGPropertyNode_ptr node = configNode->getChild(i);
  75     string childName = node->getName();
  76     if( componentForge.count(childName) == 0 ) {
  77       SG_LOG( SG_AUTOPILOT, SG_BULK, "unhandled element <" << childName << ">" << std::endl );
  78       continue;
  79     }
  80
  81     Component * component = (*componentForge[childName])(node);
  82     if( component->get_name().length() == 0 ) {
  83       std::ostringstream buf;
  84       buf <<  "unnamed_component_" << i;
  85       component->set_name( buf.str() );
  86     }
  87
  88     SG_LOG( SG_AUTOPILOT, SG_INFO, "adding  autopilot component \"" << childName << "\" as \"" << component->get_name() << "\"" );
  89     add_component(component);
  90   }
  91 }
  92
  93 Autopilot::~Autopilot()
  94 {
  95 }
  96
  97 void Autopilot::bind()
  98 {
  99   fgTie( _rootNode->getNode("serviceable", true)->getPath().c_str(), this,
 100     &Autopilot::is_serviceable, &Autopilot::set_serviceable );
 101 }
 102
 103 void Autopilot::unbind()
 104 {
 105   _rootNode->untie( "serviceable" );
 106 }
 107
 108 void Autopilot::add_component( Component * component )
 109 {
 110   if( component == NULL ) return;
 111
 112   // check for duplicate name
 113   std::string name = component->get_name();
 114   for( unsigned i = 0; get_subsystem( name.c_str() ) != NULL; i++ ) {
 115       std::ostringstream buf;
 116       buf <<  component->get_name() << "_" << i;
 117       name = buf.str();
 118   }
 119   if( name != component->get_name() )
 120     SG_LOG( SG_ALL, SG_WARN, "Duplicate autopilot component " << component->get_name() << ", renamed to " << name );
 121
 122   set_subsystem( name.c_str(), component );
 123 }
 124
 125 void Autopilot::update( double dt )
 126 {
 127   if( !_serviceable || dt <= SGLimitsd::min() )
 128     return;
 129   SGSubsystemGroup::update( dt );
 130 }