Geant4 Cross Reference

Cross-Referencing   Geant4
Geant4/examples/extended/parameterisations/Par04/src/Par04MLFastSimModel.cc

Version: [ ReleaseNotes ] [ 1.0 ] [ 1.1 ] [ 2.0 ] [ 3.0 ] [ 3.1 ] [ 3.2 ] [ 4.0 ] [ 4.0.p1 ] [ 4.0.p2 ] [ 4.1 ] [ 4.1.p1 ] [ 5.0 ] [ 5.0.p1 ] [ 5.1 ] [ 5.1.p1 ] [ 5.2 ] [ 5.2.p1 ] [ 5.2.p2 ] [ 6.0 ] [ 6.0.p1 ] [ 6.1 ] [ 6.2 ] [ 6.2.p1 ] [ 6.2.p2 ] [ 7.0 ] [ 7.0.p1 ] [ 7.1 ] [ 7.1.p1 ] [ 8.0 ] [ 8.0.p1 ] [ 8.1 ] [ 8.1.p1 ] [ 8.1.p2 ] [ 8.2 ] [ 8.2.p1 ] [ 8.3 ] [ 8.3.p1 ] [ 8.3.p2 ] [ 9.0 ] [ 9.0.p1 ] [ 9.0.p2 ] [ 9.1 ] [ 9.1.p1 ] [ 9.1.p2 ] [ 9.1.p3 ] [ 9.2 ] [ 9.2.p1 ] [ 9.2.p2 ] [ 9.2.p3 ] [ 9.2.p4 ] [ 9.3 ] [ 9.3.p1 ] [ 9.3.p2 ] [ 9.4 ] [ 9.4.p1 ] [ 9.4.p2 ] [ 9.4.p3 ] [ 9.4.p4 ] [ 9.5 ] [ 9.5.p1 ] [ 9.5.p2 ] [ 9.6 ] [ 9.6.p1 ] [ 9.6.p2 ] [ 9.6.p3 ] [ 9.6.p4 ] [ 10.0 ] [ 10.0.p1 ] [ 10.0.p2 ] [ 10.0.p3 ] [ 10.0.p4 ] [ 10.1 ] [ 10.1.p1 ] [ 10.1.p2 ] [ 10.1.p3 ] [ 10.2 ] [ 10.2.p1 ] [ 10.2.p2 ] [ 10.2.p3 ] [ 10.3 ] [ 10.3.p1 ] [ 10.3.p2 ] [ 10.3.p3 ] [ 10.4 ] [ 10.4.p1 ] [ 10.4.p2 ] [ 10.4.p3 ] [ 10.5 ] [ 10.5.p1 ] [ 10.6 ] [ 10.6.p1 ] [ 10.6.p2 ] [ 10.6.p3 ] [ 10.7 ] [ 10.7.p1 ] [ 10.7.p2 ] [ 10.7.p3 ] [ 10.7.p4 ] [ 11.0 ] [ 11.0.p1 ] [ 11.0.p2 ] [ 11.0.p3, ] [ 11.0.p4 ] [ 11.1 ] [ 11.1.1 ] [ 11.1.2 ] [ 11.1.3 ] [ 11.2 ] [ 11.2.1 ] [ 11.2.2 ] [ 11.3.0 ]

  1 //
  2 // ********************************************************************
  3 // * License and Disclaimer                                           *
  4 // *                                                                  *
  5 // * The  Geant4 software  is  copyright of the Copyright Holders  of *
  6 // * the Geant4 Collaboration.  It is provided  under  the terms  and *
  7 // * conditions of the Geant4 Software License,  included in the file *
  8 // * LICENSE and available at  http://cern.ch/geant4/license .  These *
  9 // * include a list of copyright holders.                             *
 10 // *                                                                  *
 11 // * Neither the authors of this software system, nor their employing *
 12 // * institutes,nor the agencies providing financial support for this *
 13 // * work  make  any representation or  warranty, express or implied, *
 14 // * regarding  this  software system or assume any liability for its *
 15 // * use.  Please see the license in the file  LICENSE  and URL above *
 16 // * for the full disclaimer and the limitation of liability.         *
 17 // *                                                                  *
 18 // * This  code  implementation is the result of  the  scientific and *
 19 // * technical work of the GEANT4 collaboration.                      *
 20 // * By using,  copying,  modifying or  distributing the software (or *
 21 // * any work based  on the software)  you  agree  to acknowledge its *
 22 // * use  in  resulting  scientific  publications,  and indicate your *
 23 // * acceptance of all terms of the Geant4 Software license.          *
 24 // ********************************************************************
 25 //
 26 #ifdef USE_INFERENCE
 27 #  include "Par04MLFastSimModel.hh"
 28 
 29 #  include "Par04InferenceSetup.hh"  // for Par04InferenceSetup
 30 
 31 #  include "G4Electron.hh"  // for G4Electron
 32 #  include "G4FastHit.hh"  // for G4FastHit
 33 #  include "G4FastSimHitMaker.hh"  // for G4FastSimHitMaker
 34 #  include "G4Gamma.hh"  // for G4Gamma
 35 #  include "G4Positron.hh"  // for G4Positron
 36 
 37 #  include <G4FastStep.hh>  // for G4FastStep
 38 #  include <G4FastTrack.hh>  // for G4FastTrack
 39 #  include <G4Track.hh>  // for G4Track
 40 #  include <G4VFastSimulationModel.hh>  // for G4VFastSimulationModel
 41 #  include <stddef.h>  // for size_t
 42 class G4ParticleDefinition;
 43 class G4Region;
 44 
 45 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
 46 
 47 Par04MLFastSimModel::Par04MLFastSimModel(G4String aModelName, G4Region* aEnvelope)
 48   : G4VFastSimulationModel(aModelName, aEnvelope),
 49     fInference(new Par04InferenceSetup),
 50     fHitMaker(new G4FastSimHitMaker),
 51     fParallelHitMaker(new G4FastSimHitMaker)
 52 {
 53   fParallelHitMaker->SetNameOfWorldWithSD("parallelWorldFastSim");
 54 }
 55 
 56 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
 57 
 58 Par04MLFastSimModel::Par04MLFastSimModel(G4String aModelName)
 59   : G4VFastSimulationModel(aModelName),
 60     fInference(new Par04InferenceSetup),
 61     fHitMaker(new G4FastSimHitMaker),
 62     fParallelHitMaker(new G4FastSimHitMaker)
 63 {
 64   fParallelHitMaker->SetNameOfWorldWithSD("parallelWorldFastSim");
 65 }
 66 
 67 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
 68 
 69 Par04MLFastSimModel::~Par04MLFastSimModel() {}
 70 
 71 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
 72 
 73 G4bool Par04MLFastSimModel::IsApplicable(const G4ParticleDefinition& aParticleType)
 74 {
 75   return &aParticleType == G4Electron::ElectronDefinition()
 76          || &aParticleType == G4Positron::PositronDefinition()
 77          || &aParticleType == G4Gamma::GammaDefinition();
 78 }
 79 
 80 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
 81 
 82 G4bool Par04MLFastSimModel::ModelTrigger(const G4FastTrack& aFastTrack)
 83 {
 84   return fInference->IfTrigger(aFastTrack.GetPrimaryTrack()->GetKineticEnergy());
 85 }
 86 
 87 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
 88 
 89 void Par04MLFastSimModel::DoIt(const G4FastTrack& aFastTrack, G4FastStep& aFastStep)
 90 {
 91   // remove particle from further processing by G4
 92   aFastStep.KillPrimaryTrack();
 93   aFastStep.SetPrimaryTrackPathLength(0.0);
 94   G4double energy = aFastTrack.GetPrimaryTrack()->GetKineticEnergy();
 95   aFastStep.SetTotalEnergyDeposited(energy);
 96   G4ThreeVector position = aFastTrack.GetPrimaryTrack()->GetPosition();
 97   G4ThreeVector direction = aFastTrack.GetPrimaryTrack()->GetMomentumDirection();
 98 
 99   // calculate the incident angle
100   G4float angle = direction.theta();
101 
102   // calculate how to deposit energy within the detector
103   // get it from inference model
104   fInference->GetEnergies(fEnergies, energy, angle);
105   fInference->GetPositions(fPositions, position, direction);
106 
107   // deposit energy in the detector using calculated values of energy deposits
108   // and positions
109   for (size_t iHit = 0; iHit < fPositions.size(); iHit++) {
110     if (fEnergies[iHit] > 0.0005) {
111       // Place hit in the physical readout of the detector
112       fParallelHitMaker->make(G4FastHit(fPositions[iHit], fEnergies[iHit]), aFastTrack);
113       // Place hit in the virtual grid, for validation purposes
114       fHitMaker->make(G4FastHit(fPositions[iHit], fEnergies[iHit]), aFastTrack);
115     }
116   }
117 }
118 #endif
119