Support assignment in Some

src/shogun/base/some.h

@@ -31,6 +31,7 @@ namespace shogun
 			Some(const Some<T>& other);
 			Some(Some<T>&& other);
 			Some& operator=(const Some<T>& other);
+			Some& operator=(T* other);
 			~Some();
 
 			/** Casts the underlying object back to raw pointer
@@ -45,6 +46,7 @@ namespace shogun
 			T* operator->();
 		private:
 			using std::shared_ptr<T>::get;
+			using std::shared_ptr<T>::reset;
 	};
 
 	template <typename T>
@@ -67,6 +69,13 @@ namespace shogun
 	{
 	}
 	template <typename T>
+	Some<T>& Some<T>::operator=(T* other)
+	{
+		this->reset(other);
+		SG_REF(this->get());
+		return *this;
+	}
+	template <typename T>
 	Some<T>::operator T*()
 	{
 		T* ptr = this->get();

tests/unit/base/Some_unittest.cc

@@ -27,6 +27,17 @@ TEST(Some,basic)
     SG_UNREF(raw);
 }
 
+TEST(Some,reassignment)
+{
+    auto kernel = some<CGaussianKernel>();
+    CGaussianKernel* raw = new CGaussianKernel();
+    EXPECT_EQ(1, kernel->ref_count());
+    EXPECT_EQ(0, raw->ref_count());
+    kernel = raw;
+    EXPECT_TRUE(kernel->equals(raw));
+    EXPECT_EQ(1, kernel->ref_count());
+}
+
 TEST(Some,get)
 {
     auto kernel = some<CGaussianKernel>();