SWIG для расширения C ++ до Python, неопределенное перечисление - PullRequest
1 голос
/ 26 марта 2020

У меня проблемы с расширением / выставлением моего перечисления в C ++ с помощью SWIG до Python.

Сначала я создал уменьшенную версию своего класса "MyClass" без перечислений и получил ее работу, см. Ниже:

Файл интерфейса SWIG (MyClassMini.i):

%module MyClassMini
%{
#include "MyClassMini.h"
#include <stdio.h>
#include
#include
using namespace std;
%}
%include "std_string.i"
%include "MyClassMini.h"

Выполнение команд для автоматического создания оболочек SWIG, компиляции и ссылки:

swig -python -c++ MyClassMini.i

Компиляция и компоновка:

python setup.py build_ext --inplace

Загрузка модуля в новый модуль в Python, создание и установка строки:

Вывод:

>Python
Type "help", "copyright", "credits" or "license" for more information.
>>> import _MyClassMini
>>> MyClassMini =  _MyClassMini.new_MyClassMini();
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: Wrong number or type of arguments for overloaded function 'new_MyClassMini'.
  Possible C/C++ prototypes are:
    MyClassMini::MyClassMini(int)
    MyClassMini::MyClassMini(int,int)
    MyClassMini::MyClassMini(int,int,int)
    MyClassMini::MyClassMini(int,int,int,int)
>>> MyClassMini =  _MyClassMini.new_MyClassMini(7);
>>> _MyClassMini.delete_MyClassMini(MyClassMini);
>>> MyClassMini =  _MyClassMini.new_MyClassMini(7);
>>> _MyClassMini.MyClassMini_getType(MyClassMini);
7
>>> _MyClassMini.MyClassMini_getType(MyClassMini);
7
>>> _MyClassMini.MyClassMini_setValueString(MyClassMini,"Im a string");
>>> _MyClassMini.MyClassMini_getValueString(MyClassMini);          
'Im a string'
>>> _MyClassMini.delete_MyClassMini(MyClassMini);

Все работает нормально.

Проблема с перечислениями ....

Я повторил тот же самый рецепт SWIG для моего настоящего полноценного класса «MyClass», и единственная проблема, с которой я столкнулся, была с перечислениями:

# 1 попытка:

MyClass.i:

%module MyClass
%{
#include "MyClass.h"
#include <stdio.h>
#include <iostream>
#include <cstdlib>
#include "SparseDataType.h"
using namespace std; 
%}

%include "std_string.i" 
%include "MyClass.h"
%include "SparseDataType.h"

Файл интерфейса SWIG SparseDataType.i :

%module SparseDataType
%{
  enum SparseDataType
        {
            SparseBool,
            SparseChar,
            SparseByte, // unsigned char
            SparseInt,
            SparseShort,
            SparseFloat,
            SparseDouble,
            SparseString,
            SparseComposite
        };
%}

Некоторый фрагмент MyClass.h:

#ifndef MyClass_H
#define MyClass_H
#include <stdio.h>
#include <iostream>
#include <string.h>
#include "SparseDataType.h"
using namespace std;

class MyClass
{
public:

        /*constructors:*/
               /*if type is scaler, assume size 1
         n is size (if type is scalar, ignore size)*/
               MyClass (SparseDataType SparseDataType)
        {
             size = _n3 = _n2 = _n1= 1;  
            _type =  SparseDataType;
             alloc();
        }
               MyClass  (SparseDataType SparseDataType, int size_n1)
        {
            size = _n1= size_n1; 
            _n2 = _n3 = 1; 
           _type =  SparseDataType;
           alloc();
        }
        // 2d array
               MyClass (SparseDataType SparseDataType, int size_n2 , int size_n1)  
        {
            _n1 =  size_n1; 
            _n2 = size_n2; 
            _n3 = 1; 
            size =  _n2 * _n1;  
            _type =  SparseDataType;
            alloc();
        }
        // 3d array
               MyClass  (SparseDataType SparseDataType, int size_n3 , int size_n2, int size_n1)
        {
             _n3 = size_n3;
             _n2 = size_n2;  
             _n1 = size_n1;  
            size =  _n3 * _n2 * _n1;  
            _type =  SparseDataType;
             alloc();
        }
//etc…       
}

I edited setup.py to  include SparseDataType:

from distutils.core import setup, Extension


MyClass_module = Extension('_MyClass',
                           sources=['MyClass_wrap.cxx'],
                           )


SparseDataType_module = Extension('_SparseDataType',
                           sources=['SparseDataType_wrap.cxx'],
                           )

setup (name = 'MyClass',
       version = '0.1',
       author      = "SWIG Docs",
       description = """Simple swig MyClass from docs""",
       ext_modules = [MyClass_module,SparseDataType_module],
       py_modules = ["MyClass"],
       )

**Then I ran SWIG wrapper auto-generation commands again & compilation & linking  again:**

swig -python -c++ MyClass.i

swig -python -c++ SparseDataType.i

python setup.py build_ext –inplace

Пока ошибок нет, все скомпилировано. Импорт в python прошел нормально, но при создании экземпляра возникли проблемы (тип перечисления не разрешен):

devlinux{user1}% python
Python 2.7.5 (default, Aug  2 2016, 04:20:16)
[GCC 4.8.5 20150623 (Red Hat 4.8.5-4)] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> import _MyClass
>>> import _SparseDataType
>>> MyClass = _MyClass.new_MyClass(_SparseDataType.SparseInt);
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
AttributeError: 'module' object has no attribute 'SparseInt'

Попытка № 2 изменена,% module SparseDataType.i (на основе on Wrapping C -enum в модуле Python с Swig ):

%module SparseDataType
%inline %{

    struct mySparseDataType {

        enum {
            SparseBool,
            SparseChar,
            SparseByte, // unsigned char
            SparseInt,
            SparseShort,
            SparseFloat,
            SparseDouble,
            SparseString,
            SparseComposite
        };
    };
%}

Затем я перезапускаю команды SWIG:

swig -python -c++ MyClass.i

swig -python -c++ SparseDataType.i

python setup.py build_ext –inplace

Пока ошибок нет, все скомпилировано.

devlinux{user1}% python
>>> import _MyClass                                                    
>>> import _SparseDataType                                                
 >>> MyClass = _MyClass.new_MyClass(_SparseDataType.mySparseDataType.SparseInt);
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
AttributeError: 'module' object has no attribute 'mySparseDataType'
>>> MyClass = _MyClass.new_MyClass(_SparseDataType.SparseInt);         Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
AttributeError: 'module' object has no attribute 'SparseInt'
>>> MyClass = _MyClass.new_MyClass(mySparseDataType.SparseInt);        Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
NameError: name 'mySparseDataType' is not defined

Снова не удалось разрешить enum "undefined" ??

Любые идеи о том, как вызвать enum из Python?

Спасибо за миллион!

1 Ответ

0 голосов
/ 27 марта 2020

import SparseDataType не import _SparseDataType. Последний импортирует _SparseDataType.pyd напрямую и обходит оболочку SparseDataType.py, предназначенную для импорта.

Тогда SparseDataType.mySparseDataType.SparseInt получит доступ к значению перечисления. enum находится внутри структуры, поэтому вам нужно module.structure.enum для доступа к нему.

...