смешайте VTK и SWIG Python - PullRequest
Новая
       12

смешайте VTK и SWIG Python

2 голосов
/ 17 января 2012

Вот мой класс: 

#include <vtkPolyData>

class VTKUtilities

Mesh3D MeshfromVTKPolyData(vtkPolyData* pdmesh)
{
   Mesh3D mesh;
   //...
   //my conversion code to do the actual conversion
   //...
   return mesh;
}

Я попытался обернуть это в python с помощью SWIG. но я пытаюсь вызвать мою функцию в Python так: 

import VTKUtilities
import vtk

pd = vtk.vtkPolyData()
VTKUtilities.MeshfromVTKPolyData(pd)

Я получаю сообщения об ошибках типа: 

NotImplementedError: Wrong number of arguments... for VTKUtilities_MeshfromVTKPolyData
...
Possible prototypes are VTKUtilities::MeshfromVTKPolyData(vtkPolyData *);

Я читал что-то о типографских картах, но я подумал, что мне не нужно возиться с этим, поскольку SWIG должен справиться с этим для меня?
Может кто-нибудь сказать мне, что отсутствует в моем объявлении потока, возможно, какое-то исправление?

Ответы [ 2 ]

4 голосов
/ 09 марта 2012

Мне удается обернуть функции, аргументы которых vtkPolyData, следующим образом:

Сначала вы должны включить vtkPythonUtil в файл swig .i: 

%{

#include <vtkPythonUtil.h>

}%

Затем у вас естьдля сопоставления vtkPolydata в файле swig .i: 

 %typemap(out) vtkPolyData* {

    PyImport_ImportModule("vtk");

    $result = vtkPythonUtil::GetObjectFromPointer ( (vtkPolyData*)$1 );
 }

%typemap(in) vtkPolyData* {

    $1 = (vtkPolyData*) vtkPythonUtil::GetPointerFromObject ( $input, "vtkPolyData" );

    if ( $1 == NULL ) { SWIG_fail; }
}

Я обнаружил, что в ITK itkVTKGlue.

Наконец, вам нужно связать ваш модуль с библиотекой vtkPythonCore

0 голосов
/ 13 июля 2017

Вот vtk.i , предоставляющий карты типов SWIG для классов VTK и перехватчики управления памятью для классов, определенных в вашем проекте, полученных из объектов VTK, обернутых SWIG.

Код

вот код в полном объеме.это проверено с VTK 8 и SWIG 3.7.Обратите внимание, что приведенная выше ссылка содержит примеры и может включать будущие обновления. 

/*
vtk.i a SWIG interface to VTK classes
Copyright (C)  2017 Burlen Loring

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 3 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, see <http://www.gnu.org/licenses/>.
*/
%{
#include <vtkPythonUtil.h>
%}

%include "exception.i"

/*---------------------------------------------------------------------------
macro: VTK_SWIG_INTEROP(vtk_t)

arguments:
  vtk_t - a VTK class name that is used in the SWIG generated API.

The macro defines the typemaps needed for SWIG to convert to and
from VTK's Python bindings. Use this when your API containes pointers
to classes defined in VTK.
---------------------------------------------------------------------------*/
%define VTK_SWIG_INTEROP(vtk_t)
%{
#include <vtk_t##.h>
%}
%typemap(out) vtk_t*
{
  $result = vtkPythonUtil::GetObjectFromPointer(
    static_cast<vtkObjectBase*>($1));
}
%typemap(in) vtk_t*
{
  $1 = static_cast<vtk_t*>(
    vtkPythonUtil::GetPointerFromObject($input,#vtk_t));
  if (!$1)
  {
    SWIG_exception(SWIG_TypeError,
      "an object of type " #vtk_t " is required");
  }
}
%typemap(typecheck, precedence=SWIG_TYPECHECK_POINTER) vtk_t*
{
  $1 = vtkPythonUtil::GetPointerFromObject($input,#vtk_t) ? 1 : 0;
}
%enddef

/*---------------------------------------------------------------------------
macro: VTK_DERIVED(derived_t)

arguments:
  derived_t - name of a class that derives from vtkObjectBase.

The macro causes SWIG to wrap the class and defines memory management hooks
that prevent memory leaks when SWIG creates the objects. Use this to wrap
VTK classes defined in your project.
---------------------------------------------------------------------------*/
%define VTK_DERIVED(derived_t)
%{
#include <derived_t##.h>
%}
%feature("ref") derived_t "$this->Register(nullptr);"
%feature("unref") derived_t "$this->UnRegister(nullptr);"
%newobject derived_t##::New();
%include <derived_t##.h>
%enddef

Если ваш API использует vtkObjectBase и vtkDataObject, ваш SWIG-файл .i будет содержать: 

VTK_SWIG_INTEROP(vtkObjectBase)
VTK_SWIG_INTEROP(vtkDataObject)

Там будетбыть одним вызовом макроса для каждого класса VTK, появляющегося в вашем API.

Пример использования

Если вы определите класс, производный от vtkObject или одного из его подклассов, называемого DataAdaptor, ваш SWIG-файл .i будет включать: 

VTK_DERIVED(DataAdaptor)

Обратите внимание, что вам также нужно будет вызвать VTK_SWIG_INTEROP для любых классов VTK в API вашего класса, включая vtkObjectBase.

...