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.
 
 
 
 
 
 
FFmpegAndroid/app/src/main/cpp/ffmpeg_pusher.cpp

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