CbmKFTrackFitQA.h

Go to the documentation of this file.

/*
 *====================================================================
 *
 *  CBM KF Track Quality
 *  
 *  Authors: M.Zyzak
 *
 *  e-mail : 
 *
 *====================================================================
 *
 *  KF Fit performance 
 *
 *====================================================================
 */
 
#ifndef _CbmKFTrackFitQA_h_
#define _CbmKFTrackFitQA_h_
 
#include "FairTask.h"
 
#include "CbmKFTrack.h"
#include "CbmMCTrack.h"
#include "CbmStsPoint.h"
#include "CbmMvdPoint.h"
#include "CbmTrackMatch.h"
#include "CbmStsTrack.h"
#include "CbmStsHit.h"
#include "CbmStsCluster.h"
#include "CbmStsDigi.h"
#include "CbmMvdHit.h"
#include "CbmMvdHitMatch.h"
 
#include "CbmKFTrErrMCPoints.h"
 
#include "TClonesArray.h"
#include "TH1.h"
#include "TProfile.h"
 
#include <iostream>
#include <vector>
 
class CbmKFTrackFitQA :public FairTask
{
 public:
 
  CbmKFTrackFitQA();
  ~CbmKFTrackFitQA();
 
  void StsHitMatch();
  void FillHitHistos();
  void FillHistoAtParticleVertex(CbmMCTrack *track_mc, CbmKFTrack *track_kf);
  void FillHistoAtFirstPoint(CbmKFTrErrMCPoints *mc_points, CbmMCTrack *track_mc, CbmKFTrack *track_kf);
  void Write();
  void FindBackTracks(CbmKFTrErrMCPoints *mc_points, CbmMCTrack *track_mc, CbmKFTrack *track_kf, int iEvent);
  void FindBackTracks();
  void Save();
 
  void SetOutFile(TString outname) {outfileName = outname;}
 
  virtual InitStatus ReInit();
  virtual InitStatus Init();
  void Exec(Option_t * option);
  void Finish();
 
 private:
// Arrays of points, tracks, etc
 
  TClonesArray *listStsPts;
  TClonesArray *listMvdPts;
  TClonesArray *listMCTracks;
  TClonesArray *listStsTracksMatch;
  TClonesArray *listStsTracks;
  TClonesArray *listStsHits;
  TClonesArray *listMvdHits;
  TClonesArray *listMvdHitMatches;
  TClonesArray *listStsClusters;
  TClonesArray *listStsDigi;
 
 
// Names of files
  TString outfileName;
 
  vector<int> vStsHitMatch;
 
// histograms
 
  //hit parameters
  TH1F *res_STShit_x;
  TH1F *res_STShit_y;
  TH1F *pull_STShit_x;
  TH1F *pull_STShit_y;
 
  TH1F *res_MVDhit_x;
  TH1F *res_MVDhit_y;
  TH1F *pull_MVDhit_x;
  TH1F *pull_MVDhit_y;
 
 
  //track parameters at the particle vertex
  TH1F *res_AtPV_x;
  TH1F *res_AtPV_y;
  TH1F *res_AtPV_tx;
  TH1F *res_AtPV_ty;
  TH1F *res_AtPV_qp;
 
  TH1F *pull_AtPV_x;
  TH1F *pull_AtPV_y;
  TH1F *pull_AtPV_tx;
  TH1F *pull_AtPV_ty;
  TH1F *pull_AtPV_qp;
 
  //track parameters at the first MC point of the track
  TH1F *res_AtFP_x;
  TH1F *res_AtFP_y;
  TH1F *res_AtFP_tx;
  TH1F *res_AtFP_ty;
  TH1F *res_AtFP_qp;
 
  TH1F *pull_AtFP_x;
  TH1F *pull_AtFP_y;
  TH1F *pull_AtFP_tx;
  TH1F *pull_AtFP_ty;
  TH1F *pull_AtFP_qp;
 
  //Q detertmination quality
  TProfile *q_QA;
  TProfile *dp_p;
 
  TH1F *ggg;
 
  int Nback;
  //FILE *fBack;
 
  ClassDef(CbmKFTrackFitQA,1);
 private:
  CbmKFTrackFitQA(const CbmKFTrackFitQA&);
  void operator=(const CbmKFTrackFitQA&);
};
 
#endif // _CbmKFTrackFitQA_h_