KVGeoDetectorNode.h

 
#ifndef __KVDETECTORNODE_H
#define __KVDETECTORNODE_H
 
#include "KVBase.h"
#include "KVSeqCollection.h"
 
class KVDetector;
class KVGeoDNTrajectory;
 
class KVGeoDetectorNode : public KVBase {
    KVDetector* fDetector;
    KVSeqCollection* fInFront;
    KVSeqCollection* fBehind;
    KVSeqCollection* fTraj;
    KVSeqCollection* fTrajF;
    KVSeqCollection* fTrajB;
    Int_t fNTraj;
    Int_t fNTrajForwards;
    Int_t fNTrajBackwards;
 
   void init();
 
   void CalculateForwardsTrajectories();
   void CalculateBackwardsTrajectories();
 
public:
   KVGeoDetectorNode();
   KVGeoDetectorNode(const Char_t* name);
   virtual ~KVGeoDetectorNode();
 
   void SetDetector(KVDetector*);
   KVDetector* GetDetector() const;
   const Char_t* GetName() const;
 
   void AddInFront(KVDetector*);
   void AddBehind(KVDetector*);
   Bool_t IsInFrontOf(KVDetector*);
   Bool_t IsBehind(KVDetector*);
   KVSeqCollection* GetDetectorsInFront() const
   {
      return fInFront;
   }
   KVSeqCollection* GetDetectorsBehind() const
   {
      return fBehind;
   }
   KVSeqCollection* GetTrajectories() const
   {
      return fTraj;
   }
   KVSeqCollection* GetForwardTrajectories() const;
   KVSeqCollection* GetBackwardTrajectories() const;
   Int_t GetNDetsInFront() const;
   Int_t GetNDetsBehind() const;
   Int_t GetNTraj() const;
   Int_t GetNTrajForwards() const;
   Int_t GetNTrajBackwards() const;
 
   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;
 
   ClassDef(KVGeoDetectorNode, 2) //Information on relative positions of detectors & particle trajectories
};
 
#endif