TOPPDFCollection.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2017 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Sam Cunliffe, Jan Strube, Marko Staric                   *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <framework/datastore/RelationsObject.h>
#include <framework/gearbox/Const.h>
#include <framework/logging/Logger.h>
 
#include <map>
#include <vector>
#include <tuple>
#include <utility>
 
namespace Belle2 {
  class TOPPDFCollection : public RelationsObject {
 
  public:
 
    struct Gaussian {
      float mean = 0;  
      float width = 0; 
      float area = 0;  
      Gaussian(float m, float w, float a): mean(m), width(w), area(a)
      {}
    };
 
    typedef std::vector<Gaussian> channelPDF_t; 
    typedef std::array<channelPDF_t, 512> modulePDF_t; 
    TOPPDFCollection() { }
 
    bool addHypothesisPDF(const modulePDF_t& pdf, const int hypothesis)
    {
      auto result = m_data.insert(std::make_pair(hypothesis, pdf));
      if (not result.second) {
        B2WARNING("PDF exists already for this track");
      }
      return result.second;
    }
 
    const modulePDF_t& getHypothesisPDF(const int hypothesis) const
    {
      return m_data.at(hypothesis);
    }
 
    void setLocalPositionMomentum(const TVector3& pos, const TVector3& mom, int moduleID)
    {
      m_localHitPosition.SetXYZ(pos.X(), pos.Y(), pos.Z());
      m_localHitMomentum.SetXYZ(mom.X(), mom.Y(), mom.Z());
      m_moduleID = moduleID;
    }
 
    const TVector3& getAssociatedLocalHit() const
    {
      return m_localHitPosition;
    }
 
    const TVector3& getAssociatedLocalMomentum() const
    {
      return m_localHitMomentum;
    }
 
    int getModuleID() const {return m_moduleID;}
 
  private:
    std::map<int, modulePDF_t> m_data; 
    // The following two members are useful for python modules (with no access
    // to TOPGeometryPar)
    TVector3 m_localHitPosition; 
    TVector3 m_localHitMomentum; 
    int m_moduleID = 0; 
    ClassDef(TOPPDFCollection, 3); 
  };
} // end namespace Belle2