Add helper function to retrieve available objects (shogun-toolbox#4112)

karlnapf · Feb 4, 2018 · 8af43ee · 8af43ee
1 parent a6d1f45
commit 8af43ee
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/src/shogun/base/class_list.cpp.templ b/src/shogun/base/class_list.cpp.templ
@@ -74,3 +74,14 @@ void shogun::delete_object(CSGObject* object)
 {
 	delete object;
 }
+
+std::set<std::string> available_objects()
+{
+	std::set<std::string> result;
+	for (class_list_entry_t* i=class_list; i->m_class_name != NULL;
+		 i++)
+	{
+		result.insert(i->m_class_name);
+	}
+	return result;
+}
diff --git a/src/shogun/base/class_list.h b/src/shogun/base/class_list.h
@@ -13,6 +13,9 @@
 
 #include <shogun/io/SGIO.h>
 
+#include <set>
+#include <string>
+
 namespace shogun {
 	class CSGObject;
 
@@ -50,6 +53,12 @@ namespace shogun {
 		cast->ref();
 		return cast;
 	}
+
+	/** Returns all available object names
+	 *
+	 */
+	std::set<std::string> available_objects();
+
 }
 
 #endif /* __SG_CLASS_LIST_H__  */