MuidElementNumbers.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
 
/* KLM headers. */
#include <klm/dataobjects/bklm/BKLMElementNumbers.h>
#include <klm/dataobjects/eklm/EKLMElementNumbers.h>
#include <klm/dataobjects/KLMElementNumbers.h>
 
/* C++ headers. */
#include <vector>
 
namespace Belle2 {
  class MuidElementNumbers {
 
  public:
 
    enum Outcome {
      c_NotReached = 0,
      c_StopInBarrel = 1,
      c_StopInForwardEndcap = 2,
      c_ExitBarrel = 3,
      c_ExitForwardEndcap = 4,
      c_StopInBackwardEndcap = 5,
      c_ExitBackwardEndcap = 6,
      c_CrossBarrelStopInForwardMin = 7,
      c_CrossBarrelStopInForwardMax = 21,
      c_CrossBarrelStopInBackwardMin = 22,
      c_CrossBarrelStopInBackwardMax = 36,
      c_CrossBarrelExitForwardMin = 37,
      c_CrossBarrelExitForwardMax = 51,
      c_CrossBarrelExitBackwardMin = 52,
      c_CrossBarrelExitBackwardMax = 66
    };
 
    enum DetectorsCrossed {
      c_Both = 0,
      c_OnlyBarrel = KLMElementNumbers::c_BKLM,
      c_OnlyEndcap = KLMElementNumbers::c_EKLM,
    };
 
    enum Hypothesis {
      c_NotValid = -1,
      c_Positron = 0,
      c_Electron = 1,
      c_Deuteron = 2,
      c_AntiDeuteron = 3,
      c_Proton = 4,
      c_AntiProton = 5,
      c_PionPlus = 6,
      c_PionMinus = 7,
      c_KaonPlus = 8,
      c_KaonMinus = 9,
      c_MuonPlus = 10,
      c_MuonMinus = 11,
    };
 
    enum IdentifierBits {
      c_OutcomeBit = 7,
      c_LastLayerBit = 4,
      c_DetectorBit = 2,
      c_DegreesOfFreedomBit = 6
    };
 
    MuidElementNumbers();
 
    ~MuidElementNumbers();
 
    static bool checkExtrapolationOutcome(unsigned int outcome, int lastLayer);
 
    static unsigned int calculateExtrapolationOutcome(bool isForward, bool escaped, int lastBarrelLayer, int lastEndcapLayer);
 
    static Hypothesis calculateHypothesisFromPDG(int pdg);
 
    static constexpr int getMaximalBarrelLayer()
    {
      return m_MaximalBarrelLayer;
    }
 
    static constexpr int getMaximalEndcapForwardLayer()
    {
      return m_MaximalEndcapForwardLayer;
    }
 
    static constexpr int getMaximalEndcapBackwardLayer()
    {
      return m_MaximalEndcapBackwardLayer;
    }
 
    static constexpr int getMaximalOutcome()
    {
      return m_MaximalOutcome;
    }
 
    static constexpr int getMaximalDetector()
    {
      return m_MaximalDetector;
    }
 
    static constexpr int getMaximalHalfNdof()
    {
      return m_MaximalHalfNDof;
    }
 
    static constexpr int getSizeReducedChiSquared()
    {
      return m_SizeReducedChiSquared;
    }
 
    static constexpr double getMaximalReducedChiSquared()
    {
      return m_MaximalReducedChiSquared;
    }
 
    static std::vector<int> getPDGVector(int charge);
 
    static std::vector<int> getPDGVector();
 
    static int getLongitudinalID(int hypothesis, int outcome, int lastLayer);
 
    static int getTransverseID(int hypothesis, int detector, int degreesOfFreedom);
 
  protected:
 
    static constexpr int m_MaximalBarrelLayer = BKLMElementNumbers::getMaximalLayerNumber() - 1;
 
    static constexpr int m_MaximalEndcapForwardLayer = EKLMElementNumbers::getMaximalLayerNumber() - 1;
 
    static constexpr int m_MaximalEndcapBackwardLayer = m_MaximalEndcapForwardLayer - 2;
 
    static constexpr int m_MaximalOutcome = c_CrossBarrelExitBackwardMax;
 
    static constexpr int m_MaximalDetector = c_OnlyEndcap;
 
    static constexpr int m_MaximalHalfNDof = 18;
 
    static constexpr int m_SizeReducedChiSquared = 100;
 
    static constexpr double m_MaximalReducedChiSquared = 10.;
 
  };
 
}