Ticket #2642: 2642_4.diff

source/tools/atlas/AtlasUI/ScenarioEditor/ScenarioEditor.cpp

@@ -20 +20 @@
 #include "ScenarioEditor.h"
 
 #include "wx/busyinfo.h"
+#include "wx/clipbrd.h"
 #include "wx/config.h"
 #include "wx/dir.h"
 #include "wx/evtloop.h"
 #include "wx/ffile.h"
 #include "wx/filename.h"
 #include "wx/image.h"
+#include "wx/sstream.h"
 #include "wx/sysopt.h"
 #include "wx/tooltip.h"
+#include "wx/xml/xml.h"
 
 #include "General/AtlasEventLoop.h"
 #include "General/Datafile.h"
@@ -310 +313 @@
     ID_SaveAs,
     ID_ImportHeightmap,
 
+    ID_Copy,
+    ID_Paste,
+
     ID_Wireframe,
     ID_MessageTrace,
     ID_Screenshot,
@@ -338 +344 @@
     EVT_MENU(ID_Quit, ScenarioEditor::OnQuit)
     EVT_MENU(wxID_UNDO, ScenarioEditor::OnUndo)
     EVT_MENU(wxID_REDO, ScenarioEditor::OnRedo)
+    EVT_MENU(ID_Copy, ScenarioEditor::OnCopy)
+    EVT_MENU(ID_Paste, ScenarioEditor::OnPaste)
 
     EVT_MENU(ID_Wireframe, ScenarioEditor::OnWireframe)
     EVT_MENU(ID_MessageTrace, ScenarioEditor::OnMessageTrace)
@@ -350 +358 @@
     EVT_MENU(ID_RenderPathFixed, ScenarioEditor::OnRenderPath)
     EVT_MENU(ID_RenderPathShader, ScenarioEditor::OnRenderPath)
 
+    EVT_MENU_OPEN(ScenarioEditor::OnMenuOpen)
+
     EVT_IDLE(ScenarioEditor::OnIdle)
 END_EVENT_TABLE()
 
@@ -435 +445 @@
     {
         menuEdit->Append(wxID_UNDO, _("&Undo"));
         menuEdit->Append(wxID_REDO, _("&Redo"));
+        menuEdit->AppendSeparator();
+        menuEdit->Append(ID_Copy, _("&Copy"));
+        menuEdit->Enable(ID_Copy, false);
+        menuEdit->Append(ID_Paste, _("&Paste"));
+        menuEdit->Enable(ID_Paste, false);
     }
 
+    g_SelectedObjects.RegisterObserver(0, &ScenarioEditor::OnSelectedObjectsChange, this);
+
     GetCommandProc().SetEditMenu(menuEdit);
     GetCommandProc().Initialize();
 
@@ -621 +638 @@
     GetCommandProc().Redo();
 }
 
+void ScenarioEditor::OnCopy(wxCommandEvent& WXUNUSED(event))
+{
+    if (GetToolManager().GetCurrentToolName() == _T("TransformObject"))
+        GetToolManager().GetCurrentTool()->OnCommand(_T("copy"), NULL);
+}
+
+void ScenarioEditor::OnPaste(wxCommandEvent& WXUNUSED(event))
+{
+    if (GetToolManager().GetCurrentToolName() != _T("TransformObject"))
+        GetToolManager().SetCurrentTool(_T("TransformObject"), NULL);
+
+    GetToolManager().GetCurrentTool()->OnCommand(_T("paste"), NULL);
+}
+
 //////////////////////////////////////////////////////////////////////////
 
 void ScenarioEditor::OnNew(wxCommandEvent& WXUNUSED(event))
@@ -866 +897 @@
     }
 }
 
+void ScenarioEditor::OnSelectedObjectsChange(const std::vector<ObjectID>& selectedObjects)
+{
+    GetMenuBar()->Enable(ID_Copy, !selectedObjects.empty());
+}
+
+void ScenarioEditor::OnMenuOpen(wxMenuEvent& event)
+{
+    // This could be done far more elegantly if wxMenuItem had changeable id.
+    wxMenu* pasteMenuItem = NULL;
+    event.GetMenu()->FindItem(ID_Paste, &pasteMenuItem);
+
+    GetMenuBar()->Enable(ID_Paste, false);
+
+    if (!pasteMenuItem)
+        return;
+
+    wxString content;
+    if (wxTheClipboard->Open())
+    {
+        if (wxTheClipboard->IsSupported(wxDF_TEXT))
+        {
+            wxTextDataObject data;
+            wxTheClipboard->GetData(data);
+            content = data.GetText();
+        }
+
+        wxTheClipboard->Close();
+    }
+
+    if (content.empty())
+        return;
+
+    wxInputStream* is = new wxStringInputStream(content);
+    wxXmlDocument doc;
+    {
+        wxLogNull stopComplaining;
+        static_cast<void>(stopComplaining);
+        if (!doc.Load(*is))
+            return;
+    }
+
+    wxXmlNode* root = doc.GetRoot();
+    if (!root || root->GetName() != wxT("Entities"))
+        return;
+
+    GetMenuBar()->Enable(ID_Paste, true);
+}
+
 
 //////////////////////////////////////////////////////////////////////////
 

