PndLmdQATask.h

Go to the documentation of this file.

//****************************************************************************
//*                   This file is part of PandaRoot.                        *
//*                                                                          *
//*            PandaRoot is distributed under the terms of the               *
//*              GNU General Public License (GPL) version 3,                 *
//*                 copied verbatim in the file "LICENSE".                   *
//*                                                                          *
//*  Copyright (C) 2006 - 2024 FAIR GmbH and copyright holders of PandaRoot  *
//*     The copyright holders are listed in the file "COPYRIGHTHOLDERS".     *
//*               The authors are listed in the file "AUTHORS".              *
//****************************************************************************

//-----------------------------------------------------------
//
// Description:
//      Task to generate and store some useful histograms
//
// Author List:
//  Anastasia Karavdina
//      Mathias Michel
//
//-----------------------------------------------------------

#ifndef PNDLMDQATASK_H
#define PNDLMDQATASK_H

// Base Class Headers ----------------
#include "FairTask.h"

// Collaborating Class Headers -------
#include <map>
#include "TString.h"
#include "TH2.h"
#include "TH1.h"

// Collaborating Class Declarations --
class TClonesArray;
class TGeoManager;

class PndLmdQATask : public FairTask {

 public:
  // Constructors/Destructors ---------
  PndLmdQATask(TString mcHitBranch = "LMDPoint", TString mcTrkBranch = "MCTrack", TString clusterBranch = "LMDPixelClusterCand", TString digiBrunch = "LMDPixelDigis",
               TString hitBranch = "LmdHits", TString TrkCandBranch = "LMDTrackCand", TString trackBranch = "LMDTrack", TString geaneBranch = "GeaneTrackFinal",
               TString outFile = "tmpOutput/QA.root");
#ifndef __CINT__
  PndLmdQATask(const PndLmdQATask &) = delete;
  PndLmdQATask &operator=(const PndLmdQATask &) = delete;
#endif
  virtual ~PndLmdQATask();
  void SetVerboseLevel(int verbose) { verboseLevel = verbose; };
  void WriteHists();
  virtual InitStatus Init();

  virtual void FinishTask();

  virtual void Exec(Option_t *opt);

 protected:
  int verboseLevel;
  //  double fPlab;
  // Input Data------------
  TClonesArray *fmcHitArray;
  TClonesArray *fmcTrkArray;
  TClonesArray *fHitArray;
  TClonesArray *fClusterArray;
  TClonesArray *fDigiArray;
  TClonesArray *fTrkCandArray;
  TClonesArray *fTrkArray;
  TClonesArray *fGeaneArray;

  TString fmcHitName;
  TString fmcTrkName;
  TString fHitName;
  TString fDigiName;
  TString fClusterName;
  TString fTrkCandName;
  TString fTrkName;
  TString fGeaneName;

  // storage / output
  TString foutFile;
  Double_t trkpar[4];
  int tot;
  int all, uneff, mistrk, ghosttrk;
  Int_t fEvent;

  // Analysis Functions
  void ResoAndPulls();
  bool HitReco();
  // void Acceptance(); TODO

  // histos ResoAndPulls
  // Near IP
  TH1 *hResMom;
  TH1 *hErrMom;
  TH1 *hPullMom;
  TH1 *hResTheta;
  TH2 *hResTheta_th; // resolution vs. theta
  TH2 *hResTheta_ph; // resolution vs. phi
  TH1 *hErrTheta;
  TH1 *hPullTheta;
  TH1 *hResPhi;
  TH1 *hErrPhi;
  TH1 *hPullPhi;
  TH1 *hResPointPx;
  TH1 *hErrPointPx;
  TH1 *hPullPointPx;

  TH1 *hResPointPy;
  TH1 *hErrPointPy;
  TH1 *hPullPointPy;

  TH1 *hResPointPz;
  TH1 *hErrPointPz;
  TH1 *hPullPointPz;

  TH1 *hResPointX;
  TH1 *hPullPointX;
  TH1 *hResPointY;
  TH1 *hPullPointY;
  TH1 *hResPointZ;
  TH1 *hPullPointZ;

  /*   //QA candidates */
  /*  TH1 *hPullLikePointX; */
  /*  TH1 *hPullLikePointY; */
  /*  TH1 *hPullLikePointZ; */
  /*  TH2 *hPullLikeX_Zmc; */
  /*  TH2 *hPullLikeY_Zmc; */
  /*  TH2 *hPullLikeZ_Zmc; */
  /*  TH2 *hPullLikeTh_Zmc; */
  /*  TH2 *hPullLikePh_Zmc; */
  /*  TH2 *hPullLikeP_Zmc; */
  /*  TH2 *hXrecZmc; */
  /*  TH2 *hYrecZmc; */
  /*  TH2 *hZrecZmc; */
  /*  TH2 *hThrecZmc; */
  /*  TH2 *hPhrecZmc; */
  /*  TH2 *hPrecZmc; */
  /* TH2 *hResXrecZmc; */
  /*  TH2 *hResYrecZmc; */
  /*  TH2 *hResZrecZmc; */
  /*  TH2 *hResThrecZmc; */
  /*  TH2 *hResPhrecZmc; */
  /*  TH2 *hResPrecZmc; */
  /*  TH2 *herrXrecZmc; */
  /*  TH2 *herrYrecZmc; */
  /*  TH2 *herrZrecZmc; */
  /*  TH2 *herrThrecZmc; */
  /*  TH2 *herrPhrecZmc; */
  /*  TH2 *herrPrecZmc; */

