Source code for bmtk.simulator.pointnet.gids
from collections import namedtuple
import pandas as pd
PopulationID = namedtuple('PopulationID', 'node_id population')
[docs]class GidPool(object):
def __init__(self):
# self._popid2gid = {} # (pop_name, node_id) --> nest_id
self._gid2pop_id = {} # nest_id --> (pop_name, node_id)
self._nestid_lu = {}
@property
def gids(self):
return list(self._gid2pop_id.keys())
@property
def populations(self):
return list(self._nestid_lu.keys())
[docs] def add(self, name, node_id, gid):
#self._popid2gid[name][node_id] = gid
#self._gid2pop_id[gid] = PopulationID(population=name, node_id=node_id)
raise NotImplementedError()
[docs] def get_gid(self, name, node_id):
return self.get_nestids(name=name, node_ids=[node_id])[0]
[docs] def get_pool_id(self, gid):
return self._gid2pop_id[gid]
[docs] def create_pool(self, name):
pass
[docs] def add_nestids(self, name, node_ids, nest_ids):
if name not in self._nestid_lu:
lu_table = pd.DataFrame({'nest_ids': nest_ids, 'node_ids': node_ids})
lu_table = lu_table.set_index('node_ids')
else:
new_df = pd.DataFrame({'nest_ids': nest_ids, 'node_ids': node_ids})
new_df = new_df.set_index('node_ids')
lu_table = self._nestid_lu[name]
lu_table = lu_table.append(new_df)
self._nestid_lu[name] = lu_table
for node_id, nest_id in zip(node_ids, nest_ids):
self._gid2pop_id[nest_id] = PopulationID(population=name, node_id=node_id)
[docs] def add_gids(self, name, node_ids, gids):
self.add_nestids(name=name, node_ids=node_ids, nest_ids=gids)
[docs] def get_nestids(self, name, node_ids):
nestids_table = self._nestid_lu[name]
return nestids_table.loc[node_ids]['nest_ids'].values
[docs] def get_gids(self, name, node_ids):
return self.get_nestids(name=name, node_ids=node_ids)