Motor implementation done, elements classes reorganized

1 files changed, 14 insertions, 2 deletions

diff --git a/Project/Workspace.cpp b/Project/Workspace.cpp
index d515683..519d6b3 100644
--- a/Project/Workspace.cpp
+++ b/Project/Workspace.cpp
@@ -4,7 +4,8 @@
 #include "Bus.h"
 #include "Line.h"
 #include "Transformer.h"
-#include "Generator.h"
+#include "SyncGenerator.h"
+#include "IndMotor.h"
 
 // Camera
 Camera::Camera()
@@ -594,7 +595,7 @@ void Workspace::OnKeyDown(wxKeyEvent& event)
 			case 'G':  // Insert a generator.
 			{
 			    if(m_mode != MODE_INSERT) {
-				    Generator* newGenerator = new Generator();
+				    SyncGenerator* newGenerator = new SyncGenerator();
 				    m_elementList.push_back(newGenerator);
 				    m_mode = MODE_INSERT;
 				    m_statusBar->SetStatusText(_("Insert Generator: Click on a buses, ESC to cancel."));
@@ -602,6 +603,17 @@ void Workspace::OnKeyDown(wxKeyEvent& event)
 				}
 			}
 			break;
+			case 'I':  // Insert an induction motor.
+			{
+			    if(m_mode != MODE_INSERT) {
+				    IndMotor* newIndMotor = new IndMotor();
+				    m_elementList.push_back(newIndMotor);
+				    m_mode = MODE_INSERT;
+				    m_statusBar->SetStatusText(_("Insert Induction Motor: Click on a buses, ESC to cancel."));
+				    Redraw();
+				}
+			}
+			break;
 		    default:
 			break;
 		}