Cythonの2つのタイプにC ++でテンプレート化された関数を使用するにはどうすればよいですか？

debugcn 投稿 Dev

チエル

テンプレート化されたC ++関数があり、両方のタイプを使用できるようにしたいと考えています。Pythonはオーバーロードをサポートしていないので、これを解決する方法が少し行き詰まっています。私は.pyx以下に示すように持っています。どのように私は両方でC ++の関数を使用することができますfloatし、double？

import cython
import numpy as np
cimport numpy as np

# declare the interface to the C code
cdef extern from "diff_cpp.cpp" namespace "diff":
    cdef void diff_cpp[float] (float* at, const float* a, const float visc,
                               const float dxidxi, const float dyidyi, const float dzidzi,
                               const int itot, const int jtot, const int ktot)

cdef extern from "diff_cpp.cpp" namespace "diff":
    cdef void diff_cpp[double] (double* at, const double* a, const double visc,
                                const double dxidxi, const double dyidyi, const double dzidzi,
                                const int itot, const int jtot, const int ktot)

@cython.boundscheck(False)
@cython.wraparound(False)
def diff(np.ndarray[double, ndim=3, mode="c"] at not None,
         np.ndarray[double, ndim=3, mode="c"] a not None,
         double visc, double dxidxi, double dyidyi, double dzidzi):
    cdef int ktot, jtot, itot
    ktot, jtot, itot = at.shape[0], at.shape[1], at.shape[2]
    diff_cpp[double](&at[0,0,0], &a[0,0,0], visc, dxidxi, dyidyi, dzidzi, itot, jtot, ktot)
    return None

@cython.boundscheck(False)
@cython.wraparound(False)
def diff_f(np.ndarray[float, ndim=3, mode="c"] at not None,
           np.ndarray[float, ndim=3, mode="c"] a not None,
           float visc, float dxidxi, float dyidyi, float dzidzi):
    cdef int ktot, jtot, itot
    ktot, jtot, itot = at.shape[0], at.shape[1], at.shape[2]
    diff_cpp[float](&at[0,0,0], &a[0,0,0], visc, dxidxi, dyidyi, dzidzi, itot, jtot, ktot)
    return None

ソリューションで更新

@ oz1の答えは、これを行う正しい方法を提供しました。これは、この特定の問題の解決策に関心のある人のために機能するコードです。

import cython
import numpy as np
cimport numpy as np

# declare the interface to the C code
cdef extern from "diff_cpp.cpp" namespace "diff":
    cdef void diff_cpp[T](T* at, const T* a, const T visc,
                          const T dxidxi, const T dyidyi, const T dzidzi,
                          const int itot, const int jtot, const int ktot)

ctypedef fused float_t:
    cython.float
    cython.double

@cython.boundscheck(False)
@cython.wraparound(False)
def diff(np.ndarray[float_t, ndim=3, mode="c"] at not None,
         np.ndarray[float_t, ndim=3, mode="c"] a not None,
         float_t visc, float_t dxidxi, float_t dyidyi, float_t dzidzi):
    cdef int ktot, jtot, itot
    ktot, jtot, itot = at.shape[0], at.shape[1], at.shape[2]
    diff_cpp(&at[0,0,0], &a[0,0,0], visc, dxidxi, dyidyi, dzidzi, itot, jtot, ktot)
    return None

oz1

2つのメモ：

Cythonはc ++テンプレートをサポートしています（http://docs.cython.org/en/latest/src/userguide/wrapping_CPlusPlus.html）
Cythonには融合型があります（http://docs.cython.org/en/latest/src/userguide/fusedtypes.html）

例：

// lib.cpp
template<typename T>
T arr_sum(T *arr, size_t size)
{
    T temp=0;
    for (size_t i=0; i != size; ++i){
        temp += arr[i];
    }
    return temp;
}

# lib_wrapper.pyx
cimport cython

ctypedef fused  float_t:
    cython.float
    cython.double

cdef extern from "lib.cpp" nogil:
    T arr_sum[T](T *arr, size_t size)

def py_arr_sum(float_t[:] arr not None):
    print(sizeof(arr[0]))  # check the element size
    return arr_sum(&arr[0], arr.shape[0])

# use.py
import numpy as np
from lib_wrapper import py_arr_sum

print(py_arr_sum(np.array([1,2,3], dtype=np.float32)))
print(py_arr_sum(np.array([1,2,3], dtype=np.float64)))
print(py_arr_sum(np.array([1,2,3], dtype=np.int32)))  # oops

この記事はインターネットから収集されたものであり、転載の際にはソースを示してください。

侵害の場合は、連絡してください[email protected]