X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=src%2FKML.py;h=9b589ad582c0ac4f11c16a775222716bf7fa1d2f;hb=HEAD;hp=3279c858f29d42c69f7e825529031f89b685f177;hpb=b2537c605b2ad0e6b04bf1534d569f91f30f1594;p=dead%2Fcensus-tools.git diff --git a/src/KML.py b/src/KML.py index 3279c85..9b589ad 100644 --- a/src/KML.py +++ b/src/KML.py @@ -110,6 +110,75 @@ class Description(KmlObject): +class Coordinates(KmlObject): + + OPEN_TAG = '' + CLOSE_TAG = '' + + +class LineString(KmlObject): + + OPEN_TAG = '' + CLOSE_TAG = '' + + @classmethod + def parse_linestrings(self, kml): + """ + Parse each ... block from the KML. + """ + linestrings = [] + + search_idx = kml.find(self.OPEN_TAG, 0) + + while (search_idx != -1): + # No reason to keep the tag around. + ls_start = search_idx + len(self.OPEN_TAG) + ls_tag_end = kml.find(self.CLOSE_TAG, ls_start) + ls = kml[ ls_start : ls_tag_end ] + linestrings.append(ls) + search_idx = kml.find(self.OPEN_TAG, (ls_tag_end + len(self.CLOSE_TAG))) + + return linestrings + + + @classmethod + def parse_coordinates_from_linestrings(self, linestrings): + coords = [] + + for ls in linestrings: + c_tag_start = ls.find(Coordinates.OPEN_TAG) + c_start = c_tag_start + len(Coordinates.OPEN_TAG) + c_end = ls.find(Coordinates.CLOSE_TAG) + c = ls[ c_start : c_end ] + coords.append(c) + + return coords + + + @classmethod + def tuples_from_kml(self, kml): + """ + Parse one or more linestrings from a KML document. + Return a list of tuples. + """ + ls = self.parse_linestrings(kml) + cs = self.parse_coordinates_from_linestrings(ls) + + tuples = [] + + for c in cs: + pointstrings = c.strip().split() + for point in pointstrings: + components = point.strip().split(',') + if (len(components) >= 2): + # Project the three-dimensional vector onto the + # x-y plane. I don't think we're going to run + # in to any linestrings in 3d. + tuples.append( (float(components[0]), float(components[1])) ) + + return tuples + + class Name(KmlObject): OPEN_TAG = ''