KVGroup.h

#ifndef KVGROUP_H
#define KVGROUP_H
 
#define KVGROUP_ADD_UNKNOWN_TELESCOPE "Attempt to add undefined telescope to group"
 
#include "KVGeoStrucElement.h"
#include "KVUnownedList.h"
#include "KVGeoDNTrajectory.h"
#include "KVNameValueList.h"
#include "KVDetector.h"
#include "KVNucleus.h"
 
class KVGroup : public KVGeoStrucElement {
 
protected:
   enum {
      kIsRemoving = BIT(14)     //flag set during call to RemoveTelescope
   };
    KVUnownedList fReconstructedNuclei;
    KVHashList fTrajectories;
    KVHashList fReconTraj;
    KVNameValueList fReconTrajMap;
 
public:
   enum {
      kForwards,
      kBackwards
   };
   KVGroup();
   void init();
   void SetNumber(UInt_t num) override
   {
      SetName(Form("Group_%u", num));
      KVGeoStrucElement::SetNumber(num);
   }
 
   void Reset(Option_t* opt = "")
   {
 
       fReconstructedNuclei.Clear();
       GetDetectors()->R__FOR_EACH(KVDetector, Reset)(opt);
   }
 
   inline UInt_t GetHits()
   {
      return fReconstructedNuclei.GetSize();
   }
   void ClearHitDetectors()
   {
       GetDetectors()->R__FOR_EACH(KVDetector, ClearHits)();
   }
   KVList* GetParticles()
   {
      return &fReconstructedNuclei;
   }
   void AddHit(KVNucleus* kvd)
   {
       fReconstructedNuclei.Add(kvd);
   }
   void RemoveHit(KVNucleus* kvd)
   {
       fReconstructedNuclei.Remove(kvd);
   }
 
   Bool_t IsRemoving()
   {
      return TestBit(kIsRemoving);
   }
   virtual void Sort() {}
   virtual void CountLayers() {}
   const TCollection* GetTrajectories() const
   {
      return &fTrajectories;
   }
   void AddTrajectory(KVGeoDNTrajectory* t)
   {
      fTrajectories.Add(t);
   }
   void AddTrajectories(const TCollection* c)
   {
      fTrajectories.AddAll(c);
   }
   void ReplaceTrajectories(const TCollection* c)
   {
      fTrajectories.Clear();
      fTrajectories.AddAll(c);
   }
 
   Int_t CalculateReconstructionTrajectories();
   const TSeqCollection* GetReconTrajectories() const
   {
      return &fReconTraj;
   }
   const KVGeoDNTrajectory* FindReconTraj(const KVString& path)
   {
      return (KVGeoDNTrajectory*)fReconTraj.FindObjectByTitle(path);
   }
 
   const KVGeoDNTrajectory* GetTrajectoryForReconstruction(const KVGeoDNTrajectory* t, const KVGeoDetectorNode* n) const
   {
      if (t && n) {
         TString mapped_name = fReconTrajMap.GetStringValue(Form("%s_%s", t->GetTrajectoryName(), n->GetName()));
         const KVGeoDNTrajectory* tr = (const KVGeoDNTrajectory*)fReconTraj.FindObject(mapped_name);
         return tr;
      }
      return nullptr;
   }
   void PrintData() const;
 
   ClassDefOverride(KVGroup, 2)//Group of detectors having similar angular positions.
};
#endif