CbmKFParticleFinderQA.h

Go to the documentation of this file.

//-----------------------------------------------------------
//-----------------------------------------------------------
 
#ifndef CbmKFParticleFinderQA_HH
#define CbmKFParticleFinderQA_HH
 
#include "FairTask.h"
 
#include "TString.h"
 
#include <vector>
 
class KFParticleTopoReconstructor;
class KFTopoPerformance;
class TClonesArray;
class CbmMCEventList;
class CbmMCDataArray;
class TFile;
class TObject;
 
class CbmKFParticleFinderQA : public FairTask {
public:
  // Constructors/Destructors ---------
  CbmKFParticleFinderQA(const char* name = "CbmKFParticleFinderQA",
                        Int_t iVerbose   = 0,
                        const KFParticleTopoReconstructor* tr = 0,
                        TString outFileName = "CbmKFParticleFinderQA.root");
  ~CbmKFParticleFinderQA();
 
  void SetEffFileName(const TString& name) { fEfffileName = name; }
  void SetMCTrackBranchName(const TString& name) { fMCTracksBranchName = name; }
  void SetTrackMatchBranchName(const TString& name) {
    fTrackMatchBranchName = name;
  }
 
  virtual InitStatus Init();
  virtual void Exec(Option_t* opt);
  virtual void Finish();
 
  void SetPrintEffFrequency(Int_t n);
 
  void SaveParticles(Bool_t b = 1) { fSaveParticles = b; }
  void SaveMCParticles(Bool_t b = 1) { fSaveMCParticles = b; }
 
  // Set SE analysis
  void SetSuperEventAnalysis() { fSuperEventAnalysis = 1; }
 
  //Tests
  void SetCheckDecayQA() { fCheckDecayQA = true; }
  void SetReferenceResults(TString t) { fReferenceResults = t; }
  void SetDecayToAnalyse(int iDecay) { fDecayToAnalyse = iDecay; }
  bool IsTestPassed() { return fTestOk; }
 
private:
  const CbmKFParticleFinderQA& operator=(const CbmKFParticleFinderQA&);
  CbmKFParticleFinderQA(const CbmKFParticleFinderQA&);
 
  void WriteHistosCurFile(TObject* obj);
  void FitDecayQAHistograms(float sigma[14],
                            const bool saveReferenceResults = false) const;
  void CheckDecayQA();
 
  //names of input branches
  TString fMCTracksBranchName;    
  TString fTrackMatchBranchName;  
 
  //input branches
  CbmMCDataArray* fMCTrackArray;  //mc tracks
  TClonesArray* fMCTrackArrayEvent;
  CbmMCEventList* fEventList;      //mc event list in timeslice
  TClonesArray* fTrackMatchArray;  //track match
 
  // output arrays of particles
  TClonesArray* fRecParticles;    // output array of KF Particles
  TClonesArray* fMCParticles;     // output array of MC Particles
  TClonesArray* fMatchParticles;  // output array of match objects
 
  Bool_t fSaveParticles;
  Bool_t fSaveMCParticles;
 
  bool fTimeSliceMode;
 
  //output file with histograms
  TString fOutFileName;
  TFile* fOutFile;
  TString fEfffileName;
  //KF Particle QA
  KFTopoPerformance* fTopoPerformance;
 
  Int_t fPrintFrequency;
  Int_t fNEvents;
  Double_t fTime[5];
 
  //for super event analysis
  bool fSuperEventAnalysis;
 
  //for tests
  TString fReferenceResults;
  int fDecayToAnalyse;
  bool fCheckDecayQA;
  bool fTestOk;
 
  ClassDef(CbmKFParticleFinderQA, 1);
};
 
#endif