vtk/src/PrimaryGeneratorAction.cc

0001 //
0002 // ********************************************************************
0003 // * License and Disclaimer                                           *
0004 // *                                                                  *
0005 // * The  Geant4 software  is  copyright of the Copyright Holders  of *
0006 // * the Geant4 Collaboration.  It is provided  under  the terms  and *
0007 // * conditions of the Geant4 Software License,  included in the file *
0008 // * LICENSE and available at  http://cern.ch/geant4/license .  These *
0009 // * include a list of copyright holders.                             *
0010 // *                                                                  *
0011 // * Neither the authors of this software system, nor their employing *
0012 // * institutes,nor the agencies providing financial support for this *
0013 // * work  make  any representation or  warranty, express or implied, *
0014 // * regarding  this  software system or assume any liability for its *
0015 // * use.  Please see the license in the file  LICENSE  and URL above *
0016 // * for the full disclaimer and the limitation of liability.         *
0017 // *                                                                  *
0018 // * This  code  implementation is the result of  the  scientific and *
0019 // * technical work of the GEANT4 collaboration.                      *
0020 // * By using,  copying,  modifying or  distributing the software (or *
0021 // * any work based  on the software)  you  agree  to acknowledge its *
0022 // * use  in  resulting  scientific  publications,  and indicate your *
0023 // * acceptance of all terms of the Geant4 Software license.          *
0024 // ********************************************************************
0025 //
0026 /// \file PrimaryGeneratorAction.cc
0027 /// \brief Implementation of the VtkVis::PrimaryGeneratorAction class
0028
0029 #include "PrimaryGeneratorAction.hh"
0030
0031 #include "G4Box.hh"
0032 #include "G4LogicalVolume.hh"
0033 #include "G4LogicalVolumeStore.hh"
0034 #include "G4ParticleDefinition.hh"
0035 #include "G4ParticleGun.hh"
0036 #include "G4ParticleTable.hh"
0037 #include "G4RunManager.hh"
0038 #include "G4SystemOfUnits.hh"
0039 #include "Randomize.hh"
0040
0041 namespace VtkVis
0042 {
0043
0044 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
0045
0046 PrimaryGeneratorAction::PrimaryGeneratorAction()
0047 {
0048   G4int n_particle = 1;
0049   fParticleGun = new G4ParticleGun(n_particle);
0050
0051   // default particle kinematic
0052   G4ParticleTable* particleTable = G4ParticleTable::GetParticleTable();
0053   G4String particleName;
0054   G4ParticleDefinition* particle = particleTable->FindParticle(particleName = "e-");
0055   fParticleGun->SetParticleDefinition(particle);
0056   fParticleGun->SetParticleMomentumDirection(G4ThreeVector(0., 0., 1.));
0057   fParticleGun->SetParticleEnergy(16.5 * GeV);
0058 }
0059
0060 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
0061
0062 PrimaryGeneratorAction::~PrimaryGeneratorAction()
0063 {
0064   delete fParticleGun;
0065 }
0066
0067 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
0068
0069 void PrimaryGeneratorAction::GeneratePrimaries(G4Event* anEvent)
0070 {
0071   // this function is called at the begining of ecah event
0072   //
0073
0074   // In order to avoid dependence of PrimaryGeneratorAction
0075   // on DetectorConstruction class we get Envelope volume
0076   // from G4LogicalVolumeStore.
0077
0078   G4double envSizeXY = 0;
0079   G4double envSizeZ = 0;
0080
0081   if (!fEnvelopeBox) {
0082     G4LogicalVolume* envLV = G4LogicalVolumeStore::GetInstance()->GetVolume("Envelope");
0083     if (envLV) fEnvelopeBox = dynamic_cast<G4Box*>(envLV->GetSolid());
0084   }
0085
0086   if (fEnvelopeBox) {
0087     envSizeXY = fEnvelopeBox->GetXHalfLength() * 2.;
0088     envSizeZ = fEnvelopeBox->GetZHalfLength() * 2.;
0089   }
0090   else {
0091     G4ExceptionDescription msg;
0092     msg << "Envelope volume of box shape not found.\n";
0093     msg << "Perhaps you have changed geometry.\n";
0094     msg << "The gun will be place at the center.";
0095     G4Exception("PrimaryGeneratorAction::GeneratePrimaries()", "MyCode0002", JustWarning, msg);
0096   }
0097
0098   G4double size = 0.8;
0099   G4double x0 = size * envSizeXY * (G4UniformRand() - 0.5);
0100   G4double y0 = size * envSizeXY * (G4UniformRand() - 0.5);
0101   G4double z0 = 0.5 * envSizeZ;
0102
0103   fParticleGun->SetParticlePosition(G4ThreeVector(0, 0, -z0));
0104
0105   fParticleGun->GeneratePrimaryVertex(anEvent);
0106 }
0107
0108 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
0109
0110 }  // namespace VtkVis