StepWiseHelixGeneralBrokenLineInterfaceProcessor.h

Go to the documentation of this file.

#ifndef STEPWISEHELIXGENERALBROKENLINEINTERFACEPROCESSOR_H
#define STEPWISEHELIXGENERALBROKENLINEINTERFACEPROCESSOR_H

#include "SimpleHelixTrackModel.h"
#include "GeneralBrokenLineInterfaceHelpers.h"

#include <marlin/Processor.h>
#include <marlin/Global.h>

#include <lcio.h>
#include <string>

// GBL:
#include "GblTrajectory.h"
#include "MilleBinary.h"

// LCIO:
#include <EVENT/Track.h>
#include "EVENT/TrackerHit.h"

#include "TFile.h"
#include "TTree.h"
#include "TMath.h"

namespace marlintpc {

typedef std::pair<int, double> indexArcPair;
typedef std::pair<double, double> arcVarPair;


class StepWiseHelixGeneralBrokenLineInterfaceProcessor: public marlin::Processor {
        public:

                virtual Processor* newProcessor() {
                        return new StepWiseHelixGeneralBrokenLineInterfaceProcessor;
                }

                StepWiseHelixGeneralBrokenLineInterfaceProcessor();

                virtual void init();

                virtual void processRunHeader(EVENT::LCRunHeader* run);

                virtual void processEvent(EVENT::LCEvent* evt);

                virtual void check(EVENT::LCEvent* evt);

                virtual void end();
        protected:
                std::string _inputCollectionName; 
                std::string _outputTrackCollectionName; 
                double _bfieldScaleFactor; 
                bool _outputIsPersistent;  
                std::string _fitOptions; 
                bool _writeMillepedeOut; 
                std::string _fileNameMillepedeFile; 
                int _milleCalcMethod; 
                int _milleMinHits; 
                bool _encodedModuleID; 
                double _maxStep; 
                double _x0PerUnitLength; 
                bool _thickScatterer; 
                double _defaultMomentum; 

        private:
                gbl::GblTrajectory* _trajectory; 
                gbl::MilleBinary* _milleBinary; 
                bool _curvature; 
                std::vector<unsigned int> _theLabels; 
                TVectorD* _correctionVector; 
                TMatrixDSym* _covarianceMatrix; 
                int _ndf; 
                double _chisquare; 
                double _lostweight; 
                double _refPointS; 
                unsigned int _refPointIndex; 
                double _Xcenter; 
                double _Ycenter; 

                void calcLineSeed(const std::vector<TrackerHit*>, const double*, double&, double&, double&, double&, double&) const;
};

} // namespace marlintpc
#endif // STEPWISEHELIXGENERALBROKENLINEINTERFACEPROCESSOR_H