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