source/tools/atlas/AtlasUI/ScenarioEditor/ScenarioEditor.h

@@ -46 +46 @@
     void OnQuit(wxCommandEvent& event);
     void OnUndo(wxCommandEvent& event);
     void OnRedo(wxCommandEvent& event);
+    void OnCopy(wxCommandEvent& event);
+    void OnPaste(wxCommandEvent& event);
 
     void OnWireframe(wxCommandEvent& event);
     void OnMessageTrace(wxCommandEvent& event);
@@ -55 +57 @@
     void OnCameraReset(wxCommandEvent& event);
     void OnRenderPath(wxCommandEvent& event);
     void OnDumpState(wxCommandEvent& event);
+    void OnSelectedObjectsChange(const std::vector<AtlasMessage::ObjectID>& selectedObjects);
+
+    void OnMenuOpen(wxMenuEvent& event);
 
     bool OpenFile(const wxString& name, const wxString& filename);
 

source/tools/atlas/AtlasUI/ScenarioEditor/Tools/Common/Tools.cpp

@@ -28 +28 @@
     bool OnMouse(wxMouseEvent& WXUNUSED(evt)) { return false; }
     bool OnKey(wxKeyEvent& WXUNUSED(evt), KeyEventType) { return false; }
     void OnTick(float) {}
+    void OnCommand(const wxString& WXUNUSED(comand), void* WXUNUSED(userData)) {}
 } dummy;
 
 struct ToolManagerImpl

source/tools/atlas/AtlasUI/ScenarioEditor/Tools/Common/Tools.h

@@ -35 +35 @@
     virtual bool OnMouse(wxMouseEvent& evt) = 0; // return true if handled
     virtual bool OnKey(wxKeyEvent& evt, KeyEventType dir) = 0; // return true if handled
     virtual void OnTick(float dt) = 0; // dt in seconds
+    virtual void OnCommand(const wxString& command, void* userData) = 0;
 
     virtual ~ITool() {};
 };
@@ -166 +167 @@
     {
         m_CurrentState->OnTick(static_cast<T*>(this), dt);
     }
+
+    virtual void OnCommand(const wxString& WXUNUSED(command), void* WXUNUSED(userData))
+    {
+    }
 };
 
 

source/tools/atlas/AtlasUI/ScenarioEditor/Tools/TransformObject.cpp

@@ -50 +50 @@
         SetState(&Waiting);
     }
 
+    virtual void OnCommand(const wxString& command, void* WXUNUSED(userData))
+    {
+        if (command == _T("copy"))
+            OnCopy();
+        else if (command == _T("paste"))
+            OnPasteStart();
+    }
+
     void OnDisable()
     {
         g_SelectedObjects.clear();
@@ -57 +65 @@
         POST_MESSAGE(SetSelectionPreview, (g_SelectedObjects));
     }
 
+    bool OnCopy() const
+    {
+        if (g_SelectedObjects.empty())
+            return false;
+
+        AtlasMessage::qGetObjectMapSettings info(g_SelectedObjects);
+        info.Post();
+
+        if (wxTheClipboard->Open())
+        {
+            wxString text(info.xmldata.c_str());
+            wxTheClipboard->SetData(new wxTextDataObject(text));
+            wxTheClipboard->Close();
+        }
+
+        return true;
+    }
+
     void OnPasteStart()
     {
-        //PASTE
         wxString entities;
         if (wxTheClipboard->Open())
         {
@@ -71 +96 @@
             }
             wxTheClipboard->Close();
         }
-        else
-        {
-            //TODO: Show something when the clipboard couldnt open
-        }
 
-        //First do we need to check if it is a correct xml string
+        // First do we need to check if it is a correct xml string
         wxInputStream* is = new wxStringInputStream(entities);
         wxXmlDocument doc;
         if (!doc.Load(*is))
@@ -87 +108 @@
         if (root->GetName() != wxT("Entities"))
             return;
 
-        //  Template, position,orientation
+        // Template, position,orientation
         const wxXmlNode* child = root->GetChildren();
 
         while (child)
@@ -95 +116 @@
             if (child->GetName() != wxT("Entity"))
                 return;
 
-            //TODO Validate all attributes
             child = child->GetNext();
         }
 
-        //Update selectedObjects??
         g_SelectedObjects.clear();
         POST_MESSAGE(SetSelectionPreview, (g_SelectedObjects));
 
@@ -329 +348 @@
             {
                 if (evt.GetKeyCode() == 'C')
                 {
-                    if (!g_SelectedObjects.empty())
-                    {
-                        //COPY current selections
-                        AtlasMessage::qGetObjectMapSettings info(g_SelectedObjects);
-                        info.Post();
-
-                        //In xmldata is the configuration
-                        //now send to clipboard
-                        if (wxTheClipboard->Open())
-                        {
-                            wxString text(info.xmldata.c_str());
-                            wxTheClipboard->SetData(new wxTextDataObject(text));
-                            wxTheClipboard->Close();
-                        }
-                        else
-                        {
-                            //TODO: Say something about couldnt open clipboard
-                        }
-                        return true;
-                    }
+                    return obj->OnCopy();
                 }
                 else if (evt.GetKeyCode() == 'V')
                 {