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() override;
      GeoCDCCreator(const GeoCDCCreator&) = delete;

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


      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:
      static 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);

      virtual bool getEndplateInformation(const CDCGeometry& geo, const uint iSLayer,
                                          double& rMinLeft, double& rMaxLeft, double& zBackLeft, double& zForLeft,
                                          double& rMinMiddle, double& rMaxMiddle, double& zBackMiddle, double& zForMiddle,
                                          double& rMinRight, double& rMaxRight, double& zBackRight, double& zForRight) const;

      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