Update pages/5_📍_VertXtractor.py

pages/5_📍_VertXtractor.py

@@ -40,8 +40,22 @@ def load_swiss_dem(bbox):
         data = response.json()
         
         if 'features' in data and data['features']:
-            asset_url = data['features'][0].get('assets', {}).get('data', {}).get('href')
-            if asset_url:
+            feature = data['features'][0]
+            assets = feature.get('assets', {})
+            
+            # Chercher le fichier GeoTIFF avec la meilleure résolution
+            tiff_asset = None
+            best_resolution = float('inf')
+            for asset_name, asset_info in assets.items():
+                if asset_info['type'].startswith('image/tiff'):
+                    resolution = float(asset_info.get('eo', float('inf')))
+                    if resolution < best_resolution:
+                        best_resolution = resolution
+                        tiff_asset = asset_info
+
+            if tiff_asset:
+                asset_url = tiff_asset['href']
+                st.write(f"URL de l'asset sélectionné : {asset_url}")
                 response = requests.get(asset_url)
                 response.raise_for_status()
                 with MemoryFile(response.content) as memfile:
@@ -50,7 +64,7 @@ def load_swiss_dem(bbox):
                         transform = dataset.transform
                 return dem, transform
             else:
-                st.error("URL de l'asset non trouvée dans la réponse")
+                st.error("Aucun fichier GeoTIFF trouvé dans les assets")
         else:
             st.error("Aucune donnée trouvée dans la réponse de l'API")
     except requests.RequestException as e:
@@ -62,7 +76,99 @@ def load_swiss_dem(bbox):
     
     return None, None
 
-# Les autres fonctions restent inchangées...
+def extract_dem_region(dem, transform, bbox):
+    """
+    Extrait une région spécifique du MNT basée sur une boîte englobante.
+    """
+    minx, miny, maxx, maxy = bbox
+    rows, cols = rasterio.transform.rowcol(transform, [minx, maxx], [miny, maxy])
+    window = ((min(rows), max(rows)), (min(cols), max(cols)))
+    dem_region = dem[window[0][0]:window[0][1], window[1][0]:window[1][1]]
+    new_transform = rasterio.transform.from_bounds(minx, miny, maxx, maxy, 
+                                                   dem_region.shape[1], dem_region.shape[0])
+    return dem_region, new_transform
+
+def check_invalid_values(dem):
+    invalid_values = []
+    if np.any(np.isnan(dem)):
+        invalid_values.append('Le DEM contient des valeurs NaN.')
+    if np.any(np.isinf(dem)):
+        invalid_values.append('Le DEM contient des valeurs infinies.')
+    return invalid_values
+
+def create_mesh(dem, transform, resolution):
+    """
+    Crée un maillage 3D à partir des données du MNT.
+    """
+    try:
+        height, width = dem.shape
+        if height < 3 or width < 3:
+            raise ValueError('Le MNT est trop petit pour créer un maillage.')
+
+        invalid_values = check_invalid_values(dem)
+        if invalid_values:
+            raise ValueError('\n'.join(invalid_values))
+
+        x, y = np.meshgrid(np.arange(width), np.arange(height))
+        lon, lat = rasterio.transform.xy(transform, y, x)
+        
+        vertices = np.column_stack((lon.flatten(), lat.flatten(), dem.flatten()))
+        faces = []
+        for i in range(height - 1):
+            for j in range(width - 1):
+                idx = i * width + j
+                faces.append([idx, idx + 1, idx + width])
+                faces.append([idx + 1, idx + width + 1, idx + width])
+        
+        mesh = trimesh.Trimesh(vertices=vertices, faces=faces)
+        
+        target_faces = (height * width) // (resolution ** 2)
+        mesh = mesh.simplify_quadric_decimation(target_faces)
+        
+        return mesh
+    except Exception as e:
+        logger.error(f"Erreur lors de la création du maillage: {str(e)}")
+        st.error(f"Erreur lors de la création du maillage: {str(e)}")
+        return None
+
+def visualize_mesh(mesh):
+    """
+    Crée une visualisation du maillage en 3D Trisurf.
+    """
+    fig = plt.figure()
+    ax = fig.add_subplot(111, projection='3d')
+    ax.plot_trisurf(mesh.vertices[:, 0], mesh.vertices[:, 1], mesh.vertices[:, 2], 
+                    triangles=mesh.faces, cmap='viridis')
+    ax.set_xlabel('X')
+    ax.set_ylabel('Y')
+    ax.set_zlabel('Z')
+    return fig
+
+def visualize_mesh_wireframe(mesh):
+    """
+    Crée une visualisation du maillage en wireframe.
+    """
+    fig = plt.figure()
+    ax = fig.add_subplot(111, projection='3d')
+    ax.plot_wireframe(mesh.vertices[:, 0], mesh.vertices[:, 1], mesh.vertices[:, 2], 
+                      triangles=mesh.faces, color='black')
+    ax.set_xlabel('X')
+    ax.set_ylabel('Y')
+    ax.set_zlabel('Z')
+    return fig
+
+def export_for_blender(mesh):
+    """
+    Exporte le maillage dans un format compatible avec Blender.
+    """
+    try:
+        obj_file = BytesIO()
+        mesh.export(obj_file, file_type='obj')
+        return obj_file.getvalue()
+    except Exception as e:
+        logger.error(f"Erreur lors de l'export pour Blender: {str(e)}")
+        st.error(f"Erreur lors de l'export pour Blender: {str(e)}")
+        return None
 
 def main():
     st.title("Mesh Tiler CH - Streamlit Edition")