WIP: all tests pass

examples/terrorism/TerroristNetworkModel.py

@@ -81,6 +81,26 @@ class TerroristSpreadModel(FSM, Geo):
                     return
             return self.leader
 
+    def ego_search(self, steps=1, center=False, node=None, **kwargs):
+        '''Get a list of nodes in the ego network of *node* of radius *steps*'''
+        node = as_node(node if node is not None else self)
+        G = self.subgraph(**kwargs)
+        return nx.ego_graph(G, node, center=center, radius=steps).nodes()
+
+    def degree(self, node, force=False):
+        node = as_node(node)
+        if force or (not hasattr(self.model, '_degree')) or getattr(self.model, '_last_step', 0) < self.now:
+            self.model._degree = nx.degree_centrality(self.G)
+            self.model._last_step = self.now
+        return self.model._degree[node]
+
+    def betweenness(self, node, force=False):
+        node = as_node(node)
+        if force or (not hasattr(self.model, '_betweenness')) or getattr(self.model, '_last_step', 0) < self.now:
+            self.model._betweenness = nx.betweenness_centrality(self.G)
+            self.model._last_step = self.now
+        return self.model._betweenness[node]
+
 
 class TrainingAreaModel(FSM, Geo):
     """
@@ -194,14 +214,14 @@ class TerroristNetworkModel(TerroristSpreadModel):
                     break
 
     def get_distance(self, target):
-        source_x, source_y = nx.get_node_attributes(self.topology, 'pos')[self.id]
-        target_x, target_y = nx.get_node_attributes(self.topology, 'pos')[target]
+        source_x, source_y = nx.get_node_attributes(self.G, 'pos')[self.id]
+        target_x, target_y = nx.get_node_attributes(self.G, 'pos')[target]
         dx = abs( source_x - target_x )
         dy = abs( source_y - target_y )
         return ( dx ** 2 + dy ** 2 ) ** ( 1 / 2 )
 
     def shortest_path_length(self, target):
         try:
-            return nx.shortest_path_length(self.topology, self.id, target)
+            return nx.shortest_path_length(self.G, self.id, target)
         except nx.NetworkXNoPath:
             return float('inf')

examples/terrorism/TerroristNetworkModel.yml

@@ -1,31 +1,31 @@
 name: TerroristNetworkModel_sim
-max_time: 150 
+max_steps: 150 
 num_trials: 1
-network_params:
-  generator: random_geometric_graph
-  radius: 0.2
-  # generator: geographical_threshold_graph
-  # theta: 20
-  n: 100
-network_agents:
-  - agent_class: TerroristNetworkModel.TerroristNetworkModel
-    weight: 0.8
-    state:
-      id: civilian  # Civilians
-  - agent_class: TerroristNetworkModel.TerroristNetworkModel
-    weight: 0.1
-    state:
-      id: leader    # Leaders
-  - agent_class: TerroristNetworkModel.TrainingAreaModel
-    weight: 0.05
-    state:
-      id: terrorist # Terrorism
-  - agent_class: TerroristNetworkModel.HavenModel
-    weight: 0.05
-    state:
-      id: civilian  # Civilian
+model_params:
+  network_params:
+    generator: random_geometric_graph
+    radius: 0.2
+    # generator: geographical_threshold_graph
+    # theta: 20
+    n: 100
+  network_agents:
+    - agent_class: TerroristNetworkModel.TerroristNetworkModel
+      weight: 0.8
+      state:
+        id: civilian  # Civilians
+    - agent_class: TerroristNetworkModel.TerroristNetworkModel
+      weight: 0.1
+      state:
+        id: leader    # Leaders
+    - agent_class: TerroristNetworkModel.TrainingAreaModel
+      weight: 0.05
+      state:
+        id: terrorist # Terrorism
+    - agent_class: TerroristNetworkModel.HavenModel
+      weight: 0.05
+      state:
+        id: civilian  # Civilian
 
-environment_params:
   # TerroristSpreadModel
   information_spread_intensity: 0.7
   terrorist_additional_influence: 0.035