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 {
 
  struct DihadronKinematicsVars {
    int pindex_a;
    int pindex_b;
    int pdg_a;
    int pdg_b;
    double Mh;
    double z;
    double PhPerp;
    double MX;
    double xF;
    double phiH;
    double phiR;
    double theta;
  };
 
  class DihadronKinematics : public Algorithm
  {
 
      DEFINE_IGUANA_ALGORITHM(DihadronKinematics, physics::DihadronKinematics)
 
    public:
 
      void Start(hipo::banklist& banks) override;
      void Run(hipo::banklist& banks) const override;
      void Stop() override;
 
      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_MX;
      int i_xF;
      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;
      };
 
  };
 
}