From 6830d3a81b61618b1d80c910bdd98d110c0ea363 Mon Sep 17 00:00:00 2001
From: Matthew Gordon <mgordon@sparkgeo.com>
Date: Wed, 4 Jun 2025 20:24:31 -0300
Subject: [PATCH] Import GeoTIFF as DEM. Works but needs polish and features

---
 .gitignore                  |   6 ++
 __init__.py                 |   7 +++
 blender_manifest.toml       |  26 ++++++++
 create_mesh_from_geotiff.py | 115 ++++++++++++++++++++++++++++++++++++
 4 files changed, 154 insertions(+)
 create mode 100644 .gitignore
 create mode 100644 __init__.py
 create mode 100644 blender_manifest.toml
 create mode 100644 create_mesh_from_geotiff.py

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..5b0eab9
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,6 @@
+/.venv
+/wheels/*.whl
+
+__pycache__
+
+*~
\ No newline at end of file
diff --git a/__init__.py b/__init__.py
new file mode 100644
index 0000000..0431ae2
--- /dev/null
+++ b/__init__.py
@@ -0,0 +1,7 @@
+from . import create_mesh_from_geotiff
+
+def register():
+    create_mesh_from_geotiff.register()
+
+def unregister():
+    create_mesh_from_geotiff.unregister()
diff --git a/blender_manifest.toml b/blender_manifest.toml
new file mode 100644
index 0000000..6d7bffd
--- /dev/null
+++ b/blender_manifest.toml
@@ -0,0 +1,26 @@
+schema_version = "1.0.0"
+id = "blender_mapping_tools"
+version = "0.1.0"
+name = "Blender Mapping Tools"
+tagline = "Tools for working with GIS data in Blender"
+maintainer = "Matt Gordon <matthew@gordon.earth"
+type = "add-on"
+tags = ["3D View", "Import-Export"]
+blender_version_min = "4.4.0"
+license = [
+  "SPDX:GPL-3.0-or-later",
+]
+copyright = [
+  "2025 Matthew Gordon"
+]
+wheels = [
+  "./wheels/affine-2.4.0-py3-none-any.whl",
+  "./wheels/attrs-25.3.0-py3-none-any.whl",
+  "./wheels/certifi-2025.4.26-py3-none-any.whl",
+  "./wheels/click-8.2.1-py3-none-any.whl",
+  "./wheels/click_plugins-1.1.1-py2.py3-none-any.whl",
+  "./wheels/cligj-0.7.2-py3-none-any.whl",
+  "./wheels/numpy-2.2.6-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl",
+  "./wheels/pyparsing-3.2.3-py3-none-any.whl",
+  "./wheels/rasterio-1.4.3-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl"
+]
\ No newline at end of file
diff --git a/create_mesh_from_geotiff.py b/create_mesh_from_geotiff.py
new file mode 100644
index 0000000..cf8bb80
--- /dev/null
+++ b/create_mesh_from_geotiff.py
@@ -0,0 +1,115 @@
+import bpy
+from bpy.props import StringProperty
+from bpy_extras.io_utils import ImportHelper
+import rasterio
+
+class GridVertexGenerator():
+    def __init__(self, grid_size_x, grid_size_y, cell_size_x, cell_size_y, dem_array):
+        self.cell_size_x = cell_size_x
+        if cell_size_y is not None:
+            self.cell_size_y = cell_size_y
+        else:
+            self.cell_size_y = cell_size_x
+        self.grid_size_x = grid_size_x
+        self.grid_size_y = grid_size_y
+        self.dem_array = dem_array
+        
+    def __len__(self):
+        return self.grid_size_x * self.grid_size_y
+    
+    def __iter__(self):
+        return self._generator()
+        
+    def _generator(self):
+        for j in range(self.grid_size_y):
+            for i in range(self.grid_size_x):
+                yield (i * self.cell_size_x,
+                       j * self.cell_size_y,
+                       self.dem_array[j][i])
+
+class GridTriangleGenerator():
+    def __init__(self, grid_size_x, grid_size_y, cell_size_x, cell_size_y=None):
+        self.cell_size_x = cell_size_x
+        if cell_size_y is not None:
+            self.cell_size_y = cell_size_y
+        else:
+            self.cell_size_y = cell_size_x
+        self.grid_size_x = grid_size_x
+        self.grid_size_y = grid_size_y
+        
+    def __len__(self):
+        #return (self.grid_size_x - 1) * (self.grid_size_y - 1) * 2
+        return (self.grid_size_x - 1) * (self.grid_size_y - 1)
+    
+    def __iter__(self):
+        return self._generator()
+        
+    def _generator(self):
+        for i in range(self.grid_size_x - 1):
+            for j in  range(self.grid_size_y - 1):
+                v00 = j * self.grid_size_x + i
+                v01 = (j + 1) * self.grid_size_x + i
+                v10 = j * self.grid_size_x + i + 1
+                v11 = (j + 1) * self.grid_size_x + i + 1
+                yield (v00, v10, v11, v01)
+                #yield (v00, v10, v01)
+                #yield (v01, v10, v11)
+        
+
+class CreateMeshFromGeotiffOperator(bpy.types.Operator, ImportHelper):
+    """Create a new object from a GeoTIFF"""
+    bl_idname = "object.create_mesh_from_geotiff"
+    bl_label = "Mesh from GeoTIFF"
+
+    filter_glob = StringProperty(default="*.tif;*.tiff", options={"HIDDEN"})
+
+    def execute(self, context):
+        with rasterio.open(self.properties.filepath) as geotiff:
+            grid_size_x = geotiff.width
+            grid_size_y = geotiff.height
+            print(f"GeoTIFF is {grid_size_x}x{grid_size_y}")
+            bounds = geotiff.bounds
+            print(f"GeoTIFF bounds: {bounds}")
+            cell_size_x = (bounds.right - bounds.left) / geotiff.width
+            cell_size_y = (bounds.bottom - bounds.top) / geotiff.height
+            print(f"Calculated cell size: {cell_size_x} x {cell_size_y}")
+            print("Reading GeoTIFF...")
+            dem_array = geotiff.read(1)
+            print("Processing values...")
+            print("Done.")
+        mesh = bpy.data.meshes.new("mesh")
+        dem_obj = bpy.data.objects.new("DEM", mesh)
+        collection = bpy.data.collections["Collection"]
+        collection.objects.link(dem_obj)
+        context.view_layer.objects.active = dem_obj
+        print("Creating mesh data...")
+        (vertices, edges, faces) = self.construct_mesh(grid_size_x, grid_size_y, cell_size_x, cell_size_y, dem_array)
+        print("Constructing mesh...")
+        mesh.from_pydata(vertices, edges, faces, shade_flat=False)
+        print("Done.")
+        mesh.validate(verbose=True)
+        return {'FINISHED'}
+    
+    def construct_mesh(self, grid_size_x, grid_size_y, cell_size_x, cell_size_y, dem_array):
+        #grid_size_x = 10
+        #grid_size_y = 15
+        #cell_size_x = 0.75
+        #cell_size_y = 0.5
+        return (GridVertexGenerator(grid_size_x, grid_size_y, cell_size_x, cell_size_y, dem_array), 
+                [], 
+                GridTriangleGenerator(grid_size_x, grid_size_y, cell_size_x, cell_size_y))  
+
+def menu_func(self, context):
+    self.layout.operator(CreateMeshFromGeotiffOperator.bl_idname,
+                         text=CreateMeshFromGeotiffOperator.bl_label)
+
+def register():
+    bpy.utils.register_class(CreateMeshFromGeotiffOperator)
+    bpy.types.VIEW3D_MT_add.append(menu_func)
+
+def unregister():
+    bpy.utils.unregister_class(CreateMeshFromGeotiffOperator)
+    bpy.types.VIEW3D_MT_add.remove(menu_func)
+
+if __name__ == "__main__":
+    register()