Model/Node側のメソッドでchild node 有無判定( Mapを使わない )

Author: sevendev

include/plateau/polygon_mesh/model.h

@@ -40,6 +40,7 @@ namespace plateau::polygonMesh {
 
         Node& getRootNodeAt(size_t index);
         const Node& getRootNodeAt(size_t index) const;
+        const int getRootNodeIndexByName(const std::string name) const;
 
         /// 各ノードのペアレント、ルートを設定します。
         void assignNodeHierarchy();

include/plateau/polygon_mesh/node.h

@@ -54,6 +54,8 @@ namespace plateau::polygonMesh {
         Node& getChildAt(unsigned int index);
         const Node& getChildAt(unsigned int index) const;
 
+        const int getChildIndexByName(const std::string name) const;
+
         /// Parent Node設定
         void setParentNode(Node* node);  
         const Node& getParentNode() const; //Model.assignNodeHierarchyを呼ぶことで取得可能になります。

src/polygon_mesh/model.cpp

@@ -34,6 +34,17 @@ namespace plateau::polygonMesh {
         return root_nodes_.at(index);
     }
 
+    const int Model::getRootNodeIndexByName(const std::string name) const {
+        int num_children = getRootNodeCount();
+        for (int i = 0; i < num_children; ++i) {
+            auto& child_node = getRootNodeAt(i);
+            if (child_node.getName() == name) {
+                return i;
+            }
+        }
+        return -1;
+    }
+
     void Model::eraseEmptyNodes() {
         auto new_end = std::remove_if(root_nodes_.begin(), root_nodes_.end(), [](Node& node) {
             node.eraseEmptyChildren();

src/polygon_mesh/node.cpp

@@ -121,6 +121,17 @@ namespace plateau::polygonMesh {
         return child_nodes_.at(index);
     }
 
+    const int Node::getChildIndexByName(const std::string name) const {
+        int num_children = getChildCount();
+        for (int i = 0; i < num_children; ++i) {
+            auto& child_node = getChildAt(i);
+            if (child_node.getName() == name) {
+                return i;
+            }
+        }
+        return -1;
+    }
+
     void Node::eraseEmptyChildren() {
         const auto new_end = std::remove_if(child_nodes_.begin(), child_nodes_.end(), [](Node& child) {
             child.eraseEmptyChildren();

src/polygon_mesh/tile_extractor.cpp

@@ -82,9 +82,6 @@ namespace {
         // 範囲の境界上にある地物を取り逃さないように、範囲を少し広げます。
         auto extents = MeshExtractor::extendExtents(extents_before_adjust, 1.2f);
 
-		std::map<unsigned, Node*> grid_nodes; // グリッドIDとグリッドノードのマップ
-
-        // rootNode として LODノード を作ります。
         for (unsigned lod = options.min_lod; lod <= options.max_lod; lod++) {
 
             // 3D都市モデルをグリッドに分け、グリッドごとにメッシュをマージします。
@@ -95,18 +92,33 @@ namespace {
                 // model -> GRIDノード -> LODノード -> ノード
                 for (auto& [grid_id, mesh] : result) {
 
-					if (grid_nodes.find(grid_id) == grid_nodes.end()) {
+					const auto& grid_name = "GRID_" + std::to_string(grid_id);
+                    const auto& lod_name = "LOD" + std::to_string(lod);
+
+                    const auto grid_index = out_model.getRootNodeIndexByName(grid_name);	
+                    auto node = Node("LOD" + std::to_string(lod), std::move(mesh));
+
+					if (grid_index == -1) {
 						// 存在しない場合 新しいグリッドノードとLODノードを作成
-                        auto grid_node = Node("GRID_" + std::to_string(grid_id));
-                        auto& new_grid_node = out_model.addNode(std::move(grid_node));
-						grid_nodes[grid_id] = &new_grid_node; // 新しいグリッドノードを保存
-					}
-
-                    auto outer_it = grid_nodes.find(grid_id);
-                    if (outer_it != grid_nodes.end()) {
-                        Node* grid_node = outer_it->second;
-                        auto node = Node("LOD" + std::to_string(lod), std::move(mesh));
-                        grid_node->addChildNode(std::move(node));
+						auto grid_node = Node(grid_name);
+						auto& new_grid_node = out_model.addNode(std::move(grid_node));			
+                        auto lod_node = Node(lod_name);
+                        lod_node.addChildNode(std::move(node));
+                        new_grid_node.addChildNode(std::move(lod_node));
+                    }
+                    else {
+						auto& grid_node = out_model.getRootNodeAt(grid_index);  
+                        const auto lod_index = grid_node.getChildIndexByName(lod_name);
+						if (lod_index == -1) {
+							// LODノードが存在しない場合、新しいLODノードを作成
+							auto lod_node = Node(lod_name);
+							lod_node.addChildNode(std::move(node));
+							grid_node.addChildNode(std::move(lod_node));
+						}
+						else {
+							// 既存のLODノードにノードを追加
+							grid_node.getChildAt(lod_index).addChildNode(std::move(node));
+						}
                     }
                 }
             }
@@ -123,7 +135,6 @@ namespace {
             } 
 
         }
-        grid_nodes.clear(); // グリッドノードのマップをクリア
 
         out_model.eraseEmptyNodes();
         out_model.assignNodeHierarchy();