KVGeoDetectorNode.h

 
#ifndef __KVDETECTORNODE_H
#define __KVDETECTORNODE_H
 
#include "KVBase.h"
#include "KVUniqueNameList.h"
 
class KVDetector;
class KVGeoDNTrajectory;
 
class KVGeoDetectorNode : public KVBase {
    KVDetector* fDetector;
    KVUniqueNameList fInFront;
    KVUniqueNameList fBehind;
    KVUniqueNameList fTraj;
    KVUniqueNameList fTrajF;
    KVUniqueNameList fTrajB;
   Int_t fNTraj = -1;              //number of trajectories passing through this node
   Int_t fNTrajForwards = -1;      //number of trajectories going forwards from this node
   Int_t fNTrajBackwards = -1;     //number of trajectories going backwards from this node
 
   void CalculateForwardsTrajectories();
   void CalculateBackwardsTrajectories();
 
public:
   KVGeoDetectorNode()
   {
      fTraj.SetCleanup();
   }
   KVGeoDetectorNode(const Char_t* name) : KVBase(name, "/FULL/PATH/TO/NODE")
   {
      fTraj.SetCleanup();
   }
 
   void SetDetector(KVDetector*);
   KVDetector* GetDetector() const;
   const Char_t* GetName() const override;
 
   void AddInFront(KVDetector*);
   void AddBehind(KVDetector*);
   Bool_t IsInFrontOf(KVDetector*);
   Bool_t IsBehind(KVDetector*);
   const KVSeqCollection* GetDetectorsInFront() const
   {
      return &fInFront;
   }
   const KVSeqCollection* GetDetectorsBehind() const
   {
      return &fBehind;
   }
   const KVSeqCollection* GetTrajectories() const
   {
      return &fTraj;
   }
   const KVSeqCollection* GetForwardTrajectories() const
   {
 
      if (fNTrajForwards < 0) const_cast<KVGeoDetectorNode*>(this)->CalculateForwardsTrajectories();
      return (fNTrajForwards ? &fTrajF : nullptr);
   }
   const KVSeqCollection* GetBackwardTrajectories() const
   {
 
      if (fNTrajBackwards < 0) const_cast<KVGeoDetectorNode*>(this)->CalculateBackwardsTrajectories();
      return (fNTrajBackwards ? &fTrajB : nullptr);
   }
   Int_t GetNDetsInFront() const
   {
      return fInFront.GetEntries();
   }
   Int_t GetNDetsBehind() const
   {
      return fBehind.GetEntries();
   }
   Int_t GetNTraj() const
   {
      if (fNTraj < 0) {
         const_cast<KVGeoDetectorNode*>(this)->fNTraj = fTraj.GetEntries();
      }
      return fNTraj;
   }
   Int_t GetNTrajForwards() const
   {
 
      if (fNTrajForwards < 0) const_cast<KVGeoDetectorNode*>(this)->CalculateForwardsTrajectories();
      return fNTrajForwards;
   }
   Int_t GetNTrajBackwards() const
   {
 
      if (fNTrajBackwards < 0) const_cast<KVGeoDetectorNode*>(this)->CalculateBackwardsTrajectories();
      return fNTrajBackwards;
   }
 
   void BuildTrajectoriesForwards(TSeqCollection*);
   void AddTrajectory(KVGeoDNTrajectory*);
 
   void RehashLists();
 
   KVGeoDNTrajectory* FindTrajectory(const char* title) const;
   KVGeoDNTrajectory* FindTrajectory(UInt_t number) const;
 
   const Char_t* GetFullPathToNode() const
   {
      return GetTitle();
   }
 
   void ls(Option_t* option = "") const override;
 
   ClassDefOverride(KVGeoDetectorNode, 2) //Information on relative positions of detectors & particle trajectories
};
 
#endif