1. 程式人生 > >使用FFmpeg類庫實現YUV視訊序列編碼為視訊

使用FFmpeg類庫實現YUV視訊序列編碼為視訊

也歡迎大家轉載本篇文章。分享知識,造福人民,實現我們中華民族偉大復興!

               

搞視訊處理的朋友肯定比較熟悉YUV視訊序列,很多測試庫提供的視訊資料都是YUV視訊序列,我們這裡就用用YUV視訊序列來做視訊。關於YUV視訊序列,我就不多講了,可以看書學習,通常的視訊序列都是YUV420格式的。

步驟也就那幾步,新增視訊流,開啟編碼器,開闢相應的記憶體空間,然後就可以開啟YUV序列逐幀寫入資料了,so easy!記得最後要做好檔案的關閉和記憶體的釋放,因為FFmpeg是c風格的(不知道新版本是否是c++風格的),這些工作都需要自己做好啊。過多的說明是沒用的,直接上程式碼:

這裡我補充一下,大多數的視訊格式好像只支援YUV格式的視訊幀AVFrame,我試圖直接把RGB的視訊序列直接編碼到視訊這條路好像走不通,都需要把RGB的視訊幀再轉成YUV視訊幀才行,

不知道高手有沒有其他高見。

  1. #include <stdio.h>
  2. #include <string.h>
  3. extern"C"
  4. {  
  5. #include <libavcodec\avcodec.h>
  6. #include <libavformat\avformat.h>
  7. #include <libswscale\swscale.h>
  8. };  
  9. void main(int argc, char ** argv)  
  10. {  
  11.     AVFormatContext* oc;  
  12.     AVOutputFormat* fmt;  
  13.     AVStream* video_st;  
  14.     double video_pts;  
  15.     uint8_t* video_outbuf;  
  16.     uint8_t* picture_buf;  
  17.     AVFrame* picture;  
  18. //  AVFrame* pictureRGB;
  19.     int size;  
  20.     int ret;  
  21.     int video_outbuf_size;  
  22.     FILE *fin = fopen("akiyo_qcif.yuv""rb"); //視訊原始檔 
  23.     constchar* filename = "test.mpg";  
  24. //  const char* filename;
  25. //  filename = argv[1];
  26.     av_register_all();  
  27. //  avcodec_init(); // 初始化codec庫
  28. //  avcodec_register_all(); // 註冊編碼器
  29.     fmt = guess_format(NULL, filename, NULL);  
  30.     oc = av_alloc_format_context();  
  31.     oc->oformat = fmt;  
  32.     snprintf(oc->filename, sizeof(oc->filename), "%s", filename);  
  33.     video_st = NULL;  
  34.     if (fmt->video_codec != CODEC_ID_NONE)  
  35.     {  
  36.         AVCodecContext* c;  
  37.         video_st = av_new_stream(oc, 0);  
  38.         c = video_st->codec;  
  39.         c->codec_id = fmt->video_codec;  
  40.         c->codec_type = CODEC_TYPE_VIDEO;  
  41.         c->bit_rate = 400000;  
  42.         c->width = 176;  
  43.         c->height = 144;  
  44.         c->time_base.num = 1;  
  45.         c->time_base.den = 25;   
  46.         c->gop_size = 12;  
  47.         c->pix_fmt = PIX_FMT_YUV420P;  
  48.         if (c->codec_id == CODEC_ID_MPEG2VIDEO)  
  49.         {  
  50.             c->max_b_frames = 2;  
  51.         }  
  52.         if (c->codec_id == CODEC_ID_MPEG1VIDEO)  
  53.         {  
  54.             c->mb_decision = 2;  
  55.         }  
  56.         if (!strcmp(oc->oformat->name, "mp4") || !strcmp(oc->oformat->name, "mov") || !strcmp(oc->oformat->name, "3gp"))  
  57.         {  
  58.             c->flags |= CODEC_FLAG_GLOBAL_HEADER;  
  59.         }  
  60.     }  
  61.     if (av_set_parameters(oc, NULL)<0)  
  62.     {  
  63.         return;  
  64.     }  
  65.     dump_format(oc, 0, filename, 1);  
  66.     if (video_st)  
  67.     {  
  68.         AVCodecContext* c;  
  69.         AVCodec* codec;  
  70.         c = video_st->codec;  
  71.         codec = avcodec_find_encoder(c->codec_id);  
  72.         if (!codec)  
  73.         {  
  74.             return;  
  75.         }  
  76.         if (avcodec_open(c, codec) < 0)  
  77.         {  
  78.             return;  
  79.         }  
  80.         if (!(oc->oformat->flags & AVFMT_RAWPICTURE))  
  81.         {  
  82.             video_outbuf_size = 200000;  
  83.             video_outbuf = (uint8_t*)av_malloc(video_outbuf_size);  
  84.         }  
  85.         picture = avcodec_alloc_frame();  
  86.         size = avpicture_get_size(c->pix_fmt, c->width, c->height);  
  87.         picture_buf = (uint8_t*)av_malloc(size);  
  88.         if (!picture_buf)  
  89.         {  
  90.             av_free(picture);  
  91.         }  
  92.         avpicture_fill((AVPicture*)picture, picture_buf, c->pix_fmt, c->width, c->height);  
  93.     }  
  94.     if (!(fmt->flags & AVFMT_NOFILE))  
  95.     {  
  96.         if (url_fopen(&oc->pb, filename, URL_WRONLY) < 0)  
  97.         {  
  98.             return;  
  99.         }  
  100.     }  
  101.     av_write_header(oc);  
  102.     for (int i=0; i<300; i++)  
  103.     {  
  104.         if (video_st)  
  105.         {  
  106.             video_pts = (double)(video_st->pts.val * video_st->time_base.num / video_st->time_base.den);  
  107.         }  
  108.         else
  109.         {  
  110.             video_pts = 0.0;  
  111.         }  
  112.         if (!video_st/* || video_pts >= 5.0*/)  
  113.         {  
  114.             break;  
  115.         }  
  116.         AVCodecContext* c;  
  117.         c = video_st->codec;  
  118.         size = c->width * c->height;  
  119.         if (fread(picture_buf, 1, size*3/2, fin) < 0)  
  120.         {  
  121.             break;  
  122.         }  
  123.         picture->data[0] = picture_buf;  // 亮度
  124.         picture->data[1] = picture_buf+ size;  // 色度 
  125.         picture->data[2] = picture_buf+ size*5/4; // 色度 
  126.         // 如果是rgb序列,可能需要如下程式碼
  127. //      SwsContext* img_convert_ctx;
  128. //      img_convert_ctx = sws_getContext(c->width, c->height, PIX_FMT_RGB24, c->width, c->height, c->pix_fmt, SWS_BICUBIC, NULL, NULL, NULL);
  129. //      sws_scale(img_convert_ctx, pictureRGB->data, pictureRGB->linesize, 0, c->height, picture->data, picture->linesize);
  130.         if (oc->oformat->flags & AVFMT_RAWPICTURE)  
  131.         {  
  132.             AVPacket pkt;  
  133.             av_init_packet(&pkt);  
  134.             pkt.flags |= PKT_FLAG_KEY;  
  135.             pkt.stream_index = video_st->index;  
  136.             pkt.data = (uint8_t*)picture;  
  137.             pkt.size = sizeof(AVPicture);  
  138.             ret = av_write_frame(oc, &pkt);  
  139.         }  
  140.         else
  141.         {  
  142.             int out_size = avcodec_encode_video(c, video_outbuf, video_outbuf_size, picture);  
  143.             if (out_size > 0)  
  144.             {  
  145.                 AVPacket pkt;  
  146.                 av_init_packet(&pkt);  
  147.                 pkt.pts = av_rescale_q(c->coded_frame->pts, c->time_base, video_st->time_base);  
  148.                 if (c->coded_frame->key_frame)  
  149.                 {  
  150.                     pkt.flags |= PKT_FLAG_KEY;  
  151.                 }  
  152.                 pkt.stream_index = video_st->index;  
  153.                 pkt.data = video_outbuf;  
  154.                 pkt.size = out_size;  
  155.                 ret = av_write_frame(oc, &pkt);  
  156.             }  
  157.         }  
  158.     }  
  159.     if (video_st)  
  160.     {  
  161.         avcodec_close(video_st->codec);  
  162. //      av_free(picture->data[0]);
  163.         av_free(picture);  
  164.         av_free(video_outbuf);  
  165. //      av_free(picture_buf);
  166.     }  
  167.     av_write_trailer(oc);  
  168.     for (int i=0; i<oc->nb_streams; i++)  
  169.     {  
  170.         av_freep(&oc->streams[i]->codec);  
  171.         av_freep(&oc->streams[i]);  
  172.     }  
  173.     if (!(fmt->flags & AVFMT_NOFILE))  
  174.     {  
  175.         url_fclose(oc->pb);  
  176.     }  
  177.     av_free(oc);  
  178. }  

            這裡寫圖片描述 你好! 這是你第一次使用 **Markdown編輯器** 所展示的歡迎頁。如果你想學習如何使用Markdown編輯器, 可以仔細閱讀這篇文章,瞭解一下Markdown的基本語法知識。

