CgemMilleAlign.h

Go to the documentation of this file.

/* Millepede matrix method for alignment, author: Volker Blobel  */
/* First used in BESIII drift chamber by Wu Linghui ([email protected]) */
#ifndef INICGEMALIGN_H
#define INICGEMALIGN_H
 
#include "GaudiKernel/Algorithm.h"
#include "GaudiKernel/NTuple.h"
#include "GaudiKernel/INTupleSvc.h"
#include "CgemAlignAlg/CgemAlignBase.h"
#include "CgemAlignAlg/Millepede.h"
#include "CgemAlignAlg/CgemAlignment.h"
//#include "CgemGeomSvc/CgemUtilitySvc.h"
 
#include "CgemGeomSvc/ICgemGeomSvc.h"
#include "CgemCalibFunSvc/ICgemCalibFunSvc.h"
// I need to prepare it
 
#include "TROOT.h"
#include "TFile.h"
#include "TObjArray.h"
#include "TH1F.h"
#include "TF1.h"
#include "TTree.h"
#include "TFolder.h"
#include "TGraph.h"
 
#include <fstream>
 
class CgemMilleAlign: public CgemAlignBase{
public: 
     CgemMilleAlign();
     ~CgemMilleAlign();
     //void initialize(TObjArray* hlist, TTree* tree, ICgemGeomSvc* cgemGeomSvc, ICgemCalibFunSvc* cgemFunSvc);
     void initialize(TObjArray* hlist,  ICgemGeomSvc* cgemGeomSvc, ICgemCalibFunSvc* cgemFunSvc);
     void setParam(CgemAliParams& param);
     bool fillHist(CgemAliEvent* event, CgemAlignPar* alignPar);
     bool updateAlignPar(CgemAlignPar* alignPar);
     void clear();
 
private:
     int getAlignParId(int lay, int iparHit);
     bool getDeriLoc(int Intra, int ipar, int lay, int sheet, int dir, double mPhi, double mV, double laypos[], HepVector helix, HepSymMatrix &helixErr, double &deri_phi, double &deri_v);
     bool getDeriGlo(int Intra, int iparHit, int iparGB, int lay, int sheet, int dir, double mPhi, double mV, double laypos[], HepVector helix, HepSymMatrix &helixErr,double &deri_phi, double &deri_v);
 
     bool iniGeoAlignPar();
 
     NTuple::Tuple*      m_tuple;
     NTuple::Item<double>            run;
 
     ICgemGeomSvc* m_cgemGeomSvc;
     ICgemCalibFunSvc* m_cgemFunSvc;
     
     CgemAliParams m_param;
     TObjArray* m_hlist;
     //CgemUtilitySvc*   m_cgemUtilitySvc;
 
     double m_resiCut_phi[LAYERNMAX];
     double m_resiCut_v[LAYERNMAX];
     double m_docaCut[LAYERNMAX];
     
     //double m_p[20][15];
     //double m_dxini[NEP];
     //double m_dyini[NEP];
     //double m_rzini[NEP];
     int m_GeoMethod;
     bool Plot_der;
     bool debug;
     bool re_3lay;
     bool record_pos;
     bool use_phi;
     bool use_v;
     bool mul_R;
     bool mul_Ste_ang;
     int nTrack_tot;
 
     int count[30];
     double m_r[6];
     double m_z[6];
     double Ste_ang[6];
     double sig_phi[6];
     double sig_v[6];
 
     double cut_derLC_Phi[5][2] ;
     double cut_derLC_V[5][2]   ;
     double cut_derGB_Phi[4][2] ;
     double cut_derGB_V[4][2]   ;
     
     double ini_alig_par[6][6]   ;
 
    CgemMidDriftPlane* m_middriftplane;
    CgemGeoAlign* m_geoalign;
 
     /* for millepede */
     Millepede* m_pMilleAlign;
     int m_npar;
     bool m_dofs[NDOFALIGN];
     double m_sigm[NDOFALIGN*NLAYER];
     //double m_sigm[NDOFALIGN];
     int m_nGloHit;     /* number of global pars for each hit */
     int m_nglo_on_lay;
     int m_nloc;
     int m_nlay;
     std::vector<double> m_par;
     std::vector<double> m_error;
     std::vector<double> m_pull;
     std::vector<double> m_derGB_Phi;
     std::vector<double> m_derGB_V;
     std::vector<double> m_derNonLin_Phi;
     std::vector<double> m_derNonLin_V;
     std::vector<double> m_derLC_Phi;
     std::vector<double> m_derLC_V;
 
     /* histograms */
     TH1F* m_hresAll_phi;
     TH1F* m_hresAll_v;
     TH1F* m_hresAll_z;
     //TH1F* m_hresInn;
     //TH1F* m_hresStp;
     //TH1F* m_hresOut;
     TH1F* m_hresLay_phi[LAYERNMAX];
     TH1F* m_hresLay_v[LAYERNMAX];
     TH1F* m_hresLay_z[LAYERNMAX];
 
     //TH1F* m_hresAllRec;
     //TH1F* m_hresLayRec[LAYERNMAX];
 
     TH1F* m_hddoca;        /* for debug */
     TH1F* m_hddocaLay[LAYERNMAX];      /* for debug */
 
};
inline void CgemMilleAlign::setParam(CgemAliParams& param){
     CgemAlignBase::setParam(param);
     m_param = param;
}
#endif