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

package btools.mapcreator;

import java.io.BufferedOutputStream;
import java.io.DataOutputStream;
import java.io.File;
import java.io.FileOutputStream;

import btools.util.Raster2Png;

public class TrafficData2Png
{
  private static int minLon;
  private static int minLat;
  private static int maxLon;
  private static int maxLat;
  private static int ncols;
  private static int nrows;
  private static int[] pixels;

  public static void main( String[] args ) throws Exception
  {
    if ( args.length == 8 )
    {
      doConvert( args[0], args[1], Double.parseDouble( args[2] ), Double.parseDouble( args[3] ), Double.parseDouble( args[4] ),
          Double.parseDouble( args[5] ), Integer.parseInt( args[6] ), Integer.parseInt( args[7] ) );
    }
    else if ( args.length == 4 )
    {
      int lon0 = Integer.parseInt( args[0] );
      int lat0 = Integer.parseInt( args[1] );
      String inputFile = "traffic/E" + lon0 + "_N" + lat0 + ".trf";
      for ( int lon = lon0; lon < lon0 + 5; lon++ )
        for ( int lat = lat0; lat < lat0 + 5; lat++ )
        {
          String imageFile = "traffic_pics/E" + lon + "_N" + lat + ".png";
          System.out.println( "file=" + inputFile + " image=" + imageFile );
          doConvert( inputFile, imageFile, lon, lat, lon + 1, lat + 1, Integer.parseInt( args[2] ), Integer.parseInt( args[3] ) );
        }
    }

  }

  public static void doConvert( String inputFile, String imageFile, double lon0, double lat0, double lon1, double lat1, int cols, int rows )
      throws Exception
  {
    OsmTrafficMap trafficMap = new OsmTrafficMap();
    minLon = (int) ( lon0 * 1000000 + 180000000 );
    maxLon = (int) ( lon1 * 1000000 + 180000000 );
    minLat = (int) ( lat0 * 1000000 + 90000000 );
    maxLat = (int) ( lat1 * 1000000 + 90000000 );
    ncols = cols;
    nrows = rows;

    long[] keys = trafficMap.load( new File( inputFile ), minLon, minLat, maxLon, maxLat, true );

    pixels = new int[cols * rows];

    int[] tclasses = new int[]
    { 1, 2, 3, 4, 5, 6, 7, -1 };
    for ( int tclass : tclasses )
    {
      for ( long key : keys )
      {
        OsmTrafficMap.OsmTrafficElement e = trafficMap.getElement( key );
        while (e != null)
        {
          long key2 = e.node2;
          e = e.next;
          int trafficClass = trafficMap.getTrafficClass( key, key2 );
          if ( trafficClass != tclass )
            continue;

          int[] from = getImagePosition( key );
          int[] to = getImagePosition( key2 );

          int rgb = 0;
          if ( trafficClass == -1 )     rgb = 0x0000ff; // blue
          else if ( trafficClass == 1 ) rgb = 0x404040; // dark grey
          else if ( trafficClass == 2 ) rgb = 0xa0a0a0; // light grey
          else if ( trafficClass == 3 ) rgb = 0x00ff00; // green
          else if ( trafficClass == 4 ) rgb = 0xf4e500; // yellow
          else if ( trafficClass == 5 ) rgb = 0xf18e1c; // orange
          else if ( trafficClass == 6 ) rgb = 0xe32322; // red
          else if ( trafficClass == 7 ) rgb = 0xc0327d; // pink
          if ( rgb != 0 )
          {
            drawLine( from, to, rgb );
          }
        }
      }
    }

    byte[] png = new Raster2Png().pngEncode( cols, rows, pixels );

    System.out.println( "got png of size: " + png.length );
    DataOutputStream dos = new DataOutputStream( new BufferedOutputStream( new FileOutputStream( imageFile ) ) );
    dos.write( png );
    dos.close();
  }

  private static void drawLine( int[] from, int[] to, int rgb )
  {
    int ix = from[0];
    int iy = from[1];
    int ixx = to[0];
    int iyy = to[1];

    int sx = ixx > ix ? 1 : -1;
    int sy = iyy > iy ? 1 : -1;

    int dx = ( ixx - ix ) * sx;
    int dy = ( iyy - iy ) * sy;

    int sum = 0;

    for ( ;; )
    {
      drawPixel( ix, iy, rgb );
      if ( ix == ixx && iy == iyy )
        break;

      if ( Math.abs( sum + dx ) < Math.abs( sum - dy ) )
      {
        iy += sy;
        sum += dx;
      }
      else
      {
        ix += sx;
        sum -= dy;
      }
    }
  }

  private static void drawPixel( int ix, int iy, int rgb )
  {
    if ( ix >= 0 && ix < ncols && iy >= 0 && iy < nrows )
    {
      pixels[( nrows - 1 - iy ) * ncols + ix] = rgb;
    }
  }

  private static int[] getImagePosition( long key )
  {
    int ilon = (int) ( key >> 32 );
    int ilat = (int) ( key & 0xffffffff );
    double lonDelta = maxLon - minLon;
    double latDelta = maxLat - minLat;
    int[] res = new int[2];
    res[0] = (int) ( ( ( ilon - minLon ) / lonDelta ) * ncols );
    res[1] = (int) ( ( ( ilat - minLat ) / latDelta ) * nrows );
    return res;
  }
}