- #include <stdio.h>
- #include <stdlib.h>
- #include <iostream.h>
- FILE *ifp, *ofp;
- int rgb_2_yuv422(FILE *ifp, FILE *ofp);
- int rgb_2_yuv420(FILE *ifp, FILE *ofp);
- int select_option(void);
- void main()
- {
- int result = 0,selectoption = 0;
- selectoption = select_option();
- switch(selectoption)
- {
- case 1:
- result = rgb_2_yuv422(ifp, ofp);
- break;
- case 2:
- result = rgb_2_yuv420(ifp, ofp);
- break;
- default:
- ;
- }
- if (result == -1)
- {
- cout<< "You did not change anything!!/n";
- }
- else if (result == 3)
- {
- cout<< "You convert from RGB to YUV422!!/n";
- }
- else if (result == 4)
- {
- cout<< "You convert from RGB to YUV420!!/n";
- }
- }
- int rgb_2_yuv422(FILE *ifp, FILE *ofp)
- { ifp = fopen("C://fore_cif.raw", "rb");
- ofp = fopen("C://output.raw", "wb+");
- int i,j;
- int R, G, B;
- int Y, U, V;
- unsigned char *RGBBuffer = new unsigned char[512*512*3];
- unsigned char *YBuffer = new unsigned char[512*512];
- unsigned char *UBuffer = new unsigned char[512*512/2];
- unsigned char *VBuffer = new unsigned char[512*512/2];
- unsigned char *ULine = (new unsigned char[512+2])+1;
- unsigned char *VLine = (new unsigned char[512+2])+1;
- ULine[-1]=ULine[512]=128;
- VLine[-1]=VLine[512]=128;
- fread(RGBBuffer, sizeof(char), 512*512*3, ifp);
- for (i=0; i<512; ++i)
- {
- int RGBIndex = 3*512*i;
- int YIndex = 512*i;
- int UVIndex = 512*i/2;
- for ( j=0; j<512; ++j)
- {
- R = RGBBuffer[RGBIndex++];
- G = RGBBuffer[RGBIndex++];
- B = RGBBuffer[RGBIndex++];
- //Convert RGB to YUV
- Y = (unsigned char)( ( 66 * R + 129 * G + 25 * B + 128) >> 8) + 16 ;
- U = (unsigned char)( ( -38 * R - 74 * G + 112 * B + 128) >> 8) + 128 ;
- V = (unsigned char)( ( 112 * R - 94 * G - 18 * B + 128) >> 8) + 128 ;
- YBuffer[YIndex++] = static_cast<unsigned char>( (Y<0) ? 0 : ((Y>255) ? 255 : Y) );
- VLine[j] = V;
- ULine[j] = U;
- }
- for ( j=0; j<512; j+=2)
- {
- //Filter line
- V = ((VLine[j-1]+2*VLine[j]+VLine[j+1]+2)>>2);
- U = ((ULine[j-1]+2*ULine[j]+ULine[j+1]+2)>>2);
- //Clip and copy UV to output buffer
- VBuffer[UVIndex] = static_cast<unsigned char>( (V<0) ? 0 : ((V>255) ? 255 : V) );
- UBuffer[UVIndex++] = static_cast<unsigned char>( (U<0) ? 0 : ((U>255) ? 255 : U) );
- }
- }
- fwrite(YBuffer, sizeof(char), 512*512, ofp);
- fwrite(VBuffer, sizeof(char), 512*512/2, ofp);
- fwrite(UBuffer, sizeof(char), 512*512/2, ofp);
- return 3;
- fclose(ifp);
- fclose(ofp);
- }
- int rgb_2_yuv420(FILE *ifp, FILE *ofp)
- { ifp = fopen("C://news_cif.raw", "rb");
- ofp = fopen("C://output.raw", "wb+");
- int i,j;
- int R, G, B;
- int Y, U, V;
- unsigned char *RGBBuffer = new unsigned char[512*512*3];
- unsigned char *YBuffer = new unsigned char[512*512];
- unsigned char *UBuffer = new unsigned char[512*512/4];
- unsigned char *VBuffer = new unsigned char[512*512/4];
- int *ULine = (new int[512+2])+1;
- int *VLine = (new int[512+2])+1;
- ULine[-1]=ULine[512]=128;
- VLine[-1]=VLine[512]=128;
- int *UImage = (new int[(512+2)*(512+2)])+(512+2)+1;
- int *VImage = (new int[(512+2)*(512+2)])+(512+2)+1;
- fread(RGBBuffer, sizeof(char), 512*512*3, ifp);
- for (i=0; i<512; ++i)
- {
- int RGBIndex = 3*512*i;
- int YIndex = 512*i;
- for ( j=0; j<512; ++j)
- {
- R = RGBBuffer[RGBIndex++];
- G = RGBBuffer[RGBIndex++];
- B = RGBBuffer[RGBIndex++];
- //Convert RGB to YUV
- Y = (unsigned char)( ( 66 * R + 129 * G + 25 * B + 128) >> 8) + 16 ;
- U = (unsigned char)( ( -38 * R - 74 * G + 112 * B + 128) >> 8) + 128 ;
- V = (unsigned char)( ( 112 * R - 94 * G - 18 * B + 128) >> 8) + 128 ;
- YBuffer[YIndex++] = static_cast<unsigned char>( (Y<0) ? 0 : ((Y>255) ? 255 : Y) );
- VLine[j] = V;
- ULine[j] = U;
- }
- for ( j=0; j<512; j+=2)
- {
- //Filter line
- VImage[i*(512+2)+j] = ((VLine[j-1]+2*VLine[j]+VLine[j+1]+2)>>2);
- UImage[i*(512+2)+j] = ((ULine[j-1]+2*ULine[j]+ULine[j+1]+2)>>2);
- }
- }
- for (i=0; i<512; i+=2)
- {
- int UVIndex = 512*i/4;
- for ( j=0; j<512; j+=2)
- {
- V = ((VImage[(i-1)*(512+2)+j]+2*VImage[i*(512+2)+j]+VImage[(i+1)*(512+2)+j]+2)>>2);
- U = ((UImage[(i-1)*(512+2)+j]+2*UImage[i*(512+2)+j]+UImage[(i+1)*(512+2)+j]+2)>>2);
- VBuffer[UVIndex] = static_cast<unsigned char>( (V<0) ? 0 : ((V>255) ? 255 : V) );
- UBuffer[UVIndex++] = static_cast<unsigned char>( (U<0) ? 0 : ((U>255) ? 255 : U) );
- }
- }
- fwrite(YBuffer, sizeof(char), 512*512, ofp);
- fwrite(VBuffer, sizeof(char), 512*512/4, ofp);
- fwrite(UBuffer, sizeof(char), 512*512/4, ofp);
- return 4;
- fclose(ifp);
- fclose(ofp);
- }
- int select_option(void)
- {
- int sel = -1;
- printf("1. Convert from RGB to YUV422/n");
- printf("2. Convert from RGB to YUV420/n");
- printf("Select the option and press enter key: ");
- scanf(" %d", &sel);
- return sel;
- }