BDSGeometryWriter.hh

/* 
Beam Delivery Simulation (BDSIM) Copyright (C) Royal Holloway, 
University of London 2001 - 2024.
 
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 BDSGEOMETRYWRITER_H
#define BDSGEOMETRYWRITER_H
 
#include "globals.hh" // geant4 types / globals
#include "G4VisAttributes.hh"
#include "G4String.hh"
 
#include <map>
 
class G4Colour;
class G4GDMLParser;
class G4LogicalVolume;
class G4VPhysicalVolume;
struct G4GDMLAuxStructType;
 
class BDSGeometryWriter
{
public:
  BDSGeometryWriter();
  ~BDSGeometryWriter();
 
  const static G4String auxType;
 
  void ExportGeometry(const G4String& geometryType,
                      const G4String& geometryFileName);
 
  static G4String ColourToVRGBAString(G4bool visible,
                                      const G4Colour& colour);
 
private:
#ifdef USE_GDML
  void WriteToGDML(const G4String& outputFileName, G4VPhysicalVolume* volume = nullptr);
 
  static std::map<G4LogicalVolume*, G4GDMLAuxStructType> PrepareColourInformation(G4VPhysicalVolume* volume);
 
  static void RegisterVolumeAuxiliaryInformation(G4GDMLParser& parser,
                                                 const std::map<G4LogicalVolume*, G4GDMLAuxStructType>& volToAuxMap);
 
  static void AddLVColour(std::map<G4LogicalVolume*, G4GDMLAuxStructType>&, G4LogicalVolume* lv);
#endif
};
 
#endif