package btools.mapcreator;

import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.EOFException;
import java.io.File;
import java.util.HashMap;

import btools.expressions.BExpressionContext;
import btools.util.CompactLongSet;
import btools.util.FrozenLongSet;

/**
 * RelationMerger does 1 step in map processing:
 *
 * - enrich ways with relation information
 *
 * @author ab
 */
public class RelationMerger extends MapCreatorBase
{
  private HashMap<String,CompactLongSet> routesets;
  private CompactLongSet routesetall;
  private BExpressionContext expctxReport;
  private BExpressionContext expctxCheck;

  private DataOutputStream wayOutStream;
  
  public static void main(String[] args) throws Exception
  {
    System.out.println("*** RelationMerger: merge relation bits into ways" );
    if (args.length != 6)
    {
      System.out.println("usage: java RelationMerger <way-file-in> <way-file-out> <relation-file> <lookup-file> <report-profile> <check-profile>" );

      return;
    }
    new RelationMerger().process( new File( args[0] ), new File( args[1] ), new File( args[2] ), new File( args[3] ), new File( args[4] ), new File( args[5] ) );
  }

  public void process( File wayFileIn, File wayFileOut, File relationFileIn, File lookupFile, File reportProfile, File checkProfile ) throws Exception
  {
    // read lookup + profile for relation access-check
    expctxReport = new BExpressionContext("way");
    expctxReport.readMetaData( lookupFile );
    expctxReport.parseFile( reportProfile, "global" );
    expctxCheck = new BExpressionContext("way");
    expctxCheck.readMetaData( lookupFile );
    expctxCheck.parseFile( checkProfile, "global" );
    
    // *** read the relation file into sets for each processed tag
    routesets = new HashMap<String,CompactLongSet>();
    routesetall = new CompactLongSet();
    DataInputStream dis = createInStream( relationFileIn );
    try
    {
      for(;;)
      {
        long rid = readId( dis );
        String route = dis.readUTF();
        String network = dis.readUTF();
        
        String tagname = "route_" + route + "_" + network;
        CompactLongSet routeset = null;
        if ( expctxCheck.getLookupNameIdx(tagname)  >= 0 )
        {
        	routeset = routesets.get( tagname );
        	if ( routeset == null )
        	{
        		routeset = new CompactLongSet();
        		routesets.put( tagname, routeset );
        	}
        }
        	
        for(;;)
        {
          long wid = readId( dis );
          if ( wid == -1 ) break;
          if ( routeset != null && !routeset.contains( wid ) )
          {
        	  routeset.add( wid );
        	  routesetall.add( wid );
          }
        }
      }
    }
    catch( EOFException eof )
    {
      dis.close();
    }
    for( String tagname : routesets.keySet() )
    {
    	CompactLongSet routeset = new FrozenLongSet( routesets.get( tagname ) );
    	routesets.put( tagname, routeset );
        System.out.println( "marked " + routeset.size() + " ways for tag: " + tagname );
    }

    // *** finally process the way-file
    wayOutStream = createOutStream( wayFileOut );
    new WayIterator( this, true ).processFile( wayFileIn );
    wayOutStream.close();
  }

  @Override
  public void nextWay( WayData data ) throws Exception
  {
    // propagate the route-bits
    if ( routesetall.contains( data.wid ) )
    {
      boolean ok = true;
      // check access and log a warning for conflicts
      expctxReport.evaluate( false, data.description, null );
      boolean warn = expctxReport.getCostfactor() >= 10000.;
      if ( warn )
      {
          expctxCheck.evaluate( false, data.description, null );
          ok = expctxCheck.getCostfactor() < 10000.;

          System.out.println( "** relation access conflict for wid = " + data.wid + " tags:" + expctxReport.getKeyValueDescription( data.description ) + " (ok=" + ok + ")"  );
      }
    	
      if ( ok )
      {
        for( String tagname : routesets.keySet() )
        {
    	  CompactLongSet routeset = routesets.get( tagname );
    	  if ( routeset.contains( data.wid ) ) expctxReport.addLookupValue( tagname, 2 );
    	}
    	data.description = expctxReport.encode();
      }
    }

    data.writeTo( wayOutStream );
  }

}