Ensure that data is of appropriate type and return numpy array
(data)
| 102 | |
| 103 | |
| 104 | def _check_data(data): |
| 105 | """Ensure that data is of appropriate type and return numpy array""" |
| 106 | if isinstance(data, np.ndarray): |
| 107 | return data.astype(float) |
| 108 | elif isinstance(data, (str, pathlib.PosixPath)): |
| 109 | data = nib.load(data) |
| 110 | elif isinstance(data, (nib.Cifti2Image, nib.GiftiImage)): |
| 111 | pass |
| 112 | else: |
| 113 | raise TypeError('data must be a file path to a valid GIFTI or CIFTI ' |
| 114 | 'file, or an instance of numpy.ndarray, ' |
| 115 | 'nibabel.Cifti2Image nibabel.GiftiImage') |
| 116 | |
| 117 | if isinstance(data, nib.Cifti2Image): |
| 118 | return data.get_fdata().ravel().astype(float) |
| 119 | else: |
| 120 | return data.agg_data().ravel().astype(float) |
| 121 | |
| 122 | |
| 123 | def _find_color_range(v): |