CbmStsFitPerformanceTask.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----              CbmStsFitPerformanceTask header file                -----
// -----               Created 27/01/05  by E. Kryshen                 -----
// -------------------------------------------------------------------------
 
 
#ifndef CBMSTSFITPERFORMANCE_H
#define CBMSTSFITPERFORMANCE_H
 
#include "CbmStsKFTrackFitter.h"
 
#include "CbmMCTrack.h"
#include "CbmMvdPoint.h"
#include "CbmStsPoint.h"
#include "CbmStsTrack.h"
#include "CbmVertex.h"
#include "FairTask.h"
#include "FairTrackParam.h"
 
#include "TClonesArray.h"
 
#include "TH1D.h"
#include "TH2D.h"
 
class CbmStsFitPerformanceTask : public FairTask {
 
public:
  CbmStsFitPerformanceTask(const char* name = "CbmStsFitPerformanceTask",
                           Int_t iVerbose   = 1);
 
 
  ~CbmStsFitPerformanceTask();
 
 
  InitStatus Init();
 
  InitStatus ReInit();
 
 
  void Exec(Option_t* option);
 
 
  void Finish();
 
  void FillMCStateVectors(CbmStsPoint* point, Double_t mc[], Bool_t out = 0);
  void FillMCStateVectors(CbmMvdPoint* point, Double_t mc[], Bool_t out = 0);
 
  void FillTrackHisto(const Double_t mc[6], CbmStsTrack* track, TH1D* hist[10]);
 
  void FillVertexHisto(TVector3& mc, CbmVertex* V, TH1D* hist[8]);
 
  Bool_t IsLong(CbmStsTrack* track);
 
  void DoTrackAnalysis(const bool ok = 1) { fTrackAnalysis = ok; }
  void DoVertexAnalysis(const bool ok = 1) { fVertexAnalysis = ok; }
  void DoD0Analysis(const bool ok = 1) { fD0Analysis = ok; }
 
  Double_t GetCharge(CbmMCTrack* mcTrack);
 
protected:
  void CreateTrackHisto(TH1D* hist[10], const char* name, const char* title);
  void CreateVertexHisto(TH1D* hist[9], const char* name, const char* title);
  void CreateD0Histo(TH1D* hist[14], const char* name, const char* title);
 
  Int_t fEvent;
  Bool_t fVertexAnalysis, fD0Analysis, fTrackAnalysis;
 
  TClonesArray* fMCTrackArray;      
  TClonesArray* fStsPointArray;     
  TClonesArray* fMvdPointArray;     
  TClonesArray* fRecStsTrackArray;  
  TClonesArray* fStsHitArray;       
  TClonesArray* fMvdHitArray;       
  CbmVertex* fPrimaryVertex;        
  TClonesArray* fSTSTrackMatch;     
 
  TH1D* fhChi2;  // x=chi2; y=entries for all
  TH1D* fhProb;  // x=Prob function; y=entries for all
  TH1D* fhDP;
  TH2D* fhDP2;
  TH1D* fhDsP;
  TH2D* fhDsP2;
 
  TH1D *fhZMCf, *fhZMCl;      // z first/last of MC track
  TH1D *fhZRecof, *fhZRecol;  // z first/last of Reco track
 
  TH2D *fhRes_vs_Mom_f, *fhRes_vs_Mom_l;  // resolution vs momentum
  TH2D*
    fhExtraTracks2ndMVD;  // extra tracks not detected in the second mvd chamber
 
  TH1D* fhFrst[10];
  TH1D* fhMid[10];
  TH1D* fhLast[10];
  TH1D* fhImp[10];
  TH1D* fhImpPi[10];
  TH1D* fhVfit[10];
  TH1D* fhVtx[9];
  TH1D* fhV[13][9];
  TH2D* fhPq;
  TH1D* fhD0[4][14];
 
  TH1D* fhHitDensity[10];
  TH1D* fhTrackDensity[8];
  TH1D* fhTrackDensity0L;
 
  TDirectory* histodir;
  CbmStsKFTrackFitter fFitter;
 
  ClassDef(CbmStsFitPerformanceTask, 1);
 
private:
  CbmStsFitPerformanceTask(const CbmStsFitPerformanceTask&);
  void operator=(const CbmStsFitPerformanceTask&);
};
 
#endif