#include <poincare/variable_context.h>
#include <poincare/preferences.h>
#include <assert.h>
extern "C" {
#include <math.h>
}

namespace Poincare {

template<typename T>
VariableContext<T>::VariableContext(char name, Context * parentContext) :
  m_name(name),
  m_value(Complex<T>::Float(NAN)),
  m_parentContext(parentContext)
{
}

template<typename T>
void VariableContext<T>::setExpressionForSymbolName(const Expression * expression, const Symbol * symbol) {
  if (symbol->name() == m_name) {
    if (expression == nullptr) {
      return;
    }
    /* WARNING: We assume that the evaluation of expression is a real */
    assert(expression->type() == Expression::Type::Complex);
    m_value = Complex<T>::Float(static_cast<const Complex<T> *>(expression)->toScalar());
  } else {
    m_parentContext->setExpressionForSymbolName(expression, symbol);
  }
}

template<typename T>
const Expression * VariableContext<T>::expressionForSymbol(const Symbol * symbol) {
  if (symbol->name() == m_name) {
    return &m_value;
  } else {
    return m_parentContext->expressionForSymbol(symbol);
  }
}

template class Poincare::VariableContext<float>;
template class Poincare::VariableContext<double>;

}