mirror of
https://github.com/pybind/pybind11.git
synced 2026-05-13 01:36:21 +00:00
66c9a40213
This modification taps into some newer C++14 features (if present) to generate function signatures considerably more efficiently at compile time rather than at run time. With this change, pybind11 binaries are now *2.1 times* smaller compared to the Boost.Python baseline in the benchmark. Compilation times get a nice improvement as well. Visual Studio 2015 unfortunately doesn't implement 'constexpr' well enough yet to support this change and uses a runtime fallback.
41 lines
1.1 KiB
C++
41 lines
1.1 KiB
C++
/*
|
|
pybind11/complex.h: Complex number support
|
|
|
|
Copyright (c) 2015 Wenzel Jakob <wenzel@inf.ethz.ch>
|
|
|
|
All rights reserved. Use of this source code is governed by a
|
|
BSD-style license that can be found in the LICENSE file.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "pybind11.h"
|
|
#include <complex>
|
|
|
|
NAMESPACE_BEGIN(pybind11)
|
|
|
|
PYBIND11_DECL_FMT(std::complex<float>, "Zf");
|
|
PYBIND11_DECL_FMT(std::complex<double>, "Zd");
|
|
|
|
NAMESPACE_BEGIN(detail)
|
|
template <typename T> class type_caster<std::complex<T>> {
|
|
public:
|
|
bool load(PyObject *src, bool) {
|
|
Py_complex result = PyComplex_AsCComplex(src);
|
|
if (result.real == -1.0 && PyErr_Occurred()) {
|
|
PyErr_Clear();
|
|
return false;
|
|
}
|
|
value = std::complex<T>((T) result.real, (T) result.imag);
|
|
return true;
|
|
}
|
|
|
|
static PyObject *cast(const std::complex<T> &src, return_value_policy /* policy */, PyObject * /* parent */) {
|
|
return PyComplex_FromDoubles((double) src.real(), (double) src.imag());
|
|
}
|
|
|
|
PYBIND11_TYPE_CASTER(std::complex<T>, _("complex"));
|
|
};
|
|
NAMESPACE_END(detail)
|
|
NAMESPACE_END(pybind11)
|