fill AVCodecContext with Header Infomation from rkmpp

Author: DevoutPrayer

mpp.c

@@ -1,7 +1,7 @@
 #include "mpp.h"
 static void mpp_close(MppContext* ctx);
 static void init_mpp(MppContext *mpp_enc_data);
-static _Bool write_header(MppContext *mpp_enc_data);
+static _Bool write_header(MppContext *mpp_enc_data,SpsHeader *sps_header);
 static _Bool process_image(uint8_t *p, int size,MppContext *mpp_enc_data);
 
 MppContext * alloc_mpp_context()
@@ -213,7 +213,7 @@ static void init_mpp(MppContext *mpp_enc_data)
         printf("init mpp failed!\n");
 }
 
-static _Bool write_header(MppContext *mpp_enc_data)
+static _Bool write_header(MppContext *mpp_enc_data,SpsHeader *sps_header)
 {
         int ret;
         if (mpp_enc_data->type == MPP_VIDEO_CodingAVC)
@@ -229,105 +229,85 @@ static _Bool write_header(MppContext *mpp_enc_data)
 		/* get and write sps/pps for H.264 */
 		if (packet)
 		{
-
 			void *ptr   = mpp_packet_get_pos(packet);
 			size_t len  = mpp_packet_get_length(packet);
-
-			if (mpp_enc_data->fp_output)
-                        {
-                                fwrite(ptr, 1, len, mpp_enc_data->fp_output);
-                                return 0;
-                        }
-                        // if(mpp_enc_data->write_frame)
-                        //         if(!(mpp_enc_data->write_frame)(ptr,len))
-                        //                         printf("------------sendok!\n");
+			sps_header->data = (uint8_t*)malloc(len);
+			sps_header->size = len;
+			memcpy(sps_header->data,ptr,len);
 			packet = NULL;
 		}
 	}
         return 1;
 }
 
 static _Bool process_image(uint8_t *p, int size,MppContext *mpp_enc_data)
