MdstPIDModule.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/core/Module.h>
#include <framework/datastore/StoreArray.h>
#include <framework/logging/Logger.h>
#include <framework/gearbox/Const.h>
 
#include <ecl/dataobjects/ECLPidLikelihood.h>
#include <klm/dataobjects/KLMMuidLikelihood.h>
 
#include <string>
#include <cmath>
#include <map>
 
namespace Belle2 {
  class Track;
  class TOPLikelihood;
  class ARICHLikelihood;
  class CDCDedxLikelihood;
  class VXDDedxLikelihood;
  class PIDLikelihood;


  class MdstPIDModule : public Module {
  public:
 

    MdstPIDModule();

    virtual void initialize() override;

    virtual void event() override;
 
  private:

    void setLikelihoods(const TOPLikelihood* logl);

    void setLikelihoods(const ARICHLikelihood* logl);

    void setLikelihoods(const CDCDedxLikelihood* logl);

    void setLikelihoods(const VXDDedxLikelihood* logl);

    void setLikelihoods(const ECLPidLikelihood* logl);

    void setLikelihoods(const KLMMuidLikelihood* muid);

    template<class T>
    float getLogL(const T* logl, const Const::ChargedStable& chargedStable) const
    {
      return logl->getLogL(chargedStable);
    }

    float getLogL(const ECLPidLikelihood* logl, const Const::ChargedStable& chargedStable) const
    {
      return logl->getLogLikelihood(chargedStable);
    }

    float getLogL(const KLMMuidLikelihood* logl, const Const::ChargedStable& chargedStable) const
    {
      return logl->getLogL(chargedStable.getPDGCode());
    }

    template<class T>
    bool areLikelihoodsValid(const T* logl)
    {
      std::vector<float> values;
      for (const auto& chargedStable : Const::chargedStableSet) {
        auto value = getLogL(logl, chargedStable);
        values.push_back(value);
        if (std::isnan(value) or value == INFINITY) {
          B2ERROR("MdstPID::setLikelihoods: invalid " << logl->ClassName() << " for " << m_chargedNames[chargedStable]
                  << ", is " << value
                  << ". Likelihoods ignored.");
          return false;
        }
      }
      // check if log likelihoods differ
      for (auto value : values) {
        if (value != values.back()) return true; // values differ
      }
      return false; // values are all the same - ignore Likelihoods
    }
 
    // module parameters
    bool m_subtractMaximum; 
 
    // input collections
    StoreArray<Track> m_tracks; 
    StoreArray<TOPLikelihood> m_topLikelihoods; 
    StoreArray<ARICHLikelihood> m_arichLikelihoods; 
    StoreArray<CDCDedxLikelihood> m_cdcDedxLikelihoods; 
    StoreArray<VXDDedxLikelihood> m_vxdDedxLikelihoods; 
    StoreArray<ECLPidLikelihood> m_eclLikelihoods; 
    StoreArray<KLMMuidLikelihood> m_muid; 
 
    // output collection
    StoreArray<PIDLikelihood> m_pidLikelihoods; 
 
    // other
    PIDLikelihood* m_pid; 
    std::map<Const::ChargedStable, std::string> m_chargedNames; 
    int m_CDCnLayerHitsUsed = -1; 
 
  };

} // Belle2 namespace