python_compile.cpp File Reference

#include "cmd/unit_generic.h"
#include "python_compile.h"
#include <compile.h>
#include <eval.h>
#include "configxml.h"
#include "vs_globals.h"
#include "vsfilesystem.h"
#include "init.h"
#include "universe_util.h"
#include "in_kb_data.h"

Go to the source code of this file.

Functions
char *	LoadString (const char *filename)
std::string	getCompilingName (const std::string &name)
void	InterpretPython (const std::string &name)
PyCodeObject *	CompilePython (const std::string &name)
void	CompileRunPython (const std::string &filename)
PyObject *	CreateTuple (const std::vector< PythonBasicType > &values)
static void	pySetScratchVector (const KBSTATE k)
void	RunPythonPress (const KBData &s, KBSTATE k)
void	RunPythonRelease (const KBData &s, KBSTATE k)
void	RunPythonToggle (const KBData &s, KBSTATE k)
void	RunPythonPhysicsFrame (const KBData &s, KBSTATE k)

Variables
Hashtable< string, PyCodeObject, 1023 >	compiled_python

Function Documentation

PyCodeObject* CompilePython

(

const std::string &

name)

Definition at line 49 of file python_compile.cpp.

References fprintf, Hashtable< KEY, VALUE, SIZ >::Get(), getCompilingName(), LoadString(), and Hashtable< KEY, VALUE, SIZ >::Put().

Referenced by CompileRunPython().

{
    Python::reseterrors();
    PyCodeObject *retval = compiled_python.Get( name );
    Python::reseterrors();
    if (retval)
        return retval;
    char *str = LoadString( name.c_str() );
    if (str) {
        fprintf( stdout, "Compiling python module %s\n", name.c_str() );

        std::string compiling_name = getCompilingName( name ).c_str();
        char *temp = strdup( compiling_name.c_str() );

        retval = (PyCodeObject*) Py_CompileString( str, temp, Py_file_input );
        if (retval)
            compiled_python.Put( name, retval );
        free( temp );
        free( str );
    }
    return retval;
}

void CompileRunPython

(

const std::string &

filename)

Definition at line 72 of file python_compile.cpp.

References CompilePython(), d, VegaConfig::getVariable(), InterpretPython(), XMLSupport::parse_bool(), and vs_config.

Referenced by ChooseNavPoint(), DockedScript(), PythonAI< SuperClass >::Factory(), PythonClass< PythonMissionBaseClass >::Factory(), RunPython(), RunPythonPhysicsFrame(), RunPythonPress(), RunPythonRelease(), and RunPythonToggle().

{
    static bool ndebug_libs = XMLSupport::parse_bool( vs_config->getVariable( "AI", "compile_python", "true" ) );
    if (ndebug_libs) {
        Python::reseterrors();
        PyCodeObject *CompiledProgram = CompilePython( filename );
        Python::reseterrors();
        if (CompiledProgram) {
            PyObject *m, *d;
            static char main_str[16] = "__main__"; //by chuck_starchaser, to squash a warning
            if ( ( m = PyImport_AddModule( main_str ) ) != NULL ) {
                PyObject *localdict = PyDict_New();
                if ( ( d = PyModule_GetDict( m ) ) != NULL ) {
                    PyObject *exe = PyEval_EvalCode( CompiledProgram, d, localdict );
                    Py_XDECREF( exe );
                    //unref exe?
                }
                Py_XDECREF( localdict );
            }
        }
    } else {
        Python::reseterrors();
        InterpretPython( filename );
        Python::reseterrors();
    }
}

PyObject* CreateTuple

(

const std::vector< PythonBasicType > &

values)

Definition at line 99 of file python_compile.cpp.

References i.

{
    PyObject *retval = PyTuple_New( values.size() );
    for (unsigned int i = 0; i < values.size(); i++) {
        PyObject *val = values[i].NewObject();
        if (val)
            PyTuple_SET_ITEM( retval, i, val );
    }
    return retval;
}

std::string getCompilingName

(

const std::string &

name)

Definition at line 31 of file python_compile.cpp.

References DELIMSTR, and VSFileSystem::homedir.

Referenced by CompilePython(), and InterpretPython().

{
    std::string compiling_name = VSFileSystem::homedir+DELIMSTR+name;
    return compiling_name;
}

void InterpretPython

(

const std::string &

name)

Definition at line 37 of file python_compile.cpp.

References getCompilingName(), VSFileSystem::vs_close(), and VSFileSystem::vs_open().

