FEAT: Adding function to get use_count of shared pointers

pavanky · pavanky · commit 0672f56cfac6 · 2015-06-30T11:03:33.000-04:00
diff --git a/include/af/array.h b/include/af/array.h
@@ -1269,6 +1269,15 @@ extern "C" {
     */
     AFAPI af_err af_retain_array(af_array *out, const af_array in);
 
+    /**
+       \ingroup method_mat
+       @{
+
+       Get the use count of `af_array`
+    */
+    AFAPI af_err af_get_data_ref_count(int *use_count, const af_array in);
+
+
     /**
        Evaluate any expressions in the Array
     */
diff --git a/src/api/c/data.cpp b/src/api/c/data.cpp
@@ -247,6 +247,34 @@ af_err af_copy_array(af_array *out, const af_array in)
     return AF_SUCCESS;
 }
 
+//Strong Exception Guarantee
+af_err af_get_data_ref_count(int *use_count, const af_array in)
+{
+    try {
+        ArrayInfo info = getInfo(in);
+        const af_dtype type = info.getType();
+
+        int res;
+        switch(type) {
+        case f32:   res = getArray<float   >(in).useCount(); break;
+        case c32:   res = getArray<cfloat  >(in).useCount(); break;
+        case f64:   res = getArray<double  >(in).useCount(); break;
+        case c64:   res = getArray<cdouble >(in).useCount(); break;
+        case b8:    res = getArray<char    >(in).useCount(); break;
+        case s32:   res = getArray<int     >(in).useCount(); break;
+        case u32:   res = getArray<uint    >(in).useCount(); break;
+        case u8:    res = getArray<uchar   >(in).useCount(); break;
+        case s64:   res = getArray<intl    >(in).useCount(); break;
+        case u64:   res = getArray<uintl   >(in).useCount(); break;
+        default:    TYPE_ERROR(1, type);
+        }
+        std::swap(*use_count, res);
+    }
+    CATCHALL
+    return AF_SUCCESS;
+}
+
+
 template<typename T>
 static inline af_array randn_(const af::dim4 &dims)
 {
diff --git a/src/backend/cpu/Array.hpp b/src/backend/cpu/Array.hpp
@@ -170,6 +170,12 @@ namespace cpu
             return data.get() + (withOffset ? offset : 0);
         }
 
+        int useCount() const
+        {
+            if (!isReady()) eval();
+            return data.use_count();
+        }
+
         TNJ::Node_ptr getNode() const;
 
         friend Array<T> createValueArray<T>(const af::dim4 &size, const T& value);
diff --git a/src/backend/cuda/Array.hpp b/src/backend/cuda/Array.hpp
@@ -181,6 +181,12 @@ namespace cuda
             return data.get() + (withOffset ? offset : 0);
         }
 
+        int useCount() const
+        {
+            if (!isReady()) eval();
+            return data.use_count();
+        }
+
         operator Param<T>()
         {
             Param<T> out;
diff --git a/src/backend/opencl/Array.hpp b/src/backend/opencl/Array.hpp
@@ -162,6 +162,12 @@ namespace opencl
             return data.get();
         }
 
+        int useCount() const
+        {
+            if (!isReady()) eval();
+            return data.use_count();
+        }
+
         const dim_t getOffset() const
         {
             return offset;

Original file line number	Diff line number	Diff line change
`@@ -181,6 +181,12 @@ namespace cuda`
`181`	`181`	`return data.get() + (withOffset ? offset : 0);`
`182`	`182`	`}`
`183`	`183`
	`184`	`+ int useCount() const`
	`185`	`+ {`
	`186`	`+ if (!isReady()) eval();`
	`187`	`+ return data.use_count();`
	`188`	`+ }`
	`189`	`+`
`184`	`190`	`operator Param<T>()`
`185`	`191`	`{`
`186`	`192`	`Param<T> out;`
Original file line number	Diff line number	Diff line change
`@@ -162,6 +162,12 @@ namespace opencl`
`162`	`162`	`return data.get();`
`163`	`163`	`}`
`164`	`164`
	`165`	`+ int useCount() const`
	`166`	`+ {`
	`167`	`+ if (!isReady()) eval();`
	`168`	`+ return data.use_count();`
	`169`	`+ }`
	`170`	`+`
`165`	`171`	`const dim_t getOffset() const`
`166`	`172`	`{`
`167`	`173`	`return offset;`