Fitter.h

#ifndef Fitter_h
#define Fitter_h

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

#include "Math/ProbFunc.h"


using namespace lcio;



class FitterException : public std::exception {
   
   
protected:
   std::string message ;
   
   FitterException(){  /*no_op*/ ; } 
   
public: 
   virtual ~FitterException() throw() { /*no_op*/; } 
   
   FitterException( const std::string& text ){
      message = "FitterException: " + text ;
   }
   
   virtual const char* what() const  throw() { return  message.c_str() ; } 
   
};



class TrackStatePlus{
   
   
   
public: 
   
   const TrackState* getTrackState() const {return _trackState; }
   double getChi2() const {return _chi2;}
   int getNdf() const {return _Ndf;}
   
   TrackStatePlus( const TrackState* trackState, double chi2, int Ndf ):
      _trackState( trackState ), _chi2( chi2 ), _Ndf( Ndf ){}
   
   
   
private: 
   
   const TrackState* _trackState;
   double _chi2;
   int _Ndf;
   
   
};


class Fitter{
   
   
public:
   
   Fitter( Track* track , MarlinTrk::IMarlinTrkSystem* trkSystem )throw( FitterException );
   Fitter( std::vector < TrackerHit* > trackerHits, MarlinTrk::IMarlinTrkSystem* trkSystem )throw( FitterException );
   Fitter( Track* track , MarlinTrk::IMarlinTrkSystem* trkSystem, int VXDFlag )throw( FitterException );  

   
   double getChi2Prob( int trackStateLocation ) throw( FitterException );
   double getChi2( int trackStateLocation ) throw( FitterException );
   int getNdf( int trackStateLocation ) throw( FitterException );
   
   //TODO: maybe add methods for custom points (location: TrackState::AtOther) In that case, the point would have to
   // be passed as well. (or only the point is passed)
   
   const TrackState* getTrackState( int trackStateLocation ) throw( FitterException );
   
   ~Fitter(){ 
      
      for( unsigned i=0; i<_trackStatesPlus.size(); i++ ){
         
         delete _trackStatesPlus[i]->getTrackState();
         delete _trackStatesPlus[i];
         
      }
      _trackStatesPlus.clear();
      
      delete _marlinTrk;
      
   }
   
private:
   
   void init_BField();

   const TrackStatePlus* getTrackStatePlus( int trackStateLocation ) throw( FitterException );

   void fit()throw( FitterException );

   void fitVXD()throw( FitterException );
   
   static float _bField;
   
   
   std::vector< TrackerHit* > _trackerHits;
   
   std::vector< const TrackStatePlus* > _trackStatesPlus;
   
   MarlinTrk::IMarlinTrkSystem* _trkSystem;
   
   MarlinTrk::IMarlinTrack* _marlinTrk;
   
   // No copy constructor or assignment needed so far. so private for safety
   // If they are needed, they need to be implemented in a clean way first!
   Fitter( const Fitter& f ){};
   Fitter& operator= ( Fitter const& f ){return *this;}
   
};

#endif