Blame view

build3/poincare/src/matrix_trace.cpp 1.69 KB
6663b6c9   adorian   projet complet av...
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
  #include <poincare/matrix_trace.h>
  #include <poincare/matrix.h>
  #include <poincare/undefined.h>
  #include <poincare/addition.h>
  extern "C" {
  #include <assert.h>
  }
  #include <cmath>
  
  namespace Poincare {
  
  Expression::Type MatrixTrace::type() const {
    return Type::MatrixTrace;
  }
  
  Expression * MatrixTrace::clone() const {
    MatrixTrace * a = new MatrixTrace(m_operands, true);
    return a;
  }
  
  Expression * MatrixTrace::shallowReduce(Context& context, AngleUnit angleUnit) {
    Expression * e = Expression::shallowReduce(context, angleUnit);
    if (e != this) {
      return e;
    }
    Expression * op = editableOperand(0);
  #if MATRIX_EXACT_REDUCING
    if (op->type() == Type::Matrix) {
      Matrix * m = static_cast<Matrix *>(op);
      if (m->numberOfRows() != m->numberOfColumns()) {
        return replaceWith(new Undefined(), true);
      }
      int n = m->numberOfRows();
      Addition * a = new Addition();
      for (int i = 0; i < n; i++) {
        Expression * diagEntry = m->editableOperand(i+n*i);
        m->detachOperand(diagEntry);
        a->addOperand(diagEntry);
      }
      return replaceWith(a, true)->shallowReduce(context, angleUnit);
    }
    if (!op->recursivelyMatches(Expression::IsMatrix)) {
      return replaceWith(op, true);
    }
    return this;
  #else
    return replaceWith(op, true);
  #endif
  }
  
  template<typename T>
  Expression * MatrixTrace::templatedApproximate(Context& context, AngleUnit angleUnit) const {
    Expression * input = operand(0)->approximate<T>(context, angleUnit);
    Expression * result = nullptr;
    if (input->type() == Type::Complex) {
      result = input->clone();
    } else {
      assert(input->type() == Type::Matrix);
      result = static_cast<Matrix *>(input)->createTrace<T>();
    }
    delete input;
    return result;
  }
  
  }