RegionChooserCommandLineInterface.java 3.52 KB
Newer Older
Eric Duminil's avatar
Eric Duminil committed
1
2
package eu.simstadt.regionchooser;

Eric Duminil's avatar
Eric Duminil committed
3
import java.nio.charset.StandardCharsets;
Eric Duminil's avatar
Eric Duminil committed
4
import java.nio.file.Files;
Eric Duminil's avatar
Eric Duminil committed
5
import java.nio.file.Path;
Eric Duminil's avatar
Eric Duminil committed
6
import java.nio.file.Paths;
Eric Duminil's avatar
Eric Duminil committed
7
import java.util.Scanner;
Eric Duminil's avatar
Eric Duminil committed
8
import java.util.concurrent.Callable;
Eric Duminil's avatar
Eric Duminil committed
9
import org.osgeo.proj4j.CoordinateReferenceSystem;
Eric Duminil's avatar
Eric Duminil committed
10
11
import picocli.CommandLine;
import picocli.CommandLine.Command;
12
import picocli.CommandLine.Model.CommandSpec;
Eric Duminil's avatar
Eric Duminil committed
13
import picocli.CommandLine.Option;
14
import picocli.CommandLine.Spec;
Eric Duminil's avatar
Eric Duminil committed
15

Eric Duminil's avatar
Eric Duminil committed
16
17
18
19
20
// Example usage:
// --input /home/ricou/Desktop/CGSC_Repository/Würzburg.proj/LoD2_566_5516_2_BY.gml,/home/ricou/Desktop/CGSC_Repository/Würzburg.proj/LoD2_568_5516_2_BY.gml
// --output /home/ricou/Desktop/output.gml
// --wkt /home/ricou/Desktop/grombuhl.txt

Eric Duminil's avatar
Eric Duminil committed
21

