145 lines
5 KiB
Java
145 lines
5 KiB
Java
package btools.mapcreator;
|
|
|
|
import java.io.DataInputStream;
|
|
import java.io.DataOutputStream;
|
|
import java.io.EOFException;
|
|
import java.io.File;
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
|
|
import btools.expressions.BExpressionContextWay;
|
|
import btools.expressions.BExpressionMetaData;
|
|
import btools.util.CompactLongSet;
|
|
import btools.util.FrozenLongSet;
|
|
|
|
/**
|
|
* RelationMerger does 1 step in map processing:
|
|
* <p>
|
|
* - enrich ways with relation information
|
|
*
|
|
* @author ab
|
|
*/
|
|
public class RelationMerger extends MapCreatorBase {
|
|
private Map<String, CompactLongSet> routesets;
|
|
private CompactLongSet routesetall;
|
|
private BExpressionContextWay expctxReport;
|
|
private BExpressionContextWay expctxCheck;
|
|
// private BExpressionContext expctxStat;
|
|
|
|
private DataOutputStream wayOutStream;
|
|
|
|
public static void main(String[] args) throws Exception {
|
|
System.out.println("*** RelationMerger: merge relations 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 init(File relationFileIn, File lookupFile, File reportProfile, File checkProfile) throws Exception {
|
|
// read lookup + profile for relation access-check
|
|
BExpressionMetaData metaReport = new BExpressionMetaData();
|
|
expctxReport = new BExpressionContextWay(metaReport);
|
|
metaReport.readMetaData(lookupFile);
|
|
|
|
BExpressionMetaData metaCheck = new BExpressionMetaData();
|
|
expctxCheck = new BExpressionContextWay(metaCheck);
|
|
metaCheck.readMetaData(lookupFile);
|
|
|
|
expctxReport.parseFile(reportProfile, "global");
|
|
expctxCheck.parseFile(checkProfile, "global");
|
|
// expctxStat = new BExpressionContext("way");
|
|
|
|
// *** read the relation file into sets for each processed tag
|
|
routesets = new HashMap<>();
|
|
routesetall = new CompactLongSet();
|
|
DataInputStream dis = createInStream(relationFileIn);
|
|
try {
|
|
for (; ; ) {
|
|
long rid = readId(dis);
|
|
String route = dis.readUTF();
|
|
String network = dis.readUTF();
|
|
String state = dis.readUTF();
|
|
int value = "proposed".equals(state) ? 3 : 2; // 2=yes, 3=proposed
|
|
|
|
String tagname = "route_" + route + "_" + network;
|
|
|
|
CompactLongSet routeset = null;
|
|
if (expctxCheck.getLookupNameIdx(tagname) >= 0) {
|
|
String key = tagname + "_" + value;
|
|
routeset = routesets.get(key);
|
|
if (routeset == null) {
|
|
routeset = new CompactLongSet();
|
|
routesets.put(key, routeset);
|
|
}
|
|
}
|
|
|
|
for (; ; ) {
|
|
long wid = readId(dis);
|
|
if (wid == -1) break;
|
|
// expctxStat.addLookupValue( tagname, "yes", null );
|
|
if (routeset != null && !routeset.contains(wid)) {
|
|
routeset.add(wid);
|
|
routesetall.add(wid);
|
|
}
|
|
}
|
|
}
|
|
} catch (EOFException eof) {
|
|
dis.close();
|
|
}
|
|
for (String key : routesets.keySet()) {
|
|
CompactLongSet routeset = new FrozenLongSet(routesets.get(key));
|
|
routesets.put(key, routeset);
|
|
System.out.println("marked " + routeset.size() + " routes for key: " + key);
|
|
}
|
|
}
|
|
|
|
public void process(File wayFileIn, File wayFileOut, File relationFileIn, File lookupFile, File reportProfile, File checkProfile) throws Exception {
|
|
init(relationFileIn, lookupFile, reportProfile, checkProfile);
|
|
|
|
// *** finally process the way-file
|
|
wayOutStream = createOutStream(wayFileOut);
|
|
new WayIterator(this, true).processFile(wayFileIn);
|
|
wayOutStream.close();
|
|
|
|
// System.out.println( "-------- route-statistics -------- " );
|
|
// expctxStat.dumpStatistics();
|
|
}
|
|
|
|
@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);
|
|
boolean warn = expctxReport.getCostfactor() >= 10000.;
|
|
if (warn) {
|
|
expctxCheck.evaluate(false, data.description);
|
|
ok = expctxCheck.getCostfactor() < 10000.;
|
|
|
|
System.out.println("** relation access conflict for wid = " + data.wid + " tags:" + expctxReport.getKeyValueDescription(false, data.description) + " (ok=" + ok + ")");
|
|
}
|
|
|
|
if (ok) {
|
|
expctxReport.decode(data.description);
|
|
for (String key : routesets.keySet()) {
|
|
CompactLongSet routeset = routesets.get(key);
|
|
if (routeset.contains(data.wid)) {
|
|
int sepIdx = key.lastIndexOf('_');
|
|
String tagname = key.substring(0, sepIdx);
|
|
int val = Integer.valueOf(key.substring(sepIdx + 1));
|
|
expctxReport.addSmallestLookupValue(tagname, val);
|
|
}
|
|
}
|
|
data.description = expctxReport.encode();
|
|
}
|
|
}
|
|
if (wayOutStream != null) {
|
|
data.writeTo(wayOutStream);
|
|
}
|
|
}
|
|
|
|
}
|