From e1db3331966bf1818fbb48cb8714867900dc0485 Mon Sep 17 00:00:00 2001 From: "Christopher J. Markiewicz" Date: Wed, 23 Feb 2022 11:19:23 -0500 Subject: [PATCH] ENH: Add GIFTI object __repr__ methods Helps with working on BIAP9, but long overdue --- nibabel/gifti/gifti.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/nibabel/gifti/gifti.py b/nibabel/gifti/gifti.py index 4257d1b8c7..bd8b521661 100644 --- a/nibabel/gifti/gifti.py +++ b/nibabel/gifti/gifti.py @@ -95,6 +95,9 @@ class GiftiLabelTable(xml.XmlSerializable): def __init__(self): self.labels = [] + def __repr__(self): + return f"" + def get_labels_as_dict(self): self.labels_as_dict = {} for ele in self.labels: @@ -153,6 +156,13 @@ def __init__(self, key=0, red=None, green=None, blue=None, alpha=None): self.blue = blue self.alpha = alpha + def __repr__(self): + chars = 255 * np.array( + [self.red or 0, self.green or 0, self.blue or 0, self.alpha or 0] + ) + r, g, b, a = chars.astype('u1') + return f'' + @deprecate_with_version( 'get_rgba method deprecated. ' "Use the rgba property instead.", @@ -235,6 +245,11 @@ def __init__(self, dataspace=0, xformspace=0, xform=None): else: self.xform = xform + def __repr__(self): + src = xform_codes.label[self.dataspace] + dst = xform_codes.label[self.xformspace] + return f"" + def _to_xml_element(self): coord_xform = xml.Element('CoordinateSystemTransformMatrix') if self.xform is not None: @@ -375,6 +390,9 @@ def __init__(self, self.ext_offset = ext_offset self.dims = [] if self.data is None else list(self.data.shape) + def __repr__(self): + return f"" + @property def num_dim(self): return len(self.dims)