TOPGeoPMTArray.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/TOPGeoBase.h>
#include <top/dbobjects/TOPGeoPMT.h>
#include <vector>
 
namespace Belle2 {
  class TOPGeoPMTArray: public TOPGeoBase {
  public:
 
    TOPGeoPMTArray()
    {}
 
    TOPGeoPMTArray(unsigned numColumns, unsigned numRows,
                   double gapX, double gapY,
                   const std::string& material,
                   const TOPGeoPMT& pmt,
                   const std::string& name = "TOPPMTArray"):
      TOPGeoBase(name),
      m_numRows(numRows), m_numColumns(numColumns), m_dx(gapX), m_dy(gapY),
      m_gx(gapX), m_gy(gapY), m_material(material), m_pmt(pmt)
    {
      m_dx += pmt.getSizeX() * s_unit;
      m_dy += pmt.getSizeY() * s_unit;
    }
 
    void setAirGap(double gap) {m_airGap = gap;}
 
    void setDecoupledPMT(unsigned pmtID) {m_decoupledPMTs.push_back(pmtID);}
 
    void setSiliconeCookie(double thickness, const std::string& material)
    {
      m_cookieThickness = thickness;
      m_cookieMaterial = material;
    }
 
    void setWavelengthFilter(double thickness, const std::string& material)
    {
      m_filterThickness = thickness;
      m_filterMaterial = material;
    }
 
    void generateDecoupledPMTs(double fraction);
 
    unsigned getNumColumns() const {return m_numColumns;}
 
    unsigned getNumRows() const {return m_numRows;}
 
    unsigned getSize() const {return m_numRows * m_numColumns;}
 
    unsigned getNumPixels() const {return getSize() * m_pmt.getNumPixels();}
 
    unsigned getNumPixelColumns() const {return m_numColumns * m_pmt.getNumColumns();}
 
    unsigned getNumPixelRows() const {return m_numRows * m_pmt.getNumRows();}
 
    double getDx() const {return m_dx / s_unit;}
 
    double getDy() const {return m_dy / s_unit;}
 
    double getGapX() const {return m_gx / s_unit;}
 
    double getGapY() const {return m_gy / s_unit;}
 
    double getCookieThickness() const {return m_cookieThickness / s_unit;}
 
    const std::string& getCookieMaterial() const {return m_cookieMaterial;}
 
    double getFilterThickness() const {return m_filterThickness / s_unit;}
 
    const std::string& getFilterMaterial() const {return m_filterMaterial;}
 
    double getSizeX() const {return m_numColumns * getDx();}
 
    double getSizeY() const {return m_numRows * getDy();}
 
    double getSizeZ() const
    {
      return m_pmt.getSizeZ() + getAirGap() + getCookieThickness() + getFilterThickness();
    }
 
    double getX(unsigned col) const
    {
      return (int)(m_numColumns + 1 - 2 * col) / 2.0 * getDx();
    }
 
    double getY(unsigned row) const
    {
      return (int)(2 * row - m_numRows - 1) / 2.0 * getDy();
    }
 
    const std::string& getMaterial() const {return m_material;}
 
    const TOPGeoPMT& getPMT() const {return m_pmt;}
 
    unsigned getPmtID(unsigned row, unsigned col) const;
 
    unsigned getRow(unsigned pmtID) const;
 
    unsigned getColumn(unsigned pmtID) const;
 
    int getPixelID(double x, double y, unsigned pmtID) const;
 
    int getPixelID(unsigned pmtID, unsigned pmtPixelID) const;
 
    double getAirGap() const {return m_airGap / s_unit;}
 
    const std::vector<unsigned>& getDecoupledPMTs() const {return m_decoupledPMTs;}
 
    bool isPMTDecoupled(unsigned pmtID) const;
 
    bool isConsistent() const override;
 
    void print(const std::string& title = "PMT array geometry parameters") const override;
 
  private:
 
    unsigned m_numRows = 0;     
    unsigned m_numColumns = 0;  
    float m_dx = 0; 
    float m_dy = 0; 
    float m_gx = 0; 
    float m_gy = 0; 
    std::string m_material; 
    TOPGeoPMT m_pmt; 
    float m_airGap = 0; 
    std::vector<unsigned> m_decoupledPMTs; 
    float m_cookieThickness = 0; 
    std::string m_cookieMaterial; 
    float m_filterThickness = 0; 
    std::string m_filterMaterial; 
    ClassDefOverride(TOPGeoPMTArray, 3); 
  };
 
} // end namespace Belle2