ARICHLikelihood.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Luka Santelj                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#ifndef ARICHLIKELIHOOD_H
#define ARICHLIKELIHOOD_H
 
#include <framework/datastore/RelationsObject.h>
#include <framework/gearbox/Const.h>
 
namespace Belle2 {
  class ARICHLikelihood : public RelationsObject {
 
  public:
 
    ARICHLikelihood(): m_flag(0)
    {
      for (unsigned i = 0; i < Const::ChargedStable::c_SetSize; i++) m_logL[i] = 0;
      for (unsigned i = 0; i < Const::ChargedStable::c_SetSize; i++) m_expPhot[i] = 0;
      m_detPhot = 0;
    }
 
    void setValues(int flag,
                   const double* logL,
                   int detPhot,
                   const double* expPhot
                  )
    {
      m_flag = flag;
      for (unsigned i = 0; i < Const::ChargedStable::c_SetSize; i++)
        m_logL[i] = logL[i];
      for (unsigned i = 0; i < Const::ChargedStable::c_SetSize; i++)
        m_expPhot[i] = expPhot[i];
 
      m_detPhot = detPhot;
    }
 
    int getFlag() const {return m_flag;}
 
    float getLogL(const Const::ChargedStable& part) const
    {
      return m_logL[part.getIndex()];
    }
 
    float getDetPhot() const
    {
      return m_detPhot;
    }
 
    float getExpPhot(const Const::ChargedStable& part) const
    {
      return m_expPhot[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_expPhot[Const::electron.getIndex()];}
 
    double getNphot_mu() const {return m_expPhot[Const::muon.getIndex()];}
 
    double getNphot_pi() const {return m_expPhot[Const::pion.getIndex()];}
 
    double getNphot_K() const {return m_expPhot[Const::kaon.getIndex()];}
 
    double getNphot_p() const {return m_expPhot[Const::proton.getIndex()];}
 
  private:
    int m_flag;          
    float m_logL[Const::ChargedStable::c_SetSize]; 
    float m_expPhot[Const::ChargedStable::c_SetSize]; 
    int m_detPhot; 
    ClassDef(ARICHLikelihood, 3); 
  };
 
} // end namespace Belle2
 
#endif