diff --git a/setup.py b/setup.py index 0cf45ff23f88538e7f2d80ea7136415737073b34..1150b06b3c8d112d01b26b8794e2576e0934824c 100644 --- a/setup.py +++ b/setup.py @@ -3,7 +3,7 @@ from setuptools import find_packages, setup setup( name='tvb-ebrains-data', packages=find_packages(), - version='0.2.0', + version='0.2.1', install_requires=[ 'tvb-library', ] diff --git a/tvb_ebrains_data/data.py b/tvb_ebrains_data/data.py index de4dd98d9564aecee960c8e96b8a9fa4c0ad08c7..7c67de4ebd96f2a0781e81107b8e3d76c809af5a 100644 --- a/tvb_ebrains_data/data.py +++ b/tvb_ebrains_data/data.py @@ -132,3 +132,17 @@ class Brains1000Dataset(DataProxyConnectivityDataset): fname = f'{subject}_SC_7NW100parc.txt' W = np.loadtxt(self.open_file(f'{self.version}/Schaefer/Matrices/{fname}')) return W + + def get_connectivity(self, subject, scaling_factor=124538.470647693): + SC = self.load_sc(subject) + SC = SC / scaling_factor + conn = connectivity.Connectivity( + weights = SC, + tract_lengths=np.ones_like(SC), + centres = np.zeros(np.shape(SC)[0]), + speed = np.r_[np.Inf] + ) + conn.compute_region_labels() + logger.warning("Placeholder region names!") + return conn +