VRMLWriterModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Leo Piilonen                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#ifndef VRMLWRITERMODULE_H
#define VRMLWRITERMODULE_H
 
#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
#endif