Update pages/sol.py

pages/sol.py

@@ -1,9 +1,11 @@
 import streamlit as st
 import rasterio
 import numpy as np
-import pyvista as pv
-from datetime import datetime
+from datetime import datetime, timedelta
 from pysolar.solar import get_altitude, get_azimuth
+from pytz import timezone
+import pyvista as pv
+import imageio
 
 # Lieux prédéfinis en Suisse
 locations = {
@@ -20,7 +22,7 @@ locations = {
     "Montreux": (46.4312, 6.9107),
 }
 
-# Fonction pour charger le raster
+# Fonction pour charger un raster
 def load_raster(file_path):
     with rasterio.open(file_path) as src:
         raster = src.read(1)
@@ -31,42 +33,84 @@ def load_raster(file_path):
 def calculate_shadows(raster, altitude, azimuth):
     shadow = np.zeros_like(raster)
     if altitude > 0:  # Si le soleil est au-dessus de l'horizon
-        shadow[raster > altitude] = 1  # Simplifié
+        shadow[raster > altitude] = 1  # Simplifié pour cet exemple
     return shadow
 
-# Interface Streamlit
+# Interface utilisateur Streamlit
 st.title("Simulation des Ombres en 3D - Suisse")
 
-uploaded_file = st.file_uploader("Téléchargez un fichier raster (MES ou MNT)", type=["tif"])
+# Chargement du raster
+uploaded_file = st.sidebar.file_uploader("Téléchargez un fichier raster (MES ou MNT)", type=["tif"])
 if uploaded_file:
     raster, transform = load_raster(uploaded_file)
-    st.write("Raster chargé avec succès !")
+    st.sidebar.write("Raster chargé avec succès !")
 
-    location_choice = st.selectbox("Choisissez un lieu", list(locations.keys()))
+    # Menu des paramètres utilisateur
+    st.sidebar.header("Paramètres")
+    location_choice = st.sidebar.selectbox("Choisissez un lieu", list(locations.keys()))
     latitude, longitude = locations[location_choice]
 
-    selected_date = st.date_input("Sélectionnez une date", value=datetime.now().date())
-    selected_time = st.time_input("Sélectionnez une heure", value=datetime.now().time())
-    specific_datetime = datetime.combine(selected_date, selected_time)
+    # Récupération des dates et heures
+    selected_start_date = st.sidebar.date_input("Date de début", value=datetime.now().date())
+    selected_end_date = st.sidebar.date_input("Date de fin", value=datetime.now().date() + timedelta(days=1))
+    selected_time = st.sidebar.time_input("Heure", value=datetime.now().time())
+    timezone_swiss = timezone("Europe/Zurich")
+    
+    # Combinaison avec fuseau horaire
+    naive_start_datetime = datetime.combine(selected_start_date, selected_time)
+    naive_end_datetime = datetime.combine(selected_end_date, selected_time)
+    start_datetime = timezone_swiss.localize(naive_start_datetime)
+    end_datetime = timezone_swiss.localize(naive_end_datetime)
 
-    altitude = get_altitude(latitude, longitude, specific_datetime)
-    azimuth = get_azimuth(latitude, longitude, specific_datetime)
+    # Section de visualisation statique
+    st.subheader("Visualisation statique")
+    altitude = get_altitude(latitude, longitude, start_datetime)
+    azimuth = get_azimuth(latitude, longitude, start_datetime)
     shadows = calculate_shadows(raster, altitude, azimuth)
 
-    # Rendu 3D avec PyVista
+    # Affichage 2D des ombres
+    st.image(shadows, caption=f"Ombres pour {start_datetime}", use_column_width=True)
+
+    # Rendu 3D
     st.subheader("Rendu 3D")
     grid = pv.UniformGrid()
-    grid.dimensions = raster.shape + (1,)  # Ajoute une 3ème dimension
-    grid.spacing = (transform[0], transform[4], 1)  # Espacement (pixels)
+    grid.dimensions = raster.shape + (1,)
+    grid.spacing = (transform[0], transform[4], 1)
     grid.point_arrays["Topography"] = raster.flatten(order="F")
     grid.point_arrays["Shadows"] = shadows.flatten(order="F")
 
-    # PyVista plot
-    plotter = pv.Plotter(notebook=False)
+    # PyVista rendu 3D
+    plotter = pv.Plotter(off_screen=True)
     plotter.add_mesh(grid, scalars="Topography", cmap="terrain", show_edges=False)
     plotter.add_mesh(grid, scalars="Shadows", cmap="gray", opacity=0.5)
     plotter.view_isometric()
     plotter.add_scalar_bar("Élévation (m)")
     plotter.show(screenshot="render_3d.png")
+    st.image("render_3d.png", caption="Rendu 3D des ombres", use_column_width=True)
+
+    # Timelapse
+    st.subheader("Génération d'un Timelapse")
+    if st.button("Générer un timelapse"):
+        current_time = start_datetime
+        frames = []
+        while current_time <= end_datetime:
+            altitude = get_altitude(latitude, longitude, current_time)
+            azimuth = get_azimuth(latitude, longitude, current_time)
+            shadows = calculate_shadows(raster, altitude, azimuth)
+            
+            # Ajout des ombres comme une frame
+            grid.point_arrays["Shadows"] = shadows.flatten(order="F")
+            plotter = pv.Plotter(off_screen=True)
+            plotter.add_mesh(grid, scalars="Topography", cmap="terrain", show_edges=False)
+            plotter.add_mesh(grid, scalars="Shadows", cmap="gray", opacity=0.5)
+            plotter.view_isometric()
+            plotter.add_scalar_bar("Élévation (m)")
+            filename = f"frame_{current_time.strftime('%Y%m%d%H%M')}.png"
+            plotter.show(screenshot=filename)
+            frames.append(imageio.imread(filename))
+            current_time += timedelta(hours=1)  # Intervalle d'une heure
 
-    st.image("render_3d.png", caption="Rendu 3D des ombres", use_column_width=True)
+        # Création du GIF
+        gif_path = "timelapse.gif"
+        imageio.mimsave(gif_path, frames, duration=0.5)  # 0.5s par frame
+        st.image(gif_path, caption="Timelapse des ombres", use_column_width=True)