KVINDRAFilterGroupReconstructor.h

#ifndef __KVINDRAFILTERGROUPRECONSTRUCTOR_H
#define __KVINDRAFILTERGROUPRECONSTRUCTOR_H
 
#include "KVFilterGroupReconstructor.h"
#include "KVChIo.h"
#include "KVINDRA.h"
 
class KVINDRAFilterGroupReconstructor : public KVFilterGroupReconstructor {
 
   KVChIo* theChIo = nullptr;
   mutable bool translate_csi_id_names = false;
   mutable bool check_translate = true;
    mutable std::unordered_map<KVIDTelescope*,TString> idcsi_name_translation;
protected:
   void identify_particle(KVIDTelescope* idt, KVIdentificationResult* IDR, KVReconstructedNucleus& nuc) override;
 
   TString get_id_telescope_name_for_audit(KVIDTelescope* idt) const override
   {
 
      if(!has_audit() || (!check_translate && !translate_csi_id_names))
         return idt->GetName();
 
      if(std::strncmp(idt->GetType(),"CS",2))
         return idt->GetName();
 
      if(check_translate)
      {
         if(!audit_has_telescope_with_name(idt->GetName()))
         {
            TString idt_name(idt->GetName()), new_name;
            bool old_array = idt_name.Length()>8;//array telescopes have old "CSI_R_L_xxxx" names
            if(old_array)
               new_name = idt->GetDetector(1)->GetName();//new name is same as name of CsI
            else
               new_name = Form("CSI_R_L_%04d", idt->GetDetector(1)->GetIndex());
            if(audit_has_telescope_with_name(new_name))
            {
               auto name_to_return = new_name;//no need to do the job twice
               check_translate=false;
               translate_csi_id_names=true;
               std::unique_ptr<KVSeqCollection> csi_list(GetGroup()->GetDetectorTypeList("CSI"));
               TIter it_csi(csi_list.get());
               KVDetector* dcsi;
               while( (dcsi = (KVDetector*)it_csi()) )
               {
                  if(old_array)
                  {
                     idt_name = Form("CSI_R_L_%04d", dcsi->GetIndex());
                     new_name = dcsi->GetName();
                  }
                  else
                  {
                     idt_name = dcsi->GetName();
                     new_name = Form("CSI_R_L_%04d", dcsi->GetIndex());
                  }
                  auto iidt = gIndra->GetIDTelescope(idt_name);
                  if(iidt)
                     idcsi_name_translation[iidt] = new_name;
               }
               return name_to_return;
            }
            else
            {
               return idt->GetName();
            }
         }
         else
         {
            check_translate=false;
            translate_csi_id_names=false;
            return idt->GetName();
         }
      }
      return idcsi_name_translation[idt];
   }
public:
   KVINDRAFilterGroupReconstructor(const KVGroup* g = nullptr)
      : KVFilterGroupReconstructor(g)
   {
      if (g) theChIo = dynamic_cast<KVChIo*>(g->GetDetectorByType("CI"));
   }
 
   void Reconstruct() override;
   void IdentifyParticle(KVReconstructedNucleus&) override;
 
   ClassDefOverride(KVINDRAFilterGroupReconstructor, 1) //Reconstruct simulated events after filtering with INDRA
};
 
#endif