+ out = "POINT(%s)" % self.mk_wkt_point(geom)
+ elif geom.find('(') < 0:
+ line = ','.join([self.mk_wkt_point(x) for x in geom.split(',')])
+ out = "LINESTRING(%s)" % line
+ else:
+ inner = geom.strip('() ')
+ line = ','.join([self.mk_wkt_point(x) for x in inner.split(',')])
+ out = "POLYGON((%s))" % line
+
+ return "ST_SetSRID('%s'::geometry, 4326)" % out
+
+ def mk_wkt_point(self, point):
+ geom = point.strip()
+ if geom.find(' ') >= 0:
+ return geom
+ else:
+ pt = self.grid_node(int(geom))
+ assert_is_not_none(pt, "Point not found in grid")
+ return "%f %f" % pt
+
+ def get_scene_geometry(self, default_scene, name):
+ geoms = []
+ for obj in name.split('+'):
+ oname = obj.strip()
+ if oname.startswith(':'):
+ assert_is_not_none(default_scene, "You need to set a scene")
+ defscene = self.load_scene(default_scene)
+ wkt = defscene[oname[1:]]
+ else:
+ scene, obj = oname.split(':', 2)
+ scene_geoms = self.load_scene(scene)
+ wkt = scene_geoms[obj]
+
+ geoms.append("'%s'::geometry" % wkt)
+
+ if len(geoms) == 1:
+ return geoms[0]
+ else:
+ return 'ST_LineMerge(ST_Collect(ARRAY[%s]))' % ','.join(geoms)
+
+ def load_scene(self, name):
+ if name in self.scene_cache:
+ return self.scene_cache[name]
+
+ scene = {}
+ with open(os.path.join(self.scene_path, "%s.wkt" % name), 'r') as fd:
+ for line in fd:
+ if line.strip():
+ obj, wkt = line.split('|', 2)
+ scene[obj.strip()] = wkt.strip()
+ self.scene_cache[name] = scene
+
+ return scene
+
+ def clear_grid(self):
+ self.grid = {}