新的改變

我們對Markdown編輯器進行了一些功能拓展與語法支援,除了標準的Markdown編輯器功能,我們增加了如下幾點新功能,幫助你用它寫部落格:

  1. 全新的介面設計 ,將會帶來全新的寫作體驗;
  2. 在創作中心設定你喜愛的程式碼高亮樣式,Markdown 將程式碼片顯示選擇的高亮樣式 進行展示;
  3. 增加了 圖片拖拽 功能,你可以將本地的圖片直接拖拽到編輯區域直接展示;
  4. 全新的 KaTeX數學公式 語法;
  5. 增加了支援甘特圖的mermaid語法1 功能;
  6. 增加了 多螢幕編輯 Markdown文章功能;
  7. 增加了 焦點寫作模式、預覽模式、簡潔寫作模式、左右區域同步滾輪設定 等功能,功能按鈕位於編輯區域與預覽區域中間;
  8. 增加了 檢查列表 功能。

功能快捷鍵

撤銷:Ctrl/Command + Z
重做:Ctrl/Command + Y
加粗:Ctrl/Command + B
斜體:Ctrl/Command + I
標題:Ctrl/Command + Shift + H
無序列表:Ctrl/Command + Shift + U
有序列表:Ctrl/Command + Shift + O
檢查列表:Ctrl/Command + Shift + C
插入程式碼:Ctrl/Command + Shift + K
插入連結:Ctrl/Command + Shift + L
插入圖片:Ctrl/Command + Shift + G

