Repo created

tools/topography_generator/isolines_profile.hpp

@@ -0,0 +1,135 @@
+#pragma once
+
+#include "topography_generator/isolines_utils.hpp"
+
+#include "storage/storage_defines.hpp"
+
+#include "coding/file_reader.hpp"
+#include "coding/file_writer.hpp"
+#include "coding/serdes_json.hpp"
+
+#include "base/visitor.hpp"
+
+#include <map>
+#include <string>
+#include <unordered_set>
+
+namespace topography_generator
+{
+struct IsolinesProfile
+{
+  uint32_t m_alitudesStep = 10;
+  uint8_t m_latLonStepFactor = 1;
+  uint32_t m_maxIsolinesLength = 1000;
+  uint8_t m_simplificationZoom = 17;     // Value == 0 disables simplification.
+  uint8_t m_medianFilterR = 1;           // Value == 0 disables filter.
+  double m_gaussianFilterStDev = 2.0;    // Value == 0.0 disables filter.
+  double m_gaussianFilterRFactor = 1.0;  // Value == 0.0 disables filter.
+
+  DECLARE_VISITOR_AND_DEBUG_PRINT(IsolinesProfile, visitor(m_alitudesStep, "alitudesStep"),
+                                  visitor(m_latLonStepFactor, "latLonStepFactor"),
+                                  visitor(m_maxIsolinesLength, "maxIsolinesLength"),
+                                  visitor(m_simplificationZoom, "simplificationZoom"),
+                                  visitor(m_medianFilterR, "medianFilterR"),
+                                  visitor(m_gaussianFilterStDev, "gaussianFilterStDev"),
+                                  visitor(m_gaussianFilterRFactor, "gaussianFilterRFactor"))
+};
+
+struct IsolinesProfilesCollection
+{
+  std::map<std::string, IsolinesProfile> m_profiles;
+
+  DECLARE_VISITOR_AND_DEBUG_PRINT(IsolinesProfilesCollection, visitor(m_profiles, "profiles"))
+};
+
+struct TileCoord
+{
+  TileCoord() = default;
+  TileCoord(int bottomLat, int leftLon) : m_leftLon(leftLon), m_bottomLat(bottomLat) {}
+
+  int32_t m_leftLon = 0;
+  int32_t m_bottomLat = 0;
+
+  bool operator==(TileCoord const & rhs) const { return m_leftLon == rhs.m_leftLon && m_bottomLat == rhs.m_bottomLat; }
+
+  DECLARE_VISITOR_AND_DEBUG_PRINT(TileCoord, visitor(m_bottomLat, "bottomLat"), visitor(m_leftLon, "leftLon"))
+};
+
+struct TileCoordHash
+{
+  size_t operator()(TileCoord const & coord) const
+  {
+    return (static_cast<size_t>(coord.m_leftLon) << 32u) | static_cast<size_t>(coord.m_bottomLat);
+  }
+};
+
+struct CountryIsolinesParams
+{
+  std::string m_profileName;
+  std::unordered_set<TileCoord, TileCoordHash> m_tileCoordsSubset;
+  bool m_tilesAreBanned;
+
+  bool NeedSkipTile(int lat, int lon) const
+  {
+    if (m_tileCoordsSubset.empty())
+      return false;
+    TileCoord coord(lat, lon);
+    auto const found = m_tileCoordsSubset.find(coord) != m_tileCoordsSubset.end();
+    return m_tilesAreBanned == found;
+  }
+
+  DECLARE_VISITOR_AND_DEBUG_PRINT(CountryIsolinesParams, visitor(m_profileName, "profileName"),
+                                  visitor(m_tileCoordsSubset, "tileCoordsSubset"),
+                                  visitor(m_tilesAreBanned, "tilesAreBanned"))
+};
+
+struct CountriesToGenerate
+{
+  std::map<storage::CountryId, CountryIsolinesParams> m_countryParams;
+
+  DECLARE_VISITOR_AND_DEBUG_PRINT(CountriesToGenerate, visitor(m_countryParams, "countryParams"))
+};
+
+template <typename DataType>
+bool Serialize(std::string const & fileName, DataType const & data)
+{
+  try
+  {
+    FileWriter writer(fileName);
+    coding::SerializerJson<FileWriter> ser(writer);
+    ser(data);
+    return true;
+  }
+  catch (base::Json::Exception & ex)
+  {
+    LOG(LERROR, ("Serialization to json failed, file name", fileName, ", reason:", ex.Msg()));
+  }
+  catch (FileWriter::Exception const & ex)
+  {
+    LOG(LERROR, ("Can't write file", fileName, ", reason:", ex.Msg()));
+  }
+  return false;
+}
+
+template <typename DataType>
+bool Deserialize(std::string const & fileName, DataType & data)
+{
+  try
+  {
+    FileReader reader(fileName);
+    NonOwningReaderSource source(reader);
+    coding::DeserializerJson des(source);
+    des(data);
+    return true;
+  }
+  catch (base::Json::Exception & ex)
+  {
+    LOG(LERROR, ("Deserialization from json failed, file name", fileName, ", reason:", ex.Msg()));
+  }
+  catch (FileReader::Exception const & ex)
+  {
+    LOG(LERROR, ("Can't read file", fileName, ", reason:", ex.Msg()));
+  }
+  return false;
+}
+}  // namespace topography_generator