CgemMdcFitAlg.h

Go to the documentation of this file.

#include "GaudiKernel/Algorithm.h"
#include "GaudiKernel/NTuple.h"
#include "GaudiKernel/INTupleSvc.h"
#include "MdcCalibFunSvc/MdcCalibFunSvc.h"
#include "CgemCalibFunSvc/CgemCalibFunSvc.h"
#include "CgemGeomSvc/CgemGeomSvc.h"
#include "TrkFitter/TrkContextEv.h"
#include "MdcGeom/MdcDetector.h"
#include "MdcRecEvent/RecMdcTrack.h"
#include "MdcTrkRecon/MdcTrack.h"
#include "MdcTrkRecon/MdcTrackList.h"
//#include "KalFitAlg/helix/Helix.h"
 
class CgemMdcFitAlg:public Algorithm
{
  public:
    enum TrackType{ lineType = 2, circleType = 3, helixType = 5};
    CgemMdcFitAlg(const std::string& name, ISvcLocator* pSvcLocator);
    StatusCode initialize();
    StatusCode execute();
    StatusCode finalize();
    
    void updateTracks(int trackId, TrkRecoTrk* trkRecoTrk, RecMdcTrack* recMdcTrack, int trkStat);
    void compareTracks(MdcTrackList& mdcTrackList, double dzCut);
    TrkErrCode fit(RecMdcTrack* recMdcTrack, TrkRecoTrk* trkRecoTrk, TrackType trackType);
    //template <typename T>
    //TrkErrCode fit(double& dr,double& phi0, double& kappa, double& dz, double& tanl, RecMdcTrack* recMdcTrack, TrkRecoTrk* trkRecoTrk, T &trackfactory, TrackType trkType);
    TrkErrCode check(TrkRecoTrk* trkRecoTrk, TrackType trackType);
  private:
    StatusCode storeTracks(RecMdcTrackCol*& trackList_tds ,RecMdcHitCol*& hxitList_tds, MdcTrackList& mdcTrackList);
    StatusCode bookTuple();
    void getMcTruth();
    
    int     m_debug;
    int     m_tuple;
    int     m_debugArbHit;
    int     m_changeTDS;
    int     m_fit2D;
    int     m_filter;
    int     m_compareTrack;
    int     m_removeBadTrack;
    std::string m_evtFile;
 
    double  m_drCut;
    double  m_dzCut;
    double  m_qualityFactor;
    double  m_dropTrkDrCut;
    double  m_dropTrkDzCut;
    double  m_dropTrkNhitCut;
    double  m_dropTrkChi2Cut;
    double  m_dropTrkChi2NdfCut;
    double  m_bunchT0;
    static const TrkContextEv*       m_context;
    static const MdcCalibFunSvc*     m_mdcCalibFunSvc;
    static const CgemCalibFunSvc*    m_cgemCalibFunSvc;
    static const CgemGeomSvc*        m_cgemGeomSvc;
    static const MdcDetector*        m_mdcDetector;
    TrackType trackType;
    vector<MdcHit*>  mdcHitCol;
 
    int    t_nTrkMC;
    int    t_nMdcMcHit;
    double t_t0Truth;
    int    t_pidTruth[10];
    double t_costhetaMC[10];
    double t_phiMC[10];
    double t_qMC[10];
    double t_vzMC[10];
    double t_vrMC[10];
    double t_pxMC[10];
    double t_pyMC[10];
    double t_pzMC[10];
    double t_ptMC[10];
    double t_pMC[10];
 
    //---track parameters before fitting
    NTuple::Tuple*        ntuple_evt;
    NTuple::Item<int>      m_run;
    NTuple::Item<int>      m_event;
    NTuple::Item<int>      m_nTrkRec;
    NTuple::Array<int>    m_trkidRec;                             
    NTuple::Array<double> m_drRec   ;                             
    NTuple::Array<double> m_phi0Rec ;                             
    NTuple::Array<double> m_kappaRec;                             
    NTuple::Array<double> m_dzRec   ;                             
    NTuple::Array<double> m_tanlRec ;    
    NTuple::Array<int>     m_chargeRec  ;
    NTuple::Array<int>     m_statRec    ;
    NTuple::Array<int>     m_nhitsRec   ;
    NTuple::Array<int>     m_nclusterRec;
    NTuple::Array<int>     m_nsterRec   ;
    NTuple::Array<int>     m_ndofRec    ;
    NTuple::Array<double>  m_chi2Rec    ;    
    NTuple::Array<double>  m_pxRec      ;    
    NTuple::Array<double>  m_pyRec      ;    
    NTuple::Array<double>  m_pzRec      ;    
    NTuple::Array<double>  m_pxyRec     ;    
    NTuple::Array<double>  m_pRec       ;    
    NTuple::Array<double>  m_xRec       ;    
    NTuple::Array<double>  m_yRec       ;    
    NTuple::Array<double>  m_zRec       ;    
    NTuple::Array<double>  m_rRec       ;    
    NTuple::Array<double>  m_thetaRec   ;    
    NTuple::Array<double>  m_phiRec     ;    
    NTuple::Array<double>  m_fiTermRec  ;    
    
    //---track parameters after fitting
    NTuple::Item<int>     m_nTrkFit;
    NTuple::Array<int>    m_trkidFit;                             
    NTuple::Array<double> m_drFit   ;                             
    NTuple::Array<double> m_phi0Fit ;                             
    NTuple::Array<double> m_kappaFit;                             
    NTuple::Array<double> m_dzFit   ;                             
    NTuple::Array<double> m_tanlFit ;    
    NTuple::Array<int>     m_chargeFit  ;
    NTuple::Array<int>     m_statFit    ;
    NTuple::Array<int>     m_nhitsFit   ;
    NTuple::Array<int>     m_nclusterFit;
    NTuple::Array<int>     m_nsterFit   ;
    NTuple::Array<int>     m_ndofFit    ;
    NTuple::Array<double>  m_chi2Fit    ;    
    NTuple::Array<double>  m_pxFit      ;    
    NTuple::Array<double>  m_pyFit      ;    
    NTuple::Array<double>  m_pzFit      ;    
    NTuple::Array<double>  m_pxyFit     ;    
    NTuple::Array<double>  m_pFit       ;    
    NTuple::Array<double>  m_xFit       ;    
    NTuple::Array<double>  m_yFit       ;    
    NTuple::Array<double>  m_zFit       ;    
    NTuple::Array<double>  m_rFit       ;    
    NTuple::Array<double>  m_thetaFit   ;    
    NTuple::Array<double>  m_phiFit     ;    
    NTuple::Array<double>  m_fiTermFit  ;    
    
    //--track parameters of McParticle
    NTuple::Item<int>     m_nTrkMC;
    NTuple::Array<int>    m_trkidMC;                             
    NTuple::Array<double> m_drMC   ;                             
    NTuple::Array<double> m_phi0MC ;                             
    NTuple::Array<double> m_kappaMC;                             
    NTuple::Array<double> m_dzMC   ;                             
    NTuple::Array<double> m_tanlMC ;    
    NTuple::Array<double> m_qMC         ;
    NTuple::Array<double> m_costhetaMC  ;
    NTuple::Array<double> m_phiMC       ;
    NTuple::Array<double> m_vzMC        ;
    NTuple::Array<double> m_vrMC        ;
    NTuple::Array<double> m_pxMC        ;
    NTuple::Array<double> m_pyMC        ;
    NTuple::Array<double> m_pzMC        ;
    NTuple::Array<double> m_ptMC        ;
    NTuple::Array<double> m_pMC         ;
 
    //NTuple::Tuple*      ntuple_hit;
};