[transit][generator] Fix cross mwm features numeration for experiment…

…al transit.
mapsme · Apr 22, 2021 · 4840203 · 4840203
1 parent 3fa87ad
commit 4840203
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/generator/transit_generator_experimental.cpp b/generator/transit_generator_experimental.cpp
@@ -196,11 +196,19 @@ EdgeIdToFeatureId BuildTransit(std::string const & mwmDir, CountryId const & cou
   data.CheckValid();
   data.CheckUnique();
 
+  // Transit graph numerates features according to their orderto their order..
+  EdgeIdToFeatureId edgeToFeature;
+  for (size_t i = 0; i < data.GetEdges().size(); ++i)
+  {
+    auto const & e = data.GetEdges()[i];
+    EdgeId id(e.GetStop1Id(), e.GetStop2Id(), e.GetLineId());
+    edgeToFeature[id] = i;
+  }
+
   FilesContainerW container(mwmPath, FileWriter::OP_WRITE_EXISTING);
   auto writer = container.GetWriter(TRANSIT_FILE_TAG);
   data.Serialize(*writer);
-  CHECK_EQUAL(data.GetEdgeIdToFeatureId().size(), data.GetEdges().size(), ());
-  return data.GetEdgeIdToFeatureId();
+  return edgeToFeature;
 }
 }  // namespace experimental
 }  // namespace transit