THE RETURN OF REMOVE

# this commit sounds like film name, doesn't it?) # this method unused in PlotArea but needed in DataManager
UmbrellaLeaf5 · Apr 14, 2024 · 5338bc5 · 5338bc5
1 parent 6a6e775
commit 5338bc5
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 0 deletions.
diff --git a/data_tools/data_manager/data_manager.cpp b/data_tools/data_manager/data_manager.cpp
@@ -4,6 +4,30 @@
 
 namespace data_tools {
 
+void DataManager::Remove(gui::ObjectType obj_type, size_t index) {
+  switch (obj_type) {
+    case gui::ObjectType::Targets: {
+      targets_.erase(targets_.begin() + index);
+      break;
+    }
+
+    case gui::ObjectType::Hills: {
+      hills_.erase(hills_.begin() + index);
+      break;
+    }
+
+    case gui::ObjectType::TrappyCircles: {
+      tr_circles_.erase(tr_circles_.begin() + index);
+      break;
+    }
+
+    case gui::ObjectType::TrappyLines: {
+      tr_lines_.erase(tr_lines_.begin() + index);
+      break;
+    }
+  }
+}
+
 void DataManager::Clear() {
   targets_.clear();
   tr_circles_.clear();

diff --git a/data_tools/data_manager/data_manager.h b/data_tools/data_manager/data_manager.h
@@ -12,6 +12,13 @@ class DataManager {
  public:
   DataManager() = default;
 
+  /**
+   * @brief удаляет объект из менеджера по индексу
+   * @param obj_type: тип объекта
+   * @param index: индекс объекта в его векторе
+   */
+  void Remove(gui::ObjectType obj_type, size_t index);
+
   /// @brief очищает все вектора объектов
   void Clear();