get_rgba => rgba

Author: Ben Cipollini

nibabel/gifti/gifti.py

@@ -133,10 +133,31 @@ def __init__(self, key=0, label='', red=None, green=None, blue=None,
         self.blue = blue
         self.alpha = alpha
 
+    @np.deprecate_with_doc("Use the rgba property instead.")
     def get_rgba(self):
+        return self.rgba
+
+    @property
+    def rgba(self):
         """ Returns RGBA as tuple """
         return (self.red, self.green, self.blue, self.alpha)
 
+    @rgba.setter
+    def rgba(self, rgba):
+        """ Set RGBA via tuple
+
+        Parameters
+        ----------
+        rgba : tuple (red, green, blue, alpha)
+
+        """
+        if len(rgba) != 4:
+            raise ValueError('rgba must be length 4.')
+        self.red = rgba[0]
+        self.green = rgba[1]
+        self.blue = rgba[2]
+        self.alpha = rgba[3]
+
 
 def _arr2txt(arr, elem_fmt):
     arr = np.asarray(arr)

nibabel/gifti/tests/test_gifti.py

@@ -64,3 +64,31 @@ def test_metadata():
     with clear_and_catch_warnings() as w:
         warnings.filterwarnings('once', category=DeprecationWarning)
         assert_equal(len(GiftiMetaData().get_metadata()), 0)
+
+
+def test_gifti_label_rgba():
+    rgba = np.random.rand(4)
+    kwargs = dict(zip(['red', 'green', 'blue', 'alpha'], rgba))
+
+    gl = GiftiLabel(**kwargs)
+    assert_equal(kwargs['red'], gl.rgba[0])
+    assert_equal(kwargs['green'], gl.rgba[1])
+    assert_equal(kwargs['blue'], gl.rgba[2])
+    assert_equal(kwargs['alpha'], gl.rgba[3])
+
+    gl = GiftiLabel()
+    gl.rgba = rgba
+    assert_equal(kwargs['red'], gl.rgba[0])
+    assert_equal(kwargs['green'], gl.rgba[1])
+    assert_equal(kwargs['blue'], gl.rgba[2])
+    assert_equal(kwargs['alpha'], gl.rgba[3])
+
+    with assert_raises(ValueError):
+        gl.rgba = rgba[:2]
+    with assert_raises(ValueError):
+        gl.rgba = rgba.tolist() + rgba.tolist()
+
+    # Test deprecation
+    with clear_and_catch_warnings() as w:
+        warnings.filterwarnings('once', category=DeprecationWarning)
+        assert_equal(kwargs['red'], gl.get_rgba()[0])