【轉】C# YUV轉RGB RGB轉BMP

using System;
using System.Collections.Generic;
using System.Text;
using System.IO;
using System.Drawing;
namespace yuv2rgbcs
{
    class Program
    {
        static double[,] YUV2RGB_CONVERT_MATRIX = new double[3, 3] { { 1, 0, 1.4022 }, { 1, -0.3456, -0.7145 }, { 1, 1.771, 0 } };

        static void Main(string[] args)
        {
            int width = 1280;
            int height = 720;
            int imgSize = width * height;
            int frameSize = imgSize + (imgSize >> 1);

            byte[] yuv = new byte[frameSize];
            byte[] rgb = new byte[3 * imgSize];

            using (FileStream fs = File.OpenRead("D:\\test.cif"))
            {
                int frame = (int)fs.Length / frameSize;
                using (BinaryReader br = new BinaryReader(fs))
                {
                    int index = 0;
                    while (br.PeekChar() != -1)
                    {
                        // 循環讀取每一楨
                        br.Read(yuv, 0, frameSize);

                        int start = Environment.TickCount;
                        // 轉換為 RGB
                        Bitmap bm = ConvertYUV2RGB (yuv, rgb, width, height);

                        int interval = Environment.TickCount - start;
                        Console.WriteLine("yuv 2 bmp = "+interval);
                        start = Environment.TickCount;

                        bm.Save("D:\\yuv2bmp_1.bmp");
                        // 寫 BMP 文件。
                        //WriteBMP(rgb, width, height, string.Format("D:\\yuv2bmp_{0}.bmp", index++));
                        interval = Environment.TickCount - start;
                        Console.WriteLine("rgb to bmp write bmp = " + interval);
                    }
                }
            }
        }

        /// <summary>
        /// 將轉換后的 RGB 圖像數據按照 BMP 格式寫入文件。
        /// </summary>
        /// <param name="rgbFrame">RGB 格式圖像數據。</param>
        /// <param name="width">圖像寬（單位：像素）。</param>
        /// <param name="height">圖像高（單位：像素）。</param>
        /// <param name="bmpFile"> BMP 文件名。</param>
        static void WriteBMP(byte[] rgbFrame, int width, int height, string bmpFile)
        {
            // 寫 BMP 圖像文件。
            int yu = width * 3 % 4;
            int bytePerLine = 0;
            yu = yu != 0 ? 4 - yu : yu;
            bytePerLine = width * 3 + yu;

            using (FileStream fs = File.Open(bmpFile, FileMode.Create))
            {
                using (BinaryWriter bw = new BinaryWriter(fs))
                {
                    bw.Write('B');
                    bw.Write('M');
                    bw.Write(bytePerLine * height + 54);
                    bw.Write(0);
                    bw.Write(54);
                    bw.Write(40);
                    bw.Write(width);
                    bw.Write(height);
                    bw.Write((ushort)1);
                    bw.Write((ushort)24);
                    bw.Write(0);
                    bw.Write(bytePerLine * height);
                    bw.Write(0);
                    bw.Write(0);
                    bw.Write(0);
                    bw.Write(0);

                    byte[] data = new byte[bytePerLine * height];
                    int gIndex = width * height;
                    int bIndex = gIndex * 2;

                    for (int y = height - 1, j = 0; y >= 0; y--, j++)
                    {
                        for (int x = 0, i = 0; x < width; x++)
                        {
                            data[y * bytePerLine + i++] = rgbFrame[bIndex + j * width + x];    // B
                            data[y * bytePerLine + i++] = rgbFrame[gIndex + j * width + x];    // G
                            data[y * bytePerLine + i++] = rgbFrame[j * width + x];  // R
                        }
                    }
                    bw.Write(data, 0, data.Length);
                    bw.Flush();
                }
            }
        }

        /// <summary>
        /// 將一楨 YUV 格式的圖像轉換為一楨 RGB 格式圖像並轉換成BMP圖片。
        /// </summary>
        /// <param name="yuvFrame">YUV 格式圖像數據。</param>
        /// <param name="rgbFrame">RGB 格式圖像數據。</param>
        /// <param name="width">圖像寬（單位：像素）。</param>
        /// <param name="height">圖像高（單位：像素）。</param>
        /// <returns>Bitmap圖片</returns>
        static Bitmap ConvertYUV2RGB(byte[] yuvFrame, byte[] rgbFrame, int width, int height)
        {
            int uIndex = width * height;
            int vIndex = uIndex + ((width * height) >> 2);
            int gIndex = width * height;
            int bIndex = gIndex * 2;

            int temp = 0;


             //圖片為pic1,RGB顏色的二進制數據轉換得的int r,g,b;
            Bitmap bm = new Bitmap(width, height);

            int r = 0;
            int g = 0;
            int b = 0;


            for (int y = 0; y < height; y++)
            {
                for (int x = 0; x < width; x++)
                {
                    // R分量
                    temp = (int)(yuvFrame[y * width + x] + (yuvFrame[vIndex + (y / 2) * (width / 2) + x / 2] - 128) * YUV2RGB_CONVERT_MATRIX[0, 2]);
                    rgbFrame[y * width + x] = (byte)(temp < 0 ? 0 : (temp > 255 ? 255 : temp));
                    // G分量
                    temp = (int)(yuvFrame[y * width + x] + (yuvFrame[uIndex + (y / 2) * (width / 2) + x / 2] - 128) * YUV2RGB_CONVERT_MATRIX[1, 1] + (yuvFrame[vIndex + (y / 2) * (width / 2) + x / 2] - 128) * YUV2RGB_CONVERT_MATRIX[1, 2]);
                    rgbFrame[gIndex + y * width + x] = (byte)(temp < 0 ? 0 : (temp > 255 ? 255 : temp));
                    // B分量
                    temp = (int)(yuvFrame[y * width + x] + (yuvFrame[uIndex + (y / 2) * (width / 2) + x / 2] - 128) * YUV2RGB_CONVERT_MATRIX[2, 1]);
                    rgbFrame[bIndex + y * width + x] = (byte)(temp < 0 ? 0 : (temp > 255 ? 255 : temp));
                    Color c = Color.FromArgb(rgbFrame[y * width + x], rgbFrame[gIndex + y * width + x], rgbFrame[bIndex + y * width + x]);
                    bm.SetPixel(x, y, c);
                }
            }
            return bm;
        }
    }
}