StreamCityGmlConsumer.java 8.42 KB
Newer Older
Matthias Betz's avatar
Matthias Betz committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
/*-
 *  Copyright 2020 Beuth Hochschule für Technik Berlin, Hochschule für Technik Stuttgart
 * 
 *  This file is part of CityDoctor2.
 *
 *  CityDoctor2 is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU Lesser General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  CityDoctor2 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 Lesser General Public License for more details.
 *
 *  You should have received a copy of the GNU Lesser General Public License
 *  along with CityDoctor2.  If not, see <https://www.gnu.org/licenses/>.
 */
package de.hft.stuttgart.citydoctor2.check;

import java.time.ZonedDateTime;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
Matthias Betz's avatar
Matthias Betz committed
28
29
import java.util.Set;
import java.util.UUID;
Matthias Betz's avatar
Matthias Betz committed
30
31
import java.util.concurrent.atomic.AtomicInteger;

32
33
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
Matthias Betz's avatar
Matthias Betz committed
34
35
36
import org.citygml4j.core.model.core.AbstractFeatureProperty;
import org.citygml4j.core.model.core.CityModel;
import org.citygml4j.core.util.geometry.GeometryFactory;
Matthias Betz's avatar
Matthias Betz committed
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52

import de.hft.stuttgart.citydoctor2.check.error.SchematronError;
import de.hft.stuttgart.citydoctor2.checks.SvrlContentHandler;
import de.hft.stuttgart.citydoctor2.checks.util.FeatureCheckedListener;
import de.hft.stuttgart.citydoctor2.datastructure.BridgeObject;
import de.hft.stuttgart.citydoctor2.datastructure.Building;
import de.hft.stuttgart.citydoctor2.datastructure.CityObject;
import de.hft.stuttgart.citydoctor2.datastructure.LandObject;
import de.hft.stuttgart.citydoctor2.datastructure.TransportationObject;
import de.hft.stuttgart.citydoctor2.datastructure.Vegetation;
import de.hft.stuttgart.citydoctor2.datastructure.WaterObject;
import de.hft.stuttgart.citydoctor2.parser.CityGmlConsumer;
import de.hft.stuttgart.citydoctor2.reporting.XmlStreamReporter;
import de.hft.stuttgart.citydoctor2.reporting.pdf.PdfStreamReporter;
import de.hft.stuttgart.citydoctor2.utils.Localization;
import de.hft.stuttgart.citydoctor2.utils.QualityADEUtils;
Matthias Betz's avatar
Matthias Betz committed
53
54
55
56
57
58
59
import de.hft.stuttgart.quality.model.properties.ErrorProperty;
import de.hft.stuttgart.quality.model.properties.FeatureStatisticsProperty;
import de.hft.stuttgart.quality.model.properties.StatisticsProperty;
import de.hft.stuttgart.quality.model.properties.ValidationPlanProperty;
import de.hft.stuttgart.quality.model.types.FeatureStatistics;
import de.hft.stuttgart.quality.model.types.Statistics;
import de.hft.stuttgart.quality.model.types.Validation;
Matthias Betz's avatar
Matthias Betz committed
60
61
62

public class StreamCityGmlConsumer implements CityGmlConsumer {

63
64
	private static final Logger logger = LogManager.getLogger(StreamCityGmlConsumer.class);

Matthias Betz's avatar
Matthias Betz committed
65
66
67
68
69
	private Checker c;
	private XmlStreamReporter xmlReporter;
	private PdfStreamReporter pdfReporter;
	private SvrlContentHandler handler;
	private Map<ErrorId, AtomicInteger> errorCount;
Matthias Betz's avatar
Matthias Betz committed
70
	private GeometryFactory gmlFactory;
Matthias Betz's avatar
Matthias Betz committed
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
	private ValidationConfiguration config;
	private Statistics statistics;
	private FeatureStatistics buildingStatistics;
	private FeatureStatistics bridgeStatistics;
	private FeatureStatistics transportationStatistics;
	private FeatureStatistics vegetationStatistics;
	private FeatureStatistics landStatistics;
	private FeatureStatistics waterStatistics;
	private Validation val;
	private FeatureCheckedListener l;

	public StreamCityGmlConsumer(Checker c, XmlStreamReporter xmlReporter, PdfStreamReporter pdfReporter,
			SvrlContentHandler handler, ValidationConfiguration config, FeatureCheckedListener l) {
		this.c = c;
		this.xmlReporter = xmlReporter;
		this.pdfReporter = pdfReporter;
		this.handler = handler;
		this.config = config;
		this.l = l;
		errorCount = new HashMap<>();
Matthias Betz's avatar
Matthias Betz committed
91
		gmlFactory = GeometryFactory.newInstance();
Matthias Betz's avatar
Matthias Betz committed
92
93

		val = new Validation();
Matthias Betz's avatar
Matthias Betz committed
94
		val.setId("CD" + UUID.randomUUID().toString());
Matthias Betz's avatar
Matthias Betz committed
95
96
97
98
		val.setValidationDate(ZonedDateTime.now());
		val.setValidationSoftware("CityDoctor " + Localization.getText(Localization.VERSION));
		statistics = new Statistics();
		buildingStatistics = new FeatureStatistics();
Matthias Betz's avatar
Matthias Betz committed
99
		statistics.setNumErrorBuildings(new FeatureStatisticsProperty(buildingStatistics));
Matthias Betz's avatar
Matthias Betz committed
100
		bridgeStatistics = new FeatureStatistics();
Matthias Betz's avatar
Matthias Betz committed
101
		statistics.setNumErrorBridgeObjects(new FeatureStatisticsProperty(bridgeStatistics));
Matthias Betz's avatar
Matthias Betz committed
102
		transportationStatistics = new FeatureStatistics();
Matthias Betz's avatar
Matthias Betz committed
103
		statistics.setNumErrorTransportation(new FeatureStatisticsProperty(transportationStatistics));
Matthias Betz's avatar
Matthias Betz committed
104
		vegetationStatistics = new FeatureStatistics();
Matthias Betz's avatar
Matthias Betz committed
105
		statistics.setNumErrorVegetation(new FeatureStatisticsProperty(vegetationStatistics));
Matthias Betz's avatar
Matthias Betz committed
106
		landStatistics = new FeatureStatistics();
Matthias Betz's avatar
Matthias Betz committed
107
		statistics.setNumErrorLandObjects(new FeatureStatisticsProperty(landStatistics));
Matthias Betz's avatar
Matthias Betz committed
108
		waterStatistics = new FeatureStatistics();
Matthias Betz's avatar
Matthias Betz committed
109
		statistics.setNumErrorWaterObjects(new FeatureStatisticsProperty(waterStatistics));
Matthias Betz's avatar
Matthias Betz committed
110
111
112
113
114

	}

