Skip to content

Commit

Permalink
'#2013 Accumulates exceptions for resumed warn on processing log and
Browse files Browse the repository at this point in the history 
represents unclosed linearring features as linearstring.
  • Loading branch information
@patrickdalla
patrickdalla committed May 16, 2024
1 parent ac8ff26 commit 05a096e
Showing 1 changed file with 96 additions and 63 deletions.
159 changes: 96 additions & 63 deletions iped-geo/src/main/java/iped/geo/parsers/kmlstore/KMLParser.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -17,12 +17,14 @@
import org.jdom2.input.SAXBuilder;
import org.opengis.feature.simple.SimpleFeature;
import org.opengis.feature.simple.SimpleFeatureType;
import org.slf4j.LoggerFactory;

import com.vividsolutions.jts.geom.Coordinate;
import com.vividsolutions.jts.geom.Geometry;
import com.vividsolutions.jts.geom.GeometryFactory;

public class KMLParser {
private static org.slf4j.Logger LOGGER = LoggerFactory.getLogger(KMLParser.class);

public static List<Object> parse(File file) throws SchemaException, IOException, JDOMException {
/*
Expand Down Expand Up @@ -52,6 +54,8 @@ public static List<Object> parse(File file) throws SchemaException, IOException,
Element kml = document.getRootElement();
Element placemarks = kml.getChildren().get(0);
List<Element> pms = placemarks.getChildren();

ArrayList<Exception> exList = new ArrayList<Exception>();

for (Iterator<Element> iterator = pms.iterator(); iterator.hasNext();) {
Element ele = iterator.next();
Expand All @@ -60,19 +64,19 @@ public static List<Object> parse(File file) throws SchemaException, IOException,
if (isTrack(ele)) {
parsePlacemarkTrack(ele, features, featureBuilder);
} else {
try {
features.add(parsePlacemark(ele, featureBuilder));
} catch (Exception e) {
e.printStackTrace();
}
features.add(parsePlacemark(ele, featureBuilder, exList));
}
}
if (ele.getName().toLowerCase().equals("folder")) {
Folder f = parseFolder(ele, featureBuilder);
Folder f = parseFolder(ele, featureBuilder, exList);
features.add(f);
}
}

if (exList.size() > 0) {
LOGGER.warn("GeofileParser encountered " + exList.size() + " invalid geo features.");
}

return features;
}

Expand Down Expand Up @@ -131,7 +135,8 @@ public static boolean isTrack(Element ele) {
return false;
}

public static SimpleFeature parsePlacemark(Element pm, SimpleFeatureBuilder featureBuilder) {
public static SimpleFeature parsePlacemark(Element pm, SimpleFeatureBuilder featureBuilder,
ArrayList<Exception> exList) {
List<Element> eles = pm.getChildren();
GeometryFactory geometryFactory = JTSFactoryFinder.getGeometryFactory();
Geometry geo = null;
Expand All @@ -154,7 +159,7 @@ public static SimpleFeature parsePlacemark(Element pm, SimpleFeatureBuilder feat
timestamp = ele.getChildren().get(0).getText();
continue;
}
Geometry geoInt = parseGeometry(ele, geometryFactory);
Geometry geoInt = parseGeometry(ele, geometryFactory, exList);
if (geoInt != null) {
geo = geoInt;
}
Expand Down Expand Up @@ -202,78 +207,106 @@ static public Coordinate parseCoordinate(String coordinates, String tokens) {
}
}

public static Geometry parseGeometry(Element ele, GeometryFactory geometryFactory) {
Geometry geo = null;

if (ele.getName().toLowerCase().equals("point")) {
Coordinate coord = parsePoint(ele);
geo = geometryFactory.createPoint(coord);
public static Coordinate[] closePlygonCoords(Coordinate[] coords) {
Coordinate[] closedCoords = new Coordinate[coords.length + 1];
for (int j = 0; j < coords.length; j++) {
closedCoords[j] = coords[j];
}
closedCoords[coords.length] = coords[0];
return closedCoords;
}

if (ele.getName().toLowerCase().equals("linestring") || ele.getName().toLowerCase().equals("linearring")) {
List<Element> eles = ele.getChildren();
Element coordsEle = null;
for (Iterator<Element> iterator = eles.iterator(); iterator.hasNext();) {
Element element = iterator.next();
if (element.getName().toLowerCase().equals("coordinates")) {
coordsEle = element;
break;
}
}
public static Geometry parseGeometry(Element ele, GeometryFactory geometryFactory, ArrayList<Exception> exList) {
Geometry geo = null;

String coordinates = coordsEle.getText().trim();
StringTokenizer st = new StringTokenizer(coordinates, " \t\n\r");
Coordinate[] coords = new Coordinate[st.countTokens()];
int i = 0;
while (st.hasMoreTokens()) {
String tok = st.nextToken(" ");
coords[i] = parseCoordinate(tok);
i++;
}
if (ele.getName().toLowerCase().equals("linestring")) {
geo = geometryFactory.createLineString(coords);
} else {
geo = geometryFactory.createLinearRing(coords);
try {
if (ele.getName().toLowerCase().equals("point")) {
Coordinate coord = parsePoint(ele);
geo = geometryFactory.createPoint(coord);
}
}

if (ele.getName().toLowerCase().equals("polygon")) {
Element boundary = null;
List<Element> eles = ele.getChildren();
for (Iterator<Element> iterator = eles.iterator(); iterator.hasNext();) {
Element element = iterator.next();
if (element.getName().toLowerCase().equals("outerboundaryis")) {
boundary = element;
break;
if (ele.getName().toLowerCase().equals("linestring") || ele.getName().toLowerCase().equals("linearring")) {
List<Element> eles = ele.getChildren();
Element coordsEle = null;
for (Iterator<Element> iterator = eles.iterator(); iterator.hasNext();) {
Element element = iterator.next();
if (element.getName().toLowerCase().equals("coordinates")) {
coordsEle = element;
break;
}
}
if (element.getName().toLowerCase().equals("innerboundaryis")) {
boundary = element;
break;

String coordinates = coordsEle.getText().trim();
StringTokenizer st = new StringTokenizer(coordinates, " \t\n\r");
Coordinate[] coords = new Coordinate[st.countTokens()];
int i = 0;
while (st.hasMoreTokens()) {
String tok = st.nextToken(" ");
coords[i] = parseCoordinate(tok);
i++;
}

if (coords.length < 3) {
geo = geometryFactory.createLineString(coords);
} else {
if (ele.getName().toLowerCase().equals("linestring")) {
geo = geometryFactory.createLineString(coords);
} else {
try {
geo = geometryFactory.createLinearRing(coords);
} catch (IllegalArgumentException e) {
if (e.getMessage().contains("closed linestring")) {
try {
geo = geometryFactory.createLineString(coords);
} catch (Exception e2) {
throw e2;
}
}
}
}
}
}

if (boundary != null) {
geo = parseGeometry(boundary.getChildren().get(0), geometryFactory);
if (ele.getName().toLowerCase().equals("polygon")) {
Element boundary = null;
List<Element> eles = ele.getChildren();
for (Iterator<Element> iterator = eles.iterator(); iterator.hasNext();) {
Element element = iterator.next();
if (element.getName().toLowerCase().equals("outerboundaryis")) {
boundary = element;
break;
}
if (element.getName().toLowerCase().equals("innerboundaryis")) {
boundary = element;
break;
}
}

if (boundary != null) {
geo = parseGeometry(boundary.getChildren().get(0), geometryFactory, exList);
}
}
}

if (ele.getName().toLowerCase().equals("multigeometry")) {
List<Geometry> geos = new ArrayList<Geometry>();
if (ele.getName().toLowerCase().equals("multigeometry")) {
List<Geometry> geos = new ArrayList<Geometry>();

List<Element> eles = ele.getChildren();
for (Iterator<Element> iterator = eles.iterator(); iterator.hasNext();) {
Geometry subgeo = parseGeometry(iterator.next(), geometryFactory);
if (subgeo != null) {
geos.add(subgeo);
List<Element> eles = ele.getChildren();
for (Iterator<Element> iterator = eles.iterator(); iterator.hasNext();) {
Geometry subgeo = parseGeometry(iterator.next(), geometryFactory, exList);
if (subgeo != null) {
geos.add(subgeo);
}
}
geo = geometryFactory.buildGeometry(geos);
}
geo = geometryFactory.buildGeometry(geos);
} catch (Exception e) {
exList.add(e);
}

return geo;
}

public static Folder parseFolder(Element pm, SimpleFeatureBuilder featureBuilder) {
public static Folder parseFolder(Element pm, SimpleFeatureBuilder featureBuilder, ArrayList<Exception> exList) {
List<Object> features = new ArrayList<Object>();
Folder folder = new Folder();

Expand All @@ -282,10 +315,10 @@ public static Folder parseFolder(Element pm, SimpleFeatureBuilder featureBuilder
Element ele = iterator.next();
try {
if (ele.getName().toLowerCase().equals("placemark")) {
features.add(parsePlacemark(ele, featureBuilder));
features.add(parsePlacemark(ele, featureBuilder, exList));
}
if (ele.getName().toLowerCase().equals("folder")) {
features.add(parseFolder(ele, featureBuilder));
features.add(parseFolder(ele, featureBuilder, exList));
}
if (ele.getName().toLowerCase().equals("name")) {
folder.setName(ele.getText());
Expand Down

0 comments on commit 05a096e

Please sign in to comment.