TOPGeoPMT.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2016 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Marko Staric                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <top/dbobjects/TOPGeoBase.h>
#include <geometry/dbobjects/GeoOpticalSurface.h>
#include <string>
 
namespace Belle2 {
  class TOPGeoPMT: public TOPGeoBase {
  public:
 
    TOPGeoPMT()
    {}
 
    TOPGeoPMT(double sizeX, double sizeY, double sizeZ,
              const std::string& name = "TOPPMT"):
      TOPGeoBase(name), m_sizeX(sizeX), m_sizeY(sizeY), m_sizeZ(sizeZ)
    {}
 
    void setWallThickness(double thickness) {m_wallThickness = thickness;}
 
    void setWallMaterial(const std::string& material) {m_wallMaterial = material;}
 
    void setFillMaterial(const std::string& material) {m_fillMaterial = material;}
 
    void setSensVolume(double sizeX, double sizeY, double thickness,
                       const std::string& material)
    {
      m_sensSizeX = sizeX;
      m_sensSizeY = sizeY;
      m_sensThickness = thickness;
      m_sensMaterial = material;
    }
 
    void setNumPixels(unsigned numColumns, unsigned numRows)
    {
      m_numColumns = numColumns;
      m_numRows = numRows;
    }
 
    void setWindow(double thickness, const std::string& material)
    {
      m_winThickness = thickness;
      m_winMaterial = material;
    }
 
    void setBottom(double thickness, const std::string& material)
    {
      m_botThickness = thickness;
      m_botMaterial = material;
    }
 
    void setReflEdge(double width, double thickness, const GeoOpticalSurface& surf)
    {
      m_reflEdgeWidth = width;
      m_reflEdgeThickness = thickness;
      m_reflEdgeSurface = surf;
    }
 
    double getSizeX() const {return m_sizeX / s_unit;}
 
    double getSizeY() const {return m_sizeY / s_unit;}
 
    double getSizeZ() const {return m_sizeZ / s_unit;}
 
    double getWallThickness() const {return m_wallThickness / s_unit;}
 
    const std::string& getWallMaterial() const {return m_wallMaterial;}
 
    const std::string& getFillMaterial() const {return m_fillMaterial;}
 
    double getSensSizeX() const {return m_sensSizeX / s_unit;}
 
    double getSensSizeY() const {return m_sensSizeY / s_unit;}
 
    double getSensThickness() const {return m_sensThickness / s_unit;}
 
    unsigned getNumColumns() const {return m_numColumns;}
 
    unsigned getNumRows() const {return m_numRows;}
 
    unsigned getNumPixels() const {return m_numColumns * m_numRows;}
 
    const std::string& getSensMaterial() const {return m_sensMaterial;}
 
    double getWinThickness() const {return m_winThickness / s_unit;}
 
    const std::string& getWinMaterial() const {return m_winMaterial;}
 
    double getBotThickness() const {return m_botThickness / s_unit;}
 
    const std::string& getBotMaterial() const {return m_botMaterial;}
 
    double getReflEdgeWidth() const {return m_reflEdgeWidth / s_unit;}
 
    double getReflEdgeThickness() const {return m_reflEdgeThickness / s_unit;}
 
    const GeoOpticalSurface& getReflEdgeSurface() const {return m_reflEdgeSurface;}
 
    unsigned getPixelColumn(double x) const;
 
    unsigned getPixelRow(double y) const;
 
    unsigned getPixelID(double x, double y) const;
 
    bool isConsistent() const override;
 
    void print(const std::string& title = "MCP-PMT geometry parameters") const override;
 
  private:
 
    float m_sizeX = 0; 
    float m_sizeY = 0; 
    float m_sizeZ = 0; 
    float m_wallThickness = 0;  
    std::string m_wallMaterial; 
    std::string m_fillMaterial; 
    float m_sensSizeX = 0; 
    float m_sensSizeY = 0; 
    float m_sensThickness = 0; 
    unsigned m_numColumns = 0; 
    unsigned m_numRows = 0;    
    std::string m_sensMaterial; 
    float m_winThickness = 0; 
    std::string m_winMaterial; 
    float m_botThickness = 0; 
    std::string m_botMaterial; 
    float m_reflEdgeWidth = 0; 
    float m_reflEdgeThickness = 0; 
    GeoOpticalSurface m_reflEdgeSurface; 
    ClassDefOverride(TOPGeoPMT, 1); 
  };
 
} // end namespace Belle2