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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
|
#include "ControlElementSolver.h"
#include "ControlElementContainer.h"
#include "ControlEditor.h"
#include "ConnectionLine.h"
#include "Constant.h"
#include "Exponential.h"
#include "Gain.h"
#include "IOControl.h"
#include "Limiter.h"
#include "Multiplier.h"
#include "RateLimiter.h"
#include "Sum.h"
#include "TransferFunction.h"
ControlElementSolver::ControlElementSolver(ControlEditor* controlEditor,
double timeStep,
bool startAllZero,
double input)
{
m_ctrlContainer = new ControlElementContainer();
m_ctrlContainer->FillContainer(controlEditor);
// Check if the sistem have one input and one output
bool fail = false;
wxString failMessage = "";
auto ioList = m_ctrlContainer->GetIOControlList();
if(ioList.size() != 2) {
fail = true;
failMessage = _("The control system must have one input and one output.");
}
bool haveInput, haveOutput;
haveInput = haveOutput = false;
for(auto it = ioList.begin(), itEnd = ioList.end(); it != itEnd; ++it) {
IOControl* io = *it;
if(io->GetType() == Node::NODE_OUT) {
m_inputControl = io;
haveInput = true;
} else if(io->GetType() == Node::NODE_IN) {
m_outputControl = io;
haveOutput = true;
}
}
if(!fail && !haveInput) {
fail = true;
failMessage = _("There is no input in the control system.");
}
if(!fail && !haveOutput) {
fail = true;
failMessage = _("There is no output in the control system.");
}
if(fail) {
wxMessageDialog msgDialog(controlEditor, failMessage, _("Error"), wxOK | wxCENTRE | wxICON_ERROR);
msgDialog.ShowModal();
}
m_timeStep = timeStep;
if(!startAllZero) InitializeValues(input);
}
void ControlElementSolver::InitializeValues(double input) {}
void ControlElementSolver::SolveNextStep(double input)
{
// Set all elements as not solved
auto elementList = m_ctrlContainer->GetControlElementsList();
for(auto it = elementList.begin(), itEnd = elementList.end(); it != itEnd; ++it) {
ControlElement* element = *it;
element->SetSolved(false);
}
auto connectionLineList = m_ctrlContainer->GetConnectionLineList();
for(auto it = connectionLineList.begin(), itEnd = connectionLineList.end(); it != itEnd; ++it) {
ConnectionLine* cLine = *it;
cLine->SetSolved(false);
}
// Get first node and set input value on connected lines
ConnectionLine* firstConn = static_cast<ConnectionLine*>(m_inputControl->GetChildList()[0]);
m_inputControl->SetSolved();
firstConn->SetValue(input);
firstConn->SetSolved();
FillAllConnectedChildren(firstConn);
// Set value to the connected lines in constants
auto constantList = m_ctrlContainer->GetConstantList();
for(auto it = constantList.begin(), itEnd = constantList.end(); it != itEnd; ++it) {
Constant* constant = *it;
constant->SetSolved();
ConnectionLine* child = static_cast<ConnectionLine*>(constant->GetChildList()[0]);
child->SetValue(constant->GetValue());
child->SetSolved();
FillAllConnectedChildren(child);
}
ConnectionLine* currentLine = firstConn;
while(currentLine) {
wxMessageBox(wxString::Format("%d", currentLine->GetID()));
currentLine = SolveNextElement(currentLine);
}
}
void ControlElementSolver::FillAllConnectedChildren(ConnectionLine* parent)
{
auto childList = parent->GetLineChildList();
for(auto it = childList.begin(), itEnd = childList.end(); it != itEnd; ++it) {
ConnectionLine* child = *it;
child->SetValue(parent->GetValue());
child->SetSolved();
FillAllConnectedChildren(child);
}
}
ConnectionLine* ControlElementSolver::SolveNextElement(ConnectionLine* currentLine)
{
auto parentList = currentLine->GetParentList();
for(auto it = parentList.begin(), itEnd = parentList.end(); it != itEnd; ++it) {
ControlElement* element = static_cast<ControlElement*>(*it);
// Solve the unsolved parent.
if(!element->IsSolved()) {
if(!element->Solve(currentLine->GetValue())) return NULL;
element->SetSolved();
// Get the output node (must have one or will result NULL).
Node* outNode = NULL;
auto nodeList = element->GetNodeList();
for(auto itN = nodeList.begin(), itNEnd = nodeList.end(); itN != itNEnd; ++itN) {
Node* node = *itN;
if(node->GetNodeType() == Node::NODE_OUT)
outNode = node;
}
if(!outNode) return NULL;
// Set connection line value associated with the output node.
auto childList = element->GetChildList();
for(auto itC = childList.begin(), itCEnd = childList.end(); itC != itCEnd; ++itC) {
ConnectionLine* cLine = static_cast<ConnectionLine*>(*itC);
if(!cLine->IsSolved()) { // Only check unsolved lines
// Check if the connection line have the output node on the list
auto lineNodeList = cLine->GetNodeList();
for(auto itCN = nodeList.begin(), itCNEnd = nodeList.end(); itCN != itCNEnd; ++itCN) {
Node* childNode = *itCN;
if(childNode == outNode) {
// Check if the line connect two elements, otherwise return NULL
if(cLine->GetType() != ConnectionLine::ELEMENT_ELEMENT) return NULL;
// Set the connection line value and return it.
cLine->SetValue(element->GetOutput());
cLine->SetSolved();
FillAllConnectedChildren(cLine);
return cLine;
}
}
}
}
}
}
return NULL;
}
|
