PndSTEAnalysisTask.h

Go to the documentation of this file.

#ifndef PndSTEAnalysisTask_H_
#define PndSTEAnalysisTask_H_

#include "FairTask.h"
#include "PndTrackCand.h"
#include "PndTrack.h"
#include "PndMCTrack.h"
#include "TClonesArray.h"
#include "PndSciTHit.h"
#include "FairLink.h"

#include "TH2.h"

#include <vector>
#include <map>

class TClonesArray;

class PndSTEAnalysisTask : public FairTask {
 public:
  PndSTEAnalysisTask() : FairTask("Pnd Stt Track Extrapolator Analysis Task"){};

  virtual ~PndSTEAnalysisTask(){};

  virtual InitStatus Init();

  virtual void Exec(Option_t *opt);

  virtual void FinishEvent();

  virtual void Finish();

  void SetInputTrackBranchName(TString val) { fTrackBranchName = val; };

  void SetInputHitBranchName(TString val) { fHitBranchName = val; };

  void SetAnalyzeBtofHits(bool val) { fAnalyzeBtofHits = val; };

  void SetHitCut(int val) { fCut = val; };

 private:
  void InitHitArray(TString);

  void InitTrackArray(TString);

  void InitHistograms();

  void AssociateTrackWithMCTrack(PndTrack *);

  void AssociateBToFHitWithMCTrack(PndSciTHit *);

  void AnalyzeBToFHits(PndTrack *);

  bool fAnalyzeBtofHits = false;

  TString fHitBranchName = "SciTHit"; // Default hit branch name for BTOF hits
  TString fTrackBranchName = "Track"; // Default track branch name

  int eventCount; // Counter for the number of events

  int fCorrectlyAssignedHits = 0; // Number of correctly assigned BTOF hits
  int fWronglyAssignedHits = 0;   // Number of wrongly assigned BTOF hits
  int fNotAssignedHits = 0;       // Number of BTOF hits that are not assigned

  int fCut;    // Minimum number of hits in a track so that it is well reconstructed (~15-18 for SttCellTrackFinder)
  int numHits; // Number of hits in a track, used to fill histograms

  TClonesArray *trackArray;
  TClonesArray *hitArray;

  std::map<PndTrack *, std::vector<int> > fMapRecoTrackIndices; // Maps the PndTrack to a vector with its entry number and index
  std::map<double, std::vector<int> > fMapBToFHitIndices;       // Maps the time of a BTOF hit to a vector with its entry number and index

  TH1 *hMomRes;
  TH1 *hRelMomRes;
  TH1 *hMomResWithCut;
  TH1 *hMomRelResWithCut;
  TH1 *hMomResCutAwayPart;
  TH1 *hMomRelResCutAwayPart;

  TH1 *hMomRealisticExpanded;
  TH1 *hRelMomRealisticExpanded;

  TH1 *hMomResWithCutPTCut;
  TH1 *hMomRelResWithCutPTCut;
  TH1 *hMomResCutAwayPartPTCut;
  TH1 *hMomRelResCutAwayPartPTCut;

  TH2 *hMomRealistic;
  TH2 *hRelMomRealistic;
  TH2 *hMomAngleRealistic;
  TH2 *hRelMomAngleRealistic;
  TH2 *hMomNumHitsRealistic;
  TH2 *hRelMomNumHitsRealistic;

  TH2 *hPositiveRes_NumHits;
  TH2 *hNegativeRes_NumHits;
  TH2 *hGoodRes_NumHits;

  TH2 *hPositiveRes_RecoMom;
  TH2 *hNegativeRes_RecoMom;
  TH2 *hGoodRes_RecoMom;
  TH2 *hPositiveRes_MCMom;
  TH2 *hNegativeRes_MCMom;
  TH2 *hGoodRes_MCMom;
  TH2 *hPositiveRes_MCTheta;
  TH2 *hNegativeRes_MCTheta;
  TH2 *hGoodRes_MCTheta;
  TH2 *hPositiveRes_NumMCTracks;
  TH2 *hNegativeRes_NumMCTracks;
  TH2 *hGoodRes_NumMCTracks;

  TH2 *hPositiveRelRes_NumHits;
  TH2 *hNegativeRelRes_NumHits;
  TH2 *hGoodRelRes_NumHits;
  TH2 *hPositiveRelRes_RecoMom;
  TH2 *hNegativeRelRes_RecoMom;
  TH2 *hGoodRelRes_RecoMom;
  TH2 *hPositiveRelRes_MCMom;
  TH2 *hNegativeRelRes_MCMom;
  TH2 *hGoodRelRes_MCMom;
  TH2 *hPositiveRelRes_MCTheta;
  TH2 *hNegativeRelRes_MCTheta;
  TH2 *hGoodRelRes_MCTheta;
  TH2 *hPositiveRelRes_NumMCTracks;
  TH2 *hNegativeRelRes_NumMCTracks;
  TH2 *hGoodRelRes_NumMCTracks;
  TH2 *hAllRelRes_NumMCTracks;
  TH2 *hAllRes_NumMCTracks;

  TH2 *hTotMom;
  TH2 *hTotMomRel;

  TH2 *hTotMomFiltered;
  TH2 *hTotMomRelFiltered;

  TH2 *hNumMCTracksAbsolute;
  TH2 *hNumMCTracksRelative;

  TH2 *hMomFiltered;
  TH2 *hMomRelFiltered;

  TH1 *hMomResFiltered;
  TH1 *hMomRelResFiltered;

  TH1 *hNumMcTracksPerIdealTrack;

  ClassDef(PndSTEAnalysisTask, 1);
};

#endif /* PndSTEAnalysisTask_H_ */