// // Created by frank on 2018/2/2. // #include #include #include "ffmpeg_jni_define.h" #define TAG "FFmpegPusher" #ifdef __cplusplus extern "C" { #endif #include #include #include #include PUSHER_FUNC(jint, pushStream, jstring filePath, jstring liveUrl) { AVFormatContext *in_format = nullptr, *out_format = nullptr; AVPacket packet; const char *file_path, *live_url; int video_index = -1; int ret = 0, i; int frame_index = 0; int64_t start_time = 0; file_path = env->GetStringUTFChars(filePath, nullptr); live_url = env->GetStringUTFChars(liveUrl, nullptr); LOGE(TAG, "file_path=%s", file_path); LOGE(TAG, "live_url=%s", live_url); av_register_all(); avformat_network_init(); if ((ret = avformat_open_input(&in_format, file_path, 0, 0)) < 0) { LOGE(TAG, "could not open input file..."); goto end; } 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, nullptr, "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; } //Copy context ret = avcodec_copy_context(out_stream->codec, in_stream->codec); if (ret < 0) { LOGE(TAG, "could not copy context..."); goto end; } out_stream->codec->codec_tag = 0; if (out_format->oformat->flags & AVFMT_GLOBALHEADER) { // out_stream->codec->flags |= CODEC_FLAG_GLOBAL_HEADER; } } if (!(out_format->oformat->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; } } //Write header ret = avformat_write_header(out_format, nullptr); if (ret < 0) { LOGE(TAG, "could not write header..."); goto end; } start_time = av_gettime(); while (1) { AVStream *in_stream, *out_stream; ret = av_read_frame(in_format, &packet); if (ret < 0) { break; } // calculate pts and dts 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) { AVRational time_base = in_format->streams[video_index]->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 now_time = av_gettime() - start_time; //sleep to keep waiting 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 to dts 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)); 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)); packet.duration = av_rescale_q(packet.duration, in_stream->time_base, out_stream->time_base); packet.pos = -1; if (packet.stream_index == video_index) { frame_index++; LOGI(TAG, "write frame = %d", frame_index); } //Write a frame ret = av_interleaved_write_frame(out_format, &packet); if (ret < 0) { LOGE(TAG, "could not write frame..."); break; } //Release packet av_packet_unref(&packet); } //Write tail av_write_trailer(out_format); end: if (out_format && !(out_format->flags & AVFMT_NOFILE)) { avio_close(out_format->pb); } avformat_free_context(out_format); avformat_close_input(&in_format); env->ReleaseStringUTFChars(filePath, file_path); env->ReleaseStringUTFChars(liveUrl, live_url); if (ret < 0 && ret != AVERROR_EOF) { return -1; } return 0; } #ifdef __cplusplus } #endif