CbmAnaConversionKF.h

Go to the documentation of this file.

/* Copyright (C) 2014-2016 Fakultaet fuer Mathematik und Naturwissenschaften, Bergische Universitaet Wuppertal, Wuppertal
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Sascha Reinecke [committer], Florian Uhlig */
 
#ifndef CBM_ANA_CONVERSION_KF
#define CBM_ANA_CONVERSION_KF
 
// included from ROOT
#include <TClonesArray.h>
#include <TH1.h>
#include <TH2.h>
#include <TH3.h>
#include <TStopwatch.h>
 
// included from CbmRoot
#include "CbmKFParticleFinder.h"
#include "CbmKFParticleFinderQa.h"
#include "CbmMCTrack.h"
#include "CbmVertex.h"
#include "KFPartMatch.h"
#include "KFParticle.h"
#include "KFTopoPerformance.h"
 
#include <vector>
 
class CbmAnaConversionKF {
 
 public:
  CbmAnaConversionKF();
  virtual ~CbmAnaConversionKF();
 
  void Init();
  void InitHistos();
  void Finish();
  void Exec();
 
  //void KFParticle_Analysis();
 
  void SetKF(CbmKFParticleFinder* kfparticle, CbmKFParticleFinderQa* kfparticleQA);
  void SetSignalIds(std::vector<int>* signalids);
  void SetGhostIds(std::vector<int>* ghostids);
 
 
  void test();
  Double_t Invmass_4particles(const CbmMCTrack* mctrack1, const CbmMCTrack* mctrack2, const CbmMCTrack* mctrack3,
                              const CbmMCTrack* mctrack4);
 
  void test2();
  void CombineElectrons();
  void CombinePhotons();
  void Reconstruct();
  void ReconstructGammas();
  Double_t Invmass_4particlesRECO(KFParticle part1, KFParticle part2, KFParticle part3, KFParticle part4);
  Double_t Invmass_2gamma(KFParticle part1, KFParticle part2);
  Double_t Invmass_2electrons(KFParticle part1, KFParticle part2);
 
  Double_t OpeningAngleBetweenPhotons(KFParticle part1, KFParticle part2);
 
 
 private:
  TClonesArray* fKFMcParticles;
  TClonesArray* fMcTracks;
  TClonesArray* fStsTracks;
  TClonesArray* fStsTrackMatches;
 
  CbmKFParticleFinder* fKFparticle;
  CbmKFParticleFinderQa* fKFparticleFinderQA;
 
  const KFParticleTopoReconstructor* fKFtopo;
  KFTopoPerformance* fKFtopoPerf;
  std::vector<int> trackindexarray;
  Int_t particlecounter;
  Int_t particlecounter_2daughters;
  Int_t particlecounter_3daughters;
  Int_t particlecounter_4daughters;
  Int_t particlecounter_all;
  TH1D* fhPi0_NDaughters;
 
  // for data gained from the KFParticle package
  Int_t fNofGeneratedPi0_allEvents;    // number of generated pi0 summed up over ALL EVENTS
  Int_t fNofPi0_kfparticle_allEvents;  // number of all reconstructed pi0 summed up over ALL EVENTS
  Int_t fNofGeneratedPi0;
  Int_t fNofPi0_kfparticle;
  TH1D* fhPi0Ratio;
  TH1D* fhPi0_mass;
 
 
  std::vector<int> fSignalIds;
  std::vector<int> fGhostIds;
 
 
  std::vector<TH1*> fHistoList_kfparticle;  // list of all histograms containing results from KFParticle package
 
 
  std::vector<KFParticle> particlevector;
  std::vector<KFPartMatch> particleMatch;
  std::vector<int> electronIDs;
  std::vector<int> gammaIDs;
  TH1D* fhInvMassPi0WithFullReco;
 
  TH1D* fhInvMass2Gammas;
  TH1D* fhInvMass2Gammas_cut;
 
  TH1D* fhKF_particlevector;
  TH1D* fhKF_trackvector;
 
  TH1D* fhKF_NofPi0;
  TH1D* fhKF_NofPi0_signal;
  TH1D* fhKF_NofPi0_trackvector;
 
 
  std::vector<std::vector<int>> fKF_photon_pairs;
  TH1D* fhKF_invmass_fullReco;
 
 
  // timer
  TStopwatch timer;
  Double_t fTime;
 
  CbmAnaConversionKF(const CbmAnaConversionKF&);
  CbmAnaConversionKF operator=(const CbmAnaConversionKF&);
 
  ClassDef(CbmAnaConversionKF, 1)
};
 
#endif