BDSIM
BDSIM is a Geant4 extension toolkit for simulation of particle transport in accelerator beamlines.
Loading...
Searching...
No Matches
BDSBunchGaussian.hh
1/*
2Beam Delivery Simulation (BDSIM) Copyright (C) Royal Holloway,
3University of London 2001 - 2023.
4
5This file is part of BDSIM.
6
7BDSIM is free software: you can redistribute it and/or modify
8it under the terms of the GNU General Public License as published
9by the Free Software Foundation version 3 of the License.
10
11BDSIM is distributed in the hope that it will be useful, but
12WITHOUT ANY WARRANTY; without even the implied warranty of
13MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14GNU General Public License for more details.
15
16You should have received a copy of the GNU General Public License
17along with BDSIM. If not, see <http://www.gnu.org/licenses/>.
18*/
19#ifndef BDSBUNCHGAUSSIAN_H
20#define BDSBUNCHGAUSSIAN_H
21
22#include "BDSBunch.hh"
23
24#include "globals.hh"
25#include "G4Transform3D.hh"
26
27#include "CLHEP/Matrix/SymMatrix.h"
28#include "CLHEP/Matrix/Vector.h"
29
30#include <vector>
31
32namespace CLHEP
33{
34 class HepRandomEngine;
35 class RandMultiGauss;
36}
37
38namespace GMAD
39{
40 class Beam;
41}
42
50{
51public:
52 BDSBunchGaussian() = delete;
53 explicit BDSBunchGaussian(const G4String& nameIn);
54 virtual ~BDSBunchGaussian();
55
56 virtual void SetOptions(const BDSParticleDefinition* beamParticle,
57 const GMAD::Beam& beam,
58 const BDSBunchType& distrType,
59 G4Transform3D beamlineTransformIn = G4Transform3D::Identity,
60 const G4double beamlineS = 0);
61
64 virtual void BeginOfRunAction(G4int numberOfEvents,
65 G4bool batchMode);
66
70
71protected:
74 CLHEP::RandMultiGauss* CreateMultiGauss(CLHEP::HepRandomEngine& anEngine,
75 const CLHEP::HepVector& mu,
76 CLHEP::HepSymMatrix& sigma);
77
79 void PreGenerateEvents(G4int nGenerate);
80
83
84 CLHEP::HepVector meansGM;
85 CLHEP::HepSymMatrix sigmaGM;
86
88 CLHEP::RandMultiGauss* gaussMultiGen;
89
91
93 std::vector<G4double> x0_v, xp_v, y0_v, yp_v, z0_v, zp_v,E_v,t_v,weight_v;
96
98 std::vector<std::vector<G4double>* > coordinates;
99};
100
101#endif
Common functionality for a 6D Gaussian distribution.
CLHEP::RandMultiGauss * CreateMultiGauss(CLHEP::HepRandomEngine &anEngine, const CLHEP::HepVector &mu, CLHEP::HepSymMatrix &sigma)
std::vector< G4double > z0_v
Holder for pre-calculated coordinates.
std::vector< G4double > y0_v
Holder for pre-calculated coordinates.
std::vector< G4double > xp_v
Holder for pre-calculated coordinates.
virtual BDSParticleCoordsFull GetNextParticleLocal()
std::vector< G4double > t_v
Holder for pre-calculated coordinates.
CLHEP::RandMultiGauss * gaussMultiGen
Randon number generator with sigma matrix and mean.
virtual void SetOptions(const BDSParticleDefinition *beamParticle, const GMAD::Beam &beam, const BDSBunchType &distrType, G4Transform3D beamlineTransformIn=G4Transform3D::Identity, const G4double beamlineS=0)
std::vector< std::vector< G4double > * > coordinates
Convenience vector of vectors for clearing up.
std::vector< G4double > weight_v
Holder for pre-calculated coordinates.
std::vector< G4double > zp_v
Holder for pre-calculated coordinates.
G4int iPartIteration
Iterator for reading out pre-calculate coordinates.
std::vector< G4double > yp_v
Holder for pre-calculated coordinates.
virtual void BeginOfRunAction(G4int numberOfEvents, G4bool batchMode)
virtual BDSParticleCoordsFull GetNextParticleLocalCoords()
Fire random number generator and get coordinates. Can be overloaded if required.
std::vector< G4double > E_v
Holder for pre-calculated coordinates.
void PreGenerateEvents(G4int nGenerate)
Pre-generate all the particle coordinates and subtract the sample mean.
G4bool offsetSampleMean
Whether to offset the sample mean.
std::vector< G4double > x0_v
Holder for pre-calculated coordinates.
The base class for bunch distribution generators.
Definition: BDSBunch.hh:47
G4double beamlineS
Beamline initial S position.
Definition: BDSBunch.hh:219
A set of particle coordinates including energy and weight.
Wrapper for particle definition.
Improve type-safety of native enum data type in C++.
Beam loader.
Definition: Beam.hh:37
Beam class.
Definition: beam.h:44
Parser namespace for GMAD language. Combination of Geant4 and MAD.