#include <quiz.h>
#include <poincare_layouts.h>
#include <apps/shared/global_context.h>
#include <ion.h>
#include <assert.h>
#include "helper.h"
#include "./tree/helpers.h"

using namespace Poincare;

void assert_parsed_layout_is(Layout l, Poincare::Expression r) {
  Shared::GlobalContext context;
  constexpr int bufferSize = 500;
  char buffer[bufferSize];
  l.serializeForParsing(buffer, bufferSize);
  Expression e = parse_expression(buffer);
  Expression eSimplified;
  e.clone().simplifyAndApproximate(&eSimplified, nullptr, context, Cartesian, Degree);
  if (eSimplified.isUninitialized()) {
    /* In case the simplification is impossible (if there are matrices for
     * instance), use the non simplified expression */
    eSimplified = e;
  }
  Expression rSimplified;
  r.clone().simplifyAndApproximate(&rSimplified, nullptr, context, Cartesian, Degree);
  if (rSimplified.isUninitialized()) {
    /* In case the simplification is impossible (if there are matrices for
     * instance), use the non simplified expression */
    rSimplified = r;
  }

  bool identical = eSimplified.isIdenticalTo(rSimplified);
#if POINCARE_TREE_LOG
  if (!identical) {
    std::cout << "Expecting" << std::endl;
    rSimplified.log();
    std::cout << "Got" << std::endl;
    eSimplified.log();
  }
#endif
  quiz_assert(identical);
}

QUIZ_CASE(poincare_create_all_layouts) {
  EmptyLayout e0 = EmptyLayout::Builder();
  AbsoluteValueLayout e1 = AbsoluteValueLayout::Builder(e0);
  CharLayout e2 = CharLayout::Builder('a');
  BinomialCoefficientLayout e3 = BinomialCoefficientLayout::Builder(e1, e2);
  CeilingLayout e4 = CeilingLayout::Builder(e3);
  RightParenthesisLayout e5 = RightParenthesisLayout::Builder();
  RightSquareBracketLayout e6 = RightSquareBracketLayout::Builder();
  CondensedSumLayout e7 = CondensedSumLayout::Builder(e4, e5, e6);
  ConjugateLayout e8 = ConjugateLayout::Builder(e7);
  LeftParenthesisLayout e10 = LeftParenthesisLayout::Builder();
  FloorLayout e11 = FloorLayout::Builder(e10);
  FractionLayout e12 = FractionLayout::Builder(e8, e11);
  HorizontalLayout e13 = HorizontalLayout::Builder();
  LeftSquareBracketLayout e14 = LeftSquareBracketLayout::Builder();
  IntegralLayout e15 = IntegralLayout::Builder(e11, e12, e13, e14);
  NthRootLayout e16 = NthRootLayout::Builder(e15);
  MatrixLayout e17 = MatrixLayout::Builder();
  EmptyLayout e18 = EmptyLayout::Builder();
  EmptyLayout e19 = EmptyLayout::Builder();
  EmptyLayout e20 = EmptyLayout::Builder();
  ProductLayout e21 = ProductLayout::Builder(e17, e18, e19, e20);
  EmptyLayout e22 = EmptyLayout::Builder();
  EmptyLayout e23 = EmptyLayout::Builder();
  EmptyLayout e24 = EmptyLayout::Builder();
  SumLayout e25 = SumLayout::Builder(e21, e22, e23, e24);
  VerticalOffsetLayout e26 = VerticalOffsetLayout::Builder(e25, VerticalOffsetLayoutNode::Type::Superscript);
}

Matrix BuildOneChildMatrix(Expression entry) {
  Matrix m = Matrix::Builder();
  m.addChildAtIndexInPlace(entry, 0, 0);
  return m;
}

