def test_coords_transform():
aff = EXAMPLE_COORDS['affine']
for coords in EXAMPLE_COORDS['coords']:
ijk, xyz = coords['ijk'], coords['xyz']
assert np.all(atlasreader.coord_xyz_to_ijk(aff, xyz) == ijk)
assert np.all(atlasreader.coord_ijk_to_xyz(aff, ijk) == xyz)
with pytest.raises(ValueError):
atlasreader.coord_xyz_to_ijk(aff, [[10, 10], [20, 30]])
with pytest.raises(ValueError):
atlasreader.coord_ijk_to_xyz(aff, [[10, 10], [20, 30]])
DK_dict_voxels = {}
DK_dict_counts = {}
for n in tqdm(atlasreader.get_atlas("desikan_killiany")['labels']['name']):
DK_dict_counts.update({n: {Yeo_label: 0 for Yeo_label in Yeo_labels}})
DK_dict_voxels.update({n: {Yeo_label: [] for Yeo_label in Yeo_labels}})
for DK_i in range(0, DK_data.shape[0]):
for DK_j in range(0, DK_data.shape[1]):
for DK_k in range(0, DK_data.shape[2]):
DK_voxel_label = DK_data[DK_i, DK_j, DK_k]
DK_voxel_region = atlasreader.get_label(DK_atlas, DK_voxel_label)
if DK_voxel_region != 'Unknown':
xyz = atlasreader.coord_ijk_to_xyz(DK_affine,
[DK_i, DK_j, DK_k])
Yeo_ijk = atlasreader.coord_xyz_to_ijk(Yeo_affine, xyz)[0]
Yeo_voxel_label = int(Yeo_data[Yeo_ijk[0], Yeo_ijk[1],
Yeo_ijk[2]])
Yeo_voxel_region = Yeo_labels[Yeo_voxel_label]
DK_dict_counts[DK_voxel_region][Yeo_voxel_region] += 1
DK_dict_voxels[DK_voxel_region][Yeo_voxel_region].append(
xyz[0])
# Save 17 networks map... in .mat format?
savemat('DK_dict17_voxels.mat', DK_dict_voxels)
savemat('DK_dict17_counts.mat', DK_dict_counts)
# save as python: