从 Byte 数组创建 CImage

Question

我需要创建一个 CImage一个字节数组（实际上，它是一个无符号字符数组，但我可以转换为任何需要的形式）。字节数组的格式为“RGBRGBRGB...”。新图像需要包含图像字节的副本，而不是使用字节数组本身的内存。

我尝试了许多不同的方法来实现这一目标——包括使用各种 HBITMAP 创建函数、尝试使用 BitBlt——但到目前为止还没有任何效果。

要测试该函数是否有效，它应该通过此测试：

BYTE* imgBits;
int width;
int height;
int Bpp;   // BYTES per pixel (e.g. 3)
getImage(&imgBits, &width, &height, &Bpp); // get the image bits

// This is the magic function I need!!!
CImage img = createCImage(imgBits, width, height, Bpp);

// Test the image
BYTE* data = img.GetBits();  // data should now have the same data as imgBits

到目前为止，createCImage() 的所有实现都以 data 指向一个空（零填充）数组结束。

Answer 1

CImage 非常巧妙地支持 DIB，并且有一个 SetPixel() 方法，因此您可能可以做这样的事情（前面未编译、未经测试的代码！）：

CImage img;
img.Create(width, height, 24 /* bpp */, 0 /* No alpha channel */);

int nPixel = 0;
for(int row = 0; row < height; row++)
{
    for(int col = 0; col < width; col++)
    {
        BYTE r = imgBits[nPixel++];
        BYTE g = imgBits[nPixel++];
        BYTE b = imgBits[nPixel++];
        img.SetPixel(row, col, RGB(r, g, b));
    }
}

也许不是最有效的方法，但我应该认为这是最简单的方法。

Answer 2

使用 memcpy 复制数据，然后根据需要执行 SetDIBits 或 SetDIBitsToDevice 操作。但请注意，原始图像数据的扫描线在 4 字节边界上对齐（IIRC，自从我这样做以来已经有几年了），因此从 GetDIBits 返回的数据永远不会与原始数据完全相同（嗯，可能会，具体取决于图像大小）。

因此，您很可能需要逐条扫描线进行memcpy。

Answer 3

谢谢大家，在你们的帮助下我最终成功解决了这个问题。它主要涉及 @tinman 和 @Roel 使用 SetDIBitsToDevice() 的建议，但它涉及一些额外的位调整和内存管理，所以我想我应该在这里分享我的终点。

在下面的代码中，我假设设置了 width、height 和 Bpp（每像素字节），并且data 是指向 RGB 像素值数组的指针。

// Create the header info
bmInfohdr.biSize = sizeof(BITMAPINFOHEADER);
bmInfohdr.biWidth = width;
bmInfohdr.biHeight = -height;
bmInfohdr.biPlanes = 1;
bmInfohdr.biBitCount = Bpp*8;
bmInfohdr.biCompression = BI_RGB;
bmInfohdr.biSizeImage = width*height*Bpp;
bmInfohdr.biXPelsPerMeter = 0;
bmInfohdr.biYPelsPerMeter = 0;
bmInfohdr.biClrUsed = 0;
bmInfohdr.biClrImportant = 0;

BITMAPINFO bmInfo;
bmInfo.bmiHeader = bmInfohdr;
bmInfo.bmiColors[0].rgbBlue=255;

// Allocate some memory and some pointers
unsigned char * p24Img = new unsigned char[width*height*3];
BYTE *pTemp,*ptr;
pTemp=(BYTE*)data;
ptr=p24Img;

// Convert image from RGB to BGR
for (DWORD index = 0; index < width*height ; index++)
{
    unsigned char r = *(pTemp++);
    unsigned char g = *(pTemp++);
    unsigned char b = *(pTemp++);   

    *(ptr++) = b;
    *(ptr++) = g;
    *(ptr++) = r;
}

// Create the CImage
CImage im;
im.Create(width, height, 24, NULL);

HDC dc = im.GetDC();
SetDIBitsToDevice(dc, 0,0,width,height,0,0, 0, height, p24Img, &bmInfo, DIB_RGB_COLORS);
im.ReleaseDC();

delete[] p24Img;

Answer 4

这是一个更简单的解决方案。您可以使用 GetPixelAddress(...) 代替所有这些 BITMAPHEADERINFO 和 SedDIBitsToDevice。我解决的另一个问题是 8 位图像，它需要定义颜色表。

CImage outImage;
outImage.Create(width, height, channelCount * 8);

int lineSize = width * channelCount;
if (channelCount == 1)
{
    // Define the color table
    RGBQUAD* tab = new RGBQUAD[256];
    for (int i = 0; i < 256; ++i)
    {
        tab[i].rgbRed = i;
        tab[i].rgbGreen = i;
        tab[i].rgbBlue = i;
        tab[i].rgbReserved = 0;
    }
    outImage.SetColorTable(0, 256, tab);
    delete[] tab;
}

// Copy pixel values
// Warining: does not convert from RGB to BGR
for ( int i = 0; i < height; i++ )
{
    void*       dst = outImage.GetPixelAddress(0, i);
    const void* src = /* put the pointer to the i'th source row here */;
    memcpy(dst, src, lineSize);
}