BoundaryMerger.java

/*
 * Copyright (C) 2006, 2011.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 3 or
 * version 2 as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 */
package uk.me.parabola.mkgmap.reader.osm.boundary;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.nio.channels.FileChannel;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.ListIterator;

/**
 * The class merges two directories with precompiled bounds files. Both directory may contain files covering
 * the same area. These files are merged so that the resulting file contains all boundaries of both
 * files. Boundaries with the same OSM-id that are contained in both files are merged with a union of its covered area.
 * <br/><br/>
 * Using this merger it is possible to create several smaller bounds compilations (e.g. one country only) and merge 
 * these compilations together afterwards.
 * 
 * @author WanMil
 */
public class BoundaryMerger {

	public BoundaryMerger() {
		
	}
	
	private void copy(File file, File to) {
		String filename = file.getName();
		try (FileChannel in = (new FileInputStream(file)).getChannel();
				FileChannel out = (new FileOutputStream(new File(to, filename))).getChannel()) {
			in.transferTo(0, file.length(), out);
			in.close();
			out.close();
		} catch (IOException exp) {
			System.err.println(exp);
		}
	}
	
	public void merge(File dir1, File dir2, File targetDir) throws IOException {
		List<String> fl1 = BoundaryUtil.getBoundaryDirContent(dir1.getAbsolutePath());
		List<String> fl2 = BoundaryUtil.getBoundaryDirContent(dir2.getAbsolutePath());
		
		Collections.sort(fl1);
		Collections.sort(fl2);
		
		ListIterator<String> fl1Iter = fl1.listIterator();
		ListIterator<String> fl2Iter = fl2.listIterator();
		
		BoundarySaver bSave = new BoundarySaver(targetDir, BoundarySaver.QUADTREE_DATA_FORMAT);
		bSave.setCreateEmptyFiles(false);
		
		List<File> copy = new ArrayList<File>();
		
		int processed = 0;
		int all = fl1.size()+fl2.size();
		
		while (fl1Iter.hasNext() || fl2Iter.hasNext()) {
			String f1 = (fl1Iter.hasNext() ? fl1Iter.next(): null);
			String f2 = (fl2Iter.hasNext() ? fl2Iter.next(): null);

			if (f1 == null) {
				copy.add(new File(dir2,f2));
			} else if (f2 == null) {
				copy.add(new File(dir1,f1));
			} else {
				int cmp = f1.compareTo(f2);
				if (cmp < 0) {
					copy.add(new File(dir1,f1));
					fl2Iter.previous();
				} else if (cmp > 0) {
					copy.add(new File(dir2,f2));
					fl1Iter.previous();
				} else {
					BoundaryQuadTree bqt1 = BoundaryUtil.loadQuadTree(dir1.getAbsolutePath(), f1);
					if (bqt1 == null){
						System.err.println("Failed to load quadtree for " + dir1.getAbsolutePath() + f1);
						System.exit(-1);
					}
					BoundaryQuadTree bqt2 = BoundaryUtil.loadQuadTree(dir2.getAbsolutePath(), f2);
					if (bqt2 == null){
						System.err.println("Failed to load quadtree for " + dir2.getAbsolutePath() + f2);
						System.exit(-1);
					}
					
					bqt1.merge(bqt2);
					bSave.saveQuadTree(bqt1, f1);
					processed += 2;
					System.out.println(processed+"/"+all+" processed");
				}
			}
		}
		bSave.end();
		
		for (File f: copy) {
			copy(f, targetDir);
			processed++;
			System.out.println(processed+"/"+all+" processed");
		}
	}
	
	/**
	 * @param args
	 * @throws IOException 
	 */
	public static void main(String[] args) throws IOException {
		if (args.length != 3) {
			System.err.println("java uk.me.parabola.mkgmap.reader.osm.boundary.BoundaryMerger <bounds1> <bounds2> <merge>");
			System.err.println("<bounds1> <bounds2> : directories with *.bnd files to merge");
			System.err.println("<merge> target directory, is created if it doesn't exist");
			
			return;
		}

		File b1 = new File(args[0]);
		File b2 = new File(args[1]);

		File merge = new File(args[2]);

		// TODO: maybe allow zip as input
		if (b1.exists() == false || b1.isDirectory() == false) {
			System.err.println(b1 + " does not exist or is not a directory");
			return;
		}

		if (b2.exists() == false || b2.isDirectory() == false) {
			System.err.println(b2 + " does not exist or is not a directory");
			return;
		}

		if (merge.exists() && merge.isDirectory() == false) {
			System.err.println(merge + " is not a directory");
		}

		if (merge.exists() == false) {
			merge.mkdirs();
		}
	
		BoundaryMerger merger = new BoundaryMerger();
		merger.merge(b1, b2, merge);
		
	}

}