You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
tdeedu/kig/scripting/python_scripter.cc

580 lines
18 KiB

// Copyright (C) 2003 Dominique Devriese <devriese@kde.org>
// This program is free software; you can redistribute it and/or
// modify it under the terms of the GNU General Public License
// as published by the Free Software Foundation; either version 2
// of the License, or (at your option) any later version.
// 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, write to the Free Software
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
// 02110-1301, USA.
#include "config.h"
#include "python_scripter.h"
#include <iostream>
#include <string>
#include <Python.h>
#include <boost/python.hpp>
#include <boost/mpl/bool.hpp>
#include "../misc/common.h"
#include "../misc/coordinate.h"
#include "../misc/cubic-common.h"
#include "../misc/kigtransform.h"
#include "../objects/bogus_imp.h"
#include "../objects/common.h"
#include "../objects/circle_imp.h"
#include "../objects/cubic_imp.h"
#include "../objects/line_imp.h"
#include "../objects/other_imp.h"
#include "../objects/point_imp.h"
using namespace boost::python;
BOOST_PYTHON_MODULE_INIT( kig )
{
class_<Coordinate>( "Coordinate" )
.def( init<double, double>() )
.def( init<const Coordinate&>() )
.def( "invalidCoord", &Coordinate::invalidCoord )
.staticmethod( "invalidCoord" )
.def( "valid", &Coordinate::valid )
.def( "distance", &Coordinate::distance )
.def( "length", &Coordinate::length )
.def( "squareLength", &Coordinate::squareLength )
.def( "orthogonal", &Coordinate::orthogonal )
.def( "round", &Coordinate::round )
.def( "normalize", &Coordinate::normalize )
.def( -self )
// .def( self = self )
.def( self += self )
.def( self -= self )
.def( self *= other<double>() )
.def( self *= other<int>() )
.def( self /= other<double>() )
.def( self / other<double>() )
.def( self + self )
.def( self - self )
.def( self * other<double>() )
.def( other<double>() * self )
.def( self * self )
.def_readwrite( "x", &Coordinate::x )
.def_readwrite( "y", &Coordinate::y )
;
class_<LineData>( "LineData" )
.def( init<Coordinate, Coordinate>() )
.def( "dir", &LineData::dir )
.def( "length", &LineData::length )
.def( "isParallelTo", &LineData::isParallelTo )
.def_readwrite( "a", &LineData::a )
.def_readwrite( "b", &LineData::b )
;
// we need this cause Transformation::apply is overloaded and
// otherwise using Transformation::apply would be ambiguous..
const Coordinate (Transformation::*transformapplyfunc)( const Coordinate& ) const = &Transformation::apply;
class_<Transformation>( "Transformation", no_init )
.def( "apply", transformapplyfunc )
.def( "isHomothetic", &Transformation::isHomothetic )
.def( "inverse", &Transformation::inverse )
.def( "identity", &Transformation::identity )
.def( "translation", &Transformation::translation )
.def( "rotation", &Transformation::rotation )
.def( "pointReflection", &Transformation::pointReflection )
.def( "lineReflection", &Transformation::lineReflection )
.def( "castShadow", &Transformation::castShadow )
.def( "projectiveRotation", &Transformation::projectiveRotation )
.def( "scalingOverPoint", &Transformation::scalingOverPoint )
.def( "scalingOverLine", &Transformation::scalingOverLine )
.def( self * self )
.def( self == self )
.staticmethod( "identity" )
.staticmethod( "translation" )
.staticmethod( "rotation" )
.staticmethod( "pointReflection" )
.staticmethod( "lineReflection" )
.staticmethod( "castShadow" )
.staticmethod( "projectiveRotation" )
.staticmethod( "scalingOverPoint" )
.staticmethod( "scalingOverLine" )
;
class_<ObjectImpType, boost::noncopyable>( "ObjectType", no_init )
.def( "fromInternalName", &ObjectImpType::typeFromInternalName,
return_value_policy<reference_existing_object>() )
.staticmethod( "fromInternalName" )
.def( "inherits", &ObjectImpType::inherits )
.def( "internalName", &ObjectImpType::internalName )
.def( "translatedName", &ObjectImpType::translatedName )
.def( "selectStatement", &ObjectImpType::selectStatement )
.def( "removeAStatement", &ObjectImpType::removeAStatement )
.def( "addAStatement", &ObjectImpType::addAStatement )
.def( "moveAStatement", &ObjectImpType::moveAStatement )
.def( "attachToThisStatement", &ObjectImpType::attachToThisStatement )
;
class_<ObjectImp, boost::noncopyable>( "Object", no_init )
.def( "stype", &ObjectImp::stype,
return_value_policy<reference_existing_object>() )
.staticmethod( "stype" )
.def( "inherits", &ObjectImp::inherits )
.def( "transform", &ObjectImp::transform,
return_value_policy<manage_new_object>() )
.def( "valid", &ObjectImp::valid )
.def( "copy", &ObjectImp::copy,
return_value_policy<manage_new_object>() )
.def( "equals", &ObjectImp::equals )
;
class_<CurveImp, bases<ObjectImp>, boost::noncopyable>( "Curve", no_init )
.def( "stype", &CurveImp::stype,
return_value_policy<reference_existing_object>() )
.staticmethod( "stype" )
// .def( "getParam", &CurveImp::getParam )
// .def( "getPoint", &CurveImp::getPoint );
;
class_<PointImp, bases<ObjectImp> >( "Point", init<Coordinate>() )
.def( "stype", &PointImp::stype,
return_value_policy<reference_existing_object>() )
.staticmethod( "stype" )
.def( "coordinate", &PointImp::coordinate,
return_internal_reference<1>() )
.def( "setCoordinate", &PointImp::setCoordinate )
;
class_<AbstractLineImp, bases<CurveImp>, boost::noncopyable >( "AbstractLine", no_init )
.def( "stype", &AbstractLineImp::stype,
return_value_policy<reference_existing_object>() )
.staticmethod( "stype" )
.def( "slope", &AbstractLineImp::slope )
.def( "equationString", &AbstractLineImp::equationString )
.def( "data", &AbstractLineImp::data )
;
class_<SegmentImp, bases<AbstractLineImp> >( "Segment", init<Coordinate, Coordinate>() )
.def( "stype", &SegmentImp::stype,
return_value_policy<reference_existing_object>() )
.staticmethod( "stype" )
.def( init<LineData>() )
.def( "length", &SegmentImp::length )
;
class_<RayImp, bases<AbstractLineImp> >( "Ray", init<Coordinate, Coordinate>() )
.def( "stype", &RayImp::stype,
return_value_policy<reference_existing_object>() )
.staticmethod( "stype" )
.def( init<LineData>() )
;
class_<LineImp, bases<AbstractLineImp> >( "Line", init<Coordinate, Coordinate>() )
.def( "stype", &LineImp::stype,
return_value_policy<reference_existing_object>() )
.staticmethod( "stype" )
.def( init<LineData>() )
;
class_<ConicCartesianData>( "ConicCartesianData", init<double,double,double,double,double,double>() )
.def( init<ConicPolarData>() )
.def( "invalidData", &ConicCartesianData::invalidData )
.staticmethod( "invalidData" )
.def( "valid", &ConicCartesianData::valid )
// .def( init<double[6]>() )
// .def_readwrite( "coeffs", &ConicCartesianData::coeffs )
;
class_<ConicPolarData>( "ConicPolarData", init<Coordinate, double, double, double>() )
.def( init<ConicCartesianData>() )
.def_readwrite( "focus1", &ConicPolarData::focus1 )
.def_readwrite( "pdimen", &ConicPolarData::pdimen )
.def_readwrite( "ecostheta0", &ConicPolarData::ecostheta0 )
.def_readwrite( "esintheta0", &ConicPolarData::esintheta0 )
;
class_<ConicImp, bases<CurveImp>, boost::noncopyable >( "Conic", no_init )
.def( "stype", &ConicImp::stype,
return_value_policy<reference_existing_object>() )
.staticmethod( "stype" )
.def( "conicType", &ConicImp::conicType )
// .def( "conicTypeString", &ConicImp::conicTypeString )
// .def( "cartesianEquationString", &ConicImp::cartesianEquationString )
// .def( "polarEquationString", &ConicImp::polarEquationString )
.def( "cartesianData", &ConicImp::cartesianData )
.def( "polarData", &ConicImp::polarData )
.def( "focus1", &ConicImp::focus1 )
.def( "focus2", &ConicImp::focus2 )
;
class_<ConicImpCart, bases<ConicImp> >( "CartesianConic", init<ConicCartesianData>() )
;
class_<ConicImpPolar, bases<ConicImp> >( "PolarConic", init<ConicPolarData>() )
;
class_<CircleImp, bases<ConicImp> >( "Circle", init<Coordinate, double>() )
.def( "stype", &CircleImp::stype,
return_value_policy<reference_existing_object>() )
.staticmethod( "stype" )
.def( "center", &CircleImp::center )
.def( "radius", &CircleImp::radius )
.def( "squareRadius", &CircleImp::squareRadius )
.def( "surface", &CircleImp::surface )
.def( "circumference", &CircleImp::circumference )
;
class_<VectorImp, bases<CurveImp> >( "Vector", init<Coordinate, Coordinate>() )
.def( "stype", &VectorImp::stype,
return_value_policy<reference_existing_object>() )
.staticmethod( "stype" )
.def( "length", &VectorImp::length )
.def( "dir", &VectorImp::dir )
.def( "data", &VectorImp::data )
;
class_<AngleImp, bases<ObjectImp> >( "Angle", init<Coordinate, double, double>() )
.def( "stype", &AngleImp::stype,
return_value_policy<reference_existing_object>() )
.staticmethod( "stype" )
.def( "size", &AngleImp::size )
.def( "point", &AngleImp::point )
.def( "startAngle", &AngleImp::startAngle )
.def( "angle", &AngleImp::angle )
;
class_<ArcImp, bases<ObjectImp> >( "Arc", init<Coordinate, double, double, double>() )
.def( "stype", &ArcImp::stype,
return_value_policy<reference_existing_object>() )
.staticmethod( "stype" )
.def( "startAngle", &ArcImp::startAngle )
.def( "angle", &ArcImp::angle )
.def( "radius", &ArcImp::radius )
.def( "center", &ArcImp::center )
.def( "firstEndPoint", &ArcImp::firstEndPoint )
.def( "secondEndPoint", &ArcImp::secondEndPoint )
.def( "sectorSurface", &ArcImp::sectorSurface )
;
class_<BogusImp, bases<ObjectImp>, boost::noncopyable >( "BogusObject", no_init )
.def( "stype", &BogusImp::stype,
return_value_policy<reference_existing_object>() )
.staticmethod( "stype" )
;
class_<InvalidImp, bases<BogusImp> >( "InvalidObject", init<>() )
.def( "stype", &InvalidImp::stype,
return_value_policy<reference_existing_object>() )
.staticmethod( "stype" )
;
class_<DoubleImp, bases<BogusImp> >( "DoubleObject", init<double>() )
.def( "stype", &DoubleImp::stype,
return_value_policy<reference_existing_object>() )
.staticmethod( "stype" )
.def( "data", &DoubleImp::data )
.def( "setData", &DoubleImp::setData )
;
class_<IntImp, bases<BogusImp> >( "IntObject", init<int>() )
.def( "stype", &IntImp::stype,
return_value_policy<reference_existing_object>() )
.staticmethod( "stype" )
.def( "data", &IntImp::data )
.def( "setData", &IntImp::setData )
;
class_<StringImp, bases<BogusImp> >( "StringObject", no_init )
.def( "stype", &StringImp::stype,
return_value_policy<reference_existing_object>() )
.staticmethod( "stype" )
// .def( "data", &StringImp::data )
// .def( "setData", &StringImp::setData )
;
class_<TestResultImp, bases<BogusImp> >( "TestResultObject", no_init )
.def( "stype", &TestResultImp::stype,
return_value_policy<reference_existing_object>() )
.staticmethod( "stype" )
// .def( "data", &TestResultImp::data )
;
// class_<TextImp, bases<ObjectImp> >( "Text", init<string, Coordinate, bool>() )
// .def( "stype", &TextImp::stype,
// return_value_policy<reference_existing_object>() )
// .staticmethod( "stype" )
// .def( "text", &TextImp::text )
// .def( "coordinate", &TextImp::coordinate )
// .def( "hasFrame", &TextImp::hasFrame )
// ;
class_<CubicCartesianData>( "CubicCartesianData", init<double,double,double,double,double,double,double,double,double,double>() )
.def( "invalidData", &CubicCartesianData::invalidData )
.staticmethod( "invalidData" )
.def( "valid", &CubicCartesianData::valid )
// .def( init<double[10]>() )
// .def_readwrite( "coeffs", &CubicCartesianData::coeffs )
;
class_<CubicImp, bases<CurveImp> >( "Cubic", init<CubicCartesianData>() )
.def( "stype", &CubicImp::stype,
return_value_policy<reference_existing_object>() )
.staticmethod( "stype" )
.def( "data", &CubicImp::data )
;
}
PythonScripter* PythonScripter::instance()
{
static PythonScripter t;
return &t;
}
class PythonScripter::Private
{
public:
dict mainnamespace;
};
// allocates a new string using new [], and copies contents into it..
static char* newstring( const char* contents )
{
char* ret = new char[strlen( contents ) + 1];
strcpy( ret, contents );
return ret;
}
PythonScripter::PythonScripter()
{
d = new Private;
// tell the python interpreter about our API..
// the newstring stuff is to prevent warnings about conversion from
// const char* to char*..
char* s = newstring( "kig" );
PyImport_AppendInittab( s, KIG_Python_init );
// we can't delete this yet, since python keeps a pointer to it..
// This means we have a small but harmless memory leak here, but it
// doesn't hurt at all, since it could only be freed at the end of
// the program, at which time it is freed by the system anyway if we
// don't do it..
//delete [] s;
Py_Initialize();
s = newstring( "import math; from math import *;" );
PyRun_SimpleString( s );
delete [] s;
s = newstring( "import kig; from kig import *;" );
PyRun_SimpleString( s );
delete [] s;
s = newstring( "import traceback;" );
PyRun_SimpleString( s );
delete [] s;
// find the main namespace..
s = newstring( "__main__" );
handle<> main_module( borrowed( PyImport_AddModule( s ) ) );
delete [] s;
handle<> mnh(borrowed( PyModule_GetDict(main_module.get()) ));
d->mainnamespace = extract<dict>( mnh.get() );
}
PythonScripter::~PythonScripter()
{
PyErr_Clear();
Py_Finalize();
delete d;
}
class CompiledPythonScript::Private
{
public:
int ref;
object calcfunc;
// TODO
// object movefunc;
};
ObjectImp* CompiledPythonScript::calc( const Args& args, const KigDocument& )
{
return PythonScripter::instance()->calc( *this, args );
}
CompiledPythonScript::~CompiledPythonScript()
{
--d->ref;
if ( d->ref == 0 )
delete d;
}
CompiledPythonScript::CompiledPythonScript( Private* ind )
: d( ind )
{
++d->ref;
}
CompiledPythonScript PythonScripter::compile( const char* code )
{
clearErrors();
dict retdict;
bool error = false;
try
{
(void) PyRun_String( const_cast<char*>( code ), Py_file_input,
d->mainnamespace.ptr(), retdict.ptr() );
}
catch( ... )
{
error = true;
};
error |= static_cast<bool>( PyErr_Occurred() );
if ( error )
{
saveErrors();
retdict.clear();
}
// debugging stuff, removed.
// std::string dictstring = extract<std::string>( str( retdict ) );
CompiledPythonScript::Private* ret = new CompiledPythonScript::Private;
ret->ref = 0;
ret->calcfunc = retdict.get( "calc" );
return CompiledPythonScript( ret );
}
CompiledPythonScript::CompiledPythonScript( const CompiledPythonScript& s )
: d( s.d )
{
++d->ref;
}
std::string PythonScripter::lastErrorExceptionType() const
{
return lastexceptiontype;
}
std::string PythonScripter::lastErrorExceptionValue() const
{
return lastexceptionvalue;
}
std::string PythonScripter::lastErrorExceptionTraceback() const
{
return lastexceptiontraceback;
}
ObjectImp* PythonScripter::calc( CompiledPythonScript& script, const Args& args )
{
clearErrors();
object calcfunc = script.d->calcfunc;
try
{
std::vector<object> objectvect;
objectvect.reserve( args.size() );
for ( int i = 0; i < (int) args.size(); ++i )
{
object o( boost::ref( *args[i] ) );
objectvect.push_back( o );
}
handle<> argstuph( PyTuple_New( args.size() ) );
for ( int i = 0; i < (int) objectvect.size(); ++i )
{
PyTuple_SetItem( argstuph.get(), i, (objectvect.begin() +i)->ptr() );
};
tuple argstup( argstuph );
handle<> reth( PyEval_CallObject( calcfunc.ptr(), argstup.ptr() ) );
// object resulto = calcfunc( argstup );
// handle<> reth( PyEval_CallObject( calcfunc.ptr(), args ) );
object resulto( reth );
extract<ObjectImp&> result( resulto );
if( ! result.check() ) return new InvalidImp;
else
{
ObjectImp& ret = result();
return ret.copy();
};
}
catch( ... )
{
saveErrors();
return new InvalidImp;
};
}
void PythonScripter::saveErrors()
{
erroroccurred = true;
PyObject* poexctype;
PyObject* poexcvalue;
PyObject* poexctraceback;
PyErr_Fetch( &poexctype, &poexcvalue, &poexctraceback );
handle<> exctypeh( poexctype );
handle<> excvalueh( poexcvalue );
object exctype( exctypeh );
object excvalue( excvalueh );
object exctraceback;
if ( poexctraceback )
{
handle<> exctracebackh( poexctraceback );
exctraceback = object( exctracebackh );
}
lastexceptiontype = extract<std::string>( str( exctype ) )();
lastexceptionvalue = extract<std::string>( str( excvalue ) )();
object printexcfunc = d->mainnamespace[ "traceback" ].attr( "format_exception" );
list tracebacklist = extract<list>( printexcfunc( exctype, excvalue, exctraceback ) )();
str tracebackstr( "" );
while ( true )
{
try {
str s = extract<str>( tracebacklist.pop() );
tracebackstr += s;
}
catch( ... )
{
break;
}
}
lastexceptiontraceback = extract<std::string>( tracebackstr )();
PyErr_Clear();
}
void PythonScripter::clearErrors()
{
PyErr_Clear();
lastexceptiontype.clear();
lastexceptionvalue.clear();
lastexceptiontraceback.clear();
erroroccurred = false;
}
bool CompiledPythonScript::valid()
{
return !!d->calcfunc;
}
bool PythonScripter::errorOccurred() const
{
return erroroccurred;
}