Eric Duminil's avatar
Eric Duminil committed
22
@Command(name = "region_chooser", mixinStandardHelpOptions = true, version = "regionChooser x.x", description = "Extracts a region from one or more citygmls.", sortOptions = false)
Eric Duminil's avatar
Eric Duminil committed
23
class RegionChooserCommandLineInterface implements Callable<Integer>
Eric Duminil's avatar
Eric Duminil committed
24
{
25
26
	@Spec
	CommandSpec spec;
Eric Duminil's avatar
Notes.    
Eric Duminil committed
27

28
	//TODO: Add --gui?
Eric Duminil's avatar
Notes.    
Eric Duminil committed
29

Eric Duminil's avatar
Eric Duminil committed
30
31
	@Option(names = { "-i",
			"--input" }, required = true, split = ",", description = "Citygml files to extract from", paramLabel = "input.gml")
Eric Duminil's avatar
todo    
Eric Duminil committed
32
	//TODO: Allow folders too?
Eric Duminil's avatar
Eric Duminil committed
33
	Path[] citygmls;
Eric Duminil's avatar
Eric Duminil committed
34

Eric Duminil's avatar
Eric Duminil committed
35
36
	@Option(names = { "-o",
			"--output" }, required = true, description = "Output file", paramLabel = "output.gml")
Eric Duminil's avatar
Eric Duminil committed
37
	Path outputCityGML;
Eric Duminil's avatar
Eric Duminil committed
38

Eric Duminil's avatar
Eric Duminil committed
39
40
	@Option(names = { "-e", "--epsg" }, description = "EPSG id for coordinate reference system", paramLabel = "31467")
	Integer espgId;
Eric Duminil's avatar
Eric Duminil committed
41

Eric Duminil's avatar
Eric Duminil committed
42
43
44
	@Option(names = { "-l",
			"--local" }, description = "Are WKT coordinates in local CRS?", paramLabel = "local_coordinates?")
	boolean localCoordinates;
Eric Duminil's avatar
Eric Duminil committed
45

Eric Duminil's avatar
Eric Duminil committed
46
	@Option(names = { "-w",
Eric Duminil's avatar
Eric Duminil committed
47
			"--wkt" }, required = true, description = "File containing WKT polygon, or - for stdin", paramLabel = "polygon.wkt")
Eric Duminil's avatar
Eric Duminil committed
48
49
	String wktFile = "-";

Eric Duminil's avatar
Eric Duminil committed
50
	@Override
Eric Duminil's avatar
Eric Duminil committed
51
	public Integer call() throws Exception {
Eric Duminil's avatar
Eric Duminil committed
52
		CoordinateReferenceSystem localCRS;
Eric Duminil's avatar
Eric Duminil committed
53

Eric Duminil's avatar
Eric Duminil committed
54
		if (espgId == null) {
Eric Duminil's avatar
Eric Duminil committed
55
			localCRS = RegionChooserUtils.crsFromCityGMLHeader(citygmls[0]);
Eric Duminil's avatar
Eric Duminil committed
56
		} else {
Eric Duminil's avatar
Eric Duminil committed
57
			localCRS = RegionChooserUtils.crsFromSrsName("EPSG:" + espgId);
Eric Duminil's avatar
Eric Duminil committed
58
		}
59
		logInfo("Coordinate system: " + localCRS);
Eric Duminil's avatar
Eric Duminil committed
60

Eric Duminil's avatar
Eric Duminil committed
61
		String wktPolygon;
Eric Duminil's avatar
Eric Duminil committed
62

Eric Duminil's avatar
Eric Duminil committed
63
64
		if (wktFile.equals("-")) {
			if (System.in.available() == 0) {
Eric Duminil's avatar
Eric Duminil committed
65
				throw new IllegalArgumentException("Please provide \"POLYGON((x1 y1, x2 y2, ...))\" to standard input.");
Eric Duminil's avatar
Eric Duminil committed
66
			} else {
Eric Duminil's avatar
Eric Duminil committed
67
				wktPolygon = getInput();
Eric Duminil's avatar
Eric Duminil committed
68
			}
Eric Duminil's avatar
Eric Duminil committed
69
		} else {
Eric Duminil's avatar
Eric Duminil committed
70
71
72
73
			wktPolygon = new String(Files.readAllBytes(Paths.get(wktFile)), StandardCharsets.UTF_8);
			if (wktPolygon.isEmpty()) {
				throw new IllegalArgumentException("Please write \"POLYGON((x1 y1, x2 y2, ...))\" inside " + wktFile);
			}
Eric Duminil's avatar
Eric Duminil committed
74
		}
Eric Duminil's avatar
Eric Duminil committed
75

Eric Duminil's avatar
Eric Duminil committed
76
77
		if (!localCoordinates) {
			// WKT coordinates are in WGS84, so should be first converted to srsName
Eric Duminil's avatar
Eric Duminil committed
78
			wktPolygon = RegionChooserUtils.wktPolygonToLocalCRS(wktPolygon, localCRS);
Eric Duminil's avatar
Eric Duminil committed
79
80
		}

81
		logInfo("WKT Polygon expressed in local coordinates: " + wktPolygon);
Eric Duminil's avatar
Eric Duminil committed
82

Eric Duminil's avatar
Eric Duminil committed
83
		StringBuilder sb = RegionExtractor.selectRegionDirectlyFromCityGML(wktPolygon, localCRS.toString(), citygmls);
Eric Duminil's avatar
Eric Duminil committed
84
85
86

		RegionChooserUtils.writeStringBuilderToFile(sb, outputCityGML);

Eric Duminil's avatar
Eric Duminil committed
87
		return 0;
Eric Duminil's avatar
Eric Duminil committed
88
89
	}

90
91
92
93
	private void logInfo(String message) {
		spec.commandLine().getErr().println(message);
	}

Eric Duminil's avatar
Eric Duminil committed
94
	private static String getInput() {
Eric Duminil's avatar
Eric Duminil committed
95
96
97
		try (Scanner myObj = new Scanner(System.in)) {
			return myObj.nextLine();
		}
Eric Duminil's avatar
Eric Duminil committed
98
99
	}

Eric Duminil's avatar
Eric Duminil committed
100
101
102
103
104
	// this example implements Callable, so parsing, error handling and handling user
	// requests for usage help or version help can be done with one line of code.
	public static void main(String... args) {
		int exitCode = new CommandLine(new RegionChooserCommandLineInterface()).execute(args);
		System.exit(exitCode);
Eric Duminil's avatar
Eric Duminil committed
105
	}
Eric Duminil's avatar
Eric Duminil committed
106
}