diff options
Diffstat (limited to 'ero1/src/generation/habitation_generation.py')
| -rw-r--r-- | ero1/src/generation/habitation_generation.py | 24 |
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 |
