rs_simplepython.cpp

qcad2.05可用于windows和linux的源码

/****************************************************************************
** $Id: rs_simplepython.cpp 1775 2003-11-02 12:20:46Z andrew $
**
** Copyright (C) 2001-2003 RibbonSoft. All rights reserved.
**
** This file is part of the qcadlib Library project.
**
** This file may be distributed and/or modified under the terms of the
** GNU General Public License version 2 as published by the Free Software
** Foundation and appearing in the file LICENSE.GPL included in the
** packaging of this file.
**
** Licensees holding valid qcadlib Professional Edition licenses may use 
** this file in accordance with the qcadlib Commercial License
** Agreement provided with the Software.
**
** This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
** WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
**
** See http://www.ribbonsoft.com for further details.
**
** Contact [e-mail] if any conditions of this licensing are
** not clear to you.
**
**********************************************************************/


#ifdef RS_OPT_SIMPLEPYTHON
#include "rs_simplepython.h"


RS_SimplePython* RS_SimplePython::uniqueInstance = NULL;


/**
 *  Gets the one and only RS_SimplePython instance
 *  (creates a new one on first call only)
 *
 *  @return Pointer to the single instance of this
 * singleton class
 */
RS_SimplePython* RS_SimplePython::instance() {
    if(uniqueInstance==NULL) {
        uniqueInstance = new RS_SimplePython;
    }
    return uniqueInstance;
}


/**
 * Launches the given script.
 */
int RS_SimplePython::launch(const RS_String& script) {
    long answer;
    PyObject *modname, *mod, *mdict, *func, *rslt;
    //Py_SetProgramName(argv[0]);
    Py_Initialize();
    init_pyextension();
    modname = PyString_FromString(script);
    mod = PyImport_Import(modname);
    if (mod) {
        //printf( "mod\n");
        mdict = PyModule_GetDict(mod);

        // Borrowed reference to start function
        func = PyDict_GetItemString(mdict, "start");
        if (func) {
            //printf( "func\n");
            if (PyCallable_Check(func)) {
                //printf("calling..\n");
                rslt = PyObject_CallFunction(func, "(s)", "noparam");
                //printf("calling ok\n");
                if (rslt) {
                    //printf("c: rslt\n");
                    answer = PyInt_AsLong(rslt);
                    //printf("c: answer is: %ld\n", answer);
                    Py_XDECREF(rslt);
                }
            }
        } else {
            printf("no such function: start\n");
        }
        Py_XDECREF(mod);
    } else {
        printf("no such module: %s\n", script.latin1());
    }
    Py_XDECREF(modname);
    Py_Finalize();
    return 0;
}


/**
 * A test method exposed to Python 
 */
long inc(long i) {
    printf("c: inc called\n");
    printf("c: parameter from python: %ld\n", i);
    return ++i;
}

/**
 * The magic that exposes inc(). A wrapper function.
 */
static PyObject *py_inc(PyObject* /*self*/, PyObject* args) {
    long i;
    printf("c: py_inc called\n");
    if (!PyArg_ParseTuple(args, "l", &i))
        return NULL;
    return Py_BuildValue("l", inc(i));
}

/**
 * Adds a line to the current graphic document.
 */
void rsPyAddLine(double x1, double y1, double x2, double y2) {
    //printf("c: addLine called\n");
    //printf("c: parameter from python: %f\n", x1);

    RS_Graphic* graphic = RS_SIMPLEPYTHON->getGraphic();
    if (graphic!=NULL) {
        graphic->addEntity(new RS_Line(graphic,
                                       RS_LineData(RS_Vector(x1, y1),
                                                   RS_Vector(x2, y2))));
    } else {
        std::cerr << "rsPyAddLine: No graphic object set.\n";
    }
}

/**
 * Python wrapper.
 */
static PyObject *py_rsPyAddLine(PyObject* /*self*/, PyObject* args) {
    double x1, y1, x2, y2;
    //printf("c: py_rsPyAddLine called\n");
    if (!PyArg_ParseTuple(args, "dddd", &x1, &y1, &x2, &y2)) {
        return NULL;
    }
    rsPyAddLine(x1, y1, x2, y2);
    return Py_BuildValue("d", 1);
}

/**
 * The qcadlib module's function table.
 */
static PyMethodDef rsQCadMethods[] =
    {
        {"inc",     py_inc,     1,
         "a silly example method"},
        {"rsPyAddLine", py_rsPyAddLine, 1,
         "adds a line to the current document"},
        {NULL,      NULL}       /* sentinel */
    };

/**
 * Python will call this when the qcadlib module is imported.
 */
void init_pyextension() {
    printf("c: adding module: qcad\n");
    PyImport_AddModule("qcad");
    Py_InitModule("qcad", rsQCadMethods);
    printf("c: module qcad: OK\n");
}

#endif