blob: 5bea745c8aa5207c8551498a29f2bfacc022ca6f (
plain)
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
|
#ifndef WORKSPACE_H
#define WORKSPACE_H
#include <GL/gl.h>
#include <GL/glu.h>
#include <wx/dcclient.h>
#include <wx/msgdlg.h>
#include <wx/statusbr.h>
#include "WorkspaceBase.h"
class Camera;
class Element;
class Bus;
class Line;
class Transformer;
class SyncGenerator;
class IndMotor;
class SyncMotor;
class Load;
class Inductor;
class Capacitor;
enum WorkspaceMode
{
MODE_EDIT = 0,
MODE_MOVE_ELEMENT,
MODE_MOVE_PICKBOX,
MODE_MOVE_NODE,
MODE_DRAG,
MODE_DRAG_INSERT,
MODE_INSERT,
MODE_SELECTION_RECT
};
class Workspace : public WorkspaceBase
{
public:
Workspace();
Workspace(wxWindow* parent, wxString name = wxEmptyString, wxStatusBar* statusBar = NULL);
~Workspace();
wxString GetName() const { return m_name; }
std::vector<Element*> GetElementList() const { return m_elementList; }
WorkspaceMode GetWorkspaceMode() const { return m_mode; }
Camera* GetCamera() const { return m_camera; }
void SetName(wxString name) { m_name = name; }
void SetElementList(std::vector<Element*> elementList) { m_elementList = elementList; }
void SetStatusBarText(wxString text) { m_statusBar->SetStatusText(text); }
void SetWorkspaceMode(WorkspaceMode mode) { m_mode = mode; }
void Redraw() { m_glCanvas->Refresh(); }
void RotateSelectedElements(bool clockwise = true);
void DeleteSelectedElements();
void Fit();
protected:
virtual void OnLeftDoubleClick(wxMouseEvent& event);
virtual void OnRightClickDown(wxMouseEvent& event);
virtual void OnLeftClickUp(wxMouseEvent& event);
virtual void OnScroll(wxMouseEvent& event);
virtual void OnMiddleDown(wxMouseEvent& event);
virtual void OnMiddleUp(wxMouseEvent& event);
virtual void OnMouseMotion(wxMouseEvent& event);
virtual void OnKeyDown(wxKeyEvent& event);
virtual void OnLeftClickDown(wxMouseEvent& event);
virtual void OnPaint(wxPaintEvent& event);
virtual void OnPopupClick(wxCommandEvent& event);
void SetViewport();
wxGLContext* m_glContext;
wxStatusBar* m_statusBar;
Camera* m_camera;
wxString m_name;
WorkspaceMode m_mode = MODE_EDIT;
std::vector<Element*> m_elementList;
void UpdateStatusBar();
private:
wxRect2DDouble m_selectionRect;
wxPoint2DDouble m_startSelRect;
};
class Camera
{
public:
Camera();
~Camera();
void SetScale(wxPoint2DDouble screenPoint, double delta);
void SetScale(double scale) { m_scale = scale; }
void SetTranslation(wxPoint2DDouble screenPoint);
void StartTranslation(wxPoint2DDouble startPoint) { this->m_translationStartPt = startPoint; }
void UpdateMousePosition(wxPoint2DDouble mousePosition) { this->m_mousePosition = mousePosition; }
double GetScale() const { return m_scale; }
wxPoint2DDouble GetTranslation() const { return m_translation; }
wxPoint2DDouble GetMousePosition(bool worldCoords = true) const;
wxPoint2DDouble ScreenToWorld(wxPoint2DDouble screenCoords) const;
double GetZoomMin() const { return m_zoomMin; }
double GetZoomMax() const { return m_zoomMax; }
protected:
wxPoint2DDouble m_translation;
wxPoint2DDouble m_translationStartPt;
double m_scale;
wxPoint2DDouble m_mousePosition;
double m_zoomMin = 0.05;
double m_zoomMax = 3.0;
};
#endif // WORKSPACE_H
|
