TargetDarkBremFilter.h

Go to the documentation of this file.

 
#ifndef BIASING_TARGETDARKBREMFILTER_H_
#define BIASING_TARGETDARKBREMFILTER_H_
 
//----------------//
//   C++ StdLib   //
//----------------//
#include <algorithm>
 
//------------//
//   Geant4   //
//------------//
#include "G4RunManager.hh"
#include "G4VPhysicalVolume.hh"
 
//------------//
//    LDMX    //
//------------//
#include "SimCore/G4User/PtrRetrieval.h"
#include "SimCore/G4User/UserAction.h"
 
/*~~~~~~~~~~*/
/*   Core   */
/*~~~~~~~~~~*/
#include "Framework/Logger.h"
#include "G4DarkBreM/G4APrime.h"  //checking if particles match A'
#include "G4Electron.hh"          //to check if track is electron
#include "SimCore/G4User/UserTrackInformation.h"  //make sure A' is saved
 
namespace biasing {
 
class TargetDarkBremFilter : public simcore::UserAction {
 public:
  TargetDarkBremFilter(const std::string& name,
                       framework::config::Parameters& parameters);
 
  ~TargetDarkBremFilter() = default;
 
  std::vector<simcore::TYPE> getTypes() override {
    return {simcore::TYPE::STEPPING, simcore::TYPE::EVENT};
  }
 
  void BeginOfEventAction(const G4Event* e) override;
 
  void stepping(const G4Step* step) override;
 
  void EndOfEventAction(const G4Event* event) override;
 
 private:
  inline bool isOutsideTargetRegion(const G4VPhysicalVolume* vol) const {
    return vol ? isOutsideTargetRegion(vol->GetLogicalVolume()) : true;
  }
 
  inline bool isOutsideTargetRegion(const G4LogicalVolume* vol) const {
    if (!vol) return true;
    auto target_region = simcore::g4user::ptrretrieval::getRegion("target");
    if (!target_region) {
      ldmx_log(warn) << "Region 'target' not found in Geant4 region store";
    }
    auto region = vol->GetRegion();
    return region ? (region != target_region) : true;
  }
 
  void AbortEvent(const std::string& reason) const;
 
 private:
  double threshold_;
 
  bool found_aprime_;
 
};  // TargetDarkBremFilter
}  // namespace biasing
 
#endif  // BIASING_TARGETDARKBREMFILTER_H__