global_context.cpp
3.15 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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
#include <poincare/global_context.h>
#include <poincare/matrix.h>
#include <assert.h>
#include <cmath>
#include <ion.h>
namespace Poincare {
GlobalContext::GlobalContext() :
m_pi(Complex<double>::Float(M_PI)),
m_e(Complex<double>::Float(M_E))
{
for (int i = 0; i < k_maxNumberOfScalarExpressions; i++) {
m_expressions[i] = nullptr;
}
for (int i = 0; i < k_maxNumberOfMatrixExpressions ; i++) {
m_matrixExpressions[i] = nullptr;
}
}
GlobalContext::~GlobalContext() {
for (int i = 0; i < k_maxNumberOfScalarExpressions; i++) {
if (m_expressions[i] != nullptr) {
delete m_expressions[i];
}
m_expressions[i] = nullptr;
}
for (int i = 0; i < k_maxNumberOfMatrixExpressions; i++) {
if (m_matrixExpressions[i] != nullptr) {
delete m_matrixExpressions[i];
}
m_matrixExpressions[i] = nullptr;
}
}
Complex<double> * GlobalContext::defaultExpression() {
static Complex<double> * defaultExpression = new Complex<double>(Complex<double>::Float(0.0));
return defaultExpression;
}
int GlobalContext::symbolIndex(const Symbol * symbol) const {
int index = symbol->name() - 'A';
return index;
}
const Evaluation<double> * GlobalContext::evaluationForSymbol(const Symbol * symbol) {
if (symbol->name() == Ion::Charset::SmallPi) {
return &m_pi;
}
if (symbol->name() == Ion::Charset::Exponential) {
return &m_e;
}
if (symbol->isMatrixSymbol()) {
int indexMatrix = symbol->name() - (char)Symbol::SpecialSymbols::M0;
return m_matrixExpressions[indexMatrix];
}
int index = symbolIndex(symbol);
if (index < 0 || index >= k_maxNumberOfScalarExpressions) {
return nullptr;
}
if (m_expressions[index] == nullptr) {
return defaultExpression();
}
return m_expressions[index];
}
void GlobalContext::setExpressionForSymbolName(Expression * expression, const Symbol * symbol) {
if (symbol->isMatrixSymbol()) {
int indexMatrix = symbol->name() - (char)Symbol::SpecialSymbols::M0;
assert(indexMatrix >= 0 && indexMatrix < k_maxNumberOfMatrixExpressions);
if (m_matrixExpressions[indexMatrix] != nullptr) {
delete m_matrixExpressions[indexMatrix];
m_matrixExpressions[indexMatrix] = nullptr;
}
if (expression != nullptr) {
Evaluation<double> * evaluation = expression->evaluate<double>(*this);
if (evaluation->numberOfOperands() == 1) {
m_matrixExpressions[indexMatrix] = new ComplexMatrix<double>(evaluation->complexOperand(0), 1, 1);
delete evaluation;
} else {
m_matrixExpressions[indexMatrix] = (ComplexMatrix<double> *)evaluation;
}
}
return;
}
int index = symbolIndex(symbol);
if (index < 0 || index >= k_maxNumberOfScalarExpressions) {
return;
}
if (m_expressions[index] != nullptr) {
delete m_expressions[index];
m_expressions[index] = nullptr;
}
if (expression == nullptr) {
return;
}
Evaluation<double> * evaluation = expression->evaluate<double>(*this);
if (evaluation->numberOfOperands() == 1) {
m_expressions[index] = new Complex<double>(*(evaluation->complexOperand(0)));
} else {
m_expressions[index] = new Complex<double>(Complex<double>::Float(NAN));
}
delete evaluation;
}
}