First

archive/source/sim_agents/traits_current.py

@@ -0,0 +1,91 @@
+import numpy as np
+from dataclasses import dataclass
+from typing import List, Dict, Optional
+from collections import defaultdict
+
+##https://censusatschool.ca/data-results/2017-2018/average-height-by-age/
+@dataclass
+class AgentConfig:
+    id:int
+    grade:str
+    door:int
+    speed:float
+    radius:float
+
+def agent_params():
+    gr_data = {
+        # [0]: number of students
+        # [1]: door number
+        # [2]: speed mean
+        # [3]: speed standard deviation
+        # [4]: radius mean 
+        "Kindergarden":np.array(
+            [34,0,1.21,0.24,0.407]
+            ),
+        "Grade 1":np.array(
+            [26,0,1.35,0.26,0.407]
+            ),
+        "Grade 2":np.array(
+            [42,0,1.42,0.28,0.407]
+            ),
+        "Grade 3":np.array(
+            [39,0,1.48,0.23,0.407]
+            ),
+        "Grade 4":np.array(
+            [30,1,1.58,0.26,0.417]
+            ),
+        "Grade 5":np.array(
+            [43,1,1.59,0.24,0.434]
+            ),
+        "Grade 6":np.array(
+            [29,1,1.65,0.24,0.454]
+            ),
+        "Grade 7":np.array(
+            [45,2,1.61,0.25,0.471]
+            ),
+        "Grade 8":np.array(
+            [36,2,1.66,0.24,0.488]
+            ),
+        "Grade 9":np.array(
+            [44,2,1.60,0.24,0.500]
+            ),
+        "Grade 10":np.array(
+            [36,2,1.57,0.23,0.507]
+            ),
+        "Grade 11":np.array(
+            [54,2,1.51,0.22,0.515]
+            ),
+        "Grade 12":np.array(
+            [46,2,1.54,0.23,0.520]
+            )}
+    agent_id = 1
+    rng = np.random.default_rng(seed=42)
+    all_agents = []
+    gr_agents = []
+    for grade in gr_data:
+        for num in range(int(gr_data[grade][0])):
+            door = gr_data[grade][1]
+            speed = rng.normal(
+                loc=gr_data[grade][2],
+                scale=gr_data[grade][3],
+                size=1)
+            radius = gr_data[grade][4]
+            gr_agents.append(
+                AgentConfig(
+                    id=agent_id,
+                    grade=grade,
+                    door=door,
+                    speed=speed,
+                    radius = radius
+                    ))
+            agent_id += 1
+        all_agents.append(gr_agents)
+        gr_agents = []
+    #for grade in all_agents:
+    #    for agent in grade:
+    #        print(agent)
+    return all_agents
+
+
+
+