QUIZ_CASE(poincare_parse_layouts) {
  Layout l;
  Expression e;

  // 1+2
  l = HorizontalLayout::Builder(
      CharLayout::Builder('1'),
      CharLayout::Builder('+'),
      CharLayout::Builder('2'));
  e = Addition::Builder(Rational::Builder(1), Rational::Builder(2));
  assert_parsed_layout_is(l, e);

  // |3+3/6|
  l = AbsoluteValueLayout:: Builder(
      HorizontalLayout::Builder(
        CharLayout::Builder('3'),
        CharLayout::Builder('+'),
        FractionLayout::Builder(
          CharLayout::Builder('3'),
          CharLayout::Builder('6'))));
  e = AbsoluteValue::Builder(
      Addition::Builder(
        Rational::Builder(3),
        Division::Builder(
          Rational::Builder(3),
          Rational::Builder(6))));
  assert_parsed_layout_is(l, e);

  // binCoef(4,5)
  l = BinomialCoefficientLayout::Builder(
      CharLayout::Builder('4'),
      CharLayout::Builder('5'));
  e = BinomialCoefficient::Builder(
      Rational::Builder(4),
      Rational::Builder(5));
  assert_parsed_layout_is(l, e);

  // ceil(4.6)
  l = CeilingLayout::Builder(
      HorizontalLayout::Builder(
        CharLayout::Builder('4'),
        CharLayout::Builder('.'),
        CharLayout::Builder('6')));
  e = Ceiling::Builder(
      Decimal::Builder(4.6));
  assert_parsed_layout_is(l, e);

  // floor(7.2)
  l = FloorLayout::Builder(
      HorizontalLayout::Builder(
        CharLayout::Builder('7'),
        CharLayout::Builder('.'),
        CharLayout::Builder('2')));
  e = Floor::Builder(
      Decimal::Builder(7.2));
  assert_parsed_layout_is(l, e);

  // 2^(3+4)
  l = HorizontalLayout::Builder(
      CharLayout::Builder('2'),
      VerticalOffsetLayout::Builder(
        HorizontalLayout::Builder(
          CharLayout::Builder('3'),
          CharLayout::Builder('+'),
          CharLayout::Builder('4')),
        VerticalOffsetLayoutNode::Type::Superscript));
  e = Power::Builder(
      Rational::Builder(2),
      Addition::Builder(
        Rational::Builder(3),
        Rational::Builder(4)));
  assert_parsed_layout_is(l, e);

  // log_3(2)
  HorizontalLayout l1 = HorizontalLayout::Builder();
  l1.addChildAtIndex(CharLayout::Builder('l'), l1.numberOfChildren(),  l1.numberOfChildren(), nullptr);
  l1.addChildAtIndex(CharLayout::Builder('o'), l1.numberOfChildren(),  l1.numberOfChildren(), nullptr);
  l1.addChildAtIndex(CharLayout::Builder('g'), l1.numberOfChildren(),  l1.numberOfChildren(), nullptr);
  l1.addChildAtIndex(VerticalOffsetLayout::Builder(CharLayout::Builder('3'), VerticalOffsetLayoutNode::Type::Subscript), l1.numberOfChildren(),  l1.numberOfChildren(), nullptr);
  l1.addChildAtIndex(LeftParenthesisLayout::Builder(), l1.numberOfChildren(),  l1.numberOfChildren(), nullptr);
  l1.addChildAtIndex(CharLayout::Builder('2'), l1.numberOfChildren(),  l1.numberOfChildren(), nullptr);
  l1.addChildAtIndex(RightParenthesisLayout::Builder(), l1.numberOfChildren(),  l1.numberOfChildren(), nullptr);
  l = l1;
  e = Logarithm::Builder(
      Rational::Builder(2),
      Rational::Builder(3));
  assert_parsed_layout_is(l, e);

  // root(5,3)
  l = NthRootLayout::Builder(
      CharLayout::Builder('5'),
      CharLayout::Builder('3'));
  e = NthRoot::Builder(Rational::Builder(5), Rational::Builder(3));
  assert_parsed_layout_is(l, e);

  // int(7, x, 4, 5)
  l = IntegralLayout::Builder(
      CharLayout::Builder('7'),
      CharLayout::Builder('x'),
      CharLayout::Builder('4'),
      CharLayout::Builder('5'));
  e = Integral::Builder(
      Rational::Builder(7),
      Symbol::Builder('x'),
      Rational::Builder(4),
      Rational::Builder(5));
  assert_parsed_layout_is(l, e);

  // 2^2 !
  l = HorizontalLayout::Builder(
      CharLayout::Builder('2'),
      VerticalOffsetLayout::Builder(
        CharLayout::Builder('2'),
        VerticalOffsetLayoutNode::Type::Superscript),
      CharLayout::Builder('!'));
  e = Factorial::Builder(
      Power::Builder(
        Rational::Builder(2),
        Rational::Builder(2)));
  assert_parsed_layout_is(l, e);

  // 5* 6/(7+5) *3
  l = HorizontalLayout::Builder(
      CharLayout::Builder('5'),
      FractionLayout::Builder(
        CharLayout::Builder('6'),
        HorizontalLayout::Builder(
          CharLayout::Builder('7'),
          CharLayout::Builder('+'),
          CharLayout::Builder('5'))),
      CharLayout::Builder('3'));
  e = Multiplication::Builder(
      Rational::Builder(5),
      Division::Builder(
        Rational::Builder(6),
        Addition::Builder(
          Rational::Builder(7),
          Rational::Builder(5))),
      Rational::Builder(3));
  assert_parsed_layout_is(l, e);

  // [[3^2!, 7][4,5]
  l = MatrixLayout::Builder(
      HorizontalLayout::Builder(
        CharLayout::Builder('3'),
        VerticalOffsetLayout::Builder(
          CharLayout::Builder('2'),
          VerticalOffsetLayoutNode::Type::Superscript),
        CharLayout::Builder('!')),
      CharLayout::Builder('7'),
      CharLayout::Builder('4'),
      CharLayout::Builder('5'));
  Matrix m = BuildOneChildMatrix(
      Factorial::Builder(
        Power::Builder(
          Rational::Builder(3),
          Rational::Builder(2))));
  m.addChildAtIndexInPlace(Rational::Builder(7), 1, 1);
  m.addChildAtIndexInPlace(Rational::Builder(4), 2, 2);
  m.addChildAtIndexInPlace(Rational::Builder(5), 3, 3);
  m.setDimensions(2,2);
  e = m;
  assert_parsed_layout_is(l, e);

  // 2^det([[3!, 7][4,5])
  l = HorizontalLayout::Builder(
      CharLayout::Builder('2'),
      VerticalOffsetLayout::Builder(
        MatrixLayout::Builder(
          HorizontalLayout::Builder(
            CharLayout::Builder('3'),
            CharLayout::Builder('!')),
          CharLayout::Builder('7'),
          CharLayout::Builder('4'),
          CharLayout::Builder('5')),
        VerticalOffsetLayoutNode::Type::Superscript));
  m = BuildOneChildMatrix(
      Factorial::Builder(
        Rational::Builder(3)));
  m.addChildAtIndexInPlace(Rational::Builder(7), 1, 1);
  m.addChildAtIndexInPlace(Rational::Builder(4), 2, 2);
  m.addChildAtIndexInPlace(Rational::Builder(5), 3, 3);
  m.setDimensions(2,2);
  e = Power::Builder(Rational::Builder(2), m);
  assert_parsed_layout_is(l, e);

  // 2e^3
  l = HorizontalLayout::Builder(
      CharLayout::Builder('2'),
      CharLayout::Builder(Ion::Charset::Exponential),
      VerticalOffsetLayout::Builder(
        CharLayout::Builder('3'),
        VerticalOffsetLayoutNode::Type::Superscript));
  e = Multiplication::Builder(Rational::Builder(2),Power::Builder(Constant::Builder(Ion::Charset::Exponential),Parenthesis::Builder(Rational::Builder(3))));
  assert_parsed_expression_is("2X^(3)", Multiplication::Builder(Rational::Builder(2),Power::Builder(Constant::Builder(Ion::Charset::Exponential),Parenthesis::Builder(Rational::Builder(3)))));
  assert_parsed_layout_is(l, e);
}