#include "BEAM/Main/Collider_Weight.H"

using namespace BEAM;
using namespace ATOOLS;

Collider_Weight::Collider_Weight(Kinematics_Base *kinematics)
    : Weight_Base(kinematics), m_mode(collidermode::unknown) {
  if (p_beams[0]->Type() == beamspectrum::monochromatic &&
      p_beams[1]->Type() == beamspectrum::monochromatic)
    m_mode = collidermode::monochromatic;
  else if (p_beams[0]->Type() != beamspectrum::monochromatic &&
           p_beams[1]->Type() == beamspectrum::monochromatic)
    m_mode = collidermode::spectral_1;
  else if (p_beams[0]->Type() == beamspectrum::monochromatic &&
           p_beams[1]->Type() != beamspectrum::monochromatic)
    m_mode = collidermode::spectral_2;
  else if (p_beams[0]->Type() != beamspectrum::monochromatic &&
           p_beams[1]->Type() != beamspectrum::monochromatic)
    m_mode = collidermode::both_spectral;
  if (m_mode == collidermode::unknown)
    THROW(fatal_error, "Bad settings for collider mode.");
}

Collider_Weight::~Collider_Weight() = default;

void Collider_Weight::AssignKeys(Integration_Info *const info) {
  m_sprimekey.Assign(m_keyid + std::string("s'"), 5, 0, info);
  m_ykey.Assign(m_keyid + std::string("y"), 3, 0, info);
  // Convention for m_xkey:
  // [x_{min,beam0}, x_{min,beam1}, x_{max,beam0}, x_{max,beam1}, x_{val,beam0},
  // x_{val,beam1}]. The limits, i.e. index 0,1,2,3 are saved as log(x), the
  // values are saved linearly.
  m_xkey.Assign(m_keyid + std::string("x"), 6, 0, info);
}

bool Collider_Weight::Calculate(const double &scale) {
  m_weight = 0.;
  return (p_beams[0]->CalculateWeight(m_xkey[4], scale) &&
          p_beams[1]->CalculateWeight(m_xkey[5], scale));
}

double Collider_Weight::operator()() {
  m_weight = p_beams[0]->Weight() * p_beams[1]->Weight();
  return m_weight;
}