FBXWriterModule.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 FBXWRITERMODULE_H
#define FBXWRITERMODULE_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 FBXWriterModule: public Module {
 
  public:
 
    FBXWriterModule();
 
    void initialize() override;
 
    void event() override;
 
  private:
 
    void assignName(std::vector<std::string>*, unsigned int, const G4String&, int);
 
    void writeGeometryNode(G4VSolid*, const std::string&, unsigned long long);
 
    void writeMaterialNode(int, const std::string&);
 
    void writeLVModelNode(G4LogicalVolume*, const std::string&, unsigned long long);
 
    void writePVModelNode(G4VPhysicalVolume*, const std::string&, unsigned long long);
 
    void countEntities(G4VPhysicalVolume*);
 
    void addModels(G4VPhysicalVolume*, int);
 
    void addConnections(G4VPhysicalVolume*, int);
 
    void writePreamble(int, int, int);
 
    void writePolyhedron(G4VSolid*, G4Polyhedron*, const std::string&, unsigned long long);
 
    void writeSolidToLV(const std::string&, const std::string&, bool, unsigned long long, unsigned long long, unsigned long long);
 
    void writeSolidToPV(const std::string&, const std::string&, bool, unsigned long long, unsigned long long, unsigned long long);
 
    void writeLVToPV(const std::string&, const std::string&, unsigned long long, unsigned long long);
 
    void writePVToParentLV(const std::string&, const std::string&, unsigned long long, unsigned long long);
 
    void writePVToParentPV(const std::string&, const std::string&, unsigned long long, unsigned long long);
 
    HepPolyhedron* getBooleanSolidPolyhedron(G4VSolid*);
 
    bool m_First{true};
 
    bool m_UsePrototypes{false};
 
    std::string m_Filename{"belle2.fbx"};
 
    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<unsigned long long>* m_PVID{nullptr};
 
    std::vector<unsigned long long>* m_LVID{nullptr};
 
    std::vector<unsigned long long>* m_MatID{nullptr};
 
    std::vector<unsigned long long>* m_SolidID{nullptr};
 
    std::vector<bool>* m_Visible{nullptr};
 
    std::vector<unsigned int>* m_PVCount{nullptr};
 
    std::vector<unsigned int>* m_LVCount{nullptr};
 
    std::vector<unsigned int>* m_SolidCount{nullptr};
 
    std::vector<unsigned int>* m_PVReplicas{nullptr};
 
    std::vector<unsigned int>* m_LVReplicas{nullptr};
 
    std::vector<unsigned int>* m_SolidReplicas{nullptr};
 
    std::vector<bool>* m_LVUnique{nullptr};
 
  };
 
} //Belle2 namespace
#endif