RegionExtractorTests.java

package eu.simstadt.regionchooser.test;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import org.junit.Test;
import eu.simstadt.regionchooser.RegionExtractor;


public class RegionExtractorTests
{
	public static int countRegexMatches(String str, String subStr) {
		Pattern pattern = Pattern.compile(subStr);
		Matcher matcher = pattern.matcher(str);
		int count = 0;
		while (matcher.find()) {
			count++;
		}
		return count;
	}

	@Test
	public void testExtract3BuildingsFromGSK3Model() throws Throwable {
		//NOTE: Small region around Martinskirche in Grünbühl
		String wktPolygon = "POLYGON((3515848.896028535 5415823.108586172,3515848.9512289143 5415803.590347393,3515829.0815150724 5415803.338023346,3515830.9784850604 5415793.437034622,3515842.0946056456 5415793.272282251,3515843.3515515197 5415766.204935087,3515864.1064344468 5415766.557899496,3515876.489172751 5415805.433782301,3515876.343844858 5415822.009293416,3515848.896028535 5415823.108586172))";
		Path repo = Paths.get("../TestRepository");
		Path citygmlPath = repo.resolve("Gruenbuehl.proj/20140218_Gruenbuehl_LOD2.gml");
		String churchGMLString = RegionExtractor.selectRegionDirectlyFromCityGML(citygmlPath, wktPolygon, "EPSG:31467")
				.toString();
		assertEquals(countRegexMatches(churchGMLString, "<(core:)?cityObjectMember"), 3);
		assertTrue(churchGMLString.contains("Donaustr"));
		assertTrue(churchGMLString.contains("DEBW_LOD2_203056"));
		assertTrue(churchGMLString.contains("DEBW_LOD2_2869"));
		assertTrue(churchGMLString.contains("DEBW_LOD2_2909"));
		assertTrue(churchGMLString.contains("<core:CityModel")); // Header
		assertTrue(churchGMLString.contains("</core:CityModel")); // Footer
	}

	@Test
	public void testExtract3BuildingsFromNAD83Model() throws Throwable {
		//NOTE: Small region around WashingtonSquare
		String wktPolygon = "POLYGON((300259.78663489706 62835.835907766595,300230.33294975647 62792.0482567884,300213.5667431851 62770.83143720031,300183.6592861123 62730.20347659383,300252.9947486632 62676.938468840905,300273.3862256562 62701.767105345614,300257.5250407747 62715.760413539596,300308.2754543957 62805.14198211394,300259.78663489706 62835.835907766595))";
		Path repo = Paths.get("../TestRepository");
		Path citygmlPath = repo.resolve("NewYork.proj/ManhattanSmall.gml");
		String archGMLString = RegionExtractor.selectRegionDirectlyFromCityGML(citygmlPath, wktPolygon, "EPSG:32118")
				.toString();
		assertEquals(countRegexMatches(archGMLString, "<(core:)?cityObjectMember"), 2);
		assertTrue(archGMLString.contains("WASHINGTON SQUARE"));
		assertTrue(archGMLString.contains("uuid_c0980a6e-05ea-4d09-bc83-efab226945a1"));
		assertTrue(archGMLString.contains("uuid_0985cebb-922d-4b3e-95e5-15dc6089cd28"));
		assertTrue(archGMLString.contains("<CityModel")); // Header
		assertTrue(archGMLString.contains("</CityModel")); // Footer
		assertFalse(archGMLString.contains("comment between buildings")); // Comment
		assertFalse(archGMLString.contains("comment after last building")); // Comment
	}

	@Test
	public void testExtract0BuildingsWithWrongCoordinates() throws Throwable {
		//NOTE: Small region, far away from NYC
		String wktPolygon = "POLYGON((0 0, 0 1, 1 1, 1 0, 0 0))";
		Path repo = Paths.get("../TestRepository");
		Path citygmlPath = repo.resolve("NewYork.proj/ManhattanSmall.gml");
		String emptyGMLString = RegionExtractor.selectRegionDirectlyFromCityGML(citygmlPath, wktPolygon, "EPSG:32118")
				.toString();
		assertEquals(countRegexMatches(emptyGMLString, "<(core:)?cityObjectMember"), 0);
		assertTrue(emptyGMLString.contains("<CityModel")); // Header
		assertTrue(emptyGMLString.contains("</CityModel")); // Footer
	}

	@Test
	public void testExtract0BuildingsFromWeirdGML() throws Throwable {
		//NOTE: Small region, with too many spaces between coordinates
		String wktPolygon = "POLYGON((0 0, 0 1, 1 1, 1 0, 0 0))";
		Path repo = Paths.get("../TestRepository");
		Path citygmlPath = repo.resolve("NewYork.proj/broken_nyc_lod2.gml");
		String emptyGMLString = RegionExtractor.selectRegionDirectlyFromCityGML(citygmlPath, wktPolygon, "EPSG:32118")
				.toString();
		assertEquals(countRegexMatches(emptyGMLString, "<(core:)?cityObjectMember"), 0);
		assertTrue(emptyGMLString.contains("<core:CityModel")); // Header
		assertTrue(emptyGMLString.contains("</core:CityModel")); // Footer
	}
}