合理的建立標題,有助於目錄的生成

直接輸入1次#,並按下space後,將生成1級標題。
輸入2次#,並按下space後,將生成2級標題。
以此類推,我們支援6級標題。有助於使用TOC語法後生成一個完美的目錄。

如何改變文字的樣式

強調文字 強調文字

加粗文字 加粗文字

標記文字

刪除文字

引用文字

H2O is是液體。

210 運算結果是 1024.

插入連結與圖片

連結: link.

圖片: Alt

帶尺寸的圖片: Alt

當然,我們為了讓使用者更加便捷,我們增加了圖片拖拽功能。

如何插入一段漂亮的程式碼片

部落格設定頁面,選擇一款你喜歡的程式碼片高亮樣式,下面展示同樣高亮的 程式碼片.

// An highlighted block var foo = 'bar'; 

生成一個適合你的列表

  • 專案
    • 專案
      • 專案
  1. 專案1
  2. 專案2
  3. 專案3
  • 計劃任務
  • 完成任務

建立一個表格

一個簡單的表格是這麼建立的:

專案 Value
電腦 $1600
手機 $12
導管 $1

設定內容居中、居左、居右

使用:---------:居中
使用:----------居左
使用----------:居右

第一列 第二列 第三列
第一列文字居中 第二列文字居右 第三列文字居左

SmartyPants

SmartyPants將ASCII標點字元轉換為“智慧”印刷標點HTML實體。例如:

TYPE ASCII HTML
Single backticks 'Isn't this fun?' ‘Isn’t this fun?’
Quotes "Isn't this fun?" “Isn’t this fun?”
Dashes -- is en-dash, --- is em-dash – is en-dash, — is em-dash

建立一個自定義列表

Markdown
Text-to-HTML conversion tool
Authors
John
Luke

如何建立一個註腳

一個具有註腳的文字。2

註釋也是必不可少的

Markdown將文字轉換為 HTML

KaTeX數學公式

您可以使用渲染LaTeX數學表示式 KaTeX:

Gamma公式展示 Γ(n)=(n1)!nN\Gamma(n) = (n-1)!\quad\forall n\in\mathbb N 是通過尤拉積分

Γ(z)=0tz1etdt&ThinSpace;. \Gamma(z) = \int_0^\infty t^{z-1}e^{-t}dt\,.

你可以找到更多關於的資訊 LaTeX 數學表示式here.

新的甘特圖功能,豐富你的文章

gantt
        dateFormat  YYYY-MM-DD
        title Adding GANTT diagram functionality to mermaid
        section 現有任務
        已完成               :done,    des1, 2014-01-06,2014-01-08
        進行中               :active,  des2, 2014-01-09, 3d
        計劃一               :         des3, after des2, 5d
        計劃二               :         des4, after des3, 5d
  • 關於 甘特圖 語法,參考 這兒,

UML 圖表

可以使用UML圖表進行渲染。 Mermaid. 例如下面產生的一個序列圖::

張三李四王五你好!李四, 最近怎麼樣?你最近怎麼樣,王五?我很好,謝謝!我很好,謝謝!李四想了很長時間,文字太長了不適合放在一行.打量著王五...很好... 王五, 你怎麼樣?張三李四王五

這將產生一個流程圖。:

連結長方形圓角長方形菱形
  • 關於 Mermaid 語法,參考 這兒,

FLowchart流程圖

我們依舊會支援flowchart的流程圖:

  • 關於 Flowchart流程圖 語法,參考 這兒.

匯出與匯入

匯出

如果你想嘗試使用此編輯器, 你可以在此篇文章任意編輯。當你完成了一篇文章的寫作, 在上方工具欄找到 文章匯出 ,生成一個.md檔案或者.html檔案進行本地儲存。

匯入

如果你想載入一篇你寫過的.md檔案或者.html檔案,在上方工具欄可以選擇匯入功能進行對應副檔名的檔案匯入,
繼續你的創作。