WICConvertBitmapSource + CopyPixel结果为蓝色图像

WICConvertBitmapSource + CopyPixels results in blue image

本文关键字:蓝色 图像 结果 CopyPixel WICConvertBitmapSource      更新时间:2023-10-16

我正在尝试使用 WIC 将图像加载到内存缓冲区中进行进一步处理,然后在完成后将其写回文件。 具体说来:

  1. 将图像加载到 IWICBitmapFrameDecode 中。
  2. 加载的IWICBitmapFrameDecode报告其像素格式为GUID_WICPixelFormat24bppBGR。我想在32bpp RGBA中工作,所以我打电话给WICConvertBitmapSource。
  3. 在转换后的帧上调用 CopyPixel 以获取内存缓冲区。
  4. 使用 WritePixel 将内存缓冲区写回 IWICBitmapFrameEncode。

这会产生可识别的图像,但生成的图像大多是蓝色的,就好像红色通道被解释为蓝色一样。

如果我调用 WriteSource 直接写入转换后的帧,而不是写入内存缓冲区,它可以工作。 如果我从原始未转换的帧调用 CopyPixel(并相应地更新我的步幅和像素格式(,它可以工作。 只有WICConvertBitmapSource的组合加上内存缓冲区(CopyPixel + WritePixels(的使用导致了问题,但我无法弄清楚我做错了什么。

这是我的代码。

int main() {
IWICImagingFactory *pFactory;
IWICBitmapDecoder *pDecoder = NULL;
CoInitializeEx(NULL, COINIT_MULTITHREADED);
CoCreateInstance(
CLSID_WICImagingFactory,
NULL,
CLSCTX_INPROC_SERVER,
IID_IWICImagingFactory,
(LPVOID*)&pFactory
);
// Load the image.
pFactory->CreateDecoderFromFilename(L"input.png", NULL, GENERIC_READ, WICDecodeMetadataCacheOnDemand, &pDecoder);
IWICBitmapFrameDecode *pFrame = NULL;
pDecoder->GetFrame(0, &pFrame);
// pFrame->GetPixelFormat shows that the image is 24bpp BGR.
// Convert to 32bpp RGBA for easier processing.
IWICBitmapSource *pConvertedFrame = NULL;
WICConvertBitmapSource(GUID_WICPixelFormat32bppRGBA, pFrame, &pConvertedFrame);
// Copy the 32bpp RGBA image to a buffer for further processing.
UINT width, height;
pConvertedFrame->GetSize(&width, &height);
const unsigned bytesPerPixel = 4;
const unsigned stride = width * bytesPerPixel;
const unsigned bitmapSize = width * height * bytesPerPixel;
BYTE *buffer = new BYTE[bitmapSize];
pConvertedFrame->CopyPixels(nullptr, stride, bitmapSize, buffer);
// Insert image buffer processing here.  (Not currently implemented.)
// Create an encoder to turn the buffer back into an image file.
IWICBitmapEncoder *pEncoder = NULL;
pFactory->CreateEncoder(GUID_ContainerFormatPng, nullptr, &pEncoder);
IStream *pStream = NULL;
SHCreateStreamOnFileEx(L"output.png", STGM_WRITE | STGM_CREATE, FILE_ATTRIBUTE_NORMAL, true, NULL, &pStream);
pEncoder->Initialize(pStream, WICBitmapEncoderNoCache);
IWICBitmapFrameEncode *pFrameEncode = NULL;
pEncoder->CreateNewFrame(&pFrameEncode, NULL);
pFrameEncode->Initialize(NULL);
WICPixelFormatGUID pixelFormat = GUID_WICPixelFormat32bppRGBA;
pFrameEncode->SetPixelFormat(&pixelFormat);
pFrameEncode->SetSize(width, height);
pFrameEncode->WritePixels(height, stride, bitmapSize, buffer);
pFrameEncode->Commit();
pEncoder->Commit();
pStream->Commit(STGC_DEFAULT);
return 0;
}

PNG 编码器仅支持 PNG 本机编解码器官方文档中指定的 32bppGUID_WICPixelFormat32bppBGRA(BGR(。当你用GUID_WICPixelFormat32bppRGBA调用它时,它不会进行信道切换。只会使用你的像素,因为它们是 BGR,而不是 RGB,并且不会告诉你有问题。

我不知道你想做什么,但是在你的例子中,你可以在对WICConvertBitmapSource的调用中用GUID_WICPixelFormat32bppBGRA替换GUID_WICPixelFormat32bppRGBA(也可以替换最后一个pixelFormat变量的定义以确保你的源代码是正确的,但它不会改变任何东西(。

PS:您可以使用Wic保存文件,而无需使用其他API创建流,请在此处查看我的答案: 使用DirectX捕获屏幕