Automaton.h

#ifndef Automaton_h
#define Automaton_h

#include <vector>
#include "KiTrack/Segment.h"
#include "Criteria/ICriterion.h"



namespace KiTrack{
   
   
   class Automaton{
      
      
     
   public:
      
      Automaton(): _nConnections(0){}
      
      ~Automaton();
      
      
      void addSegment ( Segment* segment );
      
      void lengthenSegments();
      
      void addCriterion ( ICriterion* criterion ){ _criteria.push_back( criterion ); }
      void addCriteria ( std::vector< ICriterion* > criteria ){ _criteria.insert( _criteria.end() , criteria.begin() , criteria.end() ); }
      void clearCriteria() { _criteria.clear(); };
      
      void doAutomaton();
      
      void cleanBadStates();
      
      void cleanBadConnections();
      
      void resetStates();
      
      
      
      //std::vector < std::vector< IHit* > > getTracks( unsigned minHits = 3 );
      std::vector < std::vector< IHit* > > getTracks( unsigned minHits = 2 ); // YV, 2 mini-vector hits can form a track     
      
      //std::vector < std::vector< IHit* > > getTracksOfSegment ( Segment* segment, std::vector< IHit* > hits , unsigned minHits = 3 );
      std::vector < std::vector< IHit* > > getTracksOfSegment ( Segment* segment, std::vector< IHit* > hits , unsigned minHits = 2 );    // YV, 2 mini-vector hits can form a track   
      
      std::vector <const Segment*> getSegments() const;
      
      unsigned getNumberOfConnections(){ return _nConnections; }
      
   private:
      
      std::vector < std::list < Segment* > > _segments;
      
      std::vector < ICriterion* > _criteria;
      
      unsigned _nConnections;
      
      
      
   };  
   
   
   
   
   
}






#endif