X-Git-Url: https://gitweb.michael.orlitzky.com/?a=blobdiff_plain;ds=inline;f=src%2FKML.py;h=9b589ad582c0ac4f11c16a775222716bf7fa1d2f;hb=856b96616bcfcbc6f5c9e04b6f6dcc07fa7a1ff7;hp=f96adb26272c56563ced80ef3636623e024db173;hpb=28d3599ef601b5ad3fff3b630cbd2b87f4c65dbd;p=dead%2Fcensus-tools.git
diff --git a/src/KML.py b/src/KML.py
index f96adb2..9b589ad 100644
--- a/src/KML.py
+++ b/src/KML.py
@@ -32,17 +32,17 @@ class KmlObject(object):
def to_kml(self):
- return self.render()
+ kml = self.OPEN_TAG
+ kml += self.render()
+ kml += self.CLOSE_TAG + "\n"
+ return kml
def render(self):
- kml = self.OPEN_TAG
- kml += escape(self.text)
+ kml = escape(self.text)
for c in self.children:
kml += c.to_kml()
-
- kml += self.CLOSE_TAG + "\n"
return kml
@@ -75,7 +75,7 @@ class Document(KmlObject):
"""
CLOSE_TAG = """
- """
+"""
def __init__(self, initial_text=''):
super(Document, self).__init__(initial_text)
@@ -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 = ''