Merge pull request abrensch#828 from afischerdev/pseudo-dbtags

Enable database access for pseudo tags

Author: afischerdev

brouter-map-creator/src/main/java/btools/mapcreator/DatabasePseudoTagProvider.java

@@ -90,7 +90,69 @@ private static void appendDBTag(StringBuilder sb, ResultSet rs, String name) thr
     }
   }
 
-  public DatabasePseudoTagProvider(String filename) {
+  public DatabasePseudoTagProvider(String filename, String jdbcurl) {
+    if (filename != null) doFileImport(filename);
+    if (jdbcurl != null) doDatabaseImport(jdbcurl);
+  }
+
+  private void doDatabaseImport(String jdbcurl) {
+
+    try (Connection conn = DriverManager.getConnection(jdbcurl)) {
+
+      System.out.println("DatabasePseudoTagProvider reading from database: " + jdbcurl);
+      conn.setAutoCommit(false);
+
+
+      Map<Map<String, String>, Map<String, String>> mapUnifier = new HashMap<>();
+      CompactLongMap<Map<String, String>> data = new CompactLongMap<>();
+
+      String sql_all_tags = "SELECT * from all_tags";
+      try(PreparedStatement psAllTags = conn.prepareStatement(sql_all_tags)) {
+
+        psAllTags.setFetchSize(100);
+
+        // process the results
+        ResultSet rs = psAllTags.executeQuery();
+
+        long dbRows = 0L;
+        while (rs.next()) {
+          long osm_id = rs.getLong("losmid");
+          Map<String, String> row = new HashMap<>(5);
+          addDBTag(row, rs, "noise_class");
+          addDBTag(row, rs, "river_class");
+          addDBTag(row, rs, "forest_class");
+          addDBTag(row, rs, "town_class");
+          addDBTag(row, rs, "traffic_class");
+
+          // apply the instance-unifier for the row-map
+          Map<String, String> knownRow = mapUnifier.get(row);
+          if (knownRow != null) {
+            row = knownRow;
+          } else {
+            mapUnifier.put(row, row);
+          }
+          data.put(osm_id, row);
+          dbRows++;
+          if (dbRows % 1000000L == 0L) {
+            System.out.println(".. from database: rows =" + dbRows);
+          }
+        }
+        System.out.println("freezing result map..");
+        dbData = new FrozenLongMap<>(data);
+        System.out.println("read from database: rows =" + dbData.size() + " unique rows=" + mapUnifier.size());
+      }
+
+    } catch (SQLException g) {
+      System.err.format("DatabasePseudoTagProvider execute sql .. SQL State: %s\n%s\n", g.getSQLState(), g.getMessage());
+      System.exit(1);
+    } catch (Exception f) {
+      f.printStackTrace();
+      System.exit(1);
+    }
+
+  }
+
+  private void doFileImport(String filename) {
 
     try (BufferedReader br = new BufferedReader(new InputStreamReader(
            filename.endsWith(".gz") ? new GZIPInputStream(new FileInputStream(filename)) : new FileInputStream(filename)))) {
@@ -158,12 +220,22 @@ private List<String> tokenize(String s) {
     return l;
   }
 
-  private static void addTag(Map<String, String> row, String s, String name) {    
+  private static void addTag(Map<String, String> row, String s, String name) {
     if (!s.isEmpty()) {
       row.put(name, s);
     }
   }
 
+  private static void addDBTag(Map<String, String> row, ResultSet rs, String name) {
+    String v = null;
+    try {
+      v = rs.getString(name);
+    } catch (Exception e) {}
+    if (v != null) {
+      row.put("estimated_" + name, v);
+    }
+  }
+
   public void addTags(long osm_id, Map<String, String> map) {
 
     if (map == null || !map.containsKey("highway")) {
@@ -194,6 +266,6 @@ public void addTags(long osm_id, Map<String, String> map) {
       pseudoTagsFound.put(key, cnt + 1L);
     }
   }
-  
-  
+
+
 }

brouter-map-creator/src/main/java/btools/mapcreator/OsmCutter.java

@@ -113,7 +113,11 @@ private String statsLine() {
   }
 
   public void setDbTagFilename(String filename) {
-    dbPseudoTagProvider = new DatabasePseudoTagProvider(filename);
+    dbPseudoTagProvider = new DatabasePseudoTagProvider(filename, null);
+  }
+
+  public void setDbTagDatabase(String jdbcurl) {
+    dbPseudoTagProvider = new DatabasePseudoTagProvider(null, jdbcurl);
   }
 
   @Override

brouter-map-creator/src/main/java/btools/mapcreator/OsmFastCutter.java

@@ -13,7 +13,7 @@ public class OsmFastCutter extends MapCreatorBase {
   public static void main(String[] args) throws Exception {
     System.out.println("*** OsmFastCutter: cut an osm map in node-tiles + way-tiles");
     if (args.length != 11 && args.length != 12 && args.length != 13) {
-      String common = "java OsmFastCutter <lookup-file> <node-dir> <way-dir> <node55-dir> <way55-dir> <border-file> <out-rel-file> <out-res-file> <filter-profile> <report-profile> <check-profile> <map-file> [db-tag-filename]";
+      String common = "java OsmFastCutter <lookup-file> <node-dir> <way-dir> <node55-dir> <way55-dir> <border-file> <out-rel-file> <out-res-file> <filter-profile> <report-profile> <check-profile> <map-file> [db-tag-filename | db-tag-jdbcurl]";
 
       System.out.println("usage: bzip2 -dc <map> | " + common);
       System.out.println("or   : " + common + " <inputfile> ");
@@ -37,10 +37,16 @@ public static void main(String[] args) throws Exception {
     );
   }
 
-  public static void doCut(File lookupFile, File nodeDir, File wayDir, File node55Dir, File way55Dir, File borderFile, File relFile, File resFile, File profileAll, File profileReport, File profileCheck, File mapFile, String dbTagFilename) throws Exception {
+  public static void doCut(File lookupFile, File nodeDir, File wayDir, File node55Dir, File way55Dir, File borderFile, File relFile, File resFile, File profileAll, File profileReport, File profileCheck, File mapFile, String dbTagInfo) throws Exception {
     // **** run OsmCutter ****
     OsmCutter cutter = new OsmCutter();
-    if (dbTagFilename != null) cutter.setDbTagFilename(dbTagFilename);
+    if (dbTagInfo != null) {
+      if (dbTagInfo.toLowerCase().startsWith("jdbc")) {
+        cutter.setDbTagDatabase(dbTagInfo);
+      } else {
+        cutter.setDbTagFilename(dbTagInfo);
+      }
+    }
 
     // ... inject WayCutter
     cutter.wayCutter = new WayCutter();