1 files changed, 29 insertions, 0 deletions
diff --git a/Project/ElementDataObject.h b/Project/ElementDataObject.h
new file mode 100644
index 0000000..646e76c
--- /dev/null
+++ b/Project/ElementDataObject.h
@@ -0,0 +1,29 @@
+#ifndef ELEMENTDATAOBJECT_H
+#define ELEMENTDATAOBJECT_H
+
+#include "Workspace.h"
+#include <wx/dataobj.h>
+
+struct ElementsLists {
+    std::vector<Element*> elementList;
+    std::vector<Bus*> parentList;
+};
+
+class ElementDataObject : public wxDataObjectSimple
+{
+public:
+    ElementDataObject();
+    ElementDataObject(std::vector<Element*> elementList);
+    ~ElementDataObject();
+
+    size_t GetDataSize() const override;
+    bool GetDataHere(void* buf) const override;
+    bool SetData(size_t len, const void* buf) override;
+
+    ElementsLists* GetElementsLists() { return m_elementsLists; }
+
+protected:
+    ElementsLists* m_elementsLists;
+};
+
+#endif // ELEMENTDATAOBJECT_H