PndAnalysis.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".              *
//****************************************************************************

#ifndef PNDANALYSIS_H
#define PNDANALYSIS_H 1

#include <string>
#include <vector>
#include <array>

#include "TString.h"
#include "TVector3.h"
#include "TLorentzVector.h"
#include "TMatrixD.h"

#include "RhoCandList.h"
#include "FairRootManager.h"
#include "FairTrackParP.h"

class TChain;
class TClonesArray;
class TBranch;

class TDatabasePDG;
class FairMCEventHeader;
class PndAnaPidSelector;
class PndAnaPidCombiner;
class PndMCTrack;
class PndTrack;
class RhoGoodPhotonSelector;
class RhoGoodTrackSelector;

// class PndEventInfo;

class TGeant3;

class PndAnalysis {
 public:
  PndAnalysis(TString tname1 = "", TString tname2 = "", TString algnamec = "PidAlgoIdealCharged", TString algnamen = "PidAlgoIdealNeutral");
  virtual ~PndAnalysis();

  void Rewind();
  void Reset() { Rewind(); }
  Int_t GetEvent(Int_t n = -1);
  void GetEventInTask();
  FairMCEventHeader *GetEventHeader();
  Int_t GetEntries();
  Bool_t FillList(RhoCandList &l, TString listkey = "All", TString pidTcaNames = "", int trackHypothesis = -1);
  void SetGoodTrackCriterion(TString crit="veryLoose"); 
  void SetVerbose(Int_t level) { fVerbose = level; }
  // void SetPidChargedName(TString s) {fChargedPidName = s;}
  // void SetPidNeutralName(TString s) {fNeutralPidName = s;}
  // void SetTracksName(TString s) {fTracksName = s;}
  // void SetTracksName2(TString s) {fTracksName2 = s;}

  Bool_t PropagateToIp(RhoCandidate *cand);
  Bool_t PropagateToZAxis(RhoCandidate *cand);
  Bool_t PropagateToPoint(RhoCandidate *cand, TVector3 mypoint);
  Bool_t PropagateToPlane(RhoCandidate *cand, TVector3 origin, TVector3 dj, TVector3 dk);

  PndTrack *GetTrack(RhoCandidate *cand);
  FairTrackParP GetFirstPar(RhoCandidate *cand);

  Bool_t ResetCandidate(RhoCandidate *cand);
  Bool_t ResetDaughters(RhoCandidate *cand);
  Bool_t McTruthMatch(RhoCandidate *cand, Int_t level = 2, bool verbose = false);
  std::vector<int> McTruthMatch(RhoCandList &list, Int_t level = 2, bool verbose = false);
  // this also allows match, when some soft photons are missing in decay
  void McMatchAllowPhotos(int maxn = 1, double thresh = 0.05)
  {
    fPhotosMax = maxn;
    fPhotosThresh = thresh;
  }

  // FIXME: This is an aweful solution to access the correct
  // track array from a fitter object. [R.K.03'11]
  // todo:removed this because fuck it
  // TClonesArray* GetTrackArrayPointer() const {return fTracks;};

  Bool_t Propagator(int mode, FairTrackParP &tStart, RhoCandidate *cand, TVector3 point = TVector3(0, 0, 0), Bool_t skipcov = kFALSE, Bool_t overwrite = kFALSE,
                    TVector3 planej = TVector3(1, 0, 0), TVector3 planek = TVector3(0, 1, 0));

 protected:
  void Init();
  virtual void BuildMcCands();
  Bool_t GetMcCandList(RhoCandList &l);
  void ReadRecoCandidates();
  void Cleanup();
  // void ReadCandidates();

  TClonesArray *ReadTCA(TString tcaname);

  Bool_t MctMatch(RhoCandidate &c, RhoCandList &mct, Int_t level = 2, bool verbose = false) { return MctMatch(&c, mct, level, verbose); };
  Bool_t MctMatch(RhoCandidate *c, RhoCandList &mct, Int_t level = 2, bool verbose = false);
  // Private Member Variables

  FairRootManager *fRootManager;
  PndAnaPidSelector *fPidSelector;
  PndAnaPidCombiner *fPidCombiner;
  RhoGoodPhotonSelector *fGoodPhotonSelector;
  RhoGoodTrackSelector *fGoodTrackSelector;
  Int_t fEvtCount;
  Int_t fChainEntries;
  Bool_t fEventRead;
  Bool_t fBuildMcCands;
  Int_t fVerbose;
  Int_t fPhotosMax;
  Double_t fPhotosThresh;

  std::map<int, RhoCandidate *> fMcPresenceMap;

  std::array<TClonesArray *, 6> fChargedCands; // 0-4 for trk hypothesis, 5 for fallback
  std::array<TClonesArray *, 6> fBremCorr;     // 0-4 for trk hypothesis, 5 for fallback
  TClonesArray *fNeutralCands;
  std::array<TClonesArray *, 6> fChargedProbability; // 0-4 for trk hypothesis, 5 for fallback
  // TClonesArray* fNeutralProbability;
  std::array<TClonesArray *, 6> fTracks;  // 0-4 for trk hypothesis, 5 for fallback
  std::array<TClonesArray *, 6> fTracks2; // 0-4 for trk hypothesis, 5 for fallback
  TClonesArray *fMcCands;
  TClonesArray *fMcTracks;

  std::array<RhoCandList, 6> fChargedCandList; // 0-4 for trk hypothesis, 5 for fallback
  RhoCandList fNeutralCandList;
  RhoCandList fAllCandList;
  RhoCandList fMcCandList;

  TString fChargedPidName;
  TString fNeutralPidName;
  TString fTracksName;
  TString fTracksName2;

  TString fPidHypoStr[6];     // 0-4 for trk hypothesis, 5 for fallback
  Int_t fHypoPdg[6];          // 0-4 for trk hypothesis, 5 for fallback
  Bool_t fHypoFlagCharged[6]; 
  Int_t fDefaultHypo;         // Default hypothesis (pion), used if no hypo is specified by user
  ClassDef(PndAnalysis, 0);
};

#endif