Function texture_to_numpy

Source/Falcor/Core/API/Texture.cpp:756–804

* Python binding wrapper for returning the content of a texture as a numpy array. */

 * Python binding wrapper for returning the content of a texture as a numpy array.
 */
inline pybind11::ndarray<pybind11::numpy> texture_to_numpy(const Texture& self, uint32_t mip_level, uint32_t array_slice)
{
    FALCOR_CHECK(
        mip_level < self.getMipCount(), "'mip_level' ({}) is out of bounds. Only {} level(s) available.", mip_level, self.getMipCount()
    );
    FALCOR_CHECK(
        array_slice < self.getArraySize(),
        "'array_slice' ({}) is out of bounds. Only {} slice(s) available.",
        array_slice,
        self.getArraySize()
    );

    // Get image dimensions.
    uint32_t width = self.getWidth(mip_level);
    uint32_t height = self.getHeight(mip_level);
    uint32_t depth = self.getDepth(mip_level);

    uint32_t subresource = self.getSubresourceIndex(array_slice, mip_level);
    Texture::SubresourceLayout layout = self.getSubresourceLayout(subresource);

    size_t subresourceSize = layout.getTotalByteSize();
    void* cpuData = new uint8_t[subresourceSize];
    self.getSubresourceBlob(subresource, cpuData, subresourceSize);

    pybind11::capsule owner(cpuData, [](void* p) noexcept { delete[] reinterpret_cast<uint8_t*>(p); });

    if (auto dtype = resourceFormatToDtype(self.getFormat()))
    {
        uint32_t channelCount = getFormatChannelCount(self.getFormat());
        std::vector<pybind11::size_t> shape;
        if (depth > 1)
            shape.push_back(depth);
        if (height > 1)
            shape.push_back(height);
        shape.push_back(width);
        if (channelCount > 1)
            shape.push_back(channelCount);
        return pybind11::ndarray<pybind11::numpy>(
            cpuData, shape.size(), shape.data(), owner, nullptr, *dtype, pybind11::device::cpu::value
        );
    }
    else
    {
        pybind11::size_t shape[1] = {subresourceSize};
        return pybind11::ndarray<pybind11::numpy>(
            cpuData, 1, shape, owner, nullptr, pybind11::dtype<uint8_t>(), pybind11::device::cpu::value
        );
    }
}

inline void texture_from_numpy(Texture& self, pybind11::ndarray<pybind11::numpy> data, uint32_t mip_level, uint32_t array_slice)
{