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
|
#include "Shunt.h"
Shunt::Shunt() : Element()
{
}
Shunt::~Shunt()
{
}
void Shunt::UpdateSwitchesPosition()
{
if(m_parentList[0]) {
m_pointList[1] = GetSwitchPoint(m_parentList[0], m_pointList[0], m_pointList[2]);
}
else
{
m_pointList[1] = m_pointList[0];
}
}
void Shunt::Move(wxPoint2DDouble position)
{
SetPosition(m_movePos + position - m_moveStartPt);
for(int i = 2; i < (int)m_pointList.size(); i++) {
m_pointList[i] = m_movePts[i] + position - m_moveStartPt;
}
if(!m_parentList[0]) {
m_pointList[0] = m_movePts[0] + position - m_moveStartPt;
}
UpdateSwitchesPosition();
}
void Shunt::MoveNode(Element* element, wxPoint2DDouble position)
{
if(element) {
if(element == m_parentList[0]) {
m_pointList[0] = m_movePts[0] + position - m_moveStartPt;
}
}
else
{
if(m_activeNodeID == 1) {
m_pointList[0] = m_movePts[0] + position - m_moveStartPt;
m_parentList[0] = NULL;
}
}
// Recalculate switches positions
UpdateSwitchesPosition();
}
void Shunt::StartMove(wxPoint2DDouble position)
{
m_moveStartPt = position;
m_movePts = m_pointList;
m_movePos = m_position;
}
void Shunt::RemoveParent(Element* parent)
{
if(parent == m_parentList[0]) {
m_parentList[0] = NULL;
UpdateSwitchesPosition();
}
}
bool Shunt::NodeContains(wxPoint2DDouble position)
{
wxRect2DDouble nodeRect(m_pointList[0].m_x - 5.0 - m_borderSize, m_pointList[0].m_y - 5.0 - m_borderSize,
10 + 2.0 * m_borderSize, 10 + 2.0 * m_borderSize);
if(nodeRect.Contains(position)) {
m_activeNodeID = 1;
return true;
}
m_activeNodeID = 0;
return false;
}
bool Shunt::SetNodeParent(Element* parent)
{
if(parent && m_activeNodeID != 0) {
wxRect2DDouble nodeRect(m_pointList[0].m_x - 5.0 - m_borderSize, m_pointList[0].m_y - 5.0 - m_borderSize,
10 + 2.0 * m_borderSize, 10 + 2.0 * m_borderSize);
if(parent->Intersects(nodeRect)) {
m_parentList[0] = parent;
// Centralize the node on bus.
wxPoint2DDouble parentPt = parent->RotateAtPosition(
m_pointList[0], -parent->GetAngle()); // Rotate click to horizontal position.
parentPt.m_y = parent->GetPosition().m_y; // Centralize on bus.
parentPt = parent->RotateAtPosition(parentPt, parent->GetAngle());
m_pointList[0] = parentPt;
UpdateSwitchesPosition();
return true;
}
else
{
m_parentList[0] = NULL;
}
}
return false;
}
void Shunt::UpdateNodes()
{
if(m_parentList[0]) {
wxRect2DDouble nodeRect(m_pointList[0].m_x - 5.0 - m_borderSize, m_pointList[0].m_y - 5.0 - m_borderSize,
10 + 2.0 * m_borderSize, 10 + 2.0 * m_borderSize);
if(!m_parentList[0]->Intersects(nodeRect)) {
m_parentList[0] = NULL;
UpdateSwitchesPosition();
}
}
}
void Shunt::RotateNode(Element* parent)
{
if(parent == m_parentList[0]) {
m_pointList[0] = parent->RotateAtPosition(m_pointList[0], m_rotationAngle);
UpdateSwitchesPosition();
}
}
|
