/** 他山之石，學習為主，版權所無，翻版不究，有錯無責 基於記憶體的格式封裝測試 使用 ./a.out a.avi a.mkv 支援的： avi mkv mp4 flv ts ... 參考： http://blog.csdn.net/leixiaohua1020/article/details/25422685 log 新版本出現： Using AVStream.codec.time_base as a timebase hint to the muxer is  deprecated. Set AVStream.time_base instead. test passed!! mp4->avi failed 出現： H.264 bitstream malformed, no startcode found, use the h264_mp4toannexb bitstream filter  解決見： http://blog.chinaunix.net/uid-11344913-id-4432752.html 官方解釋： https://www.ffmpeg.org/ffmpeg-bitstream-filters.html#h264_005fmp4toannexb ts -> avi passed 其它： 1、自定USE_MEM，則轉換後的資料在記憶體中，要用write將其寫回到檔案    如果不定義，則每一次write都會寫到檔案，seek也是檔案操作 2、傳遞給ffmpeg的avio_alloc_context中的記憶體p和大小size，可以使用32768。 如果轉換後的資料儲存在記憶體p1，這個記憶體p1一定要和前面所說的p不同。因為 在自定義的write中的buf引數，就是p，所以要拷貝到其它記憶體。 如定義p為32768，但定義p1為50MB，可以轉換50MB的視訊 測試： p為32768時，需呼叫write 1351次 2倍大小時，呼叫write 679次 p越大，呼叫次數最少，記憶體消耗越大 (用time測試，時間上沒什麼變化，前者為4.7s，後者為4.6s) 3、優化：    轉換功能介面封裝為類，把write、seek等和記憶體有關的操作放到類外部實現，    再傳遞到該類中，該類沒有記憶體管理更好一些。 */ #include <stdio.h> #include <stdlib.h> #include <unistd.h> #include <sys/types.h> #include <sys/stat.h> #include <fcntl.h> extern "C" { #include "libavcodec/avcodec.h" #include "libavformat/avformat.h" #include "libswscale/swscale.h" } #ifndef min #define min(a,b) ((a) > (b) ? (b) : (a)) #endif #define _LL_DEBUG_ // low level debug #ifdef _LL_DEBUG_     #define debug(fmt, ...) printf(fmt, ##__VA_ARGS__)     #define LL_DEBUG(fmt, ...) printf("[DEBUG %s().%d @ %s]: " fmt, \     __func__, __LINE__, P_SRC, ##__VA_ARGS__) #else     #define debug(fmt, ...)     #define LL_DEBUG(fmt, ...) #endif // 自定義的IO全部在記憶體 #define USE_MEM #define DEFAULT_MEM (10*1024*1024) #define IO_BUFFER_SIZE (32768*1) static char g_ptr1[IO_BUFFER_SIZE] = {0}; static char* g_ptr = NULL; //static char g_ptr[DEFAULT_MEM] = {0}; static int g_nPos = 0; static int g_nTotalLen = DEFAULT_MEM; static int g_nRealLen = 0; static int g_fd = 0; int my_read(void *opaque, unsigned char *buf, int buf_size) {     // no     return 0; } int my_write(void *opaque, unsigned char *buf, int size) {     if (g_nPos + size > g_nTotalLen)     {         // 重新申請         // 根據數值逐步加大         int nTotalLen = g_nTotalLen*sizeof(char) * 3 / 2;         char* ptr = (char*)av_realloc(g_ptr, nTotalLen);         if (ptr == NULL)         { // if (g_ptr) av_free(g_ptr);             return -1;         }         debug("org ptr: %p new ptr: %p size: %d(%0.fMB) ", g_ptr, ptr,                      nTotalLen, nTotalLen/1024.0/1024.0);         g_nTotalLen = nTotalLen;         g_ptr = ptr;         debug(" realloc!!!!!!!!!!!!!!!!!!!!!!!\n");         // todo         //memcpy(g_ptr + g_nPos, buf, size);     }     memcpy(g_ptr + g_nPos, buf, size);     if (g_nPos + size >= g_nRealLen)         g_nRealLen += size;          static int cnt = 1;     debug("%d write %p %p pos: %d len: %d\n", cnt++, g_ptr, buf, g_nPos, size);     //dump("org ", (char*)buf, 32);     //dump("g_ptr ", g_ptr, 32);          g_nPos += size;     return 0; } int64_t my_seek(void *opaque, int64_t offset, int whence) {     int64_t new_pos = 0; // 可以為負數     int64_t fake_pos = 0;     switch (whence)     {         case SEEK_SET:             new_pos = offset;             break;         case SEEK_CUR:             new_pos = g_nPos + offset;             break;         case SEEK_END: // 此處可能有問題             new_pos = g_nTotalLen + offset;             break;         default:             return -1;     }          fake_pos = min(new_pos, g_nTotalLen);     if (fake_pos != g_nPos)     {         g_nPos = fake_pos;     }     debug("seek pos: %d(%d)\n", offset, g_nPos);     return new_pos; } int my_write1(void *opaque, unsigned char *buf, int size) {     debug("write %p len: %d\n",  buf, size);     //dump("org ", (char*)buf, 32);     //dump("g_ptr ", (char*)g_ptr, 32);     return write(g_fd, buf, size); } int64_t my_seek1(void *opaque, int64_t offset, int whence) {     debug("seek pos: %d whence: %d\n", offset, whence);     return lseek(g_fd, offset, whence); } // 定義2個函式指標，免得後面改 static int (*write_packet)(void *opaque, uint8_t *buf, int buf_size) =  #ifdef USE_MEM my_write #else my_write1 #endif ; static int64_t (*seek)(void *opaque, int64_t offset, int whence) = #ifdef USE_MEM my_seek #else my_seek1 #endif ; int remuxer_mem(int argc, char* argv[]) {     //輸入對應一個AVFormatContext，輸出對應一個AVFormatContext     //（Input AVFormatContext and Output AVFormatContext）     AVFormatContext *ifmt_ctx = NULL, *ofmt_ctx = NULL;     AVPacket pkt;     const char *in_filename, *out_filename;     int ret = 0;     AVIOContext *avio_out = NULL;      if (argc < 3)     {         printf("usage: %s [input file] [output file]\n", argv[0]);         printf("eg %s foo.avi bar.ts\n", argv[0]);         return -1;     }     in_filename  = argv[1];     out_filename = argv[2];     // 分配空間     g_ptr = (char*)av_realloc(NULL, DEFAULT_MEM*sizeof(char));  // new     if (g_ptr == NULL)     {         debug("alloc mem failed.\n");         return -1;     }     printf("------alloc ptr: %p\n", g_ptr);     memset(g_ptr, '\0', DEFAULT_MEM);     av_register_all();     //輸入（Input）     if ((ret = avformat_open_input(&ifmt_ctx, in_filename, 0, 0)) < 0)     {         printf( "Could not open input file '%s': %s\n", in_filename, strerror(errno));         goto end;     }     if ((ret = avformat_find_stream_info(ifmt_ctx, 0)) < 0)     {         printf( "Failed to retrieve input stream information\n");         goto end;     }     printf("input format:\n");     av_dump_format(ifmt_ctx, 0, in_filename, 0);     //輸出（Output）          // 先開啟要儲存的檔案     g_fd = open(out_filename, O_CREAT|O_RDWR, 0666);          // 分配自定義的AVIOContext     // 注意，參考file協議的記憶體，使用大小32768，而g_ptr是不同一片記憶體的     avio_out =avio_alloc_context((unsigned char *)g_ptr1, IO_BUFFER_SIZE, 1,                 NULL, NULL, write_packet, seek);      if (!avio_out)     {         printf( "avio_alloc_context failed\n");         ret = AVERROR_UNKNOWN;         goto end;     }     // 分配AVFormatContext     // out_filename 根據輸出副檔名來判斷格式     // 注意該函式會分配AVOutputFormat     avformat_alloc_output_context2(&ofmt_ctx, NULL, NULL, out_filename);     if (!ofmt_ctx)     {         printf( "Could not create output context\n");         ret = AVERROR_UNKNOWN;         goto end;     }     ofmt_ctx->pb=avio_out; // 賦值自定義的IO結構體     ofmt_ctx->flags=AVFMT_FLAG_CUSTOM_IO; // 指定為自定義     debug("guess format: %s(%s) flag: %d\n", ofmt_ctx->oformat->name,              ofmt_ctx->oformat->long_name, ofmt_ctx->oformat->flags);     for (int i = 0; i < (int)ifmt_ctx->nb_streams; i++)     {         //根據輸入流建立輸出流（Create output AVStream according to input AVStream）         AVStream *in_stream = ifmt_ctx->streams[i];         AVStream *out_stream = avformat_new_stream(ofmt_ctx, in_stream->codec->codec);         if (!out_stream)         {             printf( "Failed allocating output stream\n");             ret = AVERROR_UNKNOWN;             goto end;         }         //複製AVCodecContext的設定（Copy the settings of AVCodecContext）         ret = avcodec_copy_context(out_stream->codec, in_stream->codec);         if (ret < 0)         {             printf( "Failed to copy context from input to output stream codec context\n");             goto end;         }         out_stream->codec->codec_tag = 0;         if (ofmt_ctx->oformat->flags & AVFMT_GLOBALHEADER)             out_stream->codec->flags |= CODEC_FLAG_GLOBAL_HEADER;     }     //輸出一下格式------------------     printf("output format:\n");     av_dump_format(ofmt_ctx, 0, out_filename, 1);     //開啟輸出檔案（Open output file）     // !!!! 存在於記憶體，不需要明確指定開啟，這種形式是針對檔案或網路協議的     /*     if (!(ofmt->flags & AVFMT_NOFILE))     {         ret = avio_open(&ofmt_ctx->pb, out_filename, AVIO_FLAG_WRITE);         if (ret < 0)         {             printf( "Could not open output file '%s': %s\n", out_filename, strerror(errno));             goto end;         }     }     */     //寫檔案頭（Write file header）     ret = avformat_write_header(ofmt_ctx, NULL);     if (ret < 0)     {         printf( "Error occurred when opening output file\n");         goto end;     }     while (1)     {         AVStream *in_stream, *out_stream;         //獲取一個AVPacket（Get an AVPacket）         ret = av_read_frame(ifmt_ctx, &pkt);         if (ret < 0)             break;         in_stream  = ifmt_ctx->streams[pkt.stream_index];         out_stream = ofmt_ctx->streams[pkt.stream_index];         /* copy packet */         //轉換PTS/DTS（Convert PTS/DTS）         pkt.pts = av_rescale_q_rnd(pkt.pts, in_stream->time_base,              out_stream->time_base, (AVRounding)(AV_ROUND_NEAR_INF|AV_ROUND_PASS_MINMAX));         pkt.dts = av_rescale_q_rnd(pkt.dts, in_stream->time_base,             out_stream->time_base, (AVRounding)(AV_ROUND_NEAR_INF|AV_ROUND_PASS_MINMAX));         pkt.duration = av_rescale_q(pkt.duration, in_stream->time_base, out_stream->time_base);         pkt.pos = -1;         //寫入（Write）         ret = av_interleaved_write_frame(ofmt_ctx, &pkt);         if (ret < 0) {             printf( "Error muxing packet\n");             break;         }         av_free_packet(&pkt);     }     //寫檔案尾（Write file trailer）     av_write_trailer(ofmt_ctx); end:     avformat_close_input(&ifmt_ctx);     avformat_free_context(ofmt_ctx);     #ifdef USE_MEM     write(g_fd, g_ptr, g_nRealLen);     #endif     av_freep(&avio_out);     av_freep(&g_ptr);     close(g_fd);     return ret; }