Jan Kieseler
107 lines
4.0 KiB
C++
107 lines
4.0 KiB
C++
//
|
|
// ********************************************************************
|
|
// * License and Disclaimer *
|
|
// * *
|
|
// * The Geant4 software is copyright of the Copyright Holders of *
|
|
// * the Geant4 Collaboration. It is provided under the terms and *
|
|
// * conditions of the Geant4 Software License, included in the file *
|
|
// * LICENSE and available at http://cern.ch/geant4/license . These *
|
|
// * include a list of copyright holders. *
|
|
// * *
|
|
// * Neither the authors of this software system, nor their employing *
|
|
// * institutes,nor the agencies providing financial support for this *
|
|
// * work make any representation or warranty, express or implied, *
|
|
// * regarding this software system or assume any liability for its *
|
|
// * use. Please see the license in the file LICENSE and URL above *
|
|
// * for the full disclaimer and the limitation of liability. *
|
|
// * *
|
|
// * This code implementation is the result of the scientific and *
|
|
// * technical work of the GEANT4 collaboration. *
|
|
// * By using, copying, modifying or distributing the software (or *
|
|
// * any work based on the software) you agree to acknowledge its *
|
|
// * use in resulting scientific publications, and indicate your *
|
|
// * acceptance of all terms of the Geant4 Software license. *
|
|
// ********************************************************************
|
|
//
|
|
//
|
|
/// \file B4/B4a/include/PrimaryGeneratorAction.hh
|
|
/// \brief Definition of the B4::PrimaryGeneratorAction class
|
|
|
|
#ifndef B4PrimaryGeneratorAction_h
|
|
#define B4PrimaryGeneratorAction_h 1
|
|
|
|
#include "G4VUserPrimaryGeneratorAction.hh"
|
|
#include "globals.hh"
|
|
|
|
class G4ParticleGun;
|
|
class G4Event;
|
|
|
|
namespace B4
|
|
{
|
|
|
|
/// The primary generator action class with particle gum.
|
|
///
|
|
/// It defines a single particle which hits the calorimeter
|
|
/// perpendicular to the input face. The type of the particle
|
|
/// can be changed via the G4 build-in commands of G4ParticleGun class
|
|
/// (see the macros provided with this example).
|
|
|
|
class PrimaryGeneratorAction : public G4VUserPrimaryGeneratorAction
|
|
{
|
|
public:
|
|
PrimaryGeneratorAction();
|
|
~PrimaryGeneratorAction() override;
|
|
|
|
void GeneratePrimaries(G4Event* event) override;
|
|
|
|
// set methods
|
|
void SetRandomFlag(G4bool value);
|
|
G4double getPartEnergy() const{return partEnergy;}
|
|
void setMinPartEnergy(G4double value){minPartEnergy = value;}
|
|
void setMaxPartEnergy(G4double value){maxPartEnergy = value;}
|
|
void setPartSpecies(const std::vector<std::string>& values){
|
|
partSpecies = values;
|
|
for(const auto value : values){
|
|
G4String val = value;
|
|
if(val == "e-"){
|
|
G4cout << "Particle species set to electron" << G4endl;
|
|
}
|
|
else if(val == "e+"){
|
|
G4cout << "Particle species set to positron" << G4endl;
|
|
}
|
|
else if(val == "gamma"){
|
|
G4cout << "Particle species set to gamma" << G4endl;
|
|
}
|
|
else if(val == "proton"){
|
|
G4cout << "Particle species set to proton" << G4endl;
|
|
}
|
|
else if(val == "neutron"){
|
|
G4cout << "Particle species set to neutron" << G4endl;
|
|
}
|
|
else if(val == "pi+"){
|
|
G4cout << "Particle species set to charged pion" << G4endl;
|
|
}
|
|
else if(val == "pi-"){
|
|
G4cout << "Particle species set to charged pion" << G4endl;
|
|
}
|
|
else{
|
|
G4cout << "Particle species "<< val << " not recognized" << G4endl;
|
|
throw std::invalid_argument("Particle species not recognized");
|
|
}
|
|
}
|
|
}
|
|
private:
|
|
G4ParticleGun* fParticleGun = nullptr; // G4 particle gun
|
|
G4double partEnergy;
|
|
G4double minPartEnergy=100; //MeV
|
|
G4double maxPartEnergy=100;
|
|
std::vector<std::string> partSpecies;
|
|
|
|
};
|
|
|
|
}
|
|
|
|
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
|
|
|
|
#endif
|