From 1088617b8f1c31af3ad6a87f9934f7a55240b3a2 Mon Sep 17 00:00:00 2001
From: Thales1330 <thaleslima.ufu@gmail.com>
Date: Mon, 12 Sep 2016 18:49:00 -0300
Subject: Bus form under implementation [2]

---
 Project/Workspace.h | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

(limited to 'Project/Workspace.h')

diff --git a/Project/Workspace.h b/Project/Workspace.h
index 60001b9..c5c0547 100644
--- a/Project/Workspace.h
+++ b/Project/Workspace.h
@@ -34,6 +34,22 @@ enum WorkspaceMode
     MODE_SELECTION_RECT
 };
 
+enum ElementID
+{
+	ID_BUS = 0,
+	ID_LINE,
+	ID_TRANSFORMER,
+	ID_SYNCGENERATOR,
+	ID_INDMOTOR,
+	ID_SYNCMOTOR,
+	ID_LOAD,
+	ID_CAPACITOR,
+	ID_INDUCTOR,
+	ID_TEXT,
+	
+	NUM_ELEMENTS
+};
+
 class Workspace : public WorkspaceBase
 {
    public:
@@ -55,6 +71,9 @@ class Workspace : public WorkspaceBase
 	void RotateSelectedElements(bool clockwise = true);
 	void DeleteSelectedElements();
 	void Fit();
+	
+	int GetElementNumber(ElementID elementID) { return m_elementNumber[elementID]; }
+	void IncrementElementNumber(ElementID elementID) { m_elementNumber[elementID]++; }
 
    protected:
     virtual void OnLeftDoubleClick(wxMouseEvent& event);
@@ -79,6 +98,8 @@ class Workspace : public WorkspaceBase
     WorkspaceMode m_mode = MODE_EDIT;
 
     std::vector<Element*> m_elementList;
+	//int m_numElements = NUM_ELEMENTS;
+	int m_elementNumber[NUM_ELEMENTS];
 
     void UpdateStatusBar();
 
-- 
cgit