kml / kmz exporter

__init__.py

@@ -106,7 +106,8 @@ def menu_func_import(self, context):
 
 def menu_func_export(self, context):
 	self.layout.operator('exportgis.shapefile', text="Shapefile (.shp)")
-
+	self.layout.operator('exportgis.kml_file', text="path as KML (.kml)")	
+	self.layout.operator('exportgis.kmz_file', text="path as KMZ (.kmz)")	
 
 def register():
 

operators/__init__.py

@@ -1 +1 @@
-__all__ = ["add_camera_exif", "add_camera_georef", "io_export_shp", "io_get_srtm", "io_import_georaster", "io_import_osm", "io_import_shp", "mesh_delaunay_voronoi", "nodes_terrain_analysis_builder", "nodes_terrain_analysis_reclassify", "view3d_mapviewer"]
+__all__ = ["add_camera_exif", "add_camera_georef", "io_export_shp", "io_export_kml", "io_export_kmz", "io_get_srtm", "io_import_georaster", "io_import_osm", "io_import_shp", "mesh_delaunay_voronoi", "nodes_terrain_analysis_builder", "nodes_terrain_analysis_reclassify", "view3d_mapviewer"]

operators/io_export_kml.py

@@ -0,0 +1,139 @@
+# -*- coding:utf-8 -*-
+import os
+from bpy_extras.io_utils import ExportHelper
+from bpy.props import StringProperty
+from bpy.types import Operator
+from ..geoscene import GeoScene
+from ..core.proj import Reproj
+
+
+class EXPORT_KML(Operator, ExportHelper):
+    """Export single path to KML file format (.kml)"""
+    bl_idname = "exportgis.kml_file"
+    bl_description = 'Export single path to KML file format (.kml)'
+    bl_label = "Export KML"
+    bl_options = {"UNDO"}
+
+    filename_ext = ".kml"
+    filter_glob = StringProperty(
+            default="*.kml",
+            options={'HIDDEN'},
+            )
+
+    def draw(self, context):
+        # Function used by blender to draw the panel.
+        layout = self.layout
+        layout.prop(self, 'exportType')
+
+     def execute(self, context):
+        filePath = self.filepath
+        folder = os.path.dirname(filePath)
+        filename = os.path.splitext(os.path.basename(filePath))[0]
+
+        scn = context.scene
+        geoscn = GeoScene(scn)
+        obj = context.active_object
+
+        if obj is None:
+            self.report({'INFO'}, "Selection is empty")
+            print("Selection is empty or too much object selected")
+            return {'FINISHED'}
+
+        if obj.type != 'MESH' and obj.type != 'CURVE':
+            self.report({'INFO'}, "Invalid selection")
+            return {'FINISHED'}
+
+        if geoscn.isGeoref:
+            dx, dy = geoscn.getOriginPrj()
+        elif geoscn.isBroken:
+            self.report({'ERROR'}, "Scene georef is broken, please fix it beforehand")
+            return {'FINISHED'}
+        else:
+            dx, dy = (0, 0)
+
+        tM = obj.matrix_world
+        d = obj.data
+
+        if obj.type == 'MESH':
+            verts = d.vertices
+
+        elif obj.type == 'CURVE':
+            spline = d.splines[0]
+            if spline.type == 'POLY':
+                verts = spline.points
+            elif spline.type == 'BEZIER':
+                verts = spline.bezier_points
+            else:
+                verts = []
+
+        if len(verts) == 0:
+            self.report({'ERROR'}, "No vertice to export")
+            print("No vertice to export")
+            return {'FINISHED'}
+
+        rprj = Reproj(geoscn.crs, 4326)
+        pts = []
+        for vert in verts:
+            x, y, alt = tM * vert.co.to_3d()
+            # Extract coords & adjust values against object location & shift against georef deltas
+            lon, lat = rprj.pt( x + dx, y + dy)
+            pts.append("{:.15f},{:.15f},{:.15f}".format(lon, lat, alt))
+
+        xmlString = """<?xml version="1.0" encoding="UTF-8"?>
+<kml xmlns="http://www.opengis.net/kml/2.2"
+    xmlns:gx="http://www.google.com/kml/ext/2.2"
+    xmlns:kml="http://www.opengis.net/kml/2.2"
+    xmlns:atom="http://www.w3.org/2005/Atom">
+	<Document>
+		<name>%s.kmz</name>
+		<Style id="s_ylw-pushpin">
+			<IconStyle>
+				<scale>1.1</scale>
+				<Icon>
+					<href>http://maps.google.com/mapfiles/kml/pushpin/ylw-pushpin.png</href>
+				</Icon>
+				<hotSpot x="20" y="2" xunits="pixels" yunits="pixels"/>
+			</IconStyle>
+		</Style>
+		<Style id="s_ylw-pushpin_hl">
+			<IconStyle>
+				<scale>1.3</scale>
+				<Icon>
+					<href>http://maps.google.com/mapfiles/kml/pushpin/ylw-pushpin.png</href>
+				</Icon>
+				<hotSpot x="20" y="2" xunits="pixels" yunits="pixels"/>
+			</IconStyle>
+		</Style>
+		<StyleMap id="m_ylw-pushpin">
+			<Pair>
+				<key>normal</key>
+				<styleUrl>#s_ylw-pushpin</styleUrl>
+			</Pair>
+			<Pair>
+				<key>highlight</key>
+				<styleUrl>#s_ylw-pushpin_hl</styleUrl>
+			</Pair>
+		</StyleMap>
+		<Placemark>
+			<name>%s</name>
+			<styleUrl>#m_ylw-pushpin</styleUrl>
+			<LineString>
+				<extrude>1</extrude>
+				<tessellate>1</tessellate>
+				<altitudeMode>absolute</altitudeMode>
+				<coordinates>
+                    %s
+                </coordinates>
+            </LineString>
+        </Placemark>
+    </Document>
+</kml>
+""" % (filename, filename, " ".join(pts))
+
+        f = open(folder + os.path.sep + 'doc.kml', "w")
+        f.write(xmlString)
+        f.close()
+
+        self.report({'INFO'}, "Export complete")
+        print("Export complete")
+        return {'FINISHED'}

