GeoCDCCreator.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 <geometry/CreatorBase.h>
 
#include <framework/gearbox/GearDir.h>
#include <framework/logging/Logger.h>
#include <framework/database/DBObjPtr.h>
#include <framework/database/IntervalOfValidity.h>
#include <framework/database/DBImportObjPtr.h>
 
#include <cdc/simulation/CDCSensitiveDetector.h>
#include <cdc/dbobjects/CDCGeometry.h>
 
class G4LogicalVolume;
class G4VPhysicalVolume;
class G4VisAttributes;
class G4UserLimits;
 
namespace Belle2 {
  class BkgSensitiveDetector;
 
  namespace CDC {
 
 
    class GeoCDCCreator : public geometry::CreatorBase {
 
    public:
 
      GeoCDCCreator();
 
      ~GeoCDCCreator();
 
 
      virtual void create(const GearDir& content, G4LogicalVolume& topVolume, geometry::GeometryTypes type) override
      {
        CDCGeometry config = createConfiguration(content);
        createGeometry(config, topVolume, type);
      }
 
      virtual void createFromDB(const std::string& name, G4LogicalVolume& topVolume, geometry::GeometryTypes type) override
      {
        DBObjPtr<CDCGeometry> geo;
        if (!geo) {
          B2FATAL("No configuration for " << name << " found.");
        }
        createGeometry(*geo, topVolume, type);
      }
 
      virtual void createPayloads(const GearDir& content, const IntervalOfValidity& iov) override
      {
        DBImportObjPtr<CDCGeometry> importObj;
        importObj.construct(createConfiguration(content));
        importObj.import(iov);
      }
 
      void createCovers(const GearDir& content);
 
      void createCovers(const CDCGeometry& geom);
 
      void createNeutronShields(const GearDir& content);
 
 
      void createNeutronShields(const CDCGeometry& geom);
 
      void createCover2s(const GearDir& content);
 
      void createCover2s(const CDCGeometry& geom);
 
      void createCone(const double rmin1, const double rmax1,
                      const double rmin2, const double rmax2,
                      const double thick, const double posz,
                      const int id, G4Material* med, const std::string& name);
      void createBox(const double length, const double height,
                     const double thick, const double x,
                     const double y, const double z,
                     const int id, G4Material* med,
                     const std::string& name);
      void createTube(const double rmin, const double rmax,
                      const double thick, const double posZ,
                      const int id, G4Material* med,
                      const std::string& name);
      void createTorus(const double rmin1, const double rmax1,
                       const double thick, const double posZ,
                       const int id, G4Material* med,
                       const std::string& name);
 
      void createTube2(const double rmin, const double rmax,
                       const double phis, const double phie,
                       const double thick, const double posZ,
                       const int id, G4Material* med,
                       const std::string& name);
 
    protected:
 
    private:
      CDCGeometry createConfiguration(const GearDir& param)
      {
        CDCGeometry cdcGeometry;
        cdcGeometry.read(param);
        return cdcGeometry;
      }
      void createGeometry(const CDCGeometry& parameters, G4LogicalVolume& topVolume,
                          geometry::GeometryTypes type);
      void createMapper(G4LogicalVolume& topVolume);
 
      G4LogicalVolume* m_logicalCDC;
 
      G4VPhysicalVolume* m_physicalCDC;
 
      CDCSensitiveDetector* m_sensitive = nullptr;
 
      BkgSensitiveDetector* m_bkgsensitive = nullptr;
 
      std::vector<BkgSensitiveDetector*> m_BkgSensitiveRib4 = {};
 
      std::vector<G4VisAttributes*> m_VisAttributes;
 
      std::vector<G4UserLimits*> m_userLimits;
 
    };
 
  } // end of cdc namespace
} // end of Belle2 namespace