BDSParallelWorldSampler.hh

/* 
Beam Delivery Simulation (BDSIM) Copyright (C) Royal Holloway, 
University of London 2001 - 2022.
 
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 BDSPARALLELWORLDSAMPER_H
#define BDSPARALLELWORLDSAMPER_H
 
#include "BDSBeamPipeType.hh"
#include "BDSSamplerType.hh"
 
#include "globals.hh"
#include "G4String.hh"
#include "G4VUserParallelWorld.hh"
 
#include <map>
#include <vector>
 
class G4VisAttributes;
class G4VPhysicalVolume;
class BDSBeamlineElement;
class BDSSampler;
class BDSSamplerPlane;
namespace GMAD
{
  class SamplerPlacement;
}
 
class BDSParallelWorldSampler: public G4VUserParallelWorld
{
public:
  BDSParallelWorldSampler() = delete; 
  explicit BDSParallelWorldSampler(const G4String& name);
  virtual ~BDSParallelWorldSampler();
 
  virtual void Construct();
 
  BDSSamplerPlane* GeneralPlane() const {return generalPlane;}
  G4LogicalVolume* WorldLV() const {return samplerWorldLV;}
  
  void Place(const BDSBeamlineElement* element,
             G4double                  samplerRadius);
 
private:
  void ErrorNonPositive(G4double value,
                        const G4String& variableName,
                        const G4String& objectName) const;
  
  BDSSampler* BuildSampler(const GMAD::SamplerPlacement& samplerPlacement,
                           BDSSamplerType st,
                           G4double& radius) const;
  
  void AdjustTransform(G4Transform3D& trans,
                       BDSSamplerType st) const;
  
  std::vector<G4VPhysicalVolume*> placements;
  
  G4String suffix; 
  G4VisAttributes* samplerWorldVis; 
  BDSSamplerPlane* generalPlane; 
  std::map<int, BDSSamplerPlane*> samplerInstances;
  G4LogicalVolume* samplerWorldLV;
};
 
#endif