KVGroupReconstructor.h

#ifndef __KVGROUPRECONSTRUCTOR_H
#define __KVGROUPRECONSTRUCTOR_H
 
#include "KVBase.h"
#include "KVGroup.h"
#include "KVReconstructedEvent.h"
#include <unordered_map>
#include <string>
 
class KVGroupReconstructor : public KVBase {
 
   static bool fDoIdentification;
   static bool fDoCalibration;
 
    KVGroup* fGroup;
    KVReconstructedEvent* fGrpEvent;
    TString fPartSeedCond;
    Bool_t fSecondaryAnalysis;
protected:
    mutable int nfireddets;
    KVIdentificationResult partID;
    KVIDTelescope* identifying_telescope;
    std::unordered_map<std::string, KVIdentificationResult*> id_by_type;
 
   virtual KVReconstructedNucleus* ReconstructTrajectory(const KVGeoDNTrajectory* traj, const KVGeoDetectorNode* node);
   virtual void ReconstructParticle(KVReconstructedNucleus* part, const KVGeoDNTrajectory* traj, const KVGeoDetectorNode* node);
   virtual void PostReconstructionProcessing();
   virtual void IdentifyParticle(KVReconstructedNucleus&);
   virtual void CalibrateCoherencyParticle(KVReconstructedNucleus*)
   {
      AbstractMethod("CalibrateCoherencyParticle(KVReconstructedNucleus*)");
   }
   void SetCalibrationStatus(KVReconstructedNucleus& PART, UShort_t code)
   {
      PART.SetIsCalibrated();
      PART.SetECode(code);
   }
 
   Double_t GetTargetEnergyLossCorrection(KVReconstructedNucleus* ion);
   TString GetPartSeedCond() const
   {
      return fPartSeedCond;
   }
   void TreatStatusStopFirstStage(KVReconstructedNucleus&);
 
   struct particle_to_add_from_coherency_analysis {
      KVReconstructedNucleus* original_particle;  
      KVGeoDetectorNode* stopping_detector_node;  
      KVGeoDNTrajectory* stopping_trajectory;     
    KVIDTelescope* identifying_telescope;
      Int_t first_id_result_to_copy;              
      Int_t max_id_result_index;                  
   };
   std::vector<particle_to_add_from_coherency_analysis> coherency_particles;
   virtual void AddCoherencyParticles() {}
 
   virtual KVReconNucTrajectory* get_recon_traj_for_particle(const KVGeoDNTrajectory* traj, const KVGeoDetectorNode* node);
   virtual void identify_particle(KVIDTelescope* idt, KVIdentificationResult* IDR, KVReconstructedNucleus&)
   {
      idt->Identify(IDR);
   }
 
public:
   KVGroupReconstructor(const KVGroup* g = nullptr);
   virtual ~KVGroupReconstructor();
 
   void SetReconEventClass(TClass* c);
   int GetNFiredDets() const
   {
      return nfireddets;
   }
 
   KVReconstructedEvent* GetEventFragment() const
   {
      return fGrpEvent;
   }
   void SetEventNumber(UInt_t en)
   {
      GetEventFragment()->SetNumber(en);
   }
   virtual void SetGroup(const KVGroup* g);
   KVGroup* GetGroup() const
   {
      return fGroup;
   }
 
   static KVGroupReconstructor* Factory(const TString& plugin = "", const KVGroup* g = nullptr);
 
   void Process();
   virtual void Reconstruct();
   virtual void Identify();
   void Calibrate();
   virtual void CalibrateParticle(KVReconstructedNucleus*)
   {
      AbstractMethod("CalibrateParticle(KVReconstructedNucleus*)");
   }
 
   void AnalyseParticles();
   Int_t GetNIdentifiedInGroup()
   {
      Int_t n = 0;
      if (GetEventFragment()->GetMult()) {
         for (KVReconstructedEvent::Iterator it = GetEventFragment()->begin(); it != GetEventFragment()->end(); ++it) {
            KVReconstructedNucleus& nuc = it.get_reference();
            n += (Int_t) nuc.IsIdentified();
         }
      }
      return n;
   }
   Int_t GetNUnidentifiedInGroup()
   {
      return (GetEventFragment()->GetMult() - GetNIdentifiedInGroup());
   }
   static void SetDoIdentification(bool on = kTRUE)
   {
      fDoIdentification = on;
   }
   static void SetDoCalibration(bool on = kTRUE)
   {
      fDoCalibration = on;
   }
   void Clear(Option_t* = "") override
   {
   }
 
   ClassDefOverride(KVGroupReconstructor, 0) //Base class for handling event reconstruction in detector groups
};
 
#endif