TOPLikelihood.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 <framework/datastore/RelationsObject.h>
#include <framework/gearbox/Const.h>
 
namespace Belle2 {
  class TOPLikelihood : public RelationsObject {
 
  public:
 
    TOPLikelihood()
    {}
 
    void setFlag(int flag) {m_flag = flag;}
 
    void setModuleID(int moduleID) {m_moduleID = moduleID;}
 
    void setXZ(double x, double z)
    {
      m_x = x;
      m_z = z;
    }
 
    void set(const Const::ChargedStable& part, int numPhot, double logL, double estPhot, double estBkg, double effYield)
    {
      auto i = part.getIndex();
      m_numPhot = numPhot;
      m_logL[i] = logL;
      m_estPhot[i] = estPhot;
      m_estBkg = estBkg;
      m_effYields[i] = effYield;
    }
 
    int getFlag() const {return m_flag;}
 
    int getModuleID() const {return m_moduleID;}
 
    double getX() const {return m_x;}
 
    double getZ() const {return m_z;}
 
    int getNphot() const {return m_numPhot;}
 
    float getLogL(const Const::ChargedStable& part) const
    {
      return m_logL[part.getIndex()];
    }
 
    float getEstPhot(const Const::ChargedStable& part) const
    {
      return m_estPhot[part.getIndex()];
    }
 
    float getEstBkg() const {return m_estBkg;}
 
    float getEffectiveSignalYield(const Const::ChargedStable& part) const
    {
      return m_effYields[part.getIndex()];
    }
 
    double getLogL_e() const {return m_logL[Const::electron.getIndex()];}
 
    double getLogL_mu() const {return m_logL[Const::muon.getIndex()];}
 
    double getLogL_pi() const {return m_logL[Const::pion.getIndex()];}
 
    double getLogL_K() const {return m_logL[Const::kaon.getIndex()];}
 
    double getLogL_p() const {return m_logL[Const::proton.getIndex()];}
 
    double getNphot_e() const {return m_estPhot[Const::electron.getIndex()];}
 
    double getNphot_mu() const {return m_estPhot[Const::muon.getIndex()];}
 
    double getNphot_pi() const {return m_estPhot[Const::pion.getIndex()];}
 
    double getNphot_K() const {return m_estPhot[Const::kaon.getIndex()];}
 
    double getNphot_p() const {return m_estPhot[Const::proton.getIndex()];}
 
  private:
    int m_flag = 0;     
    int m_numPhot = 0;  
    float m_logL[Const::ChargedStable::c_SetSize] = {0};    
    float m_estPhot[Const::ChargedStable::c_SetSize] = {0}; 
    float m_estBkg = 0; 
    float m_effYields[Const::ChargedStable::c_SetSize] = {0}; 
    int m_moduleID = 0; 
    float m_x = 0; 
    float m_z = 0; 
    ClassDef(TOPLikelihood, 3); 
  };
 
 
} // end namespace Belle2