translate the note pusher into English

translate the note pusher into English
pull/166/head
xufulong 5 years ago
parent d920f7974f
commit 781ec7444e
  1. 92
      app/src/main/cpp/ffmpeg_pusher.cpp

@ -34,142 +34,132 @@ PUSHER_FUNC(jint, pushStream, jstring filePath, jstring liveUrl) {
LOGE(TAG, "file_path=%s", file_path); LOGE(TAG, "file_path=%s", file_path);
LOGE(TAG, "live_url=%s", live_url); LOGE(TAG, "live_url=%s", live_url);
//注册所有组件
av_register_all(); av_register_all();
//初始化网络
avformat_network_init(); avformat_network_init();
//打开输入文件 if ((ret = avformat_open_input(&in_format, file_path, 0, 0)) < 0) {
if((ret = avformat_open_input(&in_format, file_path, 0, 0)) < 0){
LOGE(TAG, "could not open input file..."); LOGE(TAG, "could not open input file...");
goto end; goto end;
} }
//寻找流信息 if ((ret = avformat_find_stream_info(in_format, 0)) < 0) {
if((ret = avformat_find_stream_info(in_format, 0)) < 0){
LOGE(TAG, "could not find stream info..."); LOGE(TAG, "could not find stream info...");
goto end; goto end;
} }
// for (i = 0; i < in_format->nb_streams; i++) {
for(i=0; i<in_format->nb_streams; i++){ if (in_format->streams[i]->codec->codec_type == AVMEDIA_TYPE_VIDEO) {
//找到视频流
if(in_format->streams[i]->codec->codec_type == AVMEDIA_TYPE_VIDEO){
video_index = i; video_index = i;
break; break;
} }
} }
av_dump_format(in_format, 0, file_path, 0); av_dump_format(in_format, 0, file_path, 0);
//分配输出封装格式上下文, rtmp协议支持格式为flv
avformat_alloc_output_context2(&out_format, NULL, "flv", live_url); avformat_alloc_output_context2(&out_format, NULL, "flv", live_url);
if(!out_format){ if (!out_format) {
LOGE(TAG, "could not alloc output context..."); LOGE(TAG, "could not alloc output context...");
ret = AVERROR_UNKNOWN; ret = AVERROR_UNKNOWN;
goto end; goto end;
} }
//根据输入流来创建输出流 //Create output stream according to input stream
for(i=0; i<in_format->nb_streams; i++){ for (i = 0; i < in_format->nb_streams; i++) {
AVStream *in_stream = in_format->streams[i]; AVStream *in_stream = in_format->streams[i];
AVStream *out_stream = avformat_new_stream(out_format, in_stream->codec->codec); AVStream *out_stream = avformat_new_stream(out_format, in_stream->codec->codec);
if(!out_stream){ if (!out_stream) {
LOGE(TAG, "could not alloc output stream..."); LOGE(TAG, "could not alloc output stream...");
ret = AVERROR_UNKNOWN; ret = AVERROR_UNKNOWN;
goto end; goto end;
} }
//复制封装格式上下文 //Copy context
ret = avcodec_copy_context(out_stream->codec, in_stream->codec); ret = avcodec_copy_context(out_stream->codec, in_stream->codec);
if(ret < 0){ if (ret < 0) {
LOGE(TAG, "could not copy context..."); LOGE(TAG, "could not copy context...");
goto end; goto end;
} }
out_stream->codec->codec_tag = 0; out_stream->codec->codec_tag = 0;
if(out_format->oformat->flags & AVFMT_GLOBALHEADER){ if (out_format->oformat->flags & AVFMT_GLOBALHEADER) {
out_stream->codec->flags |= CODEC_FLAG_GLOBAL_HEADER; out_stream->codec->flags |= CODEC_FLAG_GLOBAL_HEADER;
} }
} }
//封装格式
output_format = out_format->oformat; output_format = out_format->oformat;
//打开输出文件/URL if (!(output_format->flags & AVFMT_NOFILE)) {
if(!(output_format->flags & AVFMT_NOFILE)){
ret = avio_open(&out_format->pb, live_url, AVIO_FLAG_WRITE); ret = avio_open(&out_format->pb, live_url, AVIO_FLAG_WRITE);
if(ret < 0){ if (ret < 0) {
LOGE(TAG, "could not open output url '%s'", live_url); LOGE(TAG, "could not open output url '%s'", live_url);
goto end; goto end;
} }
} }
//写文件头 //Write header
ret = avformat_write_header(out_format, NULL); ret = avformat_write_header(out_format, NULL);
if(ret < 0){ if (ret < 0) {
LOGE(TAG, "could not write header..."); LOGE(TAG, "could not write header...");
goto end; goto end;
} }
//获取开始时间
start_time = av_gettime(); start_time = av_gettime();
//开始循环读一帧数据 while (1) {
while (1){
AVStream *in_stream, *out_stream; AVStream *in_stream, *out_stream;
ret = av_read_frame(in_format, &packet); ret = av_read_frame(in_format, &packet);
if(ret < 0){ if (ret < 0) {
break; break;
} }
//计算帧间隔,参考时钟/采样率 //计算帧间隔,参考时钟/采样率
if(packet.pts == AV_NOPTS_VALUE){ if (packet.pts == AV_NOPTS_VALUE) {
AVRational time_base = in_format->streams[video_index]->time_base; AVRational time_base = in_format->streams[video_index]->time_base;
int64_t cal_duration = (int64_t)(AV_TIME_BASE/av_q2d(in_format->streams[video_index]->r_frame_rate)); int64_t cal_duration = (int64_t) (AV_TIME_BASE /
packet.pts = (int64_t)((frame_index * cal_duration)/(av_q2d(time_base) * AV_TIME_BASE)); av_q2d(in_format->streams[video_index]->r_frame_rate));
packet.pts = (int64_t) ((frame_index * cal_duration) /
(av_q2d(time_base) * AV_TIME_BASE));
packet.dts = packet.pts; packet.dts = packet.pts;
packet.duration = (int64_t)(cal_duration/(av_q2d(time_base) * AV_TIME_BASE)); packet.duration = (int64_t) (cal_duration / (av_q2d(time_base) * AV_TIME_BASE));
} }
//视频帧之间延时 if (packet.stream_index == video_index) {
if(packet.stream_index == video_index){
AVRational time_base = in_format->streams[video_index]->time_base; AVRational time_base = in_format->streams[video_index]->time_base;
AVRational time_base_q = {1, AV_TIME_BASE}; AVRational time_base_q = {1, AV_TIME_BASE};
int64_t pts_time = av_rescale_q(packet.dts, time_base, time_base_q); int64_t pts_time = av_rescale_q(packet.dts, time_base, time_base_q);
int64_t now_time = av_gettime() - start_time; int64_t now_time = av_gettime() - start_time;
//延时以保持同步 //sleep to keep waiting
if(pts_time > now_time){ if (pts_time > now_time) {
av_usleep((unsigned int)(pts_time - now_time)); av_usleep((unsigned int) (pts_time - now_time));
} }
} }
in_stream = in_format->streams[packet.stream_index]; in_stream = in_format->streams[packet.stream_index];
out_stream = out_format->streams[packet.stream_index]; out_stream = out_format->streams[packet.stream_index];
//pts/dts转换 //pts to dts
packet.pts = av_rescale_q_rnd(packet.pts, in_stream->time_base, out_stream->time_base, packet.pts = av_rescale_q_rnd(packet.pts, in_stream->time_base, out_stream->time_base,
(AVRounding)(AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX)); (AVRounding) (AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX));
packet.dts = av_rescale_q_rnd(packet.dts, in_stream->time_base, out_stream->time_base, packet.dts = av_rescale_q_rnd(packet.dts, in_stream->time_base, out_stream->time_base,
(AVRounding)(AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX)); (AVRounding) (AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX));
packet.duration = av_rescale_q(packet.duration, in_stream->time_base, out_stream->time_base); packet.duration = av_rescale_q(packet.duration, in_stream->time_base,
out_stream->time_base);
packet.pos = -1; packet.pos = -1;
//视频帧计数 if (packet.stream_index == video_index) {
if(packet.stream_index == video_index){ frame_index++;
frame_index ++;
LOGI(TAG, "write frame = %d", frame_index); LOGI(TAG, "write frame = %d", frame_index);
} }
//写一帧数据 //Write a frame
ret = av_interleaved_write_frame(out_format, &packet); ret = av_interleaved_write_frame(out_format, &packet);
if(ret < 0){ if (ret < 0) {
LOGE(TAG, "could not write frame..."); LOGE(TAG, "could not write frame...");
break; break;
} }
//释放包数据内存 //Release packet
av_packet_unref(&packet); av_packet_unref(&packet);
} }
//写文件尾 //Write tail
av_write_trailer(out_format); av_write_trailer(out_format);
end: end:
avformat_close_input(&in_format); avformat_close_input(&in_format);
if(out_format && !(out_format->flags & AVFMT_NOFILE)){ if (out_format && !(out_format->flags & AVFMT_NOFILE)) {
avio_close(out_format->pb); avio_close(out_format->pb);
} }
avformat_free_context(in_format); avformat_free_context(in_format);
avformat_free_context(out_format); avformat_free_context(out_format);
env->ReleaseStringUTFChars(filePath, file_path); env->ReleaseStringUTFChars(filePath, file_path);
env->ReleaseStringUTFChars(liveUrl, live_url); env->ReleaseStringUTFChars(liveUrl, live_url);
if(ret < 0 && ret != AVERROR_EOF){ if (ret < 0 && ret != AVERROR_EOF) {
return -1; return -1;
} }
return 0; return 0;
} }
#ifdef __cplusplus #ifdef __cplusplus
} }

Loading…
Cancel
Save