LineSets/optimizeCP.cpp at main · blackbird2003/LineSets · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
#include "optimizeCP.h"

#include <QDebug>


std::tuple<QPointF, QPointF, double> OptimizeControlPoints(int mode, double optPara, QPointF LP0, QPointF LP1, QPointF LP2, QPointF P0, QPointF P1, QPointF P2, QPointF P3) {
   // return std::make_tuple(QPointF(0,0), QPointF(0,0), 0);

    PyGILState_STATE gilState = PyGILState_Ensure();

    // 设置Python模块搜索路径
    PyObject* sysPath = PySys_GetObject("path");
    PyList_Append(sysPath, PyUnicode_FromString("D:\\Documents\\QtProject\\LineSets"));

    // 导入模块
    PyObject* pName = PyUnicode_FromString("optimize_control_points");
    PyObject* pModule = PyImport_Import(pName);
    Py_DECREF(pName);

    if (pModule == nullptr) {
        PyErr_Print();
        PyGILState_Release(gilState);
        throw std::runtime_error("Failed to load \"optimize_control_points\"");
    }

    //qDebug() << "pwp";
    // 获取函数
    PyObject* pFunc = PyObject_GetAttrString(pModule, "optimize_control_points");
    if (!pFunc || !PyCallable_Check(pFunc)) {
        if (PyErr_Occurred())
            PyErr_Print();
        Py_DECREF(pModule);
        PyGILState_Release(gilState);
        throw std::runtime_error("Cannot find function optimize_control_points");
    }

    // 构建参数
    PyObject* pArgs = PyTuple_Pack(9,
                                   Py_BuildValue("i", mode),
                                   Py_BuildValue("d", optPara),
                                   Py_BuildValue("[dd]", LP0.x(), LP0.y()),
                                   Py_BuildValue("[dd]", LP1.x(), LP1.y()),
                                   Py_BuildValue("[dd]", LP2.x(), LP2.y()),
                                   Py_BuildValue("[dd]", P0.x(), P0.y()),
                                   Py_BuildValue("[dd]", P1.x(), P1.y()),
                                   Py_BuildValue("[dd]", P2.x(), P2.y()),
                                   Py_BuildValue("[dd]", P3.x(), P3.y()));
    //qDebug() << "qwq";
    // 调用函数
    PyObject* pValue = PyObject_CallObject(pFunc, pArgs);
    Py_DECREF(pArgs);
    Py_DECREF(pFunc);
    Py_DECREF(pModule);

    if (pValue == nullptr) {
        PyErr_Print();
        PyGILState_Release(gilState);
        throw std::runtime_error("Call to optimize_control_points failed");
    }

    // 处理返回值
    double p1x = PyFloat_AsDouble(PyTuple_GetItem(pValue, 0));
    double p1y = PyFloat_AsDouble(PyTuple_GetItem(pValue, 1));
    double p2x = PyFloat_AsDouble(PyTuple_GetItem(pValue, 2));
    double p2y = PyFloat_AsDouble(PyTuple_GetItem(pValue, 3));
    double cost = PyFloat_AsDouble(PyTuple_GetItem(pValue, 4));

    Py_DECREF(pValue);
    PyGILState_Release(gilState);
    // qDebug() << P0 << P1 << P2 << P3;
    auto np1 = QPointF(p1x, p1y), np2 = QPointF(p2x, p2y);
    qDebug() << P0 << np1 << np2 << P3 << cost;
    return std::make_tuple(np1, np2, cost);
}