YV12转RGB24的计算转换和bmp(dib)文件的显示保存

最近又接触到图像处理这一块,翻查到一年前自己写的代码http://blog.csdn.net/gongluck93/article/details/52813042,发现有点看不懂了!

所以自己又整理了一波(YV12转RGB24,显示和保存dib):

#include "stdafx.h"
/******************************************************************* 
*  Copyright(c) 2017 
*  All rights reserved. 
* 
*  文件名称: Bmp.cpp 
*  简要描述: bmp位图相关处理 
* 
*  创建日期: 2017-08-04 
*  作者:  gongluck 
*  说明: 
* 
*  修改日期: 
*  作者:  
*  说明: 
******************************************************************/
#include "Bmp.h"

//根据文件头和dib数据保存bmp文件
bool SaveBmpWithFileInfo(const char* filename,const BITMAPFILEHEADER* fileheader,const BITMAPINFO* bmpinfo,const void* bits)
{
    FILE* fp = NULL;
    if(filename == NULL || fileheader == NULL || bmpinfo == NULL || bits == NULL)
        return false;
    if((fp = fopen(filename,"wb")) == NULL)
        return false;
    fwrite(fileheader,sizeof(BITMAPFILEHEADER),1,fp);
    fwrite(bmpinfo,sizeof(BITMAPINFO),1,fp);
    fwrite(bits,bmpinfo->bmiHeader.biSizeImage,1,fp);
    fclose(fp);
    return true;
}

//根据位深度、像素宽、像素高和dib数据保存bmp文件
//深度24以下的不能处理(没有考虑有调色板的情况)
bool SaveBmpWithBitscountAndWH(const char* filename,int nBits,int width,int height,const void* bits)
{
    BITMAPFILEHEADER header = {0};
    BITMAPINFO info = {0};

    info.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
    info.bmiHeader.biWidth = width;
    info.bmiHeader.biHeight = -height;
    info.bmiHeader.biPlanes = 1;
    info.bmiHeader.biBitCount = nBits;
    info.bmiHeader.biCompression = BI_RGB;
    info.bmiHeader.biSizeImage = (width * nBits + 31) / 32 * 4 * height;

    header.bfSize = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFO) + info.bmiHeader.biSizeImage;
    header.bfType = 0x4d42;//"BM"
    header.bfOffBits = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFO);

    return SaveBmpWithFileInfo(filename,&header,&info,bits);
}

//由位图句柄(HBITMAP)保存成bmp图片
//可以用CreateDIBitmap、CreateDIBSection等函数获取HBITMAP
bool SavePicture(HBITMAP hbitmap,const char* filepath)
{  
    HDC hdc= GetDC(NULL);  
    BITMAP bitmap = {0};  
    WORD clrbits = 0;  
    BITMAPINFO* bitmapinfo = NULL;  
    BITMAPFILEHEADER fileheader;  
    DWORD temp;  
    void* buf = NULL;  
    FILE* fp = NULL;  

    GetObject(hbitmap,sizeof(BITMAP),&bitmap);  
    clrbits = bitmap.bmBitsPixel * bitmap.bmPlanes;//每点的颜色位数*平面数  
    if(clrbits == 1) ;  
    else if(clrbits <= 4) clrbits = 4;  
    else if(clrbits <= 8) clrbits = 8;  
    else if(clrbits <= 16) clrbits = 16;  
    else if(clrbits <= 24) clrbits = 24;  
    else clrbits = 32;  

    if(clrbits < 24)//24位以下需要调色板  
       bitmapinfo = (BITMAPINFO*)malloc(sizeof(BITMAPINFOHEADER) + sizeof(RGBQUAD)*(1<<clrbits));  
    else  
       bitmapinfo = (BITMAPINFO*)malloc(sizeof(BITMAPINFOHEADER) + sizeof(RGBQUAD));  

    bitmapinfo->bmiHeader.biBitCount = clrbits;//图像颜色位数  
    bitmapinfo->bmiHeader.biClrImportant = 0;  
    bitmapinfo->bmiHeader.biClrUsed = (clrbits<24) ? (1<<clrbits) : 0;//使用的颜色数  
    bitmapinfo->bmiHeader.biCompression = BI_RGB;//压缩方式,BI_RGB不压缩  
    bitmapinfo->bmiHeader.biHeight = bitmap.bmHeight;//位图高度  
    bitmapinfo->bmiHeader.biPlanes = bitmap.bmPlanes;//设备平面数  
    bitmapinfo->bmiHeader.biSize = sizeof(BITMAPINFOHEADER);//BITMAPINFOHEADER结构体大小  
    bitmapinfo->bmiHeader.biSizeImage = (bitmap.bmWidth * clrbits + 31) / 32 * 4 * bitmap.bmHeight;//位图数据大小  
    bitmapinfo->bmiHeader.biWidth = bitmap.bmWidth;//位图宽度  
    bitmapinfo->bmiHeader.biXPelsPerMeter = bitmapinfo->bmiHeader.biYPelsPerMeter = 0;  

    fileheader.bfType = 0x4d42;  
    fileheader.bfSize = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFO) 
        + sizeof(RGBQUAD)*(bitmapinfo->bmiHeader.biClrUsed==0 ? 0 : bitmapinfo->bmiHeader.biClrUsed)  
        + bitmapinfo->bmiHeader.biSizeImage;  
    fileheader.bfReserved1 = fileheader.bfReserved2 = 0;  
    fileheader.bfOffBits = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFO) +   
        sizeof(RGBQUAD)*(bitmapinfo->bmiHeader.biClrUsed==0 ? 0 : bitmapinfo->bmiHeader.biClrUsed);  

    buf = malloc(bitmapinfo->bmiHeader.biSizeImage);  
    GetDIBits(hdc,hbitmap,0,bitmap.bmHeight,buf,bitmapinfo,DIB_RGB_COLORS);

    fp = fopen(filepath,"wb");  
    fwrite(&fileheader,sizeof(BITMAPFILEHEADER),1,fp);  
    fwrite(bitmapinfo,sizeof(BITMAPINFO),1,fp);  
    fwrite(buf,bitmapinfo->bmiHeader.biSizeImage,1,fp);  
    fclose(fp);  
    free(buf);
    free(bitmapinfo);
    ReleaseDC(NULL,hdc);  
    return true;
}  