operators/io_export_kmz.py

@@ -0,0 +1,145 @@
+# -*- coding:utf-8 -*-
+import os
+import zipfile
+from bpy_extras.io_utils import ExportHelper
+from bpy.props import StringProperty
+from bpy.types import Operator
+from ..geoscene import GeoScene
+from ..core.proj import Reproj
+
+
+class EXPORT_KMZ(Operator, ExportHelper):
+    """Export single path to KMZ file format (.kmz)"""
+    bl_idname = "exportgis.kmz_file"
+    bl_description = 'Export single path to KMZ file format (.kmz)'
+    bl_label = "Export KMZ"
+    bl_options = {"UNDO"}
+
+    # ExportHelper class properties
+    filename_ext = ".kmz"
+    filter_glob = StringProperty(
+            default="*.kmz",
+            options={'HIDDEN'},
+            )
+
+    def draw(self, context):
+        layout = self.layout
+        layout.prop(self, 'exportType')
+
+    def execute(self, context):
+        filePath = self.filepath
+        folder = os.path.dirname(filePath)
+        filename = os.path.splitext(os.path.basename(filePath))[0]
+
+        scn = context.scene
+        geoscn = GeoScene(scn)
+        obj = context.active_object
+
+        if obj is None:
+            self.report({'INFO'}, "Selection is empty")
+            print("Selection is empty or too much object selected")
+            return {'FINISHED'}
+
+        if obj.type != 'MESH' and obj.type != 'CURVE':
+            self.report({'INFO'}, "Invalid selection")
+            return {'FINISHED'}
+
+        if geoscn.isGeoref:
+            dx, dy = geoscn.getOriginPrj()
+        elif geoscn.isBroken:
+            self.report({'ERROR'}, "Scene georef is broken, please fix it beforehand")
+            return {'FINISHED'}
+        else:
+            dx, dy = (0, 0)
+
+        tM = obj.matrix_world
+        d = obj.data
+
+        if obj.type == 'MESH':
+            verts = d.vertices
+
+        elif obj.type == 'CURVE':
+            spline = d.splines[0]
+            if spline.type == 'POLY':
+                verts = spline.points
+            elif spline.type == 'BEZIER':
+                verts = spline.bezier_points
+            else:
+                verts = []
+
+        if len(verts) == 0:
+            self.report({'ERROR'}, "No vertice to export")
+            print("No vertice to export")
+            return {'FINISHED'}
+        rprj = Reproj(geoscn.crs, 4326)
+        pts = []
+        for vert in verts:
+            x, y, alt = tM * vert.co.to_3d()
+            # Extract coords & adjust values against object location & shift against georef deltas
+            lon, lat = rprj.pt( x + dx, y + dy)
+            pts.append("{:.15f},{:.15f},{:.15f}".format(lon, lat, alt))
+
+        xmlString = """<?xml version="1.0" encoding="UTF-8"?>
+<kml xmlns="http://www.opengis.net/kml/2.2"
+    xmlns:gx="http://www.google.com/kml/ext/2.2"
+    xmlns:kml="http://www.opengis.net/kml/2.2"
+    xmlns:atom="http://www.w3.org/2005/Atom">
+	<Document>
+		<name>%s.kmz</name>
+		<Style id="s_ylw-pushpin">
+			<IconStyle>
+				<scale>1.1</scale>
+				<Icon>
+					<href>http://maps.google.com/mapfiles/kml/pushpin/ylw-pushpin.png</href>
+				</Icon>
+				<hotSpot x="20" y="2" xunits="pixels" yunits="pixels"/>
+			</IconStyle>
+		</Style>
+		<Style id="s_ylw-pushpin_hl">
+			<IconStyle>
+				<scale>1.3</scale>
+				<Icon>
+					<href>http://maps.google.com/mapfiles/kml/pushpin/ylw-pushpin.png</href>
+				</Icon>
+				<hotSpot x="20" y="2" xunits="pixels" yunits="pixels"/>
+			</IconStyle>
+		</Style>
+		<StyleMap id="m_ylw-pushpin">
+			<Pair>
+				<key>normal</key>
+				<styleUrl>#s_ylw-pushpin</styleUrl>
+			</Pair>
+			<Pair>
+				<key>highlight</key>
+				<styleUrl>#s_ylw-pushpin_hl</styleUrl>
+			</Pair>
+		</StyleMap>
+		<Placemark>
+			<name>%s</name>
+			<styleUrl>#m_ylw-pushpin</styleUrl>
+			<LineString>
+				<extrude>1</extrude>
+				<tessellate>1</tessellate>
+				<altitudeMode>absolute</altitudeMode>
+				<coordinates>
+                    %s
+                </coordinates>
+            </LineString>
+        </Placemark>
+    </Document>
+</kml>
+""" % (filename, filename, " ".join(pts))
+
+        f = open(folder + os.path.sep + 'doc.kml', "w")
+        f.write(xmlString)
+        f.close()
+        zf = zipfile.ZipFile(folder + os.path.sep + filename + '.kmz', mode='w')
+        try:
+            zf.write(folder + os.path.sep + 'doc.kml', 'doc.kml')
+        finally:
+            zf.close()
+            os.remove(folder + os.path.sep + 'doc.kml')
+
+        self.report({'INFO'}, "Export complete")
+        print("Export complete")
+        return {'FINISHED'}