1 files changed, 31 insertions, 0 deletions
diff --git a/Project/Shunt.h b/Project/Shunt.h
new file mode 100644
index 0000000..cf1681d
--- /dev/null
+++ b/Project/Shunt.h
@@ -0,0 +1,31 @@
+#ifndef SHUNT_H
+#define SHUNT_H
+
+#include "Element.h"
+
+class Shunt : public Element
+{
+public:
+	Shunt();
+	~Shunt();
+	
+    virtual bool Contains(wxPoint2DDouble position) const { return m_rect.Contains(position); }
+    virtual bool Intersects(wxRect2DDouble rect) const { return m_rect.Intersects(rect); }
+    virtual void Move(wxPoint2DDouble position);
+    virtual void MoveNode(Element* element, wxPoint2DDouble position);
+    virtual void StartMove(wxPoint2DDouble position);
+    virtual void RotateNode(Element* parent, bool clockwise = true);
+    virtual void RemoveParent(Element* parent);
+    virtual bool NodeContains(wxPoint2DDouble position);
+    virtual bool SetNodeParent(Element* parent);
+    virtual void UpdateNodes();
+
+   protected:
+    void UpdateSwitchesPosition();
+    void UpdatePowerFlowArrowsPosition();
+	void DrawGround(wxPoint2DDouble position) const;
+    bool m_inserted = false;
+
+};
+
+#endif // SHUNT_H