Algorithm.h

#pragma once
 
#include "iguana/algorithms/Algorithm.h"
 
#include <Math/Vector3D.h>
#include <Math/VectorUtil.h>
 
namespace iguana::clas12 {

  class PhotonGBTFilter : public Algorithm
  {
 
      DEFINE_IGUANA_ALGORITHM(PhotonGBTFilter, clas12::PhotonGBTFilter)
 
    public:
 
      void Start(hipo::banklist& banks) override;
      void Run(hipo::banklist& banks) const override;
      void Stop() override;   

      bool ForwardDetectorFilter(float const theta) const;
 
    private:
      
      struct calo_row_data {
        double pcal_x = 0;
        double pcal_y = 0;
        double pcal_z = 0;
        double ecin_x = 0;
        double ecin_y = 0;
        double ecin_z = 0;
        double ecout_x = 0;
        double ecout_y = 0;
        double ecout_z = 0;
        double pcal_e = 0;
        double pcal_m2u = 0;
        double pcal_m2v = 0;
        double ecin_e = 0;
        double ecin_m2u = 0;
        double ecin_m2v = 0;
        double ecout_e = 0;
        double ecout_m2u = 0;
        double ecout_m2v = 0;
      }; 
      
      bool PidPurityPhotonFilter(float const E, float const Epcal, float const theta) const;

      bool Filter(hipo::bank const &particleBank, hipo::bank const &caloBank, std::map<int, PhotonGBTFilter::calo_row_data> calo_map, int const row, int const runnum) const;
      
      
      bool ClassifyPhoton(std::vector<float> const &input_data, int const runnum) const;
      
      
      std::map<int, PhotonGBTFilter::calo_row_data> GetCaloMap(hipo::bank const &bank) const;
      
      
      ROOT::Math::XYZVector GetParticleCaloVector(PhotonGBTFilter::calo_row_data calo_row) const;
      
      std::function<double(std::vector<float> const &)> getModelFunction(int runnum) const;
      
      hipo::banklist::size_type b_particle;
      hipo::banklist::size_type b_calorimeter;
      hipo::banklist::size_type b_config; // RUN::config
      
      double o_threshold = 0.78;
      
      int o_pass = 1;
    
      static std::map<std::tuple<int, int, int>, std::function<double(std::vector<float> const &)>> const modelMap;
  };
  class PhotonGBTFilter : public Algorithm {…};
    
}