KfpV0FinderDefs.h

Go to the documentation of this file.

/* Copyright (C) 2025 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Sergei Zharko [committer] */

 
#ifndef KfpV0FinderDefs_h
#define KfpV0FinderDefs_h 1
 
#include "CbmEnumDict.h"
 
#include <array>
 
namespace cbm::algo::kfp
{
  // TODO: extend
  // NOTE: a limited number of particles can be assigned to tracks
  // NOTE: ETrackPid::Undef defines the size of the enum and ALWAYS must be the last element
  enum class ETrackPid
  {
    Pip = 0,  
    Pim,      
    P,        
    Pbar,     
    Undef     
  };

  static constexpr int ToPdg(ETrackPid trackPid)
  {
    constexpr std::array<int, static_cast<size_t>(ETrackPid::Undef) + 1> mapper = {
      +211,   // ETrackPid::Pip
      -211,   // ETrackPid::Pim
      +2212,  // ETrackPid::P
      -2212,  // ETrackPid::Pbar
      -2,     // ETrackPid::Undef
    };
    return mapper[static_cast<size_t>(trackPid)];
  }

  static constexpr ETrackPid ToTrackPid(int pdg)
  {
    switch (pdg) {
      case +211: return ETrackPid::Pip;
      case -211: return ETrackPid::Pim;
      case +2212: return ETrackPid::P;
      case -2212: return ETrackPid::Pbar;
      default: return ETrackPid::Undef;
    }
    return ETrackPid::Undef;
  }

  std::string ToString(ETrackPid trackPid);

  enum class EV0Type
  {
    Lambda
  };
}  // namespace cbm::algo::kfp
 
namespace cbm::util
{
  CBM_ENUM_DICT(cbm::algo::kfp::ETrackPid, 
    {"Pip", cbm::algo::kfp::ETrackPid::Pip},
    {"Pim", cbm::algo::kfp::ETrackPid::Pim},
    {"P", cbm::algo::kfp::ETrackPid::P},
    {"Pbar", cbm::algo::kfp::ETrackPid::Pbar},
    {"Undef", cbm::algo::kfp::ETrackPid::Undef}
  );
}
 
#endif  // KfpV0FinderDefs_h