BDSFieldBuilder.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 BDSFIELDBUILDER_H
#define BDSFIELDBUILDER_H
 
#include "globals.hh" //G4 global constants & types
 
#include <map>
#include <set>
#include <vector>
 
class BDSFieldInfo;
class BDSFieldObjects;
class BDSMagnetStrength;
class G4LogicalVolume;
 
class BDSFieldBuilder
{
public:
  static BDSFieldBuilder* Instance();
 
  ~BDSFieldBuilder();
 
  void RegisterFieldForConstruction(const BDSFieldInfo*      info,
                                    G4LogicalVolume*         logicalVolume,
                                    const G4bool             propagateToDaughters     = false,
                                    const BDSMagnetStrength* magnetStrengthForScaling = nullptr,
                                    const G4String&          scalingKey               = "none");
 
  void RegisterFieldForConstruction(const BDSFieldInfo*      info,
                                    const std::vector<G4LogicalVolume*>& logicalVolumes,
                                    const G4bool             propagateToDaughters     = false,
                                    const BDSMagnetStrength* magnetStrengthForScaling = nullptr,
                                    const G4String&          scalingKey               = "none");
 
  void RegisterFieldForConstruction(const BDSFieldInfo*      info,
                                    const std::set<G4LogicalVolume*>& logicalVolumes,
                                    const G4bool             propagateToDaughters     = false,
                                    const BDSMagnetStrength* magnetStrengthForScaling = nullptr,
                                    const G4String&          scalingKey               = "none");
 
  std::vector<BDSFieldObjects*> CreateAndAttachAll();
 
private:
  BDSFieldBuilder();
 
  static BDSFieldBuilder* instance;
  
  std::vector<const BDSFieldInfo*> infos;
  std::vector<std::vector<G4LogicalVolume*> > lvs;
  std::vector<G4bool>              propagators;
 
  std::map<G4int, const BDSMagnetStrength*> scalingStrengths;
  std::map<G4int, G4String> scalingKeys;
};
 
#endif