ObserverInfo.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 <TObject.h>
 
#include <tracking/dataobjects/SpacePointInfo.h>
#include <tracking/dataobjects/FilterInfo.h>
 
namespace Belle2 {
  class ObserverInfo : public TObject {
  public:
    ObserverInfo();
    ~ObserverInfo();

    void clear()
    {
      m_filterInfos.clear();
      m_hits.clear();
      m_mainPurity = 0;
      m_mainMCParticleID = -666;
    };

    FilterInfo getFilterInfo(std::string const&   key)
    {
      for (int i = 0; i < (int) m_filterInfos.size(); i++)
        if (m_filterInfos.at(i).getName() == key) return m_filterInfos.at(i);
      //if nothing was found return an empty object
      return FilterInfo();
    };

    std::string getFilterName(int i)
    {
      return (0 <= i && i < (int)m_filterInfos.size()) ? m_filterInfos[i].getName() : std::string("");
    }

    double getFilterResult(std::string const& key)
    {
      return getFilterInfo(key).getResult();
    }

    bool getFilterAccepted(std::string const& key)
    {
      return getFilterInfo(key).getWasAccepted();
    }

    double getFilterUsed(std::string const& key)
    {
      return getFilterInfo(key).getWasUsed();
    }

    void addFilterInfo(FilterInfo info) { m_filterInfos.push_back(info); };

    std::vector<SpacePointInfo> getHits() { return m_hits; };

    SpacePointInfo getHit(int i)
    {
      if (i < 0 || i > (int)m_hits.size()) return SpacePointInfo();  // return a dummy object
      else return m_hits[i];
    };

    void setHits(const std::vector<SpacePointInfo>& newHits) { m_hits = newHits; };

    int getMainMCParticleID() {return m_mainMCParticleID;};

    void setMainMCParticleID(int id) { m_mainMCParticleID = id; };

    double getMainPurity() { return m_mainPurity; };

    void setMainPurity(double val) { m_mainPurity = val; };
  private:
    std::vector<FilterInfo> m_filterInfos;

    std::vector<SpacePointInfo>  m_hits;

    double m_mainPurity;

    int m_mainMCParticleID;

    ClassDef(ObserverInfo, 2);
  };

}// end namespace Belle2