#include <poincare/opposite.h>
#include <poincare/complex_matrix.h>
#include <poincare/complex.h>
extern "C" {
#include <assert.h>
#include <stdlib.h>
}
#include <cmath>
#include "layout/horizontal_layout.h"
#include "layout/parenthesis_layout.h"
#include "layout/string_layout.h"

namespace Poincare {

Opposite::Opposite(Expression * operand, bool cloneOperands) {
  assert(operand != nullptr);
  if (cloneOperands) {
    m_operand = operand->clone();
  } else {
    m_operand = operand;
  }
}

Opposite::~Opposite() {
  delete m_operand;
}

bool Opposite::hasValidNumberOfArguments() const {
  return m_operand->hasValidNumberOfArguments();
}

const Expression * Opposite::operand(int i) const {
  assert(i == 0);
  return m_operand;
}

int Opposite::numberOfOperands() const {
  return 1;
}

Expression * Opposite::clone() const {
  return this->cloneWithDifferentOperands((Expression**)&m_operand, 1, true);
}

template<typename T>
Complex<T> Opposite::compute(const Complex<T> c) {
  return Complex<T>::Cartesian(-c.a(), -c.b());
}

template<typename T>
Evaluation<T> * Opposite::computeOnMatrix(Evaluation<T> * m) {
  Complex<T> * operands = new Complex<T>[m->numberOfRows() * m->numberOfColumns()];
  for (int i = 0; i < m->numberOfRows() * m->numberOfColumns(); i++) {
    Complex<T> entry = *(m->complexOperand(i));
    operands[i] = Complex<T>::Cartesian(-entry.a(), -entry.b());
  }
  Evaluation<T> * matrix = new ComplexMatrix<T>(operands, m->numberOfRows(), m->numberOfColumns());
  delete[] operands;
  return matrix;
}

template<typename T>
Evaluation<T> * Opposite::templatedEvaluate(Context& context, AngleUnit angleUnit) const {
  Evaluation<T> * operandEvalutation = m_operand->evaluate<T>(context, angleUnit);
  Evaluation<T> * result = nullptr;
  if (operandEvalutation->numberOfRows() == 1 && operandEvalutation->numberOfColumns() == 1) {
    result = new Complex<T>(compute(*(operandEvalutation->complexOperand(0))));
  } else {
    result = computeOnMatrix(operandEvalutation);
  }
  delete operandEvalutation;
  return result;
}

ExpressionLayout * Opposite::privateCreateLayout(FloatDisplayMode floatDisplayMode, ComplexFormat complexFormat) const {
  assert(floatDisplayMode != FloatDisplayMode::Default);
  assert(complexFormat != ComplexFormat::Default);
  ExpressionLayout * children_layouts[2];
  char string[2] = {'-', '\0'};
  children_layouts[0] = new StringLayout(string, 1);
  children_layouts[1] = m_operand->type() == Type::Opposite ? new ParenthesisLayout(m_operand->createLayout(floatDisplayMode, complexFormat)) : m_operand->createLayout(floatDisplayMode, complexFormat);
  return new HorizontalLayout(children_layouts, 2);
}

Expression::Type Opposite::type() const {
  return Expression::Type::Opposite;
}

Expression * Opposite::cloneWithDifferentOperands(Expression** newOperands,
    int numberOfOperands, bool cloneOperands) const {
  assert(newOperands != nullptr);
  assert(numberOfOperands == 1);
  return new Opposite(newOperands[0], cloneOperands);
}

}

template Poincare::Complex<float> Poincare::Opposite::compute<float>(Poincare::Complex<float>);
template Poincare::Complex<double> Poincare::Opposite::compute<double>(Poincare::Complex<double>);