  /*  TH2 *hPullLikeX_Xmc; */
  /*  TH2 *hPullLikeY_Xmc; */
  /*  TH2 *hPullLikeZ_Xmc; */
  /*  TH2 *hPullLikeTh_Xmc; */
  /*  TH2 *hPullLikePh_Xmc; */
  /*  TH2 *hPullLikeP_Xmc; */
  /*  TH2 *hXrecXmc; */
  /*  TH2 *hYrecXmc; */
  /*  TH2 *hZrecXmc; */
  /*  TH2 *hThrecXmc; */
  /*  TH2 *hPhrecXmc; */
  /*  TH2 *hPrecXmc; */
  /* TH2 *hResXrecXmc; */
  /*  TH2 *hResYrecXmc; */
  /*  TH2 *hResZrecXmc; */
  /*  TH2 *hResThrecXmc; */
  /*  TH2 *hResPhrecXmc; */
  /*  TH2 *hResPrecXmc; */
  /*  TH2 *herrXrecXmc; */
  /*  TH2 *herrYrecXmc; */
  /*  TH2 *herrZrecXmc; */
  /*  TH2 *herrThrecXmc; */
  /*  TH2 *herrPhrecXmc; */
  /*  TH2 *herrPrecXmc; */

  /* TH2 *hPullLikeX_Ymc; */
  /*  TH2 *hPullLikeY_Ymc; */
  /*  TH2 *hPullLikeZ_Ymc; */
  /*  TH2 *hPullLikeTh_Ymc; */
  /*  TH2 *hPullLikePh_Ymc; */
  /*  TH2 *hPullLikeP_Ymc; */
  /*  TH2 *hXrecYmc; */
  /*  TH2 *hYrecYmc; */
  /*  TH2 *hZrecYmc; */
  /*  TH2 *hThrecYmc; */
  /*  TH2 *hPhrecYmc; */
  /*  TH2 *hPrecYmc; */
  /* TH2 *hResXrecYmc; */
  /*  TH2 *hResYrecYmc; */
  /*  TH2 *hResZrecYmc; */
  /*  TH2 *hResThrecYmc; */
  /*  TH2 *hResPhrecYmc; */
  /*  TH2 *hResPrecYmc; */
  /*  TH2 *herrXrecYmc; */
  /*  TH2 *herrYrecYmc; */
  /*  TH2 *herrZrecYmc; */
  /*  TH2 *herrThrecYmc; */
  /*  TH2 *herrPhrecYmc; */
  /*  TH2 *herrPrecYmc; */

  // Near 1st LMD plane
  TH1 *hResHitX;
  TH1 *hResHitY;
  TH1 *hResHitZ;
  TH1 *hhits;
  TH1 *hchi2;
  TH1 *hResLumiTrkMom;
  TH1 *hResLumiTrkTheta;
  TH1 *hResLumiTrkPhi;
  TH2 *hResLumiTrkTheta2D;
  TH2 *hMCLumiTrkTheta2D;
  TH2 *hRecLumiTrkTheta2D;
  TH1 *hResLumiTrkPointX;
  TH1 *hResLumiTrkPointY;
  TH1 *hResLumiTrkPointZ;

  TH1 *hResLumiTrkPointPx;
  TH1 *hResLumiTrkPointPy;
  TH1 *hResLumiTrkPointPz;

  TH1 *hResLumiTrkPointXErr;
  TH1 *hResLumiTrkPointYErr;
  TH1 *hResLumiTrkPointZErr;
  TH1 *hResLumiTrkPointPxErr;
  TH1 *hResLumiTrkPointPyErr;
  TH1 *hResLumiTrkPointPzErr;

  TH1 *hResLumiTrkPointXPull;
  TH1 *hResLumiTrkPointYPull;
  TH1 *hResLumiTrkPointZPull;

  TH1 *hResLumiTrkPointPxPull;
  TH1 *hResLumiTrkPointPyPull;
  TH1 *hResLumiTrkPointPzPull;
  TH1 *hResLumiTrkThetaPull;
  TH1 *hResLumiTrkPhiPull;

  // TFile *fouthists;
  ClassDef(PndLmdQATask, 1);
};

#endif