GeoTOPCreator.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 <top/dbobjects/TOPGeometry.h>
#include <top/geometry/TOPGeometryPar.h>
#include <geometry/CreatorBase.h>
#include <framework/gearbox/GearDir.h>
#include <framework/database/IntervalOfValidity.h>
 
class G4LogicalVolume;
class G4AssemblyVolume;
class G4Box;
class G4UnionSolid;
 
namespace Belle2 {
 
  class BkgSensitiveDetector;
 
  namespace TOP {
 
    class SensitivePMT;
    class SensitiveBar;
 
    typedef std::vector<std::pair<double, double> > Polygon;

    class GeoTOPCreator : public geometry::CreatorBase {
 
    public:

      GeoTOPCreator();

      virtual ~GeoTOPCreator() override;

      GeoTOPCreator(const GeoTOPCreator&) = delete;

      GeoTOPCreator& operator=(const GeoTOPCreator&) = delete;

      virtual void create(const GearDir& content, G4LogicalVolume& topVolume,
                          geometry::GeometryTypes type) override;

      virtual void createFromDB(const std::string& name, G4LogicalVolume& topVolume,
                                geometry::GeometryTypes type) override;

      virtual void createPayloads(const GearDir& content,
                                  const IntervalOfValidity& iov) override;
 
 
    private:

      enum EPanelType {c_Inner, c_Outer};

      enum ESideRailType {c_Left, c_Right};

      void createGeometry(const TOPGeometry& parameters,
                          G4LogicalVolume& topVolume,
                          geometry::GeometryTypes type);

      G4LogicalVolume* createModule(const TOPGeometry& geo, int moduleID);

      G4LogicalVolume* createModuleEnvelope(const TOPGeoQBB& geo, int moduleID);

      static G4AssemblyVolume* assembleQBB(const TOPGeoQBB& geo);

      static G4LogicalVolume* createHoneycombPanel(const TOPGeoQBB& geo, EPanelType type);

      static G4LogicalVolume* createSideRail(const TOPGeoQBB& geo, ESideRailType type);

      G4AssemblyVolume* assembleFrontEnd(const TOPGeoFrontEnd& geo, int N);

      static G4LogicalVolume* createBoardStack(const TOPGeoFrontEnd& geo, int N);

      G4AssemblyVolume* assembleOptics(const TOPGeoModule& geo);

      G4LogicalVolume* createBarSegment(const TOPGeoBarSegment& geo, int moduleID);

      G4LogicalVolume* createPrism(const TOPGeoPrism& geo, int moduleID);

      G4LogicalVolume* createMirrorSegment(const TOPGeoMirrorSegment& geo, int moduleID);

      G4LogicalVolume* createPMTArray(const TOPGeoPMTArray& geo, int moduleID);

      G4LogicalVolume* createPMT(const TOPGeoPMT& geo);

      static G4LogicalVolume* createBox(const std::string& name,
                                        double A, double B, double C,
                                        const std::string& materialName);

      static G4LogicalVolume* createBoxSphereIntersection(const std::string& name,
                                                          G4Box* box,
                                                          double Rmin, double Rmax,
                                                          double xc, double yc, double zc,
                                                          const std::string& materialName);

      static G4LogicalVolume* createExtrudedSolid(const std::string& name,
                                                  const Polygon& shape,
                                                  double length,
                                                  const std::string& materialName);

      static std::string addNumber(const std::string& str, unsigned number);
 
 
      SensitivePMT* m_sensitivePMT = 0;  
      SensitiveBar* m_sensitiveBar = 0;  
      BkgSensitiveDetector* m_sensitivePCB1 = 0;  
      BkgSensitiveDetector* m_sensitivePCB2 = 0;  
      TOPGeometryPar* m_topgp = TOPGeometryPar::Instance(); 
      int m_isBeamBkgStudy = 0; 
 
      G4UnionSolid* m_moduleEnvelope = 0; 
      G4AssemblyVolume* m_qbb = 0; 
      G4AssemblyVolume* m_frontEnd = 0; 
 
      int m_numDecoupledPMTs = 0;  
      int m_numBrokenGlues = 0;    
      int m_numPeelOffRegions = 0; 
 
    };
 
  } // namespace TOP
} // namespace Belle2