brouter/misc/pbfparser/OsmParser2.java

111 lines
2.5 KiB
Java

package btools.mapsplitter;
import java.io.*;
import java.util.*;
import java.util.zip.*;
import btools.util.*;
import org.openstreetmap.osmosis.osmbinary.Fileformat;
/**
* Parser for OSM data
*
* @author ab
*/
public class OsmParser2 extends MapCreatorBase
{
private BufferedReader _br;
private NodeListener nListener;
private WayListener wListener;
private RelationListener rListener;
public void readMap( File mapFile,
NodeListener nListener,
WayListener wListener,
RelationListener rListener ) throws Exception
{
this.nListener = nListener;
this.wListener = wListener;
this.rListener = rListener;
System.out.println( "*** PBF Parsing (2): " + mapFile );
// once more for testing
int rawBlobCount = 0;
DataInputStream dis = new DataInputStream( new BufferedInputStream ( new FileInputStream( mapFile ) ) );
for(;;)
{
int headerLength;
try
{
headerLength = dis.readInt();
}
catch (EOFException e)
{
break;
}
byte[] headerBuffer = new byte[headerLength];
dis.readFully(headerBuffer);
Fileformat.BlobHeader blobHeader = Fileformat.BlobHeader.parseFrom(headerBuffer);
byte[] blobData = new byte[blobHeader.getDatasize()];
dis.readFully(blobData);
new BPbfBlobDecoder2( blobHeader.getType(), blobData, this ).process();
rawBlobCount++;
}
dis.close();
System.out.println( "read raw blobs: " + rawBlobCount );
}
public void addNode( long nid, Map<String, String> tags, double lat, double lon )
{
NodeData n = new NodeData( nid, lon, lat );
n.setTags( (HashMap<String,String>)tags );
try
{
nListener.nextNode( n );
}
catch( Exception e )
{
throw new RuntimeException( "error writing node: " + e );
}
}
public void addWay( long wid, Map<String, String> tags, LongList nodes )
{
WayData w = new WayData( wid, nodes );
w.setTags( (HashMap<String,String>)tags );
try
{
wListener.nextWay( w );
}
catch( Exception e )
{
throw new RuntimeException( "error writing way: " + e );
}
}
public void addRelation( long rid, Map<String, String> tags, LongList wayIds, List<String> roles )
{
RelationData r = new RelationData( rid, wayIds, roles );
r.setTags( (HashMap<String,String>)tags );
try
{
rListener.nextRelation( r );
}
catch( Exception e )
{
throw new RuntimeException( "error writing relation: " + e );
}
}
}