2a8ebc4906
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
97 lines
3.7 KiB
Plaintext
97 lines
3.7 KiB
Plaintext
#include <mln/util/image+MLNAdditions.hpp>
|
|
|
|
#import <ImageIO/ImageIO.h>
|
|
|
|
#import "CFHandle.hpp"
|
|
|
|
using CGImageHandle = CFHandle<CGImageRef, CGImageRef, CGImageRelease>;
|
|
using CFDataHandle = CFHandle<CFDataRef, CFTypeRef, CFRelease>;
|
|
using CGImageSourceHandle = CFHandle<CGImageSourceRef, CFTypeRef, CFRelease>;
|
|
using CGDataProviderHandle = CFHandle<CGDataProviderRef, CGDataProviderRef, CGDataProviderRelease>;
|
|
using CGColorSpaceHandle = CFHandle<CGColorSpaceRef, CGColorSpaceRef, CGColorSpaceRelease>;
|
|
using CGContextHandle = CFHandle<CGContextRef, CGContextRef, CGContextRelease>;
|
|
|
|
CGImageRef CGImageCreateWithMLNPremultipliedImage(mln::PremultipliedImage&& src) {
|
|
// We're converting the PremultipliedImage's backing store to a CGDataProvider, and are taking
|
|
// over ownership of the memory.
|
|
CGDataProviderHandle provider(CGDataProviderCreateWithData(
|
|
NULL, src.data.get(), src.bytes(), [](void*, const void* data, size_t) {
|
|
delete[] reinterpret_cast<const decltype(src.data)::element_type*>(data);
|
|
}));
|
|
if (!provider) {
|
|
return nil;
|
|
}
|
|
|
|
// If we successfully created the provider, it will take over management of the memory segment.
|
|
src.data.release();
|
|
|
|
CGColorSpaceHandle colorSpace(CGColorSpaceCreateDeviceRGB());
|
|
if (!colorSpace) {
|
|
return nil;
|
|
}
|
|
|
|
constexpr const size_t bitsPerComponent = 8;
|
|
constexpr const size_t bytesPerPixel = 4;
|
|
constexpr const size_t bitsPerPixel = bitsPerComponent * bytesPerPixel;
|
|
const size_t bytesPerRow = bytesPerPixel * src.size.width;
|
|
const auto bitmapInfo = static_cast<uint32_t>(kCGBitmapByteOrderDefault) |
|
|
static_cast<uint32_t>(kCGImageAlphaPremultipliedLast);
|
|
|
|
return CGImageCreate(src.size.width, src.size.height, bitsPerComponent, bitsPerPixel, bytesPerRow,
|
|
*colorSpace, bitmapInfo, *provider, NULL, false, kCGRenderingIntentDefault);
|
|
}
|
|
|
|
mln::PremultipliedImage MLNPremultipliedImageFromCGImage(CGImageRef src) {
|
|
const size_t width = CGImageGetWidth(src);
|
|
const size_t height = CGImageGetHeight(src);
|
|
|
|
mln::PremultipliedImage image({static_cast<uint32_t>(width), static_cast<uint32_t>(height)});
|
|
|
|
CGColorSpaceHandle colorSpace(CGColorSpaceCreateDeviceRGB());
|
|
if (!colorSpace) {
|
|
throw std::runtime_error("CGColorSpaceCreateDeviceRGB failed");
|
|
}
|
|
|
|
constexpr const size_t bitsPerComponent = 8;
|
|
constexpr const size_t bytesPerPixel = 4;
|
|
const size_t bytesPerRow = bytesPerPixel * width;
|
|
const auto bitmapInfo = static_cast<uint32_t>(kCGBitmapByteOrderDefault) |
|
|
static_cast<uint32_t>(kCGImageAlphaPremultipliedLast);
|
|
|
|
CGContextHandle context(CGBitmapContextCreate(image.data.get(), width, height, bitsPerComponent,
|
|
bytesPerRow, *colorSpace, bitmapInfo));
|
|
if (!context) {
|
|
throw std::runtime_error("CGBitmapContextCreate failed");
|
|
}
|
|
|
|
CGContextSetBlendMode(*context, kCGBlendModeCopy);
|
|
CGContextDrawImage(*context, CGRectMake(0, 0, width, height), src);
|
|
|
|
return image;
|
|
}
|
|
|
|
namespace mln {
|
|
|
|
PremultipliedImage decodeImage(const std::string& source) {
|
|
CFDataHandle data(CFDataCreateWithBytesNoCopy(
|
|
kCFAllocatorDefault, reinterpret_cast<const unsigned char*>(source.data()), source.size(),
|
|
kCFAllocatorNull));
|
|
if (!data) {
|
|
throw std::runtime_error("CFDataCreateWithBytesNoCopy failed");
|
|
}
|
|
|
|
CGImageSourceHandle imageSource(CGImageSourceCreateWithData(*data, NULL));
|
|
if (!imageSource) {
|
|
throw std::runtime_error("CGImageSourceCreateWithData failed");
|
|
}
|
|
|
|
CGImageHandle image(CGImageSourceCreateImageAtIndex(*imageSource, 0, NULL));
|
|
if (!image) {
|
|
throw std::runtime_error("CGImageSourceCreateImageAtIndex failed");
|
|
}
|
|
|
|
return MLNPremultipliedImageFromCGImage(*image);
|
|
}
|
|
|
|
} // namespace mln
|