You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
163 lines
5.5 KiB
163 lines
5.5 KiB
//
|
|
// Created by frank on 2018/2/2.
|
|
//
|
|
|
|
#include <jni.h>
|
|
#include <string>
|
|
#include "ffmpeg_jni_define.h"
|
|
|
|
#define TAG "FFmpegPusher"
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
#include <libavcodec/avcodec.h>
|
|
#include <libavformat/avformat.h>
|
|
#include <libavfilter/avfilter.h>
|
|
#include <libavutil/time.h>
|
|
|
|
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 |