1 files changed, 39 insertions, 0 deletions
diff --git a/Project/ElementDataObject.cpp b/Project/ElementDataObject.cpp
new file mode 100644
index 0000000..a5ac165
--- /dev/null
+++ b/Project/ElementDataObject.cpp
@@ -0,0 +1,39 @@
+#include "ElementDataObject.h"
+
+ElementDataObject::ElementDataObject()
+    : wxDataObjectSimple(wxDataFormat("PSPCopy"))
+{
+    m_elementsLists = new ElementsLists();
+}
+
+ElementDataObject::ElementDataObject(std::vector<Element*> elementList)
+    : wxDataObjectSimple(wxDataFormat("PSPCopy"))
+{
+    m_elementsLists = new ElementsLists();
+    if(elementList.size() > 0) {
+        // Separate buses (parents) from the rest of the elements (childs).
+        for(auto it = elementList.begin(), itEnd = elementList.end(); it != itEnd; ++it) {
+            Element* copy = (*it)->GetCopy();
+            if(copy) {
+                if(typeid(*copy) == typeid(Bus))
+                    m_elementsLists->parentList.push_back((Bus*)copy);
+                else
+                    m_elementsLists->elementList.push_back(copy);
+            }
+        }
+    }
+}
+
+ElementDataObject::~ElementDataObject() {}
+size_t ElementDataObject::GetDataSize() const { return sizeof(void*); }
+bool ElementDataObject::GetDataHere(void* buf) const
+{
+    *(ElementsLists**)buf = m_elementsLists;
+    return true;
+}
+
+bool ElementDataObject::SetData(size_t len, const void* buf)
+{
+    m_elementsLists = *(ElementsLists**)buf;
+    return true;
+}