SensitiveDetectorBase.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#ifndef SENSITIVEDETECTOR_H_
#define SENSITIVEDETECTOR_H_
 
#include <framework/datastore/RelationArray.h>
#include <mdst/dataobjects/MCParticle.h>
#include <simulation/kernel/UserInfo.h>
#include <G4VSensitiveDetector.hh>
class G4Step;
 
#include <map>
#include <string>
 
 
namespace Belle2 {
  namespace Simulation {
 
    class SensitiveDetectorBase : public G4VSensitiveDetector {
    public:
 
      SensitiveDetectorBase(const std::string& name, Const::EDetector subdetector):
        G4VSensitiveDetector(name), m_subdetector(subdetector) {}
 
      virtual ~SensitiveDetectorBase() {}
 
      static const std::map<std::string, RelationArray::EConsolidationAction>& getMCParticleRelations() { return s_mcRelations; }
      static void setActive(bool activeStatus) { s_active = activeStatus; }
      static void registerMCParticleRelation(const std::string& name,
                                             RelationArray::EConsolidationAction ignoreAction = RelationArray::c_negativeWeight);
 
      static void registerMCParticleRelation(const RelationArray& relation,
                                             RelationArray::EConsolidationAction ignoreAction = RelationArray::c_negativeWeight) { registerMCParticleRelation(relation.getName(), ignoreAction); }
 
    protected:
      virtual bool step(G4Step* step, G4TouchableHistory* ROhist) = 0;
    private:
      virtual bool ProcessHits(G4Step* aStep, G4TouchableHistory* aROhist);
      static std::map<std::string, RelationArray::EConsolidationAction> s_mcRelations;
      static bool s_active;
      Const::EDetector m_subdetector;
    };
 
    inline bool SensitiveDetectorBase::ProcessHits(G4Step* aStep, G4TouchableHistory* aROhist)
    {
      if (!s_active) return false;
      bool result = step(aStep, aROhist);
      // Do not include hits from invalid detector (beast,teastbeam, etc.)
      if (result && (m_subdetector != Const::invalidDetector)) TrackInfo::getInfo(*aStep->GetTrack()).addSeenInDetector(m_subdetector);
      return result;
    }
 
  }  // end namespace Simulation
} // end namespace Belle2
 
#endif /* SENSITIVEDETECTOR_H_ */