From b06d4c3d76cca3ed8a54dae1d978030d687233b2 Mon Sep 17 00:00:00 2001
From: Thales1330 <thaleslima.ufu@gmail.com>
Date: Thu, 1 Sep 2016 18:23:56 -0300
Subject: generator under implementation

---
 Project/Workspace.cpp | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

(limited to 'Project/Workspace.cpp')

diff --git a/Project/Workspace.cpp b/Project/Workspace.cpp
index 5408327..d515683 100644
--- a/Project/Workspace.cpp
+++ b/Project/Workspace.cpp
@@ -4,6 +4,7 @@
 #include "Bus.h"
 #include "Line.h"
 #include "Transformer.h"
+#include "Generator.h"
 
 // Camera
 Camera::Camera()
@@ -590,6 +591,17 @@ void Workspace::OnKeyDown(wxKeyEvent& event)
 				}
 			}
 			break;
+			case 'G':  // Insert a generator.
+			{
+			    if(m_mode != MODE_INSERT) {
+				    Generator* newGenerator = new Generator();
+				    m_elementList.push_back(newGenerator);
+				    m_mode = MODE_INSERT;
+				    m_statusBar->SetStatusText(_("Insert Generator: Click on a buses, ESC to cancel."));
+				    Redraw();
+				}
+			}
+			break;
 		    default:
 			break;
 		}
@@ -649,6 +661,11 @@ void Workspace::OnPopupClick(wxCommandEvent& event)
 		    wxMessageBox("Edit line!");
 		}
 		break;
+		case ID_EDIT_TRANSFORMER:
+		{
+		    wxMessageBox("Edit transformer!");
+		}
+		break;
 	    case ID_LINE_ADD_NODE:
 		{
 		    Line* line = (Line*)element;
-- 
cgit