Copy the KDE 3.5 branch to branches/trinity for new KDE 3.5 features.

BUG:215923


git-svn-id: svn://anonsvn.kde.org/home/kde/branches/trinity/kdeedu@1054174 283d02a7-25f6-0310-bc7c-ecb5cbfe19da

1 files changed, 578 insertions, 0 deletions

diff --git a/kig/scripting/python_scripter.cc b/kig/scripting/python_scripter.cc
new file mode 100644
index 00000000..7e833d5c
--- /dev/null
+++ b/kig/scripting/python_scripter.cc
@@ -0,0 +1,578 @@
+// 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 "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, initkig );
+  //  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;
+}
+