/**
* @param args
* @throws IOException
*/
public static void main(String[] args) throws IOException {
ZoneCollection zones = new ZoneCollection();
String data =
new String(
Files.readAllBytes(Paths.get("/home/johannes/gsv/gis/nuts/de.nuts3.gk3.geojson")));
zones.addAll(ZoneGeoJsonIO.parseFeatureCollection(data));
data = null;
zones.setPrimaryKey("gsvId");
Map<String, Set<Zone>> aggZones = new HashMap<>();
for (Zone zone : zones.getZones()) {
String code = zone.getAttribute("nuts2_code");
Set<Zone> set = aggZones.get(code);
if (set == null) {
set = new HashSet<>();
aggZones.put(code, set);
}
set.add(zone);
}
Set<Zone> newZones = new HashSet<>();
for (Entry<String, Set<Zone>> entry : aggZones.entrySet()) {
Set<Zone> set = entry.getValue();
Geometry refGeo = null;
for (Zone zone : set) {
if (refGeo != null) {
Geometry geo2 = zone.getGeometry();
refGeo = refGeo.union(geo2);
} else {
refGeo = zone.getGeometry();
}
}
Zone zone = new Zone(refGeo);
zone.setAttribute("nuts2_code", entry.getKey());
newZones.add(zone);
}
data = ZoneGeoJsonIO.toJson(newZones);
Files.write(
Paths.get("/home/johannes/gsv/gis/nuts/de.nuts2.gk3.geojson"),
data.getBytes(),
StandardOpenOption.CREATE);
}
@Override
public void apply(Episode plan) {
for (Segment act : plan.getActivities()) {
String id = act.getAttribute(CommonKeys.ACTIVITY_FACILITY);
ActivityFacility facility =
facilities.getFacilities().get(Id.create(id, ActivityFacility.class));
Coordinate c = new Coordinate(facility.getCoord().getX(), facility.getCoord().getY());
CRSUtils.transformCoordinate(c, transform);
Zone zone = zones.get(c);
if (zone != null) {
act.setAttribute(ZONE_KEY, zone.getAttribute(idKey));
} else {
notfound++;
}
}
}
private void initMappings(ZoneCollection zones, String zoneIdKey) {
node2Zone = new ConcurrentHashMap<>();
zone2Node = new ConcurrentHashMap<>();
for (Node node : network.getNodes().values()) {
Coordinate c = new Coordinate(node.getCoord().getX(), node.getCoord().getY());
Zone zone = zones.get(c);
if (zone != null) {
String id = zone.getAttribute(zoneIdKey);
node2Zone.put(node.getId(), id);
Collection<Node> nodes = zone2Node.get(id);
if (nodes == null) {
nodes = new HashSet<>();
zone2Node.put(id, nodes);
}
nodes.add(node);
}
}
}
public NumericMatrix generate(
Network network, ZoneCollection zones, String zoneIdKey, int nThreads) {
this.network = network;
logger.info("Initializing node mappings...");
initMappings(zones, zoneIdKey);
logger.info("Calculation travel times...");
ExecutorService executor = Executors.newFixedThreadPool(nThreads);
Set<Future<Worker>> futures = new HashSet<>();
for (Zone zone : zones.getZones()) {
Worker worker = new Worker(zone.getAttribute(zoneIdKey));
futures.add(executor.submit(worker, worker));
}
ProgressLogger.init(futures.size(), 1, 10);
HashMatrix<String, CellData> sumMatrix = new HashMatrix<>();
int errors = 0;
for (Future<Worker> future : futures) {
try {
Worker worker = future.get();
HashMatrix<String, CellData> m = worker.getMatrix();
Set<String> keys = m.keys();
for (String i : keys) {
for (String j : keys) {
CellData cData = m.get(i, j);
if (cData != null) {
CellData cDataSum = sumMatrix.get(i, j);
if (cDataSum == null) {
cDataSum = new CellData();
sumMatrix.set(i, j, cDataSum);
}
cDataSum.count += cData.count;
cDataSum.ttSum += cData.ttSum;
}
}
}
errors += worker.getErrors();
ProgressLogger.step();
} catch (InterruptedException | ExecutionException e) {
e.printStackTrace();
}
}
ProgressLogger.terminate();
if (errors > 0) {
logger.info(String.format("%s errors occured.", errors));
}
logger.info("Collection results...");
NumericMatrix ttMatrix = new NumericMatrix();
Set<String> keys = sumMatrix.keys();
for (String i : keys) {
for (String j : keys) {
CellData cData = sumMatrix.get(i, j);
double avr = cData.ttSum / cData.count;
ttMatrix.set(i, j, avr);
}
}
return ttMatrix;
}