Vladimír Vondruš
3 years ago
11 changed files with 659 additions and 64 deletions
@ -0,0 +1,194 @@
|
||||
#ifndef magnum_accessorsForPixelFormat_h |
||||
#define magnum_accessorsForPixelFormat_h |
||||
/*
|
||||
This file is part of Magnum. |
||||
|
||||
Copyright © 2010, 2011, 2012, 2013, 2014, 2015, 2016, 2017, 2018, 2019, |
||||
2020, 2021, 2022 Vladimír Vondruš <mosra@centrum.cz> |
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a |
||||
copy of this software and associated documentation files (the "Software"), |
||||
to deal in the Software without restriction, including without limitation |
||||
the rights to use, copy, modify, merge, publish, distribute, sublicense, |
||||
and/or sell copies of the Software, and to permit persons to whom the |
||||
Software is furnished to do so, subject to the following conditions: |
||||
|
||||
The above copyright notice and this permission notice shall be included |
||||
in all copies or substantial portions of the Software. |
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL |
||||
THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING |
||||
FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER |
||||
DEALINGS IN THE SOFTWARE. |
||||
*/ |
||||
|
||||
#include <pybind11/pybind11.h> |
||||
#include <Corrade/Containers/Triple.h> |
||||
#include <Magnum/PixelFormat.h> |
||||
#include <Magnum/Math/Color.h> |
||||
#include <Magnum/Math/Half.h> |
||||
#include <Magnum/Math/Packing.h> |
||||
|
||||
#include "Magnum/StridedArrayViewPythonBindings.h" |
||||
|
||||
#include "magnum/bootstrap.h" |
||||
|
||||
/* This is used by both the root magnum module (Image, ImageView) and
|
||||
magnum.trade.ImageData. There's no easy way to call an exported symbol of |
||||
another module due to Python isolating their namespaces so it's duplicated |
||||
in both modules. */ |
||||
namespace magnum { namespace { |
||||
|
||||
Containers::Triple<const char*, py::object(*)(const char*), void(*)(char*, py::handle)> accessorsForPixelFormat(const PixelFormat format) { |
||||
switch(format) { |
||||
#define _c(format, type) \ |
||||
case PixelFormat::format: return { \
|
||||
Containers::Implementation::pythonFormatString<type>(), \
|
||||
[](const char* item) { \
|
||||
return py::cast(*reinterpret_cast<const type*>(item)); \
|
||||
}, \
|
||||
[](char* item, py::handle object) { \
|
||||
*reinterpret_cast<type*>(item) = py::cast<type>(object); \
|
||||
}}; |
||||
/* Types (such as half-floats) that need to be cast before passed
|
||||
from/to pybind that doesn't understand the type directly */ |
||||
#define _cc(format, type, castType) \ |
||||
case PixelFormat::format: return { \
|
||||
Containers::Implementation::pythonFormatString<type>(), \
|
||||
[](const char* item) { \
|
||||
return py::cast(castType(*reinterpret_cast<const type*>(item))); \
|
||||
}, \
|
||||
[](char* item, py::handle object) { \
|
||||
*reinterpret_cast<type*>(item) = type(py::cast<castType>(object)); \
|
||||
}}; |
||||
/* Normalized types that need to be packed/unpacked before passed
|
||||
from/to pybind */ |
||||
#define _cNormalized(format, type, unpackType) \ |
||||
case PixelFormat::format: return { \
|
||||
Containers::Implementation::pythonFormatString<type>(), \
|
||||
[](const char* item) { \
|
||||
return py::cast(Math::unpack<unpackType>(*reinterpret_cast<const type*>(item))); \
|
||||
}, \
|
||||
[](char* item, py::handle object) { \
|
||||
*reinterpret_cast<type*>(item) = Math::pack<type>(py::cast<unpackType>(object)); \
|
||||
}}; |
||||
/* LCOV_EXCL_START */ |
||||
_cNormalized(R8Unorm, UnsignedByte, Float) |
||||
_cNormalized(RG8Unorm, Vector2ub, Vector2) |
||||
_cNormalized(RGB8Unorm, Vector3ub, Vector3) |
||||
_cNormalized(RGBA8Unorm, Vector4ub, Vector4) |
||||
_cNormalized(R8Snorm, Byte, Float) |
||||
_cNormalized(RG8Snorm, Vector2b, Vector2) |
||||
_cNormalized(RGB8Snorm, Vector3b, Vector3) |
||||
_cNormalized(RGBA8Snorm, Vector4b, Vector4) |
||||
/* LCOV_EXCL_STOP */ |
||||
case PixelFormat::R8Srgb: return { |
||||
Containers::Implementation::pythonFormatString<UnsignedByte>(), |
||||
/** @todo have an (internal) API to convert just R/RG sRGB channels */ |
||||
[](const char* item) { |
||||
return py::cast(Color3::fromSrgb(Vector3ub{*reinterpret_cast<const UnsignedByte*>(item), 0, 0}).r()); |
||||
}, |
||||
[](char* item, py::handle object) { |
||||
*reinterpret_cast<UnsignedByte*>(item) = Color3{py::cast<Float>(object), 0.0f, 0.0f}.toSrgb<UnsignedByte>().r(); |
||||
}}; |
||||
case PixelFormat::RG8Srgb: return { |
||||
Containers::Implementation::pythonFormatString<Vector2ub>(), |
||||
/** @todo have an (internal) API to convert just R/RG sRGB channels */ |
||||
[](const char* item) { |
||||
return py::cast(Color3::fromSrgb(Vector3ub{*reinterpret_cast<const Vector2ub*>(item), 0}).rg()); |
||||
}, |
||||
[](char* item, py::handle object) { |
||||
*reinterpret_cast<Vector2ub*>(item) = Color3{py::cast<Vector2>(object), 0.0f}.toSrgb<UnsignedByte>().rg(); |
||||
}}; |
||||
case PixelFormat::RGB8Srgb: return { |
||||
Containers::Implementation::pythonFormatString<Vector3ub>(), |
||||
[](const char* item) { |
||||
return py::cast(Color3::fromSrgb(*reinterpret_cast<const Vector3ub*>(item))); |
||||
}, |
||||
[](char* item, py::handle object) { |
||||
*reinterpret_cast<Vector3ub*>(item) = py::cast<Color3>(object).toSrgb<UnsignedByte>(); |
||||
}}; |
||||
case PixelFormat::RGBA8Srgb: return { |
||||
Containers::Implementation::pythonFormatString<Vector4ub>(), |
||||
[](const char* item) { |
||||
return py::cast(Color4::fromSrgbAlpha(*reinterpret_cast<const Vector4ub*>(item))); |
||||
}, |
||||
[](char* item, py::handle object) { |
||||
*reinterpret_cast<Vector4ub*>(item) = py::cast<Color4>(object).toSrgbAlpha<UnsignedByte>(); |
||||
}}; |
||||
/* LCOV_EXCL_START */ |
||||
_cc(R8UI, UnsignedByte, UnsignedInt) |
||||
_cc(RG8UI, Vector2ub, Vector2ui) |
||||
_cc(RGB8UI, Vector3ub, Vector3ui) |
||||
_cc(RGBA8UI, Vector4ub, Vector4ui) |
||||
_cc(R8I, Byte, Int) |
||||
_cc(RG8I, Vector2b, Vector2i) |
||||
_cc(RGB8I, Vector3b, Vector3i) |
||||
_cc(RGBA8I, Vector4b, Vector4i) |
||||
_cNormalized(R16Unorm, UnsignedShort, Float) |
||||
_cNormalized(RG16Unorm, Vector2us, Vector2) |
||||
_cNormalized(RGB16Unorm, Vector3us, Vector3) |
||||
_cNormalized(RGBA16Unorm, Vector4us, Vector4) |
||||
_cNormalized(R16Snorm, Short, Float) |
||||
_cNormalized(RG16Snorm, Vector2s, Vector2) |
||||
_cNormalized(RGB16Snorm, Vector3s, Vector3) |
||||
_cNormalized(RGBA16Snorm, Vector4s, Vector4) |
||||
_cc(R16UI, UnsignedShort, UnsignedInt) |
||||
_cc(RG16UI, Vector2us, Vector2ui) |
||||
_cc(RGB16UI, Vector3us, Vector3ui) |
||||
_cc(RGBA16UI, Vector4us, Vector4ui) |
||||
_cc(R16I, Short, Int) |
||||
_cc(RG16I, Vector2s, Vector2i) |
||||
_cc(RGB16I, Vector3s, Vector3i) |
||||
_cc(RGBA16I, Vector4s, Vector4i) |
||||
_c(R32UI, UnsignedInt) |
||||
_c(RG32UI, Vector2ui) |
||||
_c(RGB32UI, Vector3ui) |
||||
_c(RGBA32UI, Vector4ui) |
||||
_c(R32I, Int) |
||||
_c(RG32I, Vector2i) |
||||
_c(RGB32I, Vector3i) |
||||
_c(RGBA32I, Vector4i) |
||||
_cc(R16F, Half, Float) |
||||
_cc(RG16F, Vector2h, Vector2) |
||||
_cc(RGB16F, Vector3h, Vector3) |
||||
_cc(RGBA16F, Vector4h, Vector4) |
||||
_c(R32F, Float) |
||||
_c(RG32F, Vector2) |
||||
_c(RGB32F, Vector3) |
||||
_c(RGBA32F, Vector4) |
||||
/* LCOV_EXCL_STOP */ |
||||
#undef _c |
||||
#undef _cc |
||||
#undef _cNormalized |
||||
|
||||
/** @todo handle depth/stencil types (yes, i'm lazy) */ |
||||
default: |
||||
return {}; |
||||
} |
||||
} |
||||
|
||||
template<UnsignedInt dimensions, class T> Containers::StridedArrayView<dimensions - 1, T> flattenPixelView(const Containers::ArrayView<T> data, const Containers::StridedArrayView<dimensions, T>& pixels) { |
||||
/** @todo have some builtin API for this, this is awful (pixels<void>()?
|
||||
flatten<dimensions>() that asserts the last dimension is contiguous and |
||||
of a POD type? transpose<3, 0, 1, 2>()[0], differently named to avoid |
||||
clashes, taking dimension order?) */ |
||||
Containers::Size<dimensions - 1> size{NoInit}; |
||||
Containers::Stride<dimensions - 1> stride{NoInit}; |
||||
for(std::size_t i = 0; i != dimensions - 1; ++i) { |
||||
size[i] = pixels.size()[i]; |
||||
stride[i] = pixels.stride()[i]; |
||||
} |
||||
return Containers::StridedArrayView<dimensions - 1, T>{ |
||||
data, |
||||
static_cast<T*>(pixels.data()), |
||||
size, |
||||
stride}; |
||||
} |
||||
|
||||
}} |
||||
|
||||
#endif |
||||
Binary file not shown.
Loading…
Reference in new issue