-{
-//     if(mpp_enc_data->fp_outputx)
-//     {
-//         printf("ok\n");
-//         //fwrite(p, 1, size, mpp_enc_data.fp_outputx);
-//     }
-    
+{   
 	MPP_RET ret = MPP_OK;
-    
-     
+	MppFrame frame = NULL;
+	MppPacket packet = NULL;
 
-        MppFrame frame = NULL;
-        MppPacket packet = NULL;
-
-        void *buf = mpp_buffer_get_ptr(mpp_enc_data->frm_buf);
-        //TODO: improve performance here?
-        memcpy(buf, p, size);
-        ret = mpp_frame_init(&frame);
-        if (ret)
-        {
-                printf("mpp_frame_init failed\n");
-                return 1;
-        }
+	void *buf = mpp_buffer_get_ptr(mpp_enc_data->frm_buf);
+	//TODO: improve performance here?
+	memcpy(buf, p, size);
+	ret = mpp_frame_init(&frame);
+	if (ret)
+	{
+		printf("mpp_frame_init failed\n");
+		return 1;
+	}
 
-        mpp_frame_set_width(frame, mpp_enc_data->width);
-        mpp_frame_set_height(frame, mpp_enc_data->height);
-        mpp_frame_set_hor_stride(frame, mpp_enc_data->hor_stride);
-        mpp_frame_set_ver_stride(frame, mpp_enc_data->ver_stride);
-        mpp_frame_set_fmt(frame, mpp_enc_data->fmt);
-        mpp_frame_set_buffer(frame, mpp_enc_data->frm_buf);
-        mpp_frame_set_eos(frame, mpp_enc_data->frm_eos);
+	mpp_frame_set_width(frame, mpp_enc_data->width);
+	mpp_frame_set_height(frame, mpp_enc_data->height);
+	mpp_frame_set_hor_stride(frame, mpp_enc_data->hor_stride);
+	mpp_frame_set_ver_stride(frame, mpp_enc_data->ver_stride);
+	mpp_frame_set_fmt(frame, mpp_enc_data->fmt);
+	mpp_frame_set_buffer(frame, mpp_enc_data->frm_buf);
+	mpp_frame_set_eos(frame, mpp_enc_data->frm_eos);
 
-        ret = mpp_enc_data->mpi->encode_put_frame(mpp_enc_data->ctx, frame);
-        if (ret)
-        {
-                printf("mpp encode put frame failed\n");
-                return 1;
-        }
+	ret = mpp_enc_data->mpi->encode_put_frame(mpp_enc_data->ctx, frame);
+	if (ret)
+	{
+		printf("mpp encode put frame failed\n");
+		return 1;
+	}
 
 mdddd:
-        ret = mpp_enc_data->mpi->encode_get_packet(mpp_enc_data->ctx, &packet);
-        if (ret)
-        {
-                printf("mpp encode get packet failed\n");
-                return 1;
-        }
-    
-        if (packet)
-        {
-                // write packet to file here
-                void *ptr   = mpp_packet_get_pos(packet);
-                size_t len  = mpp_packet_get_length(packet);
-                mpp_enc_data->pkt_eos = mpp_packet_get_eos(packet);
-                if (mpp_enc_data->fp_output)
-                {
-                	fwrite(ptr, 1, len, mpp_enc_data->fp_output);   
-                }
-                if(!mpp_enc_data->fp_output&&mpp_enc_data->write_frame)
-                        if(!(mpp_enc_data->write_frame)(ptr,len))
-                                        printf("------------sendok!\n");
-
-                mpp_packet_deinit(&packet);
-                printf("encoded frame %d size %ld\n", mpp_enc_data->frame_count, len);
-                mpp_enc_data->stream_size += len;
-                mpp_enc_data->frame_count++;
-
-                if (mpp_enc_data->pkt_eos)
-                {
-                        printf("found last packet\n");
-                }
-        }
-        else
-                goto mdddd;
-        if (mpp_enc_data->num_frames && mpp_enc_data->frame_count >= mpp_enc_data->num_frames)
-        {
-        	printf("encode max %d frames", mpp_enc_data->frame_count);
-        	return 0;
-        }
-        if (mpp_enc_data->frm_eos && mpp_enc_data->pkt_eos)
-                return 0;
-    
-        return 1;
+	ret = mpp_enc_data->mpi->encode_get_packet(mpp_enc_data->ctx, &packet);
+	if (ret)
+	{
+		printf("mpp encode get packet failed\n");
+		return 1;
+	}
+
+	if (packet)
+	{
+		// write packet to file here
+		void *ptr   = mpp_packet_get_pos(packet);
+		size_t len  = mpp_packet_get_length(packet);
+		mpp_enc_data->pkt_eos = mpp_packet_get_eos(packet);
+		if(mpp_enc_data->write_frame)
+			if(!(mpp_enc_data->write_frame)(ptr,len))
+				printf("------------sendok!\n");
+
+		mpp_packet_deinit(&packet);
+		printf("encoded frame %d size %ld\n", mpp_enc_data->frame_count, len);
+		mpp_enc_data->stream_size += len;
+		mpp_enc_data->frame_count++;
+
+		if (mpp_enc_data->pkt_eos)
+		{
+			printf("found last packet\n");
+		}
+	}
+	else
+		goto mdddd;
+	if (mpp_enc_data->num_frames && mpp_enc_data->frame_count >= mpp_enc_data->num_frames)
+	{
+		printf("encode max %d frames", mpp_enc_data->frame_count);
+		return 0;
+	}
+	if (mpp_enc_data->frm_eos && mpp_enc_data->pkt_eos)
+		return 0;
+
+	return 1;
 }

mpp.h

@@ -19,6 +19,11 @@
 
 #define MPP_ALIGN(x, a)         (((x)+(a)-1)&~((a)-1))
 
+typedef struct {
+        uint8_t *data;
+        uint32_t size;
+} SpsHeader;
+
 typedef struct {
         // global flow control flag
         RK_U32 frm_eos;
@@ -62,7 +67,7 @@ typedef struct {
         //function pointer
         void (*init_mpp)(void *mpp_enc_data);
         _Bool (*process_image)(uint8_t *p, int size,void *mpp_enc_data);
-        _Bool (*write_header)(void *mpp_enc_data);
+        _Bool (*write_header)(void *mpp_enc_data,SpsHeader *sps_header);
         void (*close)(void* ctx);
 
 } MppContext;

rtmp.c

@@ -7,17 +7,8 @@
 #include "rtmp.h"
 
 static AVPacket pkt;
-static char *outfile_name = "rtmp://192.168.1.165:1935/live/room1";
-static AVOutputFormat *ofmt = NULL;
 static AVFormatContext *ofmt_ctx = NULL;
-static AVFormatContext *ifmt_ctx = NULL;
-static int ret;
-static int i;
-static int videoindex=0;
 static int frame_index=0;
-static int64_t start_time=0;
-static char *infile_name = "/tmp/output.h264";
-static int ceshi;
 
 int write_frame(uint8_t*data,int size)
 {
@@ -29,49 +20,31 @@ int write_frame(uint8_t*data,int size)
         // pkt.dts = frame_index*pkt.duration;
         // pkt.pts = pkt.dts;
         // frame_index ++;
-        ret = av_write_frame(ofmt_ctx, &pkt);
- 
-        if (ret < 0) {
+
+        if (av_write_frame(ofmt_ctx, &pkt) < 0) {
                 printf( "Error muxing packet\n");
                 return -1;
         }
         return 0;
 }
 
-
-
-int init_rtmp_streamer(char* stream)
+int init_rtmp_streamer(char* stream,uint8_t *data,uint32_t size)
 {
         int ret;
         av_register_all();
-outfile_name=stream;
         if((ret = avformat_network_init()) < 0)
         {
                 fprintf(stderr, "avformat_network_init failed!");
                 return -1;
         }
 
-        avformat_alloc_output_context2(&ofmt_ctx,NULL,"flv",outfile_name);
+        avformat_alloc_output_context2(&ofmt_ctx,NULL,"flv",stream);
         if(!ofmt_ctx)
         {
                 fprintf(stderr, "Could not create output context\n");
                 return -1;
         }
-        ofmt = ofmt_ctx->oformat;
-
-        if((ret = avformat_open_input(&ifmt_ctx,infile_name,0,0)) < 0)
-        {
-                fprintf(stderr, "Could not open input file '%s' %d", infile_name,ret);
-                return -1;
-        }
-
-        if ((ret = avformat_find_stream_info(ifmt_ctx, 0)) < 0) {
-		fprintf(stderr, "Failed to retrieve input stream information");
-		return -1;
-        }
-        av_dump_format(ifmt_ctx, 0, infile_name, 0);
-
-
+        
 
         AVStream *out_stream = avformat_new_stream(ofmt_ctx,NULL);
         if(! out_stream)
@@ -80,43 +53,41 @@ outfile_name=stream;
                 goto end;
         }
 
-        ret = avcodec_copy_context(out_stream->codec,ifmt_ctx->streams[0]->codec);
-        if(ret < 0)
-        {
-                printf("Fail to copy context from out stream to input stream!\n");
-                goto end;
-        }
-        
-
-        out_stream->codec->codec_tag = 0;
-        if (ofmt_ctx->oformat->flags & AVFMT_GLOBALHEADER)
-                out_stream->codec->flags |= AV_CODEC_FLAG_GLOBAL_HEADER;
-
-
-           
-        av_dump_format(ofmt_ctx,0,outfile_name,1);
+        AVCodecContext *o_codec_ctx;
+        o_codec_ctx = out_stream->codec;
+        o_codec_ctx->codec_id = AV_CODEC_ID_H264;
+        o_codec_ctx->codec_type = AVMEDIA_TYPE_VIDEO;
+        o_codec_ctx->codec_tag = 0;
+        o_codec_ctx->pix_fmt = AV_PIX_FMT_YUV420P;
+        o_codec_ctx->width = 640;
+        o_codec_ctx->height = 480;
+        o_codec_ctx->extradata = data;
+        o_codec_ctx->extradata_size = size;
+                
+        av_dump_format(ofmt_ctx,0,stream,1);
 
-        //打开输出URL（Open output URL）
-	if (!(ofmt->flags & AVFMT_NOFILE)) {
-		ret = avio_open(&ofmt_ctx->pb, outfile_name, AVIO_FLAG_WRITE);
-		if (ret < 0) {
-			printf( "Could not open output URL '%s'", outfile_name);
-			goto end;
-		}
-	}
+        // 打开输出URL（Open output URL）
+        if (!(ofmt_ctx->oformat->flags & AVFMT_NOFILE)) {
+                ret = avio_open(&ofmt_ctx->pb, stream, AVIO_FLAG_WRITE);
+                if (ret < 0) {
+                        printf( "Could not open output URL '%s'", stream);
+                        goto end;
+                }
+        }
         //写文件头（Write file header）
-	ret = avformat_write_header(ofmt_ctx, NULL);
-	if (ret < 0) {
-		printf( "Error occurred when opening output URL\n");
-		goto end;
-	}
+        ret = avformat_write_header(ofmt_ctx, NULL);
+        if (ret < 0) {
+                printf( "Error occurred when opening output URL\n");
+                goto end;
+        }
+        // free(data);
         return 0;
 end:
-        if (ofmt_ctx && !(ofmt->flags & AVFMT_NOFILE))
-		avio_close(ofmt_ctx->pb);
-	avformat_free_context(ofmt_ctx);
-	if (ret < 0 && ret != AVERROR_EOF) {
-		printf( "Error occurred.\n");
-		return -1;
-	}
+        if (ofmt_ctx && !(ofmt_ctx->oformat->flags & AVFMT_NOFILE))
+                avio_close(ofmt_ctx->pb);
+        avformat_free_context(ofmt_ctx);
+        if (ret < 0 && ret != AVERROR_EOF) {
+                printf( "Error occurred.\n");
+                return -1;
+        }
 }

rtmp.h

@@ -5,7 +5,7 @@
         {
 #endif
 
-int init_rtmp_streamer(char *  stream);
+int init_rtmp_streamer(char* stream,uint8_t* data,uint32_t size);
 int write_frame(uint8_t*data,int size);
 #ifdef __cplusplus
         }