SignalPDF.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/TOPNominalTTS.h>
#include <vector>
 
namespace Belle2 {
  namespace TOP {
 
    class SignalPDF {
 
    public:
 
      enum EPeakType {
        c_Unknown = 0, 
        c_Direct = 1,   
        c_Reflected = 2 
      };
 
      struct PDFPeak {
        double t0 = 0; 
        double wid = 0; 
        double nph = 0; 
        double fic = 0; 
      };
 
      struct PDFExtra {
        double thc = 0; 
        double e = 0; 
        double sige = 0; 
        int Nxm = 0; 
        int Nxb = 0; 
        int Nxe = 0; 
        int Nym = 0; 
        int Nyb = 0; 
        int Nye = 0; 
        double xD = 0; 
        double yD = 0; 
        double zD = 0; 
        double yB = 0; 
        double kxE = 0; 
        double kyE = 0; 
        double kzE = 0; 
        double kxD = 0; 
        double kyD = 0; 
        double kzD = 0; 
        EPeakType type = c_Unknown; 
      };
 
      SignalPDF(int pixelID, const TOPNominalTTS& tts);
 
      void append(const PDFPeak& peak) { m_peaks.push_back(peak);}
 
      void append(const PDFExtra& extra) { m_extras.push_back(extra);}
 
      void normalize(double a);
 
      double getSum() const;
 
      int getPixelID() const {return m_pixelID;}
 
      const TOPNominalTTS* getTTS() const {return m_tts;}
 
      const std::vector<PDFPeak>& getPDFPeaks() const {return m_peaks;}
 
      const std::vector<PDFExtra>& getPDFExtraInfo() const {return m_extras;}
 
      bool isExtraInfoAvailable() const {return m_extras.size() == m_peaks.size();}
 
      double getPDFValue(double time, double timeErr, double sigt = 0) const;
 
      double getIntegral(double minTime, double maxTime) const;
 
    private:
 
      int m_pixelID = 0; 
      const TOPNominalTTS* m_tts = 0; 
      std::vector<PDFPeak> m_peaks; 
      std::vector<PDFExtra> m_extras; 
    };
 
    //--- inline functions ------------------------------------------------------------
 
    inline void SignalPDF::normalize(double a)
    {
      for (auto& peak : m_peaks) peak.nph /= a;
    }
 
    inline double SignalPDF::getSum() const
    {
      double s = 0;
      for (const auto& peak : m_peaks) s += peak.nph;
      return s;
    }
 
  } // namespace TOP
} // namespace Belle2