#include "finite_integral_calculation.h"
#include "../law/normal_law.h"
#include <assert.h>
#include <ion.h>
#include <cmath>

namespace Probability {

FiniteIntegralCalculation::FiniteIntegralCalculation() :
  Calculation(),
  m_lowerBound(0.0),
  m_upperBound(1.0),
  m_result(0.0)
{
  compute(0);
}

Calculation::Type FiniteIntegralCalculation::type() {
  return Type::FiniteIntegral;
}

int FiniteIntegralCalculation::numberOfParameters() {
  return 3;
}

int FiniteIntegralCalculation::numberOfEditableParameters() {
  if (m_law->type() == Law::Type::Normal) {
    return 3;
  }
  return 2;
}

I18n::Message FiniteIntegralCalculation::legendForParameterAtIndex(int index) {
  assert(index >= 0 && index < 3);
  if (index == 0) {
    return I18n::Message::RightIntegralFirstLegend;
  }
  if (index == 1) {
    return I18n::Message::FiniteIntegralLegend;
  }
  return I18n::Message::LeftIntegralSecondLegend;
}

void FiniteIntegralCalculation::setParameterAtIndex(double f, int index) {
  assert(index >= 0 && index < 3);
  if (index == 0) {
    m_lowerBound = f;
  }
  if (index == 1) {
    m_upperBound = f;
  }
  if (index == 2) {
    m_result = f;
  }
  compute(index);
}


double FiniteIntegralCalculation::parameterAtIndex(int index) {
  assert(index >= 0 && index < 3);
  if (index == 0) {
    return m_lowerBound;
  }
  if (index == 1) {
    return m_upperBound;
  }
  return m_result;
}

double FiniteIntegralCalculation::lowerBound() {
  return m_lowerBound;
}

double FiniteIntegralCalculation::upperBound() {
  return m_upperBound;
}

void FiniteIntegralCalculation::compute(int indexKnownElement) {
  if (m_law == nullptr) {
    return;
  }
  if (indexKnownElement == 2) {
    assert(m_law->type() == Law::Type::Normal);
    double p = (1.0+m_result)/2.0;
    m_upperBound = ((NormalLaw *)m_law)->cumulativeDistributiveInverseForProbability(&p);
    m_lowerBound = 2.0*m_law->parameterValueAtIndex(0)-m_upperBound;
  }
  m_result = m_law->finiteIntegralBetweenAbscissas(m_lowerBound, m_upperBound);
}

}