Merge branch 'feature/check_if_owndata' into main

tests/test_ref.py

@@ -255,3 +255,45 @@ class Prism(xo.Struct):
 
         assert prism_triangle.volume == 45
         assert prism_square.volume == 120
+
+        assert prism_triangle._has_refs
+
+def test_has_refs():
+
+    class StructWRef(xo.Struct):
+        a = xo.Ref(xo.Float64[:])
+    assert StructWRef._has_refs
+
+    class StructNoRef(xo.Struct):
+        a = xo.Float64[:]
+    assert not StructNoRef._has_refs
+
+    class NestedWRef(xo.Struct):
+        s = StructWRef
+    assert NestedWRef._has_refs
+
+    class NestedNoRef(xo.Struct):
+        s = StructNoRef
+    assert not NestedNoRef._has_refs
+
+    ArrNoRef = xo.Float64[:]
+    assert not ArrNoRef._has_refs
+
+    ArrWRef = xo.Ref(xo.Float64)[:]
+    assert ArrWRef._has_refs
+
+    class StructArrRef(xo.Struct):
+        arr = ArrWRef
+    assert StructArrRef._has_refs
+
+    class StructArrNoRef(xo.Struct):
+        arr = ArrNoRef
+    assert not StructArrNoRef._has_refs
+
+    ArrOfStructRef = NestedWRef[:]
+    assert ArrOfStructRef._has_refs
+
+    class MyUnion(xo.UnionRef):
+        _ref = [xo.Float64, xo.Int32]
+    assert MyUnion._has_refs
+

xobjects/array.py

@@ -212,6 +212,14 @@ def __new__(cls, name, bases, data):
         if "_c_type" not in data:
             data["_c_type"] = name
 
+        # determine has_refs
+        if '_itemtype' in data.keys():
+            if (hasattr(data['_itemtype'], '_has_refs')
+                 and data['_itemtype']._has_refs):
+                data['_has_refs'] = True
+            else:
+                data['_has_refs'] = False
+
         return type.__new__(cls, name, bases, data)
 
     def __getitem__(cls, shape):

xobjects/ref.py

@@ -19,6 +19,9 @@ def __getitem__(cls, reftype):
 
 
 class Ref(metaclass=MetaRef):
+
+    _has_refs = True
+
     def __init__(self, reftype):
         self._reftype = reftype
         self.__name__ = "Ref" + self._reftype.__name__
@@ -102,6 +105,8 @@ def __new__(cls, name, bases, data):
         if "_methods" not in data:
             data["_methods"] = []
 
+        data['_has_refs'] = True
+
         return type.__new__(cls, name, bases, data)
 
     def _is_member(cls, value):

xobjects/struct.py

@@ -43,7 +43,6 @@
 import logging
 from typing import Callable, Optional
 
-
 from .typeutils import (
     get_a_buffer,
     dispatch_arg,
@@ -246,6 +245,15 @@ def _inspect_args(cls, *args, **kwargs):
         if "_c_type" not in data:
             data["_c_type"] = name
 
+        # determine owndata
+        _has_refs = False
+        for ff in data['_fields']:
+            ftype = ff.ftype
+            if hasattr(ftype, '_has_refs') and ftype._has_refs:
+                _has_refs = True
+                break
+        data['_has_refs'] = _has_refs
+
         return type.__new__(cls, name, bases, data)
 
     def __getitem__(cls, shape):