PndFairHitQATask.h

Go to the documentation of this file.

#ifndef PndFairHitQATask_H
#define PndFairHitQATask_H

// Root includes
#include "TVector3.h"
#include "TString.h"
#include "TGeoManager.h"
#include "TH1.h"

// framework includes
#include "PndPersistencyTask.h"
#include "PndMeasurementBoundaries.h"

#include "FairLogger.h"
#include "FairHit.h"

#include <map>

class TClonesArray;

class PndFairHitQATask : public PndPersistencyTask {
 public:
  PndFairHitQATask(std::string hitBranchName, std::string pointBranchName);

  virtual ~PndFairHitQATask();

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

  virtual void Exec(Option_t *opt);
  virtual void FinishTask();
  void SetWorkdir(TString path) { fWorkdir = path; } 

  void SetHitBranch(std::string hitBranch) { fHitBranchName = hitBranch; }
  void SetPointBranch(std::string pointBranch) { fPointBranchName = pointBranch; }
  void SetOutputPlotName(std::string val) { fOutputPlotName = val; }

 protected:
  // These methods should be overwritten for a realistic test

  virtual void InitHistos();

  virtual void SetAllBoundaries();

  virtual void CalculateResolutions(FairHit *hit);

  virtual void FillHistos();

  virtual void AssignMeasurements();

  // No need to touch these classes
  bool Test();
  void DrawHistos();
  void SaveHistos();
  void PrintDartMeasurement();

  std::string fHitBranchName;
  std::string fPointBranchName;
  std::string fOutputPlotName{"ResolutionHistos.png"};

  TClonesArray *fMcPoints = nullptr;
  TClonesArray *fHits = nullptr;
  Bool_t fProperDifference = false; 
  TVector3 fSpatialDifference;
  double fTimeDifference = -1.0;
  TString fWorkdir;

  PndMeasurementBoundaries fBoundaries;

  std::map<std::string, TH1 *> fHistos;

  ClassDef(PndFairHitQATask, 2);
};

#endif