jyapayne/swig
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fixed an overload error with the complex type (missing typecheck)

Browse source 
git-svn-id: https://swig.svn.sourceforge.net/svnroot/swig/trunk/SWIG@4289 626c5289-ae23-0410-ae9c-e8d60b6d4f22
This commit is contained in:
Marcelo Matus 2003-02-11 01:48:54 +00:00
commit 2438a54b2a
1 changed files with 56 additions and 40 deletions
Download patch file Download diff file Expand all files Collapse all files

96
Lib/python/std_complex.i
View file

@ -5,59 +5,75 @@
%{ %{
#include <complex> #include <complex>
%}
namespace std static inline int
SwigComplex_Check(PyObject *o)
{ {
template <class T> class complex; return (PyComplex_Check(o) || PyFloat_Check(o) || PyInt_Check(o)) ? 1 : 0;
}
template <class __Complex>
__Complex
SwigComplex_As(PyObject *o)
{
if (PyComplex_Check(o)) {
return __Complex(PyComplex_RealAsDouble(o), PyComplex_ImagAsDouble(o));
} else if (PyFloat_Check(o)) {
return __Complex(PyFloat_AsDouble(o), 0);
} else if (PyInt_Check(o)) {
return __Complex(PyInt_AsLong(o), 0);
} else {
PyErr_SetString(PyExc_TypeError, "Expecting a complex or compatible type");
return __Complex(0,0);
}
}
static inline std::complex<double>
SwigComplex_AsComplexDouble(PyObject *o)
{
return SwigComplex_As<std::complex<double> >(o);
}
%}
%define swig_specialize_complex(Complex)
%define specialize_std_complex(T) %typecheck(SWIG_TYPECHECK_COMPLEX)
Complex, const Complex&
%typemap(in) complex<T> { {
if (PyComplex_Check($input)) { $1 = SwigComplex_Check($input);
$1 = std::complex<T>(PyComplex_RealAsDouble($input),
PyComplex_ImagAsDouble($input));
} else if (PyFloat_Check($input)) {
$1 = std::complex<T>(PyFloat_AsDouble($input), 0);
} else if (PyInt_Check($input)) {
$1 = std::complex<T>(PyInt_AsLong($input), 0);
}
else {
PyErr_SetString(PyExc_TypeError,"Expected a complex");
SWIG_fail;
}
}
%typemap(in) const complex<T>& (std::complex<T> temp) {
if (PyComplex_Check($input)) {
temp = std::complex<T>(PyComplex_RealAsDouble($input),
PyComplex_ImagAsDouble($input));
$1 = &temp;
} else if (PyFloat_Check($input)) {
temp = std::complex<T>(PyFloat_AsDouble($input), 0);
$1 = &temp;
} else if (PyInt_Check($input)) {
temp = std::complex<T>(PyInt_AsLong($input), 0);
$1 = &temp;
} else {
PyErr_SetString(PyExc_TypeError,"Expected a complex");
SWIG_fail;
}
} }
%typemap(out) complex<T> { %typemap(in) Complex {
$1 = SwigComplex_As< Complex >($input);
if (PyErr_Occurred()) SWIG_fail;
}
%typemap(in) const Complex& (Complex temp) {
temp = SwigComplex_As< Complex >($input);
if (PyErr_Occurred()) SWIG_fail;
$1 = &temp;
}
%typemap(out) Complex {
$result = PyComplex_FromDoubles($1.real(), $1.imag()); $result = PyComplex_FromDoubles($1.real(), $1.imag());
} }
%typemap(out) const complex<T> & { %typemap(out) const Complex & {
$result = PyComplex_FromDoubles($1->real(), $1->imag()); $result = PyComplex_FromDoubles($1->real(), $1->imag());
} }
%enddef %enddef
specialize_std_complex(double); namespace std
specialize_std_complex(float); {
template <class T> class complex;
} }
swig_specialize_complex(std::complex<float>);
swig_specialize_complex(std::complex<double>);
#endif // SWIG #endif // SWIG