Gui: Change SelectionFilterPy to new PyCXX extension type

src/Gui/SelectionFilterPy.cpp

@@ -0,0 +1,143 @@
+/***************************************************************************
+ *   Copyright (c) 2009 Jürgen Riegel <FreeCAD@juergen-riegel.net>         *
+ *                                                                         *
+ *   This file is part of the FreeCAD CAx development system.              *
+ *                                                                         *
+ *   This library is free software; you can redistribute it and/or         *
+ *   modify it under the terms of the GNU Library General Public           *
+ *   License as published by the Free Software Foundation; either          *
+ *   version 2 of the License, or (at your option) any later version.      *
+ *                                                                         *
+ *   This library  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 Library General Public License for more details.                  *
+ *                                                                         *
+ *   You should have received a copy of the GNU Library General Public     *
+ *   License along with this library; see the file COPYING.LIB. If not,    *
+ *   write to the Free Software Foundation, Inc., 59 Temple Place,         *
+ *   Suite 330, Boston, MA  02111-1307, USA                                *
+ *                                                                         *
+ ***************************************************************************/
+
+#include <App/DocumentObjectPy.h>
+
+#include "SelectionFilterPy.h"
+
+
+using namespace Gui;
+
+SelectionFilterPy::SelectionFilterPy(Py::PythonClassInstance* self, Py::Tuple& args, Py::Dict& kwds)
+    : Py::PythonClass<SelectionFilterPy>::PythonClass(self, args, kwds), filter("")
+{
+    const char* str;
+    if (!PyArg_ParseTuple(args.ptr(), "s", &str)) {
+        throw Py::Exception();
+    }
+    try {
+        filter = SelectionFilter(str);
+    }
+    catch (const Base::Exception& e) {
+        throw Py::Exception(PyExc_SyntaxError, e.what());
+    }
+}
+
+SelectionFilterPy::~SelectionFilterPy()
+{
+}
+
+Py::Object SelectionFilterPy::repr()
+{
+    std::string s;
+    std::ostringstream s_out;
+    s_out << "SelectionFilter";
+    return Py::String(s_out.str());
+}
+
+Py::Object SelectionFilterPy::match()
+{
+    return Py::Boolean(filter.match());
+}
+PYCXX_NOARGS_METHOD_DECL(SelectionFilterPy, match)
+
+Py::Object SelectionFilterPy::test(const Py::Tuple& args)
+{
+    PyObject * pcObj;
+    char* text=nullptr;
+    if (!PyArg_ParseTuple(args.ptr(), "O!|s",
+        &(App::DocumentObjectPy::Type), &pcObj, &text)) {
+        throw Py::Exception();
+    }
+
+    auto docObj = static_cast<App::DocumentObjectPy*>(pcObj);
+
+    return Py::Boolean(filter.test(docObj->getDocumentObjectPtr(),text));
+}
+PYCXX_VARARGS_METHOD_DECL(SelectionFilterPy, test)
+
+Py::Object SelectionFilterPy::result()
+{
+    Py::List list;
+    for (const auto& vec : filter.Result) {
+        Py::Tuple tuple(vec.size());
+        int index=0;
+        for (auto sel : vec) {
+            tuple[index++] = Py::asObject(sel.getPyObject());
+        }
+        list.append(tuple);
+    }
+
+    return list;
+}
+PYCXX_NOARGS_METHOD_DECL(SelectionFilterPy, result)
+
+Py::Object SelectionFilterPy::setFilter(const Py::Tuple& args)
+{
+    char* text=nullptr;
+    if (!PyArg_ParseTuple(args.ptr(), "s", &text)) {
+        throw Py::Exception();
+    }
+
+    try {
+        filter.setFilter(text);
+        return Py::None();
+    }
+    catch (const Base::Exception& e) {
+        throw Py::Exception(PyExc_SyntaxError, e.what());
+    }
+}
+PYCXX_VARARGS_METHOD_DECL(SelectionFilterPy, setFilter)
+
+Py::Object SelectionFilterPy::getFilter()
+{
+    return Py::String(filter.getFilter());
+}
+PYCXX_NOARGS_METHOD_DECL(SelectionFilterPy, getFilter)
+
+void SelectionFilterPy::init_type()
+{
+    behaviors().name("Gui.SelectionFilter");
+    behaviors().doc("Filter for certain selection\n"
+        "Example strings are:\n"
+        "\"SELECT Part::Feature SUBELEMENT Edge\",\n"
+        "\"SELECT Part::Feature\", \n"
+        "\"SELECT Part::Feature COUNT 1..5\"\n");
+    // you must have overwritten the virtual functions
+    behaviors().supportRepr();
+    behaviors().supportGetattro();
+    behaviors().supportSetattro();
+    PYCXX_ADD_NOARGS_METHOD(match, match,
+        "Check if the current selection matches the filter");
+    PYCXX_ADD_NOARGS_METHOD(result, result,
+        "If match() returns True then with result() you get a list of the matching objects");
+    PYCXX_ADD_VARARGS_METHOD(test, test,
+        "test(Feature, SubName='')\n"
+        "Test if a given object is described in the filter.\n"
+        "If SubName is not empty the sub-element gets also tested.");
+    PYCXX_ADD_VARARGS_METHOD(setFilter, setFilter,
+        "Set a new selection filter from a string");
+    PYCXX_ADD_NOARGS_METHOD(getFilter, getFilter,
+        "Get the selection filter string");
+
+    behaviors().readyType();
+}