RegionExtractorTests.java

package eu.simstadt.regionchooser;

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;


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
		assertTrue("The exported CityGML should contain a new envelope with the correct EPSG", churchGMLString
				.contains("<gml:Envelope srsName=\"EPSG:31467\" srsDimension=\"3\">"));
		assertTrue("The exported CityGML should contain a new envelope", churchGMLString
				.contains("<gml:lowerCorner>3515829.0815150724 5415766.204935087 "));
		assertTrue("The exported CityGML should contain a new envelope", churchGMLString
				.contains("<gml:upperCorner>3515876.489172751 5415823.108586172 "));
	}

	@Test
	public void testExtractBuildingsWithoutCommentsInBetween() 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("Comments between buildings shouldn't be extracted",
				archGMLString.contains("comment between buildings")); // Comment
		assertFalse("Comments after buildings shouldn't be extracted",
				archGMLString.contains("comment after last building")); // Comment
	}

	@Test
	public void testExtractBuildingsAndChangeEnvelope() throws Throwable {
		String wktPolygon = "POLYGON((299761.8123557725 61122.68126771413,299721.46983062755 61058.11626595352,299780.84627343423 61021.99295737501,299823.9079725632 61083.3979344517,299761.8123557725 61122.68126771413))";
		Path repo = Paths.get("../TestRepository");
		Path citygmlPath = repo.resolve("NewYork.proj/FamilyCourt_LOD2_with_PLUTO_attributes.gml");
		String familyCourtBuilding = RegionExtractor
				.selectRegionDirectlyFromCityGML(citygmlPath, wktPolygon, "EPSG:32118").toString();
		assertEquals(countRegexMatches(familyCourtBuilding, "<(core:)?cityObjectMember"), 1);
		assertTrue(familyCourtBuilding.contains("Bldg_12210021066"));
		assertFalse("The exported CityGML shouldn't contain the original envelope", familyCourtBuilding
				.contains("<gml:lowerCorner>298393.46959639067 59277.34021543693 -11.892070104139751</gml:lowerCorner>"));
		assertFalse("The exported CityGML shouldn't contain the original envelope", familyCourtBuilding
				.contains("<gml:upperCorner>305641.79529639013 67101.44881543722 547.7591871983744</gml:upperCorner>"));
		assertTrue("The exported CityGML should contain a new envelope with the original altitudes", familyCourtBuilding
				.contains("<gml:lowerCorner>299721.46983062755 61021.99295737501 -11.89"));
		assertTrue("The exported CityGML should contain a new envelope with the original altitudes", familyCourtBuilding
				.contains("<gml:upperCorner>299823.9079725632 61122.68126771413 547.75"));
		assertTrue("The exported CityGML should contain a new envelope with the correct EPSG", familyCourtBuilding
				.contains("<gml:Envelope srsName=\"EPSG:32118\" srsDimension=\"3\">"));
	}

	@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 testExtract0BuildingsFromEmptyGML() 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/empty_model.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
	}

	@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
	}
	//TODO: Add BuildingParts tests with StuttgartAroundCampus.
}