DataStructure root node

src/simplnx/DataStructure/BaseGroup.cpp

@@ -1,6 +1,7 @@
 #include "BaseGroup.hpp"
 
 #include "simplnx/DataStructure/DataPath.hpp"
+#include "simplnx/DataStructure/DataStructure.hpp"
 #include "simplnx/Utilities/StringUtilities.hpp"
 
 using namespace nx::core;
@@ -21,7 +22,7 @@ BaseGroup::BaseGroup(const BaseGroup& other)
 {
 }
 
-BaseGroup::BaseGroup(BaseGroup&& other)
+BaseGroup::BaseGroup(BaseGroup&& other) noexcept
 : DataObject(std::move(other))
 , m_DataMap(std::move(other.m_DataMap))
 {
@@ -105,6 +106,11 @@ bool BaseGroup::canInsert(const DataObject* obj) const
   {
     return false;
   }
+  // Do not allow adding a DataStructure to a BaseGroup
+  if(dynamic_cast<const DataStructure*>(obj) != nullptr)
+  {
+    return false;
+  }
   if(contains(obj) || contains(obj->getName()))
   {
     return false;

src/simplnx/DataStructure/BaseGroup.hpp

@@ -67,7 +67,7 @@ class SIMPLNX_EXPORT BaseGroup : public DataObject
    * the provided BaseGroup.
    * @param other
    */
-  BaseGroup(BaseGroup&& other);
+  BaseGroup(BaseGroup&& other) noexcept;
 
   /**
    * @brief Destroys the BaseGroup and removes it from the list of it's
@@ -278,7 +278,7 @@ class SIMPLNX_EXPORT BaseGroup : public DataObject
   /**
    * @brief Clears the group of all children.
    */
-  void clear();
+  virtual void clear();
 
   /**
    * @brief Returns an iterator to the beginning of the container.

src/simplnx/DataStructure/DataObject.cpp

@@ -76,7 +76,7 @@ DataObject& DataObject::operator=(DataObject&& rhs) noexcept
 
 DataObject::~DataObject() noexcept
 {
-  if(m_DataStructure == nullptr)
+  if(m_DataStructure == nullptr || m_DataStructure == this)
   {
     return;
   }

src/simplnx/DataStructure/DataStructure.cpp

@@ -29,13 +29,14 @@
 namespace nx::core
 {
 DataStructure::DataStructure()
-: m_IsValid(true)
+: BaseGroup(*this, k_TypeName)
+, m_IsValid(true)
 {
 }
 
 DataStructure::DataStructure(const DataStructure& dataStructure)
-: m_DataObjects(dataStructure.m_DataObjects)
-, m_RootGroup(dataStructure.m_RootGroup)
+: BaseGroup(dataStructure)
+, m_DataObjects(dataStructure.m_DataObjects)
 , m_IsValid(dataStructure.m_IsValid)
 , m_NextId(dataStructure.m_NextId)
 {
@@ -54,16 +55,16 @@
   }
   // Updates all DataMaps with the corresponding m_DataObjects pointers.
   // Updates all DataObjects with their new DataStructure
-  m_RootGroup.setDataStructure(this);
+  setDataStructure(this);
 }
 
 DataStructure::DataStructure(DataStructure&& dataStructure) noexcept
-: m_DataObjects(std::move(dataStructure.m_DataObjects))
-, m_RootGroup(std::move(dataStructure.m_RootGroup))
+: BaseGroup(std::move(dataStructure))
+, m_DataObjects(std::move(dataStructure.m_DataObjects))
 , m_IsValid(dataStructure.m_IsValid)
 , m_NextId(dataStructure.m_NextId)
 {
-  m_RootGroup.setDataStructure(this);
+  getDataMap().setDataStructure(this);
 }
 
 DataStructure::~DataStructure()
@@ -98,7 +99,7 @@
 
 void DataStructure::clear()
 {
-  auto topDataIds = m_RootGroup.getKeys();
+  auto topDataIds = getDataMap().getKeys();
   for(auto dataId : topDataIds)
   {
     removeData(dataId);
@@ -125,7 +126,7 @@
   try
   {
     std::vector<DataObject::IdType> pathIds;
-    const DataObject* data = m_RootGroup[path[0]];
+    const DataObject* data = getDataMap()[path[0]];
     const BaseGroup* parent = dynamic_cast<const BaseGroup*>(data);
     pathIds.push_back(data->getId());
 
@@ -163,7 +164,7 @@
   {
     std::vector<DataObject::IdType> pathIds;
     std::string name = path[0];
-    const DataObject* data = m_RootGroup[name];
+    const DataObject* data = getDataMap()[name];
     if(data == nullptr)
     {
       data = nx::core::DataGroup::Create(*this, name);
@@ -272,7 +273,7 @@
   {
     return nullptr;
   }
-  DataObject* targetObject = m_RootGroup[path[0]];
+  DataObject* targetObject = getDataMap()[path[0]];
   for(usize index = 1; index < path.getLength(); index++)
   {
     if(targetObject == nullptr)
@@ -351,7 +352,7 @@
   {
     return nullptr;
   }
-  const DataObject* targetObject = m_RootGroup[path[0]];
+  const DataObject* targetObject = getDataMap()[path[0]];
   for(usize index = 1; index < path.getLength(); index++)
   {
     if(targetObject == nullptr)
@@ -508,47 +509,43 @@
   notify(msg);
 }
 
+DataMap& DataStructure::getRootGroup()
+{
+  return getDataMap();
+}
+
 std::vector<DataObject*> DataStructure::getTopLevelData() const
 {
   std::vector<DataObject*> topLevel(m_RootGroup.getSize(), nullptr);
   usize index = 0;
   for(auto& iter : m_RootGroup)
   {
     auto obj = iter.second;
     topLevel[index++] = (obj.get());
   }
   return topLevel;
 }
 
-const DataMap& DataStructure::getDataMap() const
-{
-  return m_RootGroup;
-}
-
-DataMap& DataStructure::getRootGroup()
-{
-  return m_RootGroup;
-}
-
 bool DataStructure::insertTopLevel(const std::shared_ptr<DataObject>& obj)
 {
   if(obj == nullptr)
   {
     return false;
   }
 
-  if(m_RootGroup.contains(obj.get()) || m_RootGroup.contains(obj->getName()))
+  auto& dataMap = getDataMap();
+  if(dataMap.contains(obj.get()) || dataMap.contains(obj->getName()))
   {
     return false;
   }
 
-  return m_RootGroup.insert(obj);
+  return dataMap.insert(obj);
 }
 
 bool DataStructure::removeTopLevel(DataObject* data)
 {
   std::string name = data->getName();
-  if(!m_RootGroup.remove(data))
+  if(!getDataMap().remove(data))
   {
     return false;
   }
@@ -585,22 +582,22 @@
 
 DataStructure::Iterator DataStructure::begin()
 {
-  return m_RootGroup.begin();
+  return getDataMap().begin();
 }
 
 DataStructure::Iterator DataStructure::end()
 {
-  return m_RootGroup.end();
+  return getDataMap().end();
 }
 
 DataStructure::ConstIterator DataStructure::begin() const
 {
-  return m_RootGroup.begin();
+  return getDataMap().begin();
 }
 
 DataStructure::ConstIterator DataStructure::end() const
 {
-  return m_RootGroup.end();
+  return getDataMap().end();
 }
 
 bool DataStructure::insert(const std::shared_ptr<DataObject>& dataObject, const DataPath& dataPath)
@@ -644,7 +641,7 @@
     return false;
   }
 
-  if(!m_RootGroup.insert(dataObject))
+  if(!getDataMap().insert(dataObject))
   {
     return false;
   }
@@ -734,7 +731,7 @@
 DataStructure& DataStructure::operator=(const DataStructure& rhs)
 {
   m_DataObjects = rhs.m_DataObjects;
-  m_RootGroup = rhs.m_RootGroup;
+  getDataMap() = rhs.getDataMap();
   m_IsValid = rhs.m_IsValid;
   m_NextId = rhs.m_NextId;
 
@@ -760,17 +757,33 @@
 DataStructure& DataStructure::operator=(DataStructure&& rhs) noexcept
 {
   m_DataObjects = std::move(rhs.m_DataObjects);
-  m_RootGroup = std::move(rhs.m_RootGroup);
+  getDataMap() = std::move(rhs.getDataMap());
   m_IsValid = std::move(rhs.m_IsValid);
   m_NextId = std::move(rhs.m_NextId);
 
   applyAllDataStructure();
   return *this;
 }
 
+std::shared_ptr<DataObject> DataStructure::deepCopy(const DataPath& copyPath)
+{
+  // DataStructure cannot be contained in another DataStructure...
+  return nullptr;
+}
+
+DataObject* DataStructure::shallowCopy()
+{
+  return new DataStructure(*this);
+}
+
+std::string DataStructure::getTypeName() const
+{
+  return k_TypeName;
+}
+
 void DataStructure::applyAllDataStructure()
 {
-  m_RootGroup.setDataStructure(this);
+  setDataStructure(this);
 }
 
 nonstd::expected<void, std::string> DataStructure::validateNumberOfTuples(const std::vector<DataPath>& dataPaths) const
@@ -861,7 +874,7 @@
       dataObjectPtr->checkUpdatedIds(updatedIdsMap);
     }
   }
-  m_RootGroup.updateIds(updatedIdsMap);
+  getDataMap().updateIds(updatedIdsMap);
 }
 
 void DataStructure::exportHierarchyAsGraphViz(std::ostream& outputStream) const
@@ -1015,7 +1028,7 @@
    * the compile will fail if the unit tests are enabled. Which they are on all the CI machines.
    */
   Result<> result;
-  for(const auto& dataObject : m_RootGroup)
+  for(const auto& dataObject : getDataMap())
   {
     auto dataObjectType = dataObject.second->getDataObjectType();
     if(dataObjectType >= DataObject::Type::IGeometry && dataObjectType <= DataObject::Type::TetrahedralGeom)

src/simplnx/DataStructure/DataStructure.hpp

@@ -1,6 +1,7 @@
 #pragma once
 
 #include "simplnx/Common/Result.hpp"
+#include "simplnx/DataStructure/BaseGroup.hpp"
 #include "simplnx/DataStructure/DataMap.hpp"
 #include "simplnx/DataStructure/DataObject.hpp"
 #include "simplnx/DataStructure/LinkedPath.hpp"
@@ -40,7 +41,7 @@
  * geometries, and scalars are added to the structure. The DataStructure allows
  * parents to be added to or removed from DataObjects.
  */
-class SIMPLNX_EXPORT DataStructure
+class SIMPLNX_EXPORT DataStructure : public BaseGroup
 {
   using WeakCollectionType = std::map<DataObject::IdType, std::weak_ptr<DataObject>>;
 
@@ -56,6 +57,8 @@
   bool finishAddingObject(const std::shared_ptr<DataObject>& obj, const std::optional<DataObject::IdType>& parent = {});
 
 public:
+  static constexpr StringLiteral k_TypeName = "DataStructure";
+
   using SignalType = nod::signal<void(DataStructure*, const std::shared_ptr<AbstractDataStructureMessage>&)>;
   using Iterator = DataMap::Iterator;
   using ConstIterator = DataMap::ConstIterator;
@@ -83,7 +86,7 @@
   /**
    * @brief
    */
   ~DataStructure();
 
   /**
    * @brief Returns the number of unique DataObjects in the DataStructure.
@@ -95,7 +98,7 @@
    * @brief Clears the DataStructure by removing all DataObjects. The next
    * DataObject ID remains unchanged after the operation.
    */
-  void clear();
+  void clear() override;
 
   /**
    * @brief Returns the IdType for the DataObject found at the specified DataPath. The
@@ -745,12 +748,6 @@
    */
   std::vector<DataObject*> getTopLevelData() const;
 
-  /**
-   * @brief Returns a reference to the DataMap backing the top level of the DataStructure.
-   * @return const DataMap&
-   */
-  const DataMap& getDataMap() const;
-
   /**
    * @brief Inserts a new DataObject into the DataStructure nested under the given
    * DataPath. If the DataPath is empty, the DataObject is added directly to
@@ -868,6 +865,24 @@
    */
   DataStructure& operator=(DataStructure&& rhs) noexcept;
 
+  /**
+   * @brief Returns a deep copy of the DataStructure.
+   * @return DataObject*
+   */
+  std::shared_ptr<DataObject> deepCopy(const DataPath& copyPath) override;
+
+  /**
+   * @brief Returns a shallow copy of the DataObject.
+   * @return DataObject*
+   */
+  DataObject* shallowCopy() override;
+
+  /**
+   * @brief Returns typename of the DataObject as a std::string.
+   * @return std::string
+   */
+  std::string getTypeName() const override;
+
   /**
    * @brief Sets the next ID to use when constructing a DataObject.
    * Because IDs are created to be unique, this should only be called when
@@ -887,9 +902,9 @@
    * @brief Flushes all DataObjects to their respective target by calling their respective flush() method.
    * In-memory DataObjects are not affected.
    */
   void flush() const;
 
   uint64 memoryUsage() const;
 
   /**
    * @brief Transfers array data to OOC if available.
@@ -1025,7 +1040,6 @@
   // Variables
   SignalType m_Signal;
   WeakCollectionType m_DataObjects;
-  DataMap m_RootGroup;
   bool m_IsValid = false;
   DataObject::IdType m_NextId = 1;
 };