Create Vent

pages/Vent

@@ -0,0 +1,138 @@
+import streamlit as st
+import folium
+from streamlit_folium import folium_static
+import cdsapi
+import xarray as xr
+import numpy as np
+import tempfile
+import os
+
+# Fonction pour obtenir les données de vent depuis Copernicus
+def get_copernicus_wind_data(date):
+    c = cdsapi.Client()
+    
+    with tempfile.NamedTemporaryFile(delete=False, suffix='.nc') as tmp:
+        c.retrieve(
+            'reanalysis-era5-single-levels',
+            {
+                'product_type': 'reanalysis',
+                'variable': ['10m_u_component_of_wind', '10m_v_component_of_wind'],
+                'year': date.strftime('%Y'),
+                'month': date.strftime('%m'),
+                'day': date.strftime('%d'),
+                'time': '12:00',
+                'area': [
+                    47, 6, 45,
+                    7,
+                ],
+                'format': 'netcdf',
+            },
+            tmp.name)
+    
+    ds = xr.open_dataset(tmp.name)
+    os.unlink(tmp.name)
+    return ds
+
+# Fonction pour formater les données de vent pour leaflet-velocity
+def format_wind_data(ds):
+    u10 = ds['u10'].values[0]
+    v10 = ds['v10'].values[0]
+    
+    formatted_data = {
+        "data": [
+            {
+                "header": {
+                    "parameterNumberName": "eastward_wind",
+                    "parameterUnit": "m.s-1",
+                    "parameterNumber": 2,
+                    "parameterCategory": 2,
+                    "nx": u10.shape[1],
+                    "ny": u10.shape[0],
+                    "lo1": float(ds.longitude.min()),
+                    "la1": float(ds.latitude.max()),
+                    "lo2": float(ds.longitude.max()),
+                    "la2": float(ds.latitude.min()),
+                    "dx": float(ds.longitude[1] - ds.longitude[0]),
+                    "dy": float(ds.latitude[0] - ds.latitude[1])
+                },
+                "data": u10.flatten().tolist()
+            },
+            {
+                "header": {
+                    "parameterNumberName": "northward_wind",
+                    "parameterUnit": "m.s-1",
+                    "parameterNumber": 3,
+                    "parameterCategory": 2,
+                    "nx": v10.shape[1],
+                    "ny": v10.shape[0],
+                    "lo1": float(ds.longitude.min()),
+                    "la1": float(ds.latitude.max()),
+                    "lo2": float(ds.longitude.max()),
+                    "la2": float(ds.latitude.min()),
+                    "dx": float(ds.longitude[1] - ds.longitude[0]),
+                    "dy": float(ds.latitude[0] - ds.latitude[1])
+                },
+                "data": v10.flatten().tolist()
+            }
+        ]
+    }
+    
+    return formatted_data
+
+# Configuration de la page Streamlit
+st.set_page_config(page_title="Vent sur la région lémanique", layout="wide")
+
+# Titre de l'application
+st.title("Visualisation du vent sur la région lémanique")
+
+# Sélection de la date
+selected_date = st.date_input("Sélectionnez une date", value=datetime.now())
+
+# Obtention et formatage des données de vent
+wind_data = get_copernicus_wind_data(selected_date)
+formatted_wind_data = format_wind_data(wind_data)
+
+# Création de la carte avec folium
+m = folium.Map(location=[46.4, 6.5], zoom_start=9)
+
+# Ajout de la couche leaflet-velocity
+folium.JavascriptLink("https://unpkg.com/leaflet-velocity@1.7.0/dist/leaflet-velocity.js").add_to(m)
+folium.CssLink("https://unpkg.com/leaflet-velocity@1.7.0/dist/leaflet-velocity.css").add_to(m)
+
+# Ajout du script pour initialiser leaflet-velocity
+folium.Element(f"""
+    <script>
+        var windData = {json.dumps(formatted_wind_data)};
+        var velocityLayer = L.velocityLayer({{
+            displayValues: true,
+            displayOptions: {{
+                velocityType: "Vent",
+                position: "bottomleft",
+                emptyString: "Pas de données de vent",
+                angleConvention: "bearingCW",
+                speedUnit: "km/h"
+            }},
+            data: windData.data,
+            maxVelocity: 10,
+            velocityScale: 0.1
+        }});
+        velocityLayer.addTo(map);
+    </script>
+""").add_to(m)
+
+# Affichage de la carte dans Streamlit
+folium_static(m, width=1000, height=600)
+
+# Affichage des données brutes
+if st.checkbox("Afficher les données brutes"):
+    st.write(wind_data)
+
+# Instructions d'utilisation
+st.markdown("""
+## Instructions d'utilisation
+1. Sélectionnez une date dans le calendrier ci-dessus.
+2. La carte affichera automatiquement les données de vent pour la région lémanique à la date sélectionnée.
+3. Vous pouvez zoomer et vous déplacer sur la carte pour explorer différentes zones.
+4. Les flèches sur la carte indiquent la direction et la vitesse du vent.
+5. Cochez la case "Afficher les données brutes" pour voir les données de vent non formatées.
+""")