BDSPrimaryGeneratorAction.hh

/* 
Beam Delivery Simulation (BDSIM) Copyright (C) Royal Holloway, 
University of London 2001 - 2023.
 
This file is part of BDSIM.
 
BDSIM is free software: you can redistribute it and/or modify 
it under the terms of the GNU General Public License as published 
by the Free Software Foundation version 3 of the License.
 
BDSIM is distributed in the hope that it will be useful, but 
WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.
 
You should have received a copy of the GNU General Public License
along with BDSIM.  If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef BDSPRIMARYGENERATORACTION_H
#define BDSPRIMARYGENERATORACTION_H
 
#include "BDSExtent.hh"
 
#include "globals.hh"
#include "G4VUserPrimaryGeneratorAction.hh"
 
class BDSBunch;
class BDSOutputLoader;
class BDSPrimaryGeneratorFile;
class BDSPTCOneTurnMap;
class G4Event;
class G4ParticleGun;
 
namespace GMAD
{
  class Beam;
}
 
 
class BDSPrimaryGeneratorAction: public G4VUserPrimaryGeneratorAction
{
public:
  BDSPrimaryGeneratorAction(BDSBunch*         bunchIn,
                            const GMAD::Beam& beam,
                            G4bool            batchMode);
  virtual ~BDSPrimaryGeneratorAction();
 
  virtual void GeneratePrimaries(G4Event*);
 
  inline void SetWorldExtent(const BDSExtent worldExtentIn) {worldExtent = worldExtentIn;}
  void RegisterPTCOneTurnMap(BDSPTCOneTurnMap* otmIn) {oneTurnMap = otmIn;}
  
private:
  void GeneratePrimariesFromFile(G4Event* anEvent);
  
  G4ParticleGun* particleGun;     
  BDSBunch* bunch;                
  G4bool writeASCIISeedState;     
  BDSOutputLoader* recreateFile;  
  G4bool   recreate;              
  G4int    eventOffset;           
  G4bool   useASCIISeedState;     
  G4bool   ionPrimary;            
  G4bool   distrFileMatchLength;  
  
  BDSExtent worldExtent;
 
  G4bool ionCached; 
 
  BDSPTCOneTurnMap* oneTurnMap;
 
  BDSPrimaryGeneratorFile* generatorFromFile;
};
 
#endif