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;
      bool Run(hipo::banklist& banks) const override;
      void Stop() override;

      bool Run(
          hipo::bank& particleBank,
          hipo::bank const& caloBank,
          hipo::bank const& configBank) const;

      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;
  };
 
}