TrueTrack.h

#ifndef TrueTrack_h
#define TrueTrack_h


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

#include "RecoTrack.h"

using namespace lcio;


class TrueTrack{
   
public:
   
   
   TrueTrack( Track* trueTrack , MCParticle* mcp , MarlinTrk::IMarlinTrkSystem* trkSystem):
   _trueTrack(trueTrack), _mcp(mcp), _trkSystem(trkSystem) {}
   
   Track* getTrueTrack() const { return _trueTrack; }
   
   const MCParticle* getMCP() const { return _mcp; }
   
   std::vector< const RecoTrack* > getRecoTracks() const{ return _recoTracks; }
   
   
   void addRecoTrack( RecoTrack* recoTrack ){ _recoTracks.push_back( recoTrack ); }
   
   
   
   
   bool isLost() const;
   
   bool isFoundCompletely() const;
   
   bool completeVersionExists() const;
   

   
   std::string getMCPInfo() const;
   
   std::string getTrueTrackInfo() const;
   
   std::string getRelatedTracksInfo() const;
   
   std::string getFoundInfo() const;
   
   std::string getCutInfo() const;
   
   unsigned getNumberOfTracksWithType( TrackType type ) const;   
   
   std::vector< std::string > getCuts() const {return _cuts; }
   
   void addCut( std::string cut ){ _cuts.push_back( cut ); }
   
private:
   
   Track* _trueTrack;
   MCParticle* _mcp;
   std::vector< const RecoTrack* > _recoTracks;
   std::vector< std::string > _cuts;
   
   MarlinTrk::IMarlinTrkSystem* _trkSystem; // for fitting
   
   
};


#endif