ECLPidLikelihood.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>
#include <cmath>
 
namespace Belle2 {
  class ECLPidLikelihood : public RelationsObject {
  public:
    ECLPidLikelihood(): RelationsObject()
    {
      //for all particles
      for (unsigned int i = 0; i < Const::ChargedStable::c_SetSize; i++) {
        m_logl[i] = 0.0;
      }
      setVariables(0, 0, 0, 0, 0, 0, 0, 0, 0);
    }
 
    explicit ECLPidLikelihood(const float* logl, float energy = 0, float eop = 0, float e9e25 = 0, float lat = 0, float dist = 0,
                              float trkDepth = 0, float shDepth = 0, int ncrystals = 0, int nclusters = 0): RelationsObject()
    {
      //for all particles
      for (unsigned int i = 0; i < Const::ChargedStable::c_SetSize; i++) {
        m_logl[i] = logl[i];
      }
      setVariables(energy, eop, e9e25, lat, dist, trkDepth, shDepth, ncrystals, nclusters);
    }
 
    float getLogLikelihood(const Const::ChargedStable& type) const
    {
      return m_logl[type.getIndex()];
    }
 
    double getLikelihood(const Const::ChargedStable& type) const
    {
      return exp((double)m_logl[type.getIndex()]);
    }
 
    void setLogLikelihood(const Const::ChargedStable& type, float logl)
    {
      m_logl[type.getIndex()] = logl;
    }
 
    void setVariables(float energy, float eop, float e9e25, float lat, float dist, float trkDepth, float shDepth, int ncrystals,
                      int nclusters)
    {
      m_energy = energy; m_eop = eop; m_e9e25 = e9e25; m_nCrystals = ncrystals; m_nClusters = nclusters;
      m_lat = lat; m_dist = dist; m_trkDepth = trkDepth; m_shDepth = shDepth;
    }
 
    double energy() const { return m_energy; } 
    double eop() const { return m_eop; } 
    double e9e25() const { return m_e9e25; } 
    float lat() const { return m_lat; } 
    float dist() const { return m_dist; } 
    float trkDepth() const { return m_trkDepth; } 
    float shDepth() const { return m_shDepth; } 
    int nCrystals() const { return m_nCrystals; } 
    int nClusters() const { return m_nClusters; } 
  private:
 
    float m_logl[Const::ChargedStable::c_SetSize]; 
    float m_energy;  
    float m_eop; 
    float m_e9e25; 
    int m_nCrystals; 
    int m_nClusters;
    float m_lat; 
    float m_dist; 
    float m_trkDepth;  
    float m_shDepth;  
    // 3: Added DeltaL track match variable (GDN, increment by TF)
    ClassDef(ECLPidLikelihood, 3); 
  };
}