add: added projectsHEADmain

1 files changed, 24 insertions, 0 deletions

diff --git a/ero1/src/generation/habitation_generation.py b/ero1/src/generation/habitation_generation.py
new file mode 100644
index 0000000..35cddaa
--- /dev/null
+++ b/ero1/src/generation/habitation_generation.py
@@ -0,0 +1,24 @@
+# FICHIER DEPRECATED : Ce fichier n'est plus utilisé.
+
+import random
+import parameters as params
+
+def place_habitation(G):
+    """
+    Place des zones d'habitation sur les arêtes du graphe.
+    Parameters:
+        G (voir quel type on veut utiliser): le graphe des routes
+    Returns:
+        None: le graphe est modifié en place
+    """
+    edges = list(G.edges(keys=True, data=True))
+    n = len(edges)
+    amount_to_place = int(params.HABITATION_PERCENTAGE * n)
+
+    # Trie les arrêtes par leur longueur. (x[3] = data)
+    edges_sorted = sorted(edges, key=lambda x: x[3].get('length', 0), reverse=True)
+
+    # Place les habitations sur les plus longues arêtes ? (tentative - juste pour tester les drones)
+    # WARNING : peut être remplacer la fonction plus tard.
+    for (u, v, k, data) in edges_sorted[:amount_to_place]:
+        data['habitation'] = True