mirror of
https://github.com/gsi-upm/soil
synced 2025-09-15 12:42:21 +00:00
05f7f49233
See CHANGELOG.md for a full list of changes * Removed nxsim * Refactored `agents.NetworkAgent` and `agents.BaseAgent` * Refactored exporters * Added stats to history
39 lines
1.0 KiB
Python
39 lines
1.0 KiB
Python
from . import NetworkAgent
|
|
|
|
|
|
class CounterModel(NetworkAgent):
|
|
"""
|
|
Dummy behaviour. It counts the number of nodes in the network and neighbors
|
|
in each step and adds it to its state.
|
|
"""
|
|
|
|
def step(self):
|
|
# Outside effects
|
|
total = len(list(self.get_agents()))
|
|
neighbors = len(list(self.get_neighboring_agents()))
|
|
self['times'] = self.get('times', 0) + 1
|
|
self['neighbors'] = neighbors
|
|
self['total'] = total
|
|
|
|
|
|
class AggregatedCounter(NetworkAgent):
|
|
"""
|
|
Dummy behaviour. It counts the number of nodes in the network and neighbors
|
|
in each step and adds it to its state.
|
|
"""
|
|
|
|
defaults = {
|
|
'times': 0,
|
|
'neighbors': 0,
|
|
'total': 0
|
|
}
|
|
|
|
def step(self):
|
|
# Outside effects
|
|
self['times'] += 1
|
|
neighbors = len(list(self.get_neighboring_agents()))
|
|
self['neighbors'] += neighbors
|
|
total = len(list(self.get_agents()))
|
|
self['total'] += total
|
|
self.debug('Running for step: {}. Total: {}'.format(self.now, total))
|