KVSegmentedDetector.h

#ifndef KVSEGMENTEDDETECTOR_H
#define KVSEGMENTEDDETECTOR_H
 
#include "KVDetector.h"
#include <unordered_map>
class KVSegmentedDetector : public KVDetector {
 
   mutable std::unordered_map<int, std::unique_ptr<KVDetector>> fSubDetectors;
   std::vector<int> fHitSubDetectors;
 
public:
   KVSegmentedDetector()
   {
   }
 
   void AddSubDetector(int i, KVDetector* D)
   {
      fSubDetectors[i].reset(D);
   }
 
   UInt_t GetNSegments() const
   {
      return fSubDetectors.size();
   }
   KVDetector* GetSubDetector(int i) const
   {
      return fSubDetectors[i].get();
   }
 
   Bool_t IsSegmented() const override
   {
      return kTRUE;
   }
 
   void AddHitSubDetector(int i)
   {
      fHitSubDetectors.push_back(i);
   }
 
   const std::vector<int>& GetHitSubDetectors() const
   {
      return fHitSubDetectors;
   }
 
   void AddEnergyLossInSubDetector(int i, double e) override;
 
   void Clear(Option_t* opt = "") override;
 
   ClassDefOverride(KVSegmentedDetector, 1)
};
 
#endif // KVSEGMENTEDDETECTOR_H