RecoTrack.h

#ifndef RecoTrack_h
#define RecoTrack_h

#include "EVENT/Track.h"
#include "lcio.h"
#include "MarlinTrk/IMarlinTrkSystem.h"
#include "MarlinTrk/IMarlinTrack.h"

using namespace lcio;


enum TrackType { COMPLETE , COMPLETE_PLUS , INCOMPLETE , INCOMPLETE_PLUS , GHOST , LOST };


class TrueTrack;


class RecoTrack{
   
   
public:
   
   RecoTrack( Track* track, MarlinTrk::IMarlinTrkSystem* trkSystem ): _track( track ), _trkSystem( trkSystem )
      { _type = GHOST; } 
   
   Track* getTrack(){ return _track; }
   
   
   TrackType getType() const { return _type; }
   
   void setType( TrackType type ){ _type = type; }
   
   
   std::vector< const TrueTrack* > getTrueTracks() const { return _trueTracks; }
   
   void addTrueTrack( TrueTrack* trueTrack ){ _trueTracks.push_back( trueTrack ); }
   
   
   
   std::string getRecoTrackInfo() const;
   
   
   static std::string cellIDInfo( TrackerHit* hit );
   
   static std::string positionInfo( TrackerHit* hit );
   
   
   
   
protected:
   
   Track* _track;
   std::vector< const TrueTrack* > _trueTracks;
   TrackType _type;
   
   
   MarlinTrk::IMarlinTrkSystem* _trkSystem; // for fitting
   
   
   
   
   
};



#endif