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 <framework/database/DBObjPtr.h>
 
#include <analysis/dataobjects/Particle.h>
#include <analysis/dataobjects/ParticleList.h>
#include <analysis/DecayDescriptor/DecayDescriptor.h>
#include <analysis/dbobjects/PIDDetectorWeights.h>
 
 
namespace Belle2 {
  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);
 
    };
 
  };
}