1 files changed, 82 insertions, 0 deletions
diff --git a/Project/MainFrame.h b/Project/MainFrame.h
new file mode 100644
index 0000000..241551f
--- /dev/null
+++ b/Project/MainFrame.h
@@ -0,0 +1,82 @@
+#ifndef MAINFRAME_H
+#define MAINFRAME_H
+
+#include <wx/menu.h>
+#include <wx/msgdlg.h>
+
+#include "MainFrameBase.h"
+
+class MainFrameBase;
+class wxRibbonMetroArtProvider;
+class Workspace;
+
+enum
+{
+    ID_ADDMENU_BUS = 20000,
+    ID_ADDMENU_LINE,
+    ID_ADDMENU_TRANSFORMER,
+    ID_ADDMENU_GENERATOR,
+    ID_ADDMENU_LOAD,
+    ID_ADDMENU_CAPACITOR,
+    ID_ADDMENU_INDUCTOR,
+    ID_ADDMENU_INDMOTOR,
+    ID_ADDMENU_SYNCCOMP
+};
+
+class MainFrame : public MainFrameBase
+{
+   public:
+    MainFrame();
+    MainFrame(wxWindow* parent, wxLocale* locale);
+    ~MainFrame();
+
+   protected:
+    virtual void NotebookPageClosed(wxAuiNotebookEvent& event);
+    virtual void NotebookPageClosing(wxAuiNotebookEvent& event);
+    virtual void OnAboutClick(wxRibbonButtonBarEvent& event);
+    virtual void OnAddElementDropdown(wxRibbonButtonBarEvent& event);
+    virtual void OnChartsClick(wxRibbonButtonBarEvent& event);
+    virtual void OnCloseClick(wxRibbonButtonBarEvent& event);
+    virtual void OnCopyClick(wxRibbonButtonBarEvent& event);
+    virtual void OnDataReportClick(wxRibbonButtonBarEvent& event);
+    virtual void OnDeleteClick(wxRibbonButtonBarEvent& event);
+    virtual void OnDisableSolutionClick(wxRibbonButtonBarEvent& event);
+    virtual void OnDragClick(wxRibbonButtonBarEvent& event);
+    virtual void OnEnableSolutionClick(wxRibbonButtonBarEvent& event);
+    virtual void OnExitClick(wxRibbonButtonBarEvent& event) { this->Close(); };
+    virtual void OnExpImpClick(wxRibbonButtonBarEvent& event);
+    virtual void OnFaultClick(wxRibbonButtonBarEvent& event);
+    virtual void OnFitClick(wxRibbonButtonBarEvent& event);
+    virtual void OnMoveClick(wxRibbonButtonBarEvent& event);
+    virtual void OnOpenClick(wxRibbonButtonBarEvent& event);
+    virtual void OnPSPGuideClick(wxRibbonButtonBarEvent& event);
+    virtual void OnPasteClick(wxRibbonButtonBarEvent& event);
+    virtual void OnPowerFlowClick(wxRibbonButtonBarEvent& event);
+    virtual void OnProjectSettingsClick(wxRibbonButtonBarEvent& event);
+    virtual void OnRedoClick(wxRibbonButtonBarEvent& event);
+    virtual void OnResetVoltagesClick(wxRibbonButtonBarEvent& event);
+    virtual void OnRunStabilityClick(wxRibbonButtonBarEvent& event);
+    virtual void OnSCPowerClick(wxRibbonButtonBarEvent& event);
+    virtual void OnSaveAsClick(wxRibbonButtonBarEvent& event);
+    virtual void OnSaveClick(wxRibbonButtonBarEvent& event);
+    virtual void OnSnapshotClick(wxRibbonButtonBarEvent& event);
+    virtual void OnStabilitySettingsClick(wxRibbonButtonBarEvent& event);
+    virtual void OnUndoClick(wxRibbonButtonBarEvent& event);
+    virtual void OnNewClick(wxRibbonButtonBarEvent& event);
+
+   private:
+    std::vector<Workspace*> m_workspaceList;
+    int m_projectNumber = 1;
+
+    wxRibbonMetroArtProvider* m_artMetro;
+    wxMenu* m_addElementsMenu;
+    wxLocale* m_locale;
+
+    void Init();
+    void EnableCurrentProjectRibbon(bool enable = true);
+    void CreateAddElementsMenu();
+
+    void OnAddElementsClick(wxCommandEvent& event);
+};
+
+#endif  // MAINFRAME_H