PndTrackingQATask.h

Go to the documentation of this file.

#include "PndTrackFunctor.h"
#include "PndGeoSttPar.h"
#include "PndTrackingQASummary.h"

#include "FairTask.h"

#include "TObject.h"
#include "TString.h"

#include <vector>

class TClonesArray;

#pragma once

class PndTrackingQATask : public FairTask {
 public:
  PndTrackingQATask(TString trackBranchName, TString idealBranchName, Bool_t pndTrackData = kTRUE);
  virtual ~PndTrackingQATask();

  virtual void SetParContainers();
  virtual InitStatus Init();

  virtual void Exec(Option_t *opt);

  virtual void Finish();

  void SetIdealTrackBranchName(TString name) { fIdealTrackBranchName = name; };

  void SetTrackBranchName(TString name, Bool_t pndTrackData = kTRUE)
  {
    fPndTrackNotTrackCand = pndTrackData;
    fTrackBranchName = name;
  } // alternatively PndTrackCandData can be analysed. Therefore pndTrackData has to be set to kFALSE

  void AddHitsBranchName(TString name) { fBranchNames.push_back(name); }

  void SetOutputMCInfoBranchName(TString name) { fMCInfoBranchName = name; }
  void SetOutputRecoInfoBranchName(TString name) { fRecoInfoBranchName = name; }

  void SetFunctorName(TString name) { fPossibleTrackFunctorName = name; }
  void SetFunctor();

 protected:
  void FillQASummary();
  int GetNGhosts();

 private:
  std::vector<TString> fBranchNames;              
  TString fMCInfoBranchName, fRecoInfoBranchName; 

  PndTrackingQASummary fSummary;

  TClonesArray *fTrack = nullptr;
  TClonesArray *fMCTrack = nullptr;
  TClonesArray *fSttHitArray = nullptr;
  TClonesArray *fMCTrackInfo = nullptr;
  TClonesArray *fRecoTrackInfo = nullptr;
  TClonesArray *fIdealTrack = nullptr;
  TClonesArray *fSttTubeArray = nullptr;
  PndGeoSttPar *fSttParameters = nullptr;

  FairRootManager *ioman = nullptr;

  TString fPossibleTrackFunctorName;
  PndTrackFunctor *fPossibleTrackFunctor = nullptr;

  TString fTrackBranchName;
  TString fIdealTrackBranchName;
  Bool_t fPndTrackNotTrackCand = kTRUE; // kTRUE if track and kFALSE if track cand

  ClassDef(PndTrackingQATask, 1);
};