我在Mac OS X上的C ++项目中实现了以下代码,用于捕获桌面屏幕:
int ScreenCaptureRoutines::CaptureImage(int width, int height)
{
CGRect captureRect;
captureRect.origin.x = 0;
captureRect.origin.y = 0;
captureRect.size.width = width;
captureRect.size.height = height;
CGImageRef img = CGWindowListCreateImage(captureRect, kCGWindowListOptionOnScreenOnly, kCGNullWindowID, kCGWindowImageDefault);
if(img == NULL)
{
fprintf(stderr, "CGWindowListCreateImage failed\n!");
return -1;
}
/* get pixels */
CGDataProviderRef provider = CGImageGetDataProvider(img);
CFDataRef dataRef = CGDataProviderCopyData(provider);
uint8_t* pixels = (uint8_t*)CFDataGetBytePtr(dataRef);
BitmapUtility::SaveBitmapToFile(pixels, width, height, 32, "/Users/Main/test32.bmp");
}
该代码适用于1024x768,1280x1024,1280x960,1280x768等分辨率,但它完全扰乱了1366x768的图像。
这是用于保存像素的SaveBitmapFile方法:
void BitmapUtility::SaveBitmapToFile(u8* pBitmapBits, long lWidth, long lHeight, unsigned short wBitsPerPixel, char* lpszFileName)
{
BITMAPINFOHEADER bmpInfoHeader = {0};
bmpInfoHeader.biSize = sizeBITMAPINFOHEADER;
bmpInfoHeader.biBitCount = wBitsPerPixel;
bmpInfoHeader.biClrImportant = 0;
bmpInfoHeader.biClrUsed = 0;
bmpInfoHeader.biCompression = BI_RGB;
bmpInfoHeader.biHeight = lHeight;
bmpInfoHeader.biWidth = lWidth;
bmpInfoHeader.biPlanes = 1;
bmpInfoHeader.biSizeImage = lWidth* lHeight * (wBitsPerPixel / 8);
BITMAPFILEHEADER bfh = {0};
bfh.bfType=0x4D42;
bfh.bfOffBits = sizeBITMAPINFOHEADER + sizeBITMAPFILEHEADER;
bfh.bfSize = bfh.bfOffBits + bmpInfoHeader.biSizeImage;
//Creating bitmap file
int fileDescriptor = open(lpszFileName, O_CREAT | O_WRONLY);
if (fileDescriptor == -1)
{
//returning if error encountered when opening file
return;
}
ByteBufferOutputStream bufferInfoHeader;
bufferInfoHeader.write(&bmpInfoHeader.biSize, 4);
bufferInfoHeader.write(&bmpInfoHeader.biWidth, 4);
bufferInfoHeader.write(&bmpInfoHeader.biHeight, 4);
bufferInfoHeader.write(&bmpInfoHeader.biPlanes, 2);
bufferInfoHeader.write(&bmpInfoHeader.biBitCount, 2);
bufferInfoHeader.write(&bmpInfoHeader.biCompression, 4);
bufferInfoHeader.write(&bmpInfoHeader.biSizeImage, 4);
bufferInfoHeader.write(&bmpInfoHeader.biXPelsPerMeter, 4);
bufferInfoHeader.write(&bmpInfoHeader.biYPelsPerMeter, 4);
bufferInfoHeader.write(&bmpInfoHeader.biClrUsed, 4);
bufferInfoHeader.write(&bmpInfoHeader.biClrImportant, 4);
ByteBufferOutputStream bufferFileHeader;
bufferFileHeader.write(&bfh.bfType, 2);
bufferFileHeader.write(&bfh.bfSize, 4);
bufferFileHeader.write(&bfh.bfReserved1, 2);
bufferFileHeader.write(&bfh.bfReserved2, 2);
bufferFileHeader.write(&bfh.bfOffBits, 4);
//Writing bitmap to file
ssize_t bytesWritten = write(fileDescriptor, bufferFileHeader.get()->getData(), sizeBITMAPFILEHEADER);
bytesWritten = write(fileDescriptor, bufferInfoHeader.get()->getData(), sizeBITMAPINFOHEADER);
bytesWritten = write(fileDescriptor, pBitmapBits, bmpInfoHeader.biSizeImage);
//Closing bitmap file
close(fileDescriptor);
}
SaveBitmapToFile方法仅用于测试。在现实生活中,我实际上是通过libavcodec传递捕获的像素来转换并将帧保存为FLV格式,但最终结果是相同的,图像得到加扰。
有没有人知道为什么代码不能用于1366x768(或其他非标准的4:3或16:9分辨率)?
答案 0 :(得分:3)
首先,您应该尝试在捕获后立即将CGImageRef渲染到屏幕,以确保数据有效。
查看捕获的bmp文件。如果图像看起来像伪随机噪声,则您的像素格式是错误的。您需要调用 CGImageGetWidth , CGImageGetHeight 和 CGImageGetBitsPerPixel (以及CGImage.h中的其他函数)以确保像素格式为正确的。
如果你仍然可以看出一些像素,但行未对齐,那么你的字节跨度是错误的。另一方面,1024和1280是倍数32,64和128. 1366,2除数的最大幂是2.驱动器通常使用64字节对齐的像素步长,因为这是公共高速缓存行大小。步幅等于宽度+零填充。调用 CGImageGetBytesPerRow 从前一行的起始地址获取字节偏移量,以获得下一行的起始地址。
使用步幅写入图像的伪代码:
ptr = pixels
bytesPerRow = CGImageGetBytesPerRow()
bytesPerPixel = (CGImageGetBitsPerPixel() + 7) / 8
for row = 0 to height:
write(ptr, width*bytesPerPixel)
ptr += bytesPerRow