forked from xufuji456/FFmpegAndroid
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathffmpeg_pusher.cpp
More file actions
176 lines (164 loc) · 6.38 KB
/
ffmpeg_pusher.cpp
File metadata and controls
176 lines (164 loc) · 6.38 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
//
// Created by frank on 2018/2/2.
//
#include <jni.h>
#include <string>
#include <android/log.h>
#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) {
AVOutputFormat *output_format = NULL;
AVFormatContext *in_format = NULL, *out_format = NULL;
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, NULL);
live_url = env->GetStringUTFChars(liveUrl, NULL);
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);
//分配输出封装格式上下文， rtmp协议支持格式为flv
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;
}
//根据输入流来创建输出流
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;
}
//复制封装格式上下文
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;
}
}
//封装格式
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(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;
}
//计算帧间隔，参考时钟/采样率
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;
//延时以保持同步
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转换
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);
}
//写一帧数据
ret = av_interleaved_write_frame(out_format, &packet);
if(ret < 0){
LOGE(TAG, "could not write frame...");
break;
}
//释放包数据内存
av_packet_unref(&packet);
}
//写文件尾
av_write_trailer(out_format);
end:
avformat_close_input(&in_format);
if(out_format && !(out_format->flags & AVFMT_NOFILE)){
avio_close(out_format->pb);
}
avformat_free_context(in_format);
avformat_free_context(out_format);
env->ReleaseStringUTFChars(filePath, file_path);
env->ReleaseStringUTFChars(liveUrl, live_url);
if(ret < 0 && ret != AVERROR_EOF){
return -1;
}
return 0;
}
#ifdef __cplusplus
}
#endif