android推送rtmp,ffmpeg android 推送rtmp

static int (const char *input, const char *output){

AVFormatContext *pInAvFormatContext = NULL;

AVFormatContext *pOutAvFormatContext = NULL;

AVOutputFormat *pAvOutputFormat = NULL;

int videoIndex = 0;

//1. 注册

av_register_all();

LOGE("%s", output);

//2. 获取输入的文件信息

//打开文件

if (avformat_open_input(&pInAvFormatContext, input, NULL, NULL) != 0) {

LOGE("打开文件失败!");

return -1;

}

//获取流信息

if (avformat_find_stream_info(pInAvFormatContext, NULL) < 0) {

LOGE("获取文件流失败!");

return -1;

}

//找到video的对应位置

int i = 0;

for (; i < pInAvFormatContext->nb_streams; i++) {

if (pInAvFormatContext->streams[i]->codec->codec_type == AVMEDIA_TYPE_VIDEO) {

videoIndex = i;

break;

}

}

//打印信息

// av_dump_format(pAvFormatContext,0,input,0);

//3. 输出设置

//初始化输出AVFormatContext获取 AVOutputFormat进行设置

avformat_alloc_output_context2(&pOutAvFormatContext, NULL, "flv", output);

if (!pOutAvFormatContext) {

LOGE("初始化输出AVFormatContext失败");

return -1;

}

pAvOutputFormat = pOutAvFormatContext->oformat;

i = 0;

for (; i < pInAvFormatContext->nb_streams; i++) {

AVStream *in_stream = pInAvFormatContext->streams[i];

AVStream *out_stream = avformat_new_stream(pOutAvFormatContext, in_stream->codec->codec);

if (!out_stream) {

LOGE("初始化out_stream失败");

return -1;

}

//复制AVCodecContext的设置(Copy the settings of AVCodecContext)

if (avcodec_copy_context(out_stream->codec, in_stream->codec) != 0) {

LOGE("copy AVCodecContext设置失败");

return -1;

}

out_stream->codec->codec_tag = 0;

if (pOutAvFormatContext->oformat->flags & AVFMT_GLOBALHEADER)

out_stream->codec->flags |= CODEC_FLAG_GLOBAL_HEADER;

}

//打印

// av_dump_format(pOutAvFormatContext,0,output,0);

//打开输出文件/url

if (!(pAvOutputFormat->flags & AVFMT_NOFILE)) {

int ret = avio_open(&pOutAvFormatContext->pb, output, AVIO_FLAG_WRITE);

if (ret < 0) {

LOGE("打开输出文件或者url失败 %d", ret);

return -1;

}

}

//4.写入数据

//写文件头

if (avformat_write_header(pOutAvFormatContext, NULL) != 0) {

LOGE("写入头数据失败");

return -1;

}

int64_t start_time = av_gettime();

AVPacket pkt;

int frameIndex = 0;

//写入数据源

while (av_read_frame(pInAvFormatContext, &pkt) >= 0) {

AVStream *in_stream, *out_stream;

if (pkt.pts == AV_NOPTS_VALUE) {

//Write PTS

AVRational time_base1 = pInAvFormatContext->streams[videoIndex]->time_base;

//Duration between 2 frames (us)

int64_t calc_duration = (double) AV_TIME_BASE /

av_q2d(pInAvFormatContext->streams[videoIndex]->r_frame_rate);

//Parameters

pkt.pts = (double) (frameIndex * calc_duration) /

(double) (av_q2d(time_base1) * AV_TIME_BASE);

pkt.dts = pkt.pts;

pkt.duration = (double) calc_duration / (double) (av_q2d(time_base1) * AV_TIME_BASE);

}

//Important:Delay

if (pkt.stream_index == videoIndex) {

AVRational time_base = pInAvFormatContext->streams[videoIndex]->time_base;

AVRational time_base_q = {1, AV_TIME_BASE};

int64_t pts_time = av_rescale_q(pkt.dts, time_base, time_base_q);

int64_t now_time = av_gettime() - start_time;

if (pts_time > now_time)

av_usleep(pts_time - now_time);

}

in_stream = pInAvFormatContext->streams[pkt.stream_index];

out_stream = pOutAvFormatContext->streams[pkt.stream_index];

/* copy packet */

//转换PTS/DTS(Convert PTS/DTS)

pkt.pts = av_rescale_q_rnd(pkt.pts, in_stream->time_base, out_stream->time_base,

(AVRounding) (AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX));

pkt.dts = av_rescale_q_rnd(pkt.dts, in_stream->time_base, out_stream->time_base,

(AVRounding) (AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX));

pkt.duration = av_rescale_q(pkt.duration, in_stream->time_base, out_stream->time_base);

pkt.pos = -1;

//Print to Screen

if (pkt.stream_index == videoIndex) {

LOGE("Send %8d video frames to output URLn", frameIndex);

frameIndex++;

}

if (av_interleaved_write_frame(pOutAvFormatContext, &pkt) < 0) {

LOGE("Error muxing packetn");

break;

}

av_packet_unref(&pkt);

}

//写文件尾(Write file trailer)

av_write_trailer(pOutAvFormatContext);

avformat_close_input(&pInAvFormatContext);

/* close output */

if (pOutAvFormatContext && !(pAvOutputFormat->flags & AVFMT_NOFILE))

avio_close(pOutAvFormatContext->pb);

avformat_free_context(pOutAvFormatContext);

return 0;

}

JNIEXPORT jint JNICALL

Java_zzw_com_ffmpegdemo_VideoUtils_publish_1file_1stream(JNIEnv *env, jclass type, jstring input_,

jstring output_) {

const char *input = env->GetStringUTFChars(input_, 0);

const char *output = env->GetStringUTFChars(output_, 0);

//input: /storage/emulated/0/aaaaa/dst.mp4

//output: rtmp://192.168.18.231:8082/live/room

int ret = publish_file_stream(input, output);

env->ReleaseStringUTFChars(input_, input);

env->ReleaseStringUTFChars(output_, output);

return ret;

}