TOPGeometry.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 <top/dbobjects/TOPGeoModule.h>
#include <top/dbobjects/TOPGeoFrontEnd.h>
#include <top/dbobjects/TOPGeoQBB.h>
#include <top/dbobjects/TOPNominalQE.h>
#include <top/dbobjects/TOPNominalTTS.h>
#include <top/dbobjects/TOPNominalTDC.h>
#include <top/dbobjects/TOPSignalShape.h>
#include <top/dbobjects/TOPWavelengthFilter.h>
#include <framework/gearbox/Unit.h>
#include <vector>
#include <map>
 
 
namespace Belle2 {
  class TOPGeometry: public TOPGeoBase {
  public:
 
    TOPGeometry()
    {}
 
    explicit TOPGeometry(const std::string& name): TOPGeoBase(name)
    {}
 
    static void useBasf2Units() {s_unit = Unit::cm; s_unitName = "cm";}
 
    static void useGeantUnits() {s_unit = Unit::mm; s_unitName = "mm";}
 
    void appendModule(const TOPGeoModule& module);
 
    void setFrontEnd(const TOPGeoFrontEnd& frontEnd, unsigned num = 4)
    {
      m_frontEnd = frontEnd;
      m_numBoardStacks = num;
    }
 
    void setQBB(const TOPGeoQBB& QBB) {m_QBB = QBB;}
 
    void setNominalQE(const TOPNominalQE& nominalQE) {m_nominalQE = nominalQE;}
 
    void setNominalTTS(const TOPNominalTTS& nominalTTS) {m_nominalTTS = nominalTTS;}
 
    void appendTTS(const TOPNominalTTS& tts) {m_tts[tts.getPMTType()] = tts;}
 
    void appendPDETuningFactor(unsigned type, double factor) {m_tuneFactorsPDE[type] = factor;}
 
    void setNominalTDC(const TOPNominalTDC& nominalTDC) {m_nominalTDC = nominalTDC;}
 
    void setSignalShape(const TOPSignalShape& signalShape) {m_signalShape = signalShape;}
 
    void setCalPulseShape(const TOPSignalShape& shape) {m_calPulseShape = shape;}
 
    void setWavelengthFilter(const TOPWavelengthFilter& filter) {m_wavelengthFilter = filter;}
 
    unsigned getNumModules() const {return m_modules.size();}
 
    const TOPGeoModule& getModule(int moduleID) const;
 
    const std::vector<TOPGeoModule>& getModules() const {return m_modules;}
 
    bool isModuleIDValid(int moduleID) const;
 
    const TOPGeoFrontEnd& getFrontEnd() const {return m_frontEnd;}
 
    unsigned getNumBoardStacks() const {return m_numBoardStacks;}
 
    const TOPGeoQBB& getQBB() const {return m_QBB;}
 
    const TOPNominalQE& getNominalQE() const {return m_nominalQE;}
 
    const TOPNominalTTS& getNominalTTS() const {return m_nominalTTS;}
 
    const TOPNominalTTS& getTTS(unsigned type) const;
 
    double getPDETuningFactor(unsigned type) const;
 
    bool arePDETuningFactorsEmpty() const {return m_tuneFactorsPDE.empty();}
 
    const std::map<unsigned, TOPNominalTTS>& getTTSes() const {return m_tts;}
 
    const TOPNominalTDC& getNominalTDC() const {return m_nominalTDC;}
 
    const TOPSignalShape& getSignalShape() const {return m_signalShape;}
 
    const TOPSignalShape& getCalPulseShape() const {return m_calPulseShape;}
 
    const TOPWavelengthFilter& getWavelengthFilter() const {return m_wavelengthFilter;}
 
    double getInnerRadius() const;
 
    double getOuterRadius() const;
 
    double getRadius() const;
 
    double getBackwardZ() const;
 
    double getForwardZ() const;
 
    bool isConsistent() const override;
 
    void print(const std::string& title = "TOP geometry parameters") const override;
 
 
  private:
 
    std::vector<TOPGeoModule> m_modules; 
    TOPGeoFrontEnd m_frontEnd;  
    TOPGeoQBB m_QBB;  
    unsigned m_numBoardStacks = 0;  
    TOPNominalQE m_nominalQE; 
    TOPNominalTTS m_nominalTTS; 
    TOPNominalTDC m_nominalTDC; 
    TOPSignalShape m_signalShape; 
    TOPSignalShape m_calPulseShape; 
    TOPWavelengthFilter m_wavelengthFilter; 
    std::map<unsigned, TOPNominalTTS> m_tts; 
    std::map<unsigned, float> m_tuneFactorsPDE; 
    ClassDefOverride(TOPGeometry, 8); 
  };
 
} // end namespace Belle2