KVFilterGroupReconstructor.h

#ifndef __KVFILTERGROUPRECONSTRUCTOR_H
#define __KVFILTERGROUPRECONSTRUCTOR_H
 
#include "KVGroupReconstructor.h"
#include "KVDataQualityAudit.h"
#include <unordered_map>
class KVFilterGroupReconstructor : public KVGroupReconstructor {
 
   const KVDataQualityAudit* fDataQAudit = nullptr; 
 
protected:
   std::unique_ptr<KVEvent> fSimEvent;
   std::unordered_map<std::string, int> hits; 
   std::unordered_map<KVReconstructedNucleus*, TList> part_correspond; 
   KVReconstructedNucleus* current_nuc_recon;
 
   std::unordered_map<std::string, double> energy_loss; 
   std::unordered_map<std::string, int> number_uncalibrated; 
   std::unordered_map<std::string, int> number_unidentified; 
   KVReconNucTrajectory* get_recon_traj_for_particle(const KVGeoDNTrajectory*, const KVGeoDetectorNode* node) override;
   void identify_particle(KVIDTelescope* idt, KVIdentificationResult* IDR, KVReconstructedNucleus& nuc) override;
 
   void PerformSecondaryAnalysis();
 
public:
   KVFilterGroupReconstructor(const KVGroup* g = nullptr) : KVGroupReconstructor(g) {}
 
   void SetSimEvent(KVEvent* e)
   {
 
      fSimEvent.reset(static_cast<KVEvent*>(e->IsA()->New()));
   }
   void AddSimParticle(const KVNucleus* n)
   {
 
      KVNucleus* nuc;
      n->Copy(*(nuc = fSimEvent->AddNucleus()));
      auto traj = GetGroup()->FindReconTraj(n->GetParameters()->GetStringValue("TRAJECTORY"));
      traj->IterateFrom();
      while (auto node = traj->GetNextNode()) {
         ++hits[node->GetName()];
         ++number_unidentified[node->GetName()];
      }
   }
   void Clear(Option_t* = "") override
   {
      fSimEvent->Clear();
      hits.clear();
      part_correspond.clear();
      energy_loss.clear();
      number_uncalibrated.clear();
      number_unidentified.clear();
   }
   void ReconstructParticle(KVReconstructedNucleus* part, const KVGeoDNTrajectory* traj, const KVGeoDetectorNode* node) override;
   void IdentifyParticle(KVReconstructedNucleus& PART) override;
   void CalibrateParticle(KVReconstructedNucleus* PART) override;
 
   void SetDataQualityAudit(const KVDataQualityAudit* DQA)
   {
      fDataQAudit = DQA;
   }
 
   ClassDefOverride(KVFilterGroupReconstructor, 1) //Reconstruct simulated events after filtering
};
 
#endif