	@Override
	public void accept(CityObject co) {
115
116
117
118
119
120
121
122
		if (logger.isDebugEnabled()) {
			logger.debug(Localization.getText("Checker.checkFeature"), co);
		}

		// check without reporters, nothing will be put into the reports like that
		// this will also clear old check results if any (though in stream mode there
		// won't be any)
		c.checkFeature(null, null, co);
Matthias Betz's avatar
Matthias Betz committed
123

124
		// now add the schematron errors
Matthias Betz's avatar
Matthias Betz committed
125
126
127
128
129
130
131
		if (handler != null) {
			List<SchematronError> errors = handler.getFeatureErrors().get(co.getGmlId().getGmlString());
			if (errors != null) {
				Checker.handleSchematronErrorsForCityObject(errors, co);
			}
		}

132
133
134
135
136
137
138
139
140
		// now add the city object to the reports so it includes the schematron errors
		// as well
		if (xmlReporter != null) {
			xmlReporter.report(co);
		}
		if (pdfReporter != null) {
			pdfReporter.report(co);
		}

Matthias Betz's avatar
Matthias Betz committed
141
		// store quality ade datastructures in cityobject
Matthias Betz's avatar
Matthias Betz committed
142
		QualityADEUtils.writeQualityAde(co, val);
Matthias Betz's avatar
Matthias Betz committed
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
		// recreate geometry
		co.reCreateGeometries(gmlFactory, config.getParserConfiguration());

		// store result in statistics
		applyToStatistics(buildingStatistics, bridgeStatistics, transportationStatistics, vegetationStatistics,
				landStatistics, waterStatistics, co);

		// add errors to statistics
		List<CheckError> errorList = new ArrayList<>();
		co.collectContainedErrors(errorList);
		Set<CheckError> errors = new HashSet<>(errorList);
		for (CheckError e : errors) {
			errorCount.compute(e.getErrorId(), (k, v) -> {
				if (v == null) {
					return new AtomicInteger(1);
				}
				v.incrementAndGet();
				return v;
			});
		}

		if (l != null) {
			l.featureChecked(co);
		}
	}

	@Override
	public void accept(CityModel cm) {
		for (Entry<ErrorId, AtomicInteger> e : errorCount.entrySet()) {
Matthias Betz's avatar
Matthias Betz committed
172
173
174
			var stats = new de.hft.stuttgart.quality.model.types.Error();
			stats.setOccurrences(e.getValue().get());
			de.hft.stuttgart.quality.model.enums.ErrorId adeId = QualityADEUtils.mapErrorIdToAdeId(e.getKey());
Matthias Betz's avatar
Matthias Betz committed
175
176
177
178
			if (adeId == null) {
				// error that is not part of the ade standard
				continue;
			}
Matthias Betz's avatar
Matthias Betz committed
179
180
			stats.setName(adeId);
			statistics.getErrors().add(new ErrorProperty(stats));
Matthias Betz's avatar
Matthias Betz committed
181
		}
Matthias Betz's avatar
Matthias Betz committed
182
183
		val.setStatistics(new StatisticsProperty(statistics));
		val.setValidationPlan(new ValidationPlanProperty(c.createValidationPlan()));
Matthias Betz's avatar
Matthias Betz committed
184

Matthias Betz's avatar
Matthias Betz committed
185
		cm.getFeatureMembers().add(new AbstractFeatureProperty(val));
Matthias Betz's avatar
Matthias Betz committed
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
	}

	private static void applyToStatistics(FeatureStatistics buildingStatistics, FeatureStatistics bridgeStatistics,
			FeatureStatistics transportationStatistics, FeatureStatistics vegetationStatistics,
			FeatureStatistics landStatistics, FeatureStatistics waterStatistics, CityObject co) {
		if (co.isValidated()) {
			if (co instanceof Building) {
				countForFeatureStatistics(buildingStatistics, co);
			} else if (co instanceof TransportationObject) {
				countForFeatureStatistics(transportationStatistics, co);
			} else if (co instanceof BridgeObject) {
				countForFeatureStatistics(bridgeStatistics, co);
			} else if (co instanceof WaterObject) {
				countForFeatureStatistics(waterStatistics, co);
			} else if (co instanceof LandObject) {
				countForFeatureStatistics(landStatistics, co);
			} else if (co instanceof Vegetation) {
				countForFeatureStatistics(vegetationStatistics, co);
			}
		}
	}

	private static void countForFeatureStatistics(FeatureStatistics featureStatistics, CityObject co) {
		featureStatistics.setNumChecked(featureStatistics.getNumChecked() + 1);
		if (co.containsAnyError()) {
			featureStatistics.setNumErrors(featureStatistics.getNumErrors() + 1);
		}
	}

}