Referenced by CompileRunPython().

{
    char *temp = strdup( getCompilingName( name ).c_str() );
    FILE *fp   = VSFileSystem::vs_open( name.c_str(), "r" );
    if (fp) {
        PyRun_SimpleFile( fp, temp );
        Python::reseterrors();
        VSFileSystem::vs_close( fp );
    }
    free( temp );
}

char* LoadString

(

const char *

filename)

Definition at line 13 of file python_compile.cpp.

References VSFileSystem::vs_close(), VSFileSystem::vs_fseek(), VSFileSystem::vs_ftell(), VSFileSystem::vs_open(), and VSFileSystem::vs_read().

Referenced by CompilePython().

{
    FILE *fp = VSFileSystem::vs_open( filename, "r" );
    if (!fp)
        return NULL;
    VSFileSystem::vs_fseek( fp, 0, SEEK_END );
    long  len    = VSFileSystem::vs_ftell( fp );
    char *retval = NULL;
    VSFileSystem::vs_fseek( fp, 0, SEEK_SET );
    if (len) {
        retval = (char*) malloc( (len+2)*sizeof (char) );
        len    = VSFileSystem::vs_read( retval, 1, len, fp );
        retval[len] = '\0';
    }
    VSFileSystem::vs_close( fp );
    return retval;
}

static void pySetScratchVector ( const KBSTATE  k)

static

Definition at line 110 of file python_compile.cpp.

References DOWN, PRESS, RELEASE, UniverseUtil::setScratchVector(), UP, and Vector.

Referenced by RunPythonPhysicsFrame(), RunPythonPress(), RunPythonRelease(), and RunPythonToggle().

{
    switch (k)
    {
    case PRESS:
        UniverseUtil::setScratchVector( Vector( 1, 1, 0 ) );
        break;
    case RELEASE:
        UniverseUtil::setScratchVector( Vector( 0, 1, 0 ) );
        break;
    case UP:
        UniverseUtil::setScratchVector( Vector( 0, 0, 0 ) );
        break;
    case DOWN:
        UniverseUtil::setScratchVector( Vector( 1, 0, 0 ) );
        break;
    default:
        break;
    }
}

void RunPythonPhysicsFrame	(	const KBData &	s,
		KBSTATE	k
	)

Definition at line 158 of file python_compile.cpp.

References CompileRunPython(), KBData::data, DOWN, pySetScratchVector(), UniverseUtil::setScratchVector(), UP, and Vector.

Referenced by initGlobalCommandMap().

{
    if ( (k == DOWN || k == UP) && s.data.length() ) {
        pySetScratchVector( k );
        CompileRunPython( s.data );
        UniverseUtil::setScratchVector( Vector( 0, 0, 0 ) );
    }
}

void RunPythonPress	(	const KBData &	s,
		KBSTATE	k
	)

Definition at line 131 of file python_compile.cpp.

References CompileRunPython(), KBData::data, PRESS, pySetScratchVector(), UniverseUtil::setScratchVector(), and Vector.

Referenced by initGlobalCommandMap().

{
    if ( k == PRESS && s.data.length() ) {
        pySetScratchVector( k );
        CompileRunPython( s.data );
        UniverseUtil::setScratchVector( Vector( 0, 0, 0 ) );
    }
}

void RunPythonRelease	(	const KBData &	s,
		KBSTATE	k
	)

Definition at line 140 of file python_compile.cpp.

References CompileRunPython(), KBData::data, pySetScratchVector(), RELEASE, UniverseUtil::setScratchVector(), and Vector.

Referenced by initGlobalCommandMap().

{
    if ( k == RELEASE && s.data.length() ) {
        pySetScratchVector( k );
        CompileRunPython( s.data );
        UniverseUtil::setScratchVector( Vector( 0, 0, 0 ) );
    }
}

void RunPythonToggle	(	const KBData &	s,
		KBSTATE	k
	)

Definition at line 149 of file python_compile.cpp.

References CompileRunPython(), KBData::data, PRESS, pySetScratchVector(), RELEASE, UniverseUtil::setScratchVector(), and Vector.

Referenced by initGlobalCommandMap().

{
    if ( (k == RELEASE || k == PRESS) && s.data.length() ) {
        pySetScratchVector( k );
        CompileRunPython( s.data );
        UniverseUtil::setScratchVector( Vector( 0, 0, 0 ) );
    }
}

Variable Documentation

Hashtable< string, PyCodeObject, 1023 > compiled_python

Definition at line 11 of file python_compile.cpp.