//将YV12转BGR24,没有效率可言(推荐使用ffmpeg)
bool YV12ToBGR24_Native(unsigned char* pYUV,unsigned char* pBGR24,int width,int height)
{
    if (width < 1 || height < 1 || pYUV == NULL || pBGR24 == NULL)
        return false;
    const long len = width * height;
    unsigned char* yData = pYUV;
    unsigned char* vData = &yData[len];
    unsigned char* uData = &vData[len >> 2];

    int bgr[3];
    int yIdx,uIdx,vIdx,idx;
    for (int i = 0;i < height;i++){
        for (int j = 0;j < width;j++){
            yIdx = i * width + j;
            vIdx = (i/2) * (width/2) + (j/2);
            uIdx = vIdx;

            bgr[0] = (int)(yData[yIdx] + 1.732446 * (uData[vIdx] - 128));                                    // b分量
            bgr[1] = (int)(yData[yIdx] - 0.698001 * (uData[uIdx] - 128) - 0.703125 * (vData[vIdx] - 128));    // g分量
            bgr[2] = (int)(yData[yIdx] + 1.370705 * (vData[uIdx] - 128));                                    // r分量

            for (int k = 0;k < 3;k++){
                idx = (i * width + j) * 3 + k;
                if(bgr[k] >= 0 && bgr[k] <= 255)
                    pBGR24[idx] = bgr[k];
                else
                    pBGR24[idx] = (bgr[k] < 0)?0:255;
            }
        }
    }
    return true;
}

github地址

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏蜕变

#哆啦A梦

write('by dongdong', font=("Bradley Hand ITC", 30, "bold"))

940
来自专栏机器人网

别让接线这件小事,拉开你与工程师的差距

导线与导线的连接、线头与接线桩的连接,事情小,责任大。本文图文并茂,让你清清楚楚看懂! 导线与导线的连接 导线的连接情况有:单股铜芯导线的直线连接、T字形连接;...

3337
来自专栏Android点滴积累

Android高效内存2:让图片占用尽可能少的内存

Android高效内存:让图片占用尽可能少的内存 一、让你的图片最小化 1.1 大图小图内存使用情况对比 大图:440 * 336    小图:220 * 16...

25311
来自专栏GIS讲堂

基于openlayers实现聚类统计展示

在前面的博文中讲述过基于Arcgis for js如何实现聚类统计展示,在本文中讲述如何基于openlayers实现聚类统计的效果,Arcgis for js聚...

1052
来自专栏小特工作室

基于iTextSharp的PDF文档操作

  公司是跨境电商,需要和各种物流打交道,需要把东西交给物流,让他们发到世界各地。其中需要物流公司提供一个运单号,来追踪货物到达哪里?!   最近在和DHL物流...

22010
来自专栏用户2442861的专栏

Python-OpenCV 处理图像(二):滤镜和图像运算

喜欢自拍的人肯定都知道滤镜了,下面代码尝试使用一些简单的滤镜,包括图片的平滑处理、灰度化、二值化等:

1071
来自专栏叁金大数据

EmguCV学习——简单使用

公司是视觉方面的业务,我又不会c++(好想会啊,正在学习中)。由于各种需求,自己觉得对c++不是特别感冒,所以选用了net下的opencv的封装EmguCV。p...

1451
来自专栏GIS讲堂

ArcGIS Image Server简介以及OL2中的加载

本文讲述Arcgis Image Server相关以及在OL2中如何加载Arcgis Server发布的影像服务。

932
来自专栏HansBug's Lab

算法模板——单个值欧拉函数

输入N,输出phi(N) 这样的单个值欧拉函数程序一般见于部分数论题,以及有时候求逆元且取模的数不是质数的情况(逆元:A/B=A*Bphi(p)-1 (mod ...

3435
来自专栏菩提树下的杨过

ORACLE:写Function时,传入参数变量名的注意事项

oracle中开发人员写自己的sql function时,入口参数名不要与select中table的字段名重复,否则虽然编译能正常通过,但运行的结果往往是错误的...

1875

扫码关注云+社区