#include "two_parameter_law.h"
#include <assert.h>

namespace Probability {

TwoParameterLaw::TwoParameterLaw(EvaluateContext * evaluateContext) :
  Law(evaluateContext),
  m_parameter1(0.0f),
  m_parameter2(0.0f)
{
  m_evaluateContext->setOverridenValueForFirstParameter(m_parameter1);
  m_evaluateContext->setOverridenValueForSecondParameter(m_parameter2);
}

int TwoParameterLaw::numberOfParameter() {
  return 2;
}

float TwoParameterLaw::parameterValueAtIndex(int index) {
  assert(index >= 0 && index < 2);
  if (index == 0) {
    return m_parameter1;
  }
  return m_parameter2;
}

void TwoParameterLaw::setParameterAtIndex(float f, int index) {
  assert(index >= 0 && index < 2);
  if (index == 0) {
    m_parameter1 = f;
    m_evaluateContext->setOverridenValueForFirstParameter(f);
  } else {
    m_parameter2 = f;
    m_evaluateContext->setOverridenValueForSecondParameter(f);
  }
}

}