matrix_trace.cpp
1.69 KB
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
66
#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;
}
}