Function ComputeTreeWeight

src/colmap/math/spanning_tree_test.cc:38–51

Helper to compute total weight of edges in the spanning tree.

// Helper to compute total weight of edges in the spanning tree.
float ComputeTreeWeight(const SpanningTree& tree,
                        const std::vector<std::pair<int, int>>& edges,
                        const std::vector<float>& weights) {
  float total = 0;
  for (size_t i = 0; i < edges.size(); ++i) {
    int u = edges[i].first;
    int v = edges[i].second;
    // Check if this edge is in the tree (either direction).
    if (tree.parents[u] == v || tree.parents[v] == u) {
      total += weights[i];
    }
  }
  return total;
}

TEST(SpanningTree, Nominal) {
  // Triangle: edges with weights 1, 2, 3.