Algorithm.h

#pragma once
 
#include "iguana/algorithms/Algorithm.h"
#include "iguana/algorithms/physics/Tools.h"
#include <Math/Vector3D.h>
#include <Math/Vector4D.h>
 
namespace iguana::physics {

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

      bool Run(
          hipo::bank const& particle_bank,
          hipo::bank const& inc_kin_bank,
          hipo::bank& result_bank) const;

      std::vector<std::pair<int, int>> PairHadrons(hipo::bank const& particle_bank) const;
 
    private:
 
      // banklist indices
      hipo::banklist::size_type b_particle;
      hipo::banklist::size_type b_inc_kin;
      hipo::banklist::size_type b_result;
 
      // `b_result` bank item indices
      int i_pindex_a;
      int i_pindex_b;
      int i_pdg_a;
      int i_pdg_b;
      int i_Mh;
      int i_z;
      int i_PhPerp;
      int i_MX2;
      int i_xF;
      int i_yB;
      int i_phiH;
      int i_phiR;
      int i_theta;
 
      // config options
      std::set<int> o_hadron_a_pdgs;
      std::set<int> o_hadron_b_pdgs;
      std::string o_phi_r_method;
      std::string o_theta_method;
      enum { e_RT_via_covariant_kT } m_phi_r_method;
      enum { e_hadron_a } m_theta_method;
 
      // storage for a single hadron
      struct Hadron {
          int row;
          int pdg;
          ROOT::Math::PxPyPzMVector p;
          double z;
          std::optional<ROOT::Math::XYZVector> p_perp;
      };
  };
 
}