Method Save

generator/road_penalty_generator.cpp:167–228

}

void RoadPenaltyCollector::Save()
{
  auto const fileName = GetFilename();
  LOG(LINFO, ("Saving road penalty values to", fileName));
  FileWriter writer(fileName);

  // All vehicles use the same penalty types

  // Group node penalties by way
  struct NodePenaltyEntry
  {
    uint32_t nodeIdx = 0;
    m2::PointU coord;
    RoadPenalty::Type type = RoadPenalty::Type::None;
  };

  std::map<uint64_t, std::vector<NodePenaltyEntry>> penaltiesByWay;

  m_roads.ForEachWayWithIndex([&](uint64_t wayID, std::vector<uint64_t> const & nodes, size_t idx)
  {
    std::vector<NodePenaltyEntry> wayNodePenalties;

    for (uint32_t nodeIdx = 0; nodeIdx < nodes.size(); ++nodeIdx)
    {
      uint64_t const nodeID = nodes[nodeIdx];
      auto const * barrierEntry = m_nodesWithType.Find(nodeID);
      if (barrierEntry == nullptr)
        continue;

      NodePenaltyEntry entry;
      entry.nodeIdx = nodeIdx;
      entry.coord = barrierEntry->m_coord;
      entry.type = barrierEntry->m_t;
      wayNodePenalties.push_back(entry);
    }

    if (!wayNodePenalties.empty())
      penaltiesByWay[wayID] = std::move(wayNodePenalties);
  }, m_cache);

  // Write node penalties grouped by way
  uint64_t const wayCount = penaltiesByWay.size();
  WriteToSink(writer, wayCount);
  for (auto const & [wayID, entries] : penaltiesByWay)
  {
    WriteToSink(writer, wayID);
    WriteToSink(writer, static_cast<uint32_t>(entries.size()));

    for (auto const & entry : entries)
    {
      WriteToSink(writer, entry.nodeIdx);
      WriteToSink(writer, entry.coord.x);
      WriteToSink(writer, entry.coord.y);
      WriteToSink(writer, kAllVehiclesMask);

      // Store only penalty type (time derived from vehicle type when loading)
      routing::Save(writer, entry.type);
    }