FlavorTaggerInfoMap.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 <map>
#include <string>
 
namespace Belle2 {
 
  // forward declarations
 
  class Track;

 
  class FlavorTaggerInfoMap : public RelationsObject {
 
  public:

    FlavorTaggerInfoMap():
      m_targetTrackLevel(),
      m_probTrackLevel(),
      m_targetEventLevel(),
      m_probEventLevel(),
      m_qpCategory(),
      m_hasTrueTarget(),
      m_isTrueCategory(),
      m_qrCombined(-2),
      m_B0Probability(-2),
      m_B0barProbability(-2)
    {}


    void setTargetTrackLevel(const std::string& category, const Track* track)
    {
      m_targetTrackLevel.insert(std::pair<std::string, const Track*>(category, track));
    }

    void setProbTrackLevel(const std::string& category, float probability)
    {
      m_probTrackLevel.insert(std::pair<std::string, float>(category, probability));
    }

    void setTargetEventLevel(const std::string& category, const Track* track)
    {
      m_targetEventLevel.insert(std::pair<std::string, const Track*>(category, track));
    }

    void setProbEventLevel(const std::string& category, float probability)
    {
      m_probEventLevel.insert(std::pair<std::string, float>(category, probability));
    }

    void setQpCategory(const std::string& category, float qr)
    {
      m_qpCategory.insert(std::pair<std::string, float>(category, qr));
    }

    void setHasTrueTarget(const std::string& category, float isTrue)
    {
      m_hasTrueTarget.insert(std::pair<std::string, float>(category, isTrue));
    }

    void setIsTrueCategory(const std::string& category, float isTrue)
    {
      m_isTrueCategory.insert(std::pair<std::string, float>(category, isTrue));
    }

    void setQrCombined(float qr) { m_qrCombined = qr; }

    void setB0Probability(float B0Probability) { m_B0Probability = B0Probability; }

    void setB0barProbability(float B0barProbability) { m_B0barProbability = B0barProbability; }

    std::map<std::string, const Track*> getTargetTrackLevel() const { return m_targetTrackLevel; }

    std::map<std::string, float> getProbTrackLevel() const { return m_probTrackLevel; }

    std::map<std::string, const Track*> getTargetEventLevel() const { return m_targetEventLevel; }

    std::map<std::string, float> getProbEventLevel() const { return m_probEventLevel; }

    std::map<std::string, float> getQpCategory() const { return m_qpCategory; }

    std::map<std::string, float> getHasTrueTarget() const { return m_hasTrueTarget; }

    std::map<std::string, float> getIsTrueCategory() const { return m_isTrueCategory; }

    float getQrCombined() const { return m_qrCombined; }

    float getB0Probability() const { return m_B0Probability; }

    float getB0barProbability() const { return m_B0barProbability; }
 
  private:
 
    // persistent data members
 
    // Track, Event and Combiner Level Flavor Tagger Info
 
    std::map<std::string, const Track*>
    m_targetTrackLevel; 
    std::map<std::string, float>
    m_probTrackLevel; 
    std::map<std::string, const Track*>
    m_targetEventLevel; 
    std::map<std::string, float>
    m_probEventLevel; 
    std::map<std::string, float>
    m_qpCategory; 
    std::map<std::string, float>
    m_hasTrueTarget; 
    std::map<std::string, float>
    m_isTrueCategory; 
 
    float m_qrCombined; 
    float m_B0Probability; 
    float m_B0barProbability; 
 
    ClassDef(FlavorTaggerInfoMap, 3) 
  };

} // end namespace Belle2