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/logging/LogConfig.h>
#include <analysis/dataobjects/Particle.h>
#include <analysis/dataobjects/ParticleList.h>
#include <analysis/VariableManager/Manager.h>
#include <analysis/VariableManager/Utility.h>
 
 
namespace Belle2 {
  class DetSurfCylBoundaries {
 
  public:
    DetSurfCylBoundaries() {};
    DetSurfCylBoundaries(float rho, float zfwd, float zbwd, float th_fwd, float th_fwd_brl, float th_bwd_brl, float th_bwd)
    {
      m_rho = rho;
      m_zfwd = zfwd;
      m_zbwd = zbwd;
      m_th_fwd = th_fwd;
      m_th_fwd_brl = th_fwd_brl;
      m_th_bwd_brl = th_bwd_brl;
      m_th_bwd = th_bwd;
    };
    float m_rho; 
    float m_zfwd; 
    float m_zbwd; 
    float m_th_fwd; 
    float m_th_fwd_brl; 
    float m_th_bwd_brl; 
    float m_th_bwd; 
  };
 
  class TrackIsoCalculatorModule : public Module {
 
  public:
 
    TrackIsoCalculatorModule();
 
    ~TrackIsoCalculatorModule() override;
 
    void initialize() override;
 
    void event() override;
 
    void terminate() override;
 
    double get3DDistAtDetSurface(Particle* iParticle, Particle* jParticle);
 
    double get2DRhoPhiDistAsChordLength(Particle* iParticle, Particle* jParticle);
 
  private:
 
    std::string m_pListName;
 
    std::string m_detInnerSurface;
 
    bool m_use2DRhoPhiDist = false;
 
    std::string m_extraInfoName;
 
    std::unordered_map<std::string, DetSurfCylBoundaries> m_detSurfBoundaries = {
      {"CDC", DetSurfCylBoundaries(16.0, 150.0, -75.0, 0.0, 0.29, 2.61, 3.14)},
      {"PID", DetSurfCylBoundaries(116.4, 167.0, -335.1, 0.24, 0.52, 2.23, 3.14)}, // PID : TOP, ARICH
      {"ECL", DetSurfCylBoundaries(125.0, 196.0, -102.0, 0.21, 0.56, 2.24, 2.70)},
      {"KLM", DetSurfCylBoundaries(202.0, 283.9, -189.9, 0.40, 0.82, 2.13, 2.60)},
    };
 
    StoreObjPtr<EventMetaData> m_event_metadata;
 
    StoreObjPtr<ParticleList> m_pList;
 
    inline bool isStdChargedList()
    {
      return (Const::chargedStableSet.find(abs(m_pList->getPDGCode())) != Const::invalidParticle);
    };
 
    void printDistancesArr(const std::vector<std::vector<double>>& arr, int size);
 
  };
}