initial commit of BRouter Version 0.98

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

@@ -0,0 +1,205 @@
+/**
+ * This program
+ * - reads an *.osm from stdin
+ * - writes 45*30 degree node tiles + a way file + a rel file
+ *
+ * @author ab
+ */
+package btools.mapcreator;
+
+import java.io.BufferedOutputStream;
+import java.io.BufferedReader;
+import java.io.DataOutputStream;
+import java.io.File;
+import java.io.FileOutputStream;
+import java.io.FileInputStream;
+import java.io.InputStreamReader;
+import java.util.HashMap;
+import java.util.zip.GZIPInputStream;
+
+import btools.expressions.BExpressionContext;
+
+public class OsmCutter extends MapCreatorBase
+{
+  private long recordCnt;
+  private long nodesParsed;
+  private long waysParsed;
+  private long relsParsed;
+  private long changesetsParsed;
+
+  private DataOutputStream wayDos;
+  private DataOutputStream cyclewayDos;
+
+  public static void main(String[] args) throws Exception
+  {
+    System.out.println("*** OsmCutter: cut an osm map in node-tiles + a way file");
+    if (args.length != 4 && args.length != 5)
+    {
+      System.out.println("usage: bzip2 -dc <map> | java OsmCutter <lookup-file> <out-tile-dir> <out-way-file> <out-rel-file>");
+      System.out.println("or   : java OsmCutter <lookup-file> <out-tile-dir> <out-way-file> <out-rel-file> <inputfile>");
+      return;
+    }
+
+    new OsmCutter().process(
+                   new File( args[0] )
+                 , new File( args[1] )
+                 , new File( args[2] )
+                 , new File( args[3] )
+                 , args.length > 4 ? new File( args[4] ) : null );
+  }
+
+  private BExpressionContext _expctxWay;
+  private BExpressionContext _expctxNode;
+
+  public void process (File lookupFile, File outTileDir, File wayFile, File relFile, File mapFile) throws Exception
+  {
+    if ( !lookupFile.exists() )
+    {
+      throw new IllegalArgumentException( "lookup-file: " +  lookupFile + " does not exist" );
+    }
+
+    _expctxWay = new BExpressionContext("way");
+    _expctxWay.readMetaData( lookupFile );
+
+    _expctxNode = new BExpressionContext("node");
+    _expctxNode.readMetaData( lookupFile );
+
+    this.outTileDir = outTileDir;
+    if ( !outTileDir.isDirectory() ) throw new RuntimeException( "out tile directory " + outTileDir + " does not exist" );
+
+    wayDos = new DataOutputStream( new BufferedOutputStream( new FileOutputStream( wayFile ) ) );
+    cyclewayDos = new DataOutputStream( new BufferedOutputStream( new FileOutputStream( relFile ) ) );
+
+    // read the osm map into memory
+    long t0 = System.currentTimeMillis();
+    new OsmParser().readMap( mapFile, this, this, this );
+    long t1 = System.currentTimeMillis();
+    
+    System.out.println( "parsing time (ms) =" + (t1-t0) );
+
+    // close all files
+    closeTileOutStreams();
+    wayDos.close();
+    cyclewayDos.close();
+
+/*    System.out.println( "-------- way-statistics -------- " );
+    _expctxWay.dumpStatistics();
+    System.out.println( "-------- node-statistics -------- " );
+    _expctxNode.dumpStatistics();
+*/
+    System.out.println( statsLine() );
+  }
+
+  private void checkStats()
+  {
+    if ( (++recordCnt % 100000) == 0 ) System.out.println( statsLine() );
+  }
+
+  private String statsLine()
+  {
+    return "records read: " + recordCnt + " nodes=" + nodesParsed + " ways=" + waysParsed + " rels=" + relsParsed + " changesets=" + changesetsParsed;
+  }
+
+
+  @Override
+  public void nextNode( NodeData n ) throws Exception
+  {
+    nodesParsed++;
+    checkStats();
+
+    if ( n.getTagsOrNull() != null )
+    {
+      int[] lookupData = _expctxNode.createNewLookupData();
+      for( String key : n.getTagsOrNull().keySet() )
+      {
+        String value = n.getTag( key );
+        _expctxNode.addLookupValue( key, value, lookupData );
+      }
+      n.description = _expctxNode.encode(lookupData);
+    }
+    // write node to file
+    int tileIndex = getTileIndex( n.ilon, n.ilat );
+    if ( tileIndex >= 0 )
+    {
+      DataOutputStream dos = getOutStreamForTile( tileIndex );
+      n.writeTo( dos );
+    }
+  }
+
+
+  @Override
+  public void nextWay( WayData w ) throws Exception
+  {
+    waysParsed++;
+    checkStats();
+
+    // filter out non-highway ways
+    if ( w.getTag( "highway" ) == null )
+    {
+      // ... but eventually fake a ferry tag
+      if ( "ferry".equals( w.getTag( "route" ) ) )
+      {
+        w.putTag( "highway", "ferry" );
+      }
+      else
+      {
+        return;
+      }
+    }
+
+    // encode tags
+    if ( w.getTagsOrNull() != null )
+    {
+      int[] lookupData = _expctxWay.createNewLookupData();
+      for( String key : w.getTagsOrNull().keySet() )
+      {
+        String value = w.getTag( key );
+        _expctxWay.addLookupValue( key, value, lookupData );
+      }
+      w.description = _expctxWay.encode(lookupData);
+    }
+
+    w.writeTo( wayDos );
+  }
+
+  @Override
+  public void nextRelation( RelationData r ) throws Exception
+  {
+    relsParsed++;
+    checkStats();
+
+    // filter out non-cycle relations
+    if ( ! "bicycle".equals( r.getTag( "route" ) ) )
+    {
+      return;
+    }
+
+    for ( int i=0; i<r.ways.size();i++ )
+    {
+      long wid = r.ways.get(i);
+      writeId( cyclewayDos, wid );
+    }
+  }
+
+
+  private int getTileIndex( int ilon, int ilat )
+  {
+     int lon = ilon / 45000000;
+     int lat = ilat / 30000000;
+     if ( lon < 0 || lon > 7 || lat < 0 || lat > 5 )
+     {
+       System.out.println( "warning: ignoring illegal pos: " + ilon + "," + ilat );
+       return -1;
+     }
+     return lon*6 + lat;
+  }
+
+  protected String getNameForTile( int tileIndex )
+  {
+    int lon = (tileIndex / 6 ) * 45 - 180;
+    int lat = (tileIndex % 6 ) * 30 - 90;
+    String slon = lon < 0 ? "W" + (-lon) : "E" + lon;
+    String slat = lat < 0 ? "S" + (-lat) : "N" + lat;
+    return slon + "_" + slat + ".tls";
+  }
+}