ARICHPhoton.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.                *
 **************************************************************************/
 
#pragma once
#ifndef ARICHPHOTON_H
#define ARICHPHOTON_H
 
#include <framework/datastore/RelationsObject.h>
#include <framework/gearbox/Const.h>
 
namespace Belle2 {
  class ARICHPhoton : public RelationsObject {
 
  public:
 
    ARICHPhoton() {};
 
    ARICHPhoton(int hitID, float thetaCer, float phiCer, int mirror): m_hitID(hitID), m_thetaCer(thetaCer),
      m_phiCer(phiCer),
      m_mirror(mirror) {};
 
    void setHitID(int id)
    {
      m_hitID = id;
    }
 
    void setPhiCerTrk(float phi)
    {
      m_phiCerTrk = phi;
    }
 
    void setXY(float x, float y)
    {
      m_x = x;
      m_y = y;
    }
 
    void setModuleID(int modID)
    {
      m_module = modID;
    }
 
    void setChannel(int chn)
    {
      m_channel = chn;
    }
 
 
    void setSigExp(const double* sigExp)
    {
      m_sigExp_e = (float)sigExp[0];
      m_sigExp_mu = (float)sigExp[1];
      m_sigExp_pi = (float)sigExp[2];
      m_sigExp_K = (float)sigExp[3];
      m_sigExp_p = (float)sigExp[4];
    }
 
    void setBkgExp(const double* bkgExp)
    {
      m_bkgExp_e = (float)bkgExp[0];
      m_bkgExp_mu = (float)bkgExp[1];
      m_bkgExp_pi = (float)bkgExp[2];
      m_bkgExp_K = (float)bkgExp[3];
      m_bkgExp_p = (float)bkgExp[4];
    }
 
    void setNCosThetaCh(const double* n_cos_theta_ch)
    {
      m_nCosThetaCh_e = (float)n_cos_theta_ch[0];
      m_nCosThetaCh_mu = (float)n_cos_theta_ch[1];
      m_nCosThetaCh_pi = (float)n_cos_theta_ch[2];
      m_nCosThetaCh_K = (float)n_cos_theta_ch[3];
      m_nCosThetaCh_p = (float)n_cos_theta_ch[4];
    }
 
    void setPhiCh(const double* phi_ch)
    {
      m_phiCh_e = (float)phi_ch[0];
      m_phiCh_mu = (float)phi_ch[1];
      m_phiCh_pi = (float)phi_ch[2];
      m_phiCh_K = (float)phi_ch[3];
      m_phiCh_p = (float)phi_ch[4];
    }
 
    float getSigExp(const Const::ChargedStable& part) const
    {
      if (part == Const::electron) return m_sigExp_e;
      if (part == Const::muon) return m_sigExp_mu;
      if (part == Const::pion) return m_sigExp_pi;
      if (part == Const::kaon) return m_sigExp_K;
      if (part == Const::proton) return m_sigExp_p;
      else return 0;
    }
 
    float getBkgExp(const Const::ChargedStable& part) const
    {
      if (part == Const::electron) return m_bkgExp_e;
      if (part == Const::muon) return m_bkgExp_mu;
      if (part == Const::pion) return m_bkgExp_pi;
      if (part == Const::kaon) return m_bkgExp_K;
      if (part == Const::proton) return m_bkgExp_p;
      else return 0;
    }
 
    float getNCosThetaCh(const Const::ChargedStable& part) const
    {
      if (part == Const::electron) return m_nCosThetaCh_e;
      if (part == Const::muon) return m_nCosThetaCh_mu;
      if (part == Const::pion) return m_nCosThetaCh_pi;
      if (part == Const::kaon) return m_nCosThetaCh_K;
      if (part == Const::proton) return m_nCosThetaCh_p;
      else return 0;
    }
 
    float getPhiCh(const Const::ChargedStable& part) const
    {
      if (part == Const::electron) return m_phiCh_e;
      if (part == Const::muon) return m_phiCh_mu;
      if (part == Const::pion) return m_phiCh_pi;
      if (part == Const::kaon) return m_phiCh_K;
      if (part == Const::proton) return m_phiCh_p;
      else return 0;
    }
 
    double getThetaCer()
    {
      return (double)m_thetaCer;
    }
 
    double getPhiCer()
    {
      return (double)m_phiCer;
    }
 
    double getPhiCerTrk()
    {
      return (double)m_phiCerTrk;
    }
 
    int getMirror()
    {
      return m_mirror;
    }
 
    int getHitID()
    {
      return m_hitID;
    }
 
    double getX()
    {
      return (double)m_x;
    }
 
    double getY()
    {
      return (double)m_y;
    }
 
    int getModuleID()
    {
      return m_module;
    }
 
    int getChannel()
    {
      return m_channel;
    }
 
 
  private:
 
    int m_hitID = 0;     
    float m_thetaCer = 0.; 
    float m_phiCer = 0.;  
    int m_mirror = 0;    
    float m_sigExp_e = 0.; 
    float m_bkgExp_e = 0.; 
    float m_sigExp_mu = 0.; 
    float m_bkgExp_mu = 0.; 
    float m_sigExp_pi = 0.; 
    float m_bkgExp_pi = 0.; 
    float m_sigExp_K = 0.; 
    float m_bkgExp_K = 0.; 
    float m_sigExp_p = 0.; 
    float m_bkgExp_p = 0.; 
    float m_nCosThetaCh_e = 0.; 
    float m_nCosThetaCh_mu = 0.; 
    float m_nCosThetaCh_pi = 0.; 
    float m_nCosThetaCh_K = 0.; 
    float m_nCosThetaCh_p = 0.; 
    float m_phiCh_e = 0.; 
    float m_phiCh_mu = 0.; 
    float m_phiCh_pi = 0.; 
    float m_phiCh_K = 0.; 
    float m_phiCh_p = 0.; 
    float m_x = 0; 
    float m_y = 0; 
    int m_module = 0; 
    int m_channel = 0; 
    float m_phiCerTrk = 0.;  
    ClassDef(ARICHPhoton, 5); 
  };
} //Belle2 namespace
 
#endif // ARICHPHOTON_H