Added core.get_core_modules()

Author: Ayuto

addons/source-python/packages/source-python/core/__init__.py

@@ -49,6 +49,7 @@
 #   Core
 from _core import console_message
 from _core import get_interface
+from _core import get_core_modules
 from _core import OutputReturn
 from _core import SOURCE_ENGINE
 from _core import SOURCE_ENGINE_BRANCH
@@ -69,6 +70,7 @@
            'console_message',
            'create_checksum',
            'echo_console',
+           'get_core_modules',
            'get_interface',
            'get_public_ip',
            'ignore_unicode_errors',

src/CMakeLists.txt

@@ -149,6 +149,7 @@ Set(SOURCEPYTHON_CORE_MODULE_HEADERS
 )
 
 Set(SOURCEPYTHON_CORE_MODULE_SOURCES
+    core/modules/core/core.cpp
     core/modules/core/core_wrap.cpp
 )
 

src/core/modules/core/core.cpp

@@ -0,0 +1,80 @@
+/**
+* =============================================================================
+* Source Python
+* Copyright (C) 2012-2015 Source Python Development Team.  All rights reserved.
+* =============================================================================
+*
+* This program is free software; you can redistribute it and/or modify it under
+* the terms of the GNU General Public License, version 3.0, as published by the
+* Free Software Foundation.
+*
+* This program is distributed in the hope that it will be useful, but WITHOUT
+* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+* FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
+* details.
+*
+* You should have received a copy of the GNU General Public License along with
+* this program.  If not, see <http://www.gnu.org/licenses/>.
+*
+* As a special exception, the Source Python Team gives you permission
+* to link the code of this program (as well as its derivative works) to
+* "Half-Life 2," the "Source Engine," and any Game MODs that run on software
+* by the Valve Corporation.  You must obey the GNU General Public License in
+* all respects for all other code used.  Additionally, the Source.Python
+* Development Team grants this exception to all derivative works.
+*/
+
+//-----------------------------------------------------------------------------
+// Includes.
+//-----------------------------------------------------------------------------
+#include "core.h"
+#include "export_main.h"
+
+
+void ConsoleMessage(const char* msg)
+{
+	char* pMsg = (char*) msg;
+	int iLen = strlen(msg);
+
+	while(iLen > 0) {
+		ConMsg(pMsg);
+		pMsg += MAX_CON_MSG-1;
+		iLen -= MAX_CON_MSG-1;
+	}
+}
+
+void* GetInterface(const char* library, const char* interface_name)
+{
+	DLLib* lib = dlLoadLibrary(library);
+	if (!lib)
+		BOOST_RAISE_EXCEPTION(PyExc_ValueError, "Unable to load library '%s'.", library)
+
+	CreateInterfaceFn pCreateInterface = (CreateInterfaceFn) dlFindSymbol(lib, CREATEINTERFACE_PROCNAME);
+	dlFreeLibrary(lib);
+
+	if (!pCreateInterface)
+		BOOST_RAISE_EXCEPTION(PyExc_ValueError, "Unable to retrieve interface function '%s'.", CREATEINTERFACE_PROCNAME)
+
+	int return_code;
+	void* result = pCreateInterface(interface_name, &return_code);
+	if (!result)
+		BOOST_RAISE_EXCEPTION(PyExc_ValueError, "Unable to find interface '%s'. Return code: %i.", interface_name, return_code)
+
+	return result;
+}
+
+list GetCoreModules()
+{
+	list result;
+
+	for (int i=0; i < MAX_SOURCEPYTHON_MODULES; ++i)
+	{
+		const char* module = g_SourcePythonModules[i].szName;
+		if (!module)
+			break;
+
+		result.append(module);
+	}
+
+	return result;
+}

src/core/modules/core/core.h

@@ -33,6 +33,8 @@
 #include "dbg.h"
 #include "interface.h"
 #include "dynload.h"
+#include "boost/python/list.hpp"
+using namespace boost::python;
 
 // Maximum console message size including null terminating char
 #define MAX_CON_MSG 1024
@@ -64,36 +66,9 @@ enum OutputReturn
 //-----------------------------------------------------------------------------
 // ConMsg wrapper
 //-----------------------------------------------------------------------------
-inline void ConsoleMessage(const char* msg)
-{
-	char* pMsg = (char*) msg;
-	int iLen = strlen(msg);
-
-	while(iLen > 0) {
-		ConMsg(pMsg);
-		pMsg += MAX_CON_MSG-1;
-		iLen -= MAX_CON_MSG-1;
-	}
-}
-
-inline void* GetInterface(const char* library, const char* interface_name)
-{
-	DLLib* lib = dlLoadLibrary(library);
-	if (!lib)
-		BOOST_RAISE_EXCEPTION(PyExc_ValueError, "Unable to load library '%s'.", library)
-
-	CreateInterfaceFn pCreateInterface = (CreateInterfaceFn) dlFindSymbol(lib, CREATEINTERFACE_PROCNAME);
-	dlFreeLibrary(lib);
-
-	if (!pCreateInterface)
-		BOOST_RAISE_EXCEPTION(PyExc_ValueError, "Unable to retrieve interface function '%s'.", CREATEINTERFACE_PROCNAME)
-
-	int return_code;
-	void* result = pCreateInterface(interface_name, &return_code);
-	if (!result)
-		BOOST_RAISE_EXCEPTION(PyExc_ValueError, "Unable to find interface '%s'. Return code: %i.", interface_name, return_code)
+void ConsoleMessage(const char* msg);
+void* GetInterface(const char* library, const char* interface_name);
+list GetCoreModules();
 
-	return result;
-}
 
 #endif // _CORE_H

src/core/modules/core/core_wrap.cpp

@@ -126,4 +126,10 @@ void export_functions(scope _core)
 		return_by_value_policy(),
 		"Retrieve an interface from a library."
 	);
+
+	def(
+		"get_core_modules",
+		&GetCoreModules,
+		"Return a list of all modules exposed by Source.Python's core.\n\n"
+		":rtype: list");
 }