TrackIsoCalculatorModule.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/StoreObjPtr.h>
#include <framework/dataobjects/EventMetaData.h>
#include <framework/datastore/StoreArray.h>
#include <framework/gearbox/Const.h>
 
#include <analysis/dataobjects/Particle.h>
#include <analysis/dataobjects/ParticleList.h>
#include <analysis/DecayDescriptor/DecayDescriptor.h>
 
 
namespace Belle2 {
  template<class T> class DBObjPtr;
  class PIDDetectorWeights;

  class TrackIsoCalculatorModule : public Module {
 
  public:

    TrackIsoCalculatorModule();

    ~TrackIsoCalculatorModule() override;

    void initialize() override;

    void event() override;
 
 
  private:

    StoreArray<Particle> m_particles;

    std::string m_decayString;

    unsigned short m_nSelectedDaughters;

    std::string m_pListReferenceName;

    std::vector<std::string> m_detNames;

    std::unordered_map<std::string, std::string>  m_detLayerToDistVariable;

    std::unordered_map<std::string, std::string>  m_detLayerToRefPartIdxVariable;

    std::string m_isoScoreVariable;

    std::string m_isoScoreVariableAsWeightedAvg;

    std::map<std::pair<std::string, int>, double> m_distThreshPerDetLayer = {
      { {Const::parseDetectors(Const::CDC), 0}, 5.0 },
      { {Const::parseDetectors(Const::CDC), 1}, 5.0 },
      { {Const::parseDetectors(Const::CDC), 2}, 5.0 },
      { {Const::parseDetectors(Const::CDC), 3}, 5.0 },
      { {Const::parseDetectors(Const::CDC), 4}, 5.0 },
      { {Const::parseDetectors(Const::CDC), 5}, 5.0 },
      { {Const::parseDetectors(Const::CDC), 6}, 5.0 },
      { {Const::parseDetectors(Const::CDC), 7}, 5.0 },
      { {Const::parseDetectors(Const::CDC), 8}, 5.0 },
      { {Const::parseDetectors(Const::TOP), 0}, 22.0 },
      { {Const::parseDetectors(Const::ARICH), 0}, 10.0 },
      { {Const::parseDetectors(Const::ECL), 0}, 36.0 },
      { {Const::parseDetectors(Const::ECL), 1}, 36.0 },
      { {Const::parseDetectors(Const::KLM), 0}, 20.0 }
    };

    StoreObjPtr<EventMetaData> m_event_metadata;

    StoreObjPtr<ParticleList> m_pListTarget;

    DecayDescriptor m_decaydescriptor;

    StoreObjPtr<ParticleList> m_pListReference;

    bool m_useHighestProbMassForExt;

    bool m_excludePIDDetWeights;

    std::string m_payloadName;

    std::unique_ptr<DBObjPtr<PIDDetectorWeights>> m_DBWeights;

    double getDistAtDetSurface(const Particle* iParticle, const Particle* jParticle, const std::string& detLayerName) const;

    double getDetectorWeight(const Particle* iParticle, const std::string& detName) const;

    double getWeightedSumNonIsoLayers(const Particle* iParticle, const std::string& detName, const float detWeight) const;

    double getWeightedSumInvDists(const Particle* iParticle, const std::string& detName, const float detWeight) const;

    double getDistThreshold(Const::EDetector det, int layer) const
    {
      auto detAndLayer = std::make_pair(Const::parseDetectors(det), layer);
      return m_distThreshPerDetLayer.at(detAndLayer);
    };

    bool onlySelectedStdChargedInDecay();

    Const::EDetector getDetEnum(const std::string& detName) const
    {
 
      if (detName == "CDC") return Const::CDC;
      else if (detName == "TOP") return Const::TOP;
      else if (detName == "ARICH") return Const::ARICH;
      else if (detName == "ECL") return Const::ECL;
      else if (detName == "KLM") return Const::KLM;
      else B2FATAL("Unknown detector component: " << detName);
 
    };
 
  };

}