PndPidCalorimeterInfo.h

Go to the documentation of this file.

#ifndef PID_CALORIMETER_INFO_H
#define PID_CALORIMETER_INFO_H

#include <map>
#include <memory>

#include "TClonesArray.h"

#include "FairRootManager.h"
#include "FairRuntimeDb.h"
#include "FairPropagator.h"
#include "PndPropagator.h"
#include "FairGeane.h"
#include "FairLogger.h"

#include "PndTrack.h"
#include "PndEmcErrorMatrix.h"
#include "PndEmcClusterCalibrator.h"
#include "PndEmcErrorMatrixPar.h"
#include "PndEmcGeoPar.h"
#include "PndEmcClusterEnergySums.h"
#include "PndEmcCluster.h"
#include "PndEmcBump.h"

#include "PndPidDetectorInfo.h"

class PndPidCalorimeterInfo : public PndPidDetectorInfo {
 protected:

  std::map<Int_t, Bool_t> fClusterList{}; 
  std::map<Int_t, Double_t> fClusterQ{};  
  std::map<Int_t, std::pair<Double_t, Int_t>> fCorrelatedTracks; 
  Bool_t fTracksCorrelated = kFALSE;

  struct EmcQualityEntry_t {
    Int_t EmcIndex;
    Double_t Quality;
    bool operator<(const EmcQualityEntry_t &other) const {return Quality<other.Quality;}
  };

  struct TrackQualities_t {
    Int_t TrackId;
    std::vector<EmcQualityEntry_t> Qualities;
    bool operator<(const TrackQualities_t &other) const {return Qualities[0].Quality < other.Qualities[0].Quality;}
    EmcQualityEntry_t & operator [](Int_t index) {return Qualities[index];}
    const EmcQualityEntry_t & operator [](Int_t index) const {return Qualities[index];}
  };

  struct TrackQuality_t: public EmcQualityEntry_t {
    Int_t TrackId;
  };

  struct QualityTable_t {
    std::vector<TrackQualities_t> Table;

    TrackQuality_t GetNextBestFit();
    void Remove(Int_t trackId, Int_t emcIndex);
    void Clear();
  };

  QualityTable_t fQualityTable;

 private:

  virtual void FillCorrelationMap() = 0;

 public:
  PndPidCalorimeterInfo(); 
  ~PndPidCalorimeterInfo(){}; 

  virtual Int_t GetClusterCount() {
    LOG(warn) << "PndPidCalorimeterInfo::GetClusterCount: Function not implemented in base class.";
    return 0;
  };

  Bool_t SetParContainers(FairRuntimeDb *db) = 0;

  void Reset() = 0;

  virtual Int_t SetDetectorHits(FairRootManager *manager) = 0;

  Bool_t Setup(FairRootManager *manager, PndPropagator *propagator, PndPidDetectorParams::DetectorParams *params);

  virtual void ActivateDebugNtuple(Bool_t debug_mode);

  void GetNeutralInfo(Int_t nbr_tracks, TClonesArray *neutralArray) = 0;

  virtual Bool_t GetInfo(PndTrack *track, PndPidCandidate *pidCand) = 0;

  ClassDef(PndPidCalorimeterInfo, 2);
};

#endif // PID_CALORIMETER_INFO_H