数据压缩第一次实验报告（rgb与yuv的转换）

用C语言实现的图片格式YUV和RGB的转换

RGB to YUV程序如下

main.c

---

#include <stdio.h>
#include <stdlib.h>
#include <malloc.h>
#include "rgb2yuv.h"

#define u_int8_t unsigned __int8
#define u_int unsigned __int32
#define u_int32_t unsigned __int32
#define FALSE false
#define TRUE true

/*
* rgb2yuv
* required arg1 should be the input RAW RGB24 file
* required arg2 should be the output RAW YUV12 file
*/
int main(int argc, char** argv)
{
/* variables controlable from command line */
u_int frameWidth = 352; /* –width= */
u_int frameHeight = 240; /* –height= */
bool flip = TRUE; /* –flip */
unsigned int i;

/* internal variables */
char* rgbFileName = NULL;
char* yuvFileName = NULL;
FILE* rgbFile = NULL;
FILE* yuvFile = NULL;
u_int8_t* rgbBuf = NULL;
u_int8_t* yBuf = NULL;
u_int8_t* uBuf = NULL;
u_int8_t* vBuf = NULL;
u_int32_t videoFramesWritten = 0;

/* begin process command line */
/* point to the specified file names */
rgbFileName = argv[1];
yuvFileName = argv[2];

frameWidth = atoi(argv[3]);
frameHeight = atoi(argv[4]);

/* open the RGB file */
rgbFile = fopen(rgbFileName, "rb");
if (rgbFile == NULL)
{
    printf("cannot find rgb file\n");
    exit(1);
}
else
{
    printf("The input rgb file is %s\n", rgbFileName);
}

/* open the RAW file */
yuvFile = fopen(yuvFileName, "wb");
if (yuvFile == NULL)
{
    printf("cannot find yuv file\n");
    exit(1);
}
else
{
    printf("The output yuv file is %s\n", yuvFileName);
}

/* get an input buffer for a frame */
rgbBuf = (u_int8_t*)malloc(frameWidth * frameHeight * 3);

/* get the output buffers for a frame */
yBuf = (u_int8_t*)malloc(frameWidth * frameHeight);
uBuf = (u_int8_t*)malloc((frameWidth * frameHeight) / 4);
vBuf = (u_int8_t*)malloc((frameWidth * frameHeight) / 4);

if (rgbBuf == NULL || yBuf == NULL || uBuf == NULL || vBuf == NULL)
{
    printf("no enought memory\n");
    exit(1);
}

while (fread(rgbBuf, 1, frameWidth * frameHeight * 3, rgbFile)) 
{
    if(RGB2YUV(frameWidth, frameHeight, rgbBuf, yBuf, uBuf, vBuf, flip))
    {
        printf("error");
        return 0;
                }

    for (i = 0; i < frameWidth*frameHeight; i++)
    {
        if (yBuf[i] < 16) yBuf[i] = 16;
        if (yBuf[i] > 235) yBuf[i] = 235;
    }

    for (i = 0; i < frameWidth*frameHeight/4; i++)
    {
        if (uBuf[i] < 16) uBuf[i] = 16;
        if (uBuf[i] > 240) uBuf[i] = 240;

        if (vBuf[i] < 16) vBuf[i] = 16;
        if (vBuf[i] > 240) vBuf[i] = 240;
    }

    fwrite(yBuf, 1, frameWidth * frameHeight, yuvFile);
    fwrite(uBuf, 1, (frameWidth * frameHeight) / 4, yuvFile);
    fwrite(vBuf, 1, (frameWidth * frameHeight) / 4, yuvFile);

    printf("\r...%d", ++videoFramesWritten);
}

printf("\n%u %ux%u video frames written\n", 
    videoFramesWritten, frameWidth, frameHeight);

/* cleanup */

fclose(rgbFile);
fclose(yuvFile);

return(0);

}

`

rgb2yuv.c

---

#include "stdlib.h"
#include "rgb2yuv.h"
static float RGBYUV02990[256], RGBYUV05870[256], RGBYUV01140[256];
static float RGBYUV01684[256], RGBYUV03316[256];
static float RGBYUV04187[256], RGBYUV00813[256];

int RGB2YUV (int x_dim, int y_dim, void *bmp, void *y_out, void *u_out, void *v_out, int flip)
{
static int init_done = 0;

long i, j, size;
unsigned char *r, *g, *b;
unsigned char *y, *u, *v;
unsigned char *pu1, *pu2, *pv1, *pv2, *psu, *psv;
unsigned char *y_buffer, *u_buffer, *v_buffer;
unsigned char *sub_u_buf, *sub_v_buf;

if (init_done == 0)
{
    InitLookupTable();
    init_done = 1;
}

// check to see if x_dim and y_dim are divisible by 2
if ((x_dim % 2) || (y_dim % 2)) return 1;
size = x_dim * y_dim;

// allocate memory
y_buffer = (unsigned char *)y_out;
sub_u_buf = (unsigned char *)u_out;
sub_v_buf = (unsigned char *)v_out;
u_buffer = (unsigned char *)malloc(size * sizeof(unsigned char));
v_buffer = (unsigned char *)malloc(size * sizeof(unsigned char));
if (!(u_buffer && v_buffer))
{
    if (u_buffer) free(u_buffer);
    if (v_buffer) free(v_buffer);
    return 2;
}

b = (unsigned char *)bmp;
y = y_buffer;
u = u_buffer;
v = v_buffer;

// convert RGB to YUV
if (!flip) {
    for (j = 0; j < y_dim; j ++)
    {
        y = y_buffer + (y_dim - j - 1) * x_dim;
        u = u_buffer + (y_dim - j - 1) * x_dim;
        v = v_buffer + (y_dim - j - 1) * x_dim;

        for (i = 0; i < x_dim; i ++) {
            g = b + 1;
            r = b + 2;
            *y = (unsigned char)(  R