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 "FairTrackParam.h"
#include "CbmStsPoint.h"
#include "CbmMvdPoint.h"
#include "CbmStsTrack.h"
#include "FairTask.h"
#include "CbmMCTrack.h"
#include "CbmVertex.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