VRMLWriterModule.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <framework/core/Module.h>
 
#include <string>
#include <fstream>
 
class G4VPhysicalVolume;
class G4LogicalVolume;
class G4VSolid;
class G4Polyhedron;
class G4String;
class HepPolyhedron;
 
namespace Belle2 {
  class VRMLWriterModule: public Module {
 
  public:
 
    VRMLWriterModule();
 
    void initialize() override;
 
    void event() override;
 
  private:
 
    void assignName(std::vector<std::string>*, unsigned int, const G4String&, int);
 
    void describeSolid(G4VSolid*, const std::string&, bool);
 
    void describeLogicalVolume(G4LogicalVolume*, const std::string&, const std::string&, bool);
 
    void describePhysicalVolume(G4VPhysicalVolume*);
 
    void descendAndDescribe(G4VPhysicalVolume*, const std::string&, int);
 
    void writePreamble(void);
 
    void writePolyhedron(const G4Polyhedron*, const std::string&);
 
    void writePhysicalVolume(const G4VPhysicalVolume*, const std::string&, const std::string&, bool);
 
    HepPolyhedron* getBooleanSolidPolyhedron(G4VSolid*);
 
    bool m_First{true};
 
    std::string m_Filename{"belle2.wrl"};
 
    std::ofstream m_File;
 
    std::vector<std::string>* m_PVName{nullptr};
 
    std::vector<std::string>* m_LVName{nullptr};
 
    std::vector<std::string>* m_SolidName{nullptr};
 
    std::vector<bool>* m_IsCylinder{nullptr};
 
    std::vector<std::vector<int> >* m_PVIndex{nullptr};
 
    std::vector<bool>* m_LVWritten{nullptr};
 
    std::vector<bool>* m_PVWritten{nullptr};
 
  };
 
} //Belle2 namespace