接上一篇。
activity_flv.xml
xmlns:tools&＃61;"http://schemas.android.com/tools"
android:layout_width&＃61;"match_parent"
android:layout_height&＃61;"match_parent"
tools:context&＃61;".FlvActivity">
android:layout_width&＃61;"match_parent"
android:layout_height&＃61;"match_parent"
android:orientation&＃61;"horizontal">
android:id&＃61;"&＃64;&＃43;id/take_button"
android:layout_width&＃61;"wrap_content"
android:layout_height&＃61;"match_parent"
android:text&＃61;"open"/>
android:id&＃61;"&＃64;&＃43;id/surfaceView1"
android:layout_width&＃61;"0dp"
android:layout_height&＃61;"match_parent"
android:layout_weight&＃61;"1"/>
最后偷懒贴个图build.gradle
build.gradle
忘记上jni文件了&＃xff0c;直接走起
#include
#include"libavcodec/avcodec.h"
#include"libavformat/avformat.h"
#include"libavutil/time.h"
#ifdefANDROID
#include
#include
#defineLOGE(format, ...) __android_log_print(ANDROID_LOG_ERROR,"(>_
#defineLOGI(format, ...) __android_log_print(ANDROID_LOG_INFO,"(&＃61;_&＃61;)",format, ##__VA_ARGS__)
#else
#define LOGE(format, ...) printf("(>_
#define LOGI(format, ...) printf("(^_^) "format"\n", ##__VA_ARGS__)
#endif
AVFormatContext*ofmt_ctx;
AVStream* video_st;
AVCodecContext* pCodecCtx;
AVCodec* pCodec;
AVPacketenc_pkt;
AVFrame*pFrameYUV;
intframecnt &＃61;0;
intyuv_width;
intyuv_height;
inty_length;
intuv_length;
int64_tstart_time;
//Output FFmpeg&＃39;s av_log()
voidcustom_log(void*ptr,intlevel,const char* fmt,va_listvl){
FILE*fp&＃61;fopen("/storage/emulated/0/av_log.txt","a&＃43;");
if(fp){
vfprintf(fp,fmt,vl);
fflush(fp);
fclose(fp);
}
}
JNIEXPORTjintJNICALLJava_csupport_lyjq_com_csupport_FlvActivity_initial
(JNIEnv*env,jobjectobj,jintwidth,jintheight)
{
const char* out_path &＃61;"/storage/emulated/0/testffmpeg.flv";
yuv_width&＃61;width;
yuv_height&＃61;height;
y_length&＃61;width*height;
uv_length&＃61;width*height/4;
//FFmpeg av_log() callback
av_log_set_callback(custom_log);
av_register_all();
//output initialize
avformat_alloc_output_context2(&ofmt_ctx,NULL,"flv", out_path);
//output encoder initialize
pCodec &＃61; avcodec_find_encoder(AV_CODEC_ID_H264);
if(!pCodec){
LOGE("Can not find encoder!\n");
return-1;
}
pCodecCtx &＃61; avcodec_alloc_context3(pCodec);
pCodecCtx->pix_fmt&＃61;AV_PIX_FMT_YUV420P;
pCodecCtx->width&＃61; width;
pCodecCtx->height&＃61; height;
pCodecCtx->time_base.num&＃61;1;
pCodecCtx->time_base.den&＃61;30;
pCodecCtx->bit_rate&＃61;800000;
pCodecCtx->gop_size&＃61;300;
/* Some formats want stream headers to be separate. */
if(ofmt_ctx->oformat->flags&AVFMT_GLOBALHEADER)
pCodecCtx->flags|&＃61;CODEC_FLAG_GLOBAL_HEADER;
//H264 codec param
//pCodecCtx->me_range &＃61; 16;
//pCodecCtx->max_qdiff &＃61; 4;
//pCodecCtx->qcompress &＃61; 0.6;
pCodecCtx->qmin&＃61;10;
pCodecCtx->qmax&＃61;51;
//Optional Param
pCodecCtx->max_b_frames&＃61;3;
// Set H264 preset and tune
AVDictionary*param &＃61;0;
av_dict_set(¶m,"preset","ultrafast",0);
av_dict_set(¶m,"tune","zerolatency",0);
if(avcodec_open2(pCodecCtx, pCodec, ¶m) <0){
LOGE("Failed to open encoder!\n");
return-1;
}
//Add a new stream to output,should be called by the user before avformat_write_header() for muxing
video_st &＃61; avformat_new_stream(ofmt_ctx, pCodec);
if(video_st &＃61;&＃61;NULL){
return-1;
}
video_st->time_base.num&＃61;1;
video_st->time_base.den&＃61;30;
video_st->codec&＃61; pCodecCtx;
//Open output URL,set before avformat_write_header() for muxing
if(avio_open(&ofmt_ctx->pb, out_path,AVIO_FLAG_READ_WRITE) <0){
LOGE("Failed to open output file!\n");
return-1;
}
//Write File Header
avformat_write_header(ofmt_ctx,NULL);
start_time &＃61; av_gettime();
return0;
}
JNIEXPORTjintJNICALLJava_csupport_lyjq_com_csupport_FlvActivity_encode
(JNIEnv*env,jobjectobj,jbyteArrayyuv)
{
intret;
intenc_got_frame&＃61;0;
inti&＃61;0;
pFrameYUV &＃61; av_frame_alloc();
uint8_t*out_buffer &＃61; (uint8_t*)av_malloc(avpicture_get_size(AV_PIX_FMT_YUV420P, pCodecCtx->width, pCodecCtx->height));
avpicture_fill((AVPicture*)pFrameYUV, out_buffer,AV_PIX_FMT_YUV420P, pCodecCtx->width, pCodecCtx->height);
//&＃xfffd;&＃xfffd;&＃xfffd;&＃xfffd;&＃xfffd;&＃xfffd;ͷ&＃xfffd;&＃xfffd;&＃xfffd;&＃xfffd;ΪNV21&＃xfffd;&＃xfffd;ʽ&＃xfffd;&＃xfffd;&＃xfffd;˴&＃xfffd;&＃xfffd;&＃xfffd;&＃xfffd;&＃xfffd;ת&＃xfffd;&＃xfffd;ΪYUV420P&＃xfffd;&＃xfffd;ʽ
jbyte* in&＃61; (jbyte*)(*env)->GetByteArrayElements(env,yuv,0);
memcpy(pFrameYUV->data[0],in,y_length);
for(i&＃61;0;i
{
*(pFrameYUV->data[2]&＃43;i)&＃61;*(in&＃43;y_length&＃43;i*2);
*(pFrameYUV->data[1]&＃43;i)&＃61;*(in&＃43;y_length&＃43;i*2&＃43;1);
}
pFrameYUV->format&＃61;AV_PIX_FMT_YUV420P;
pFrameYUV->width&＃61; yuv_width;
pFrameYUV->height&＃61; yuv_height;
enc_pkt.data&＃61;NULL;
enc_pkt.size&＃61;0;
av_init_packet(&enc_pkt);
ret &＃61; avcodec_encode_video2(pCodecCtx, &enc_pkt, pFrameYUV, &enc_got_frame);
av_frame_free(&pFrameYUV);
if(enc_got_frame &＃61;&＃61;1){
LOGI("Succeed to encode frame: %5d\tsize:%5d\n", framecnt, enc_pkt.size);
framecnt&＃43;&＃43;;
enc_pkt.stream_index&＃61; video_st->index;
//Write PTS
AVRationaltime_base &＃61; ofmt_ctx->streams[0]->time_base;//{ 1, 1000 };
AVRationalr_framerate1 &＃61; {60,2};//{ 50, 2 };
AVRationaltime_base_q &＃61; {1,AV_TIME_BASE};
//Duration between 2 frames (us)
int64_tcalc_duration &＃61; (double)(AV_TIME_BASE)*(1/ av_q2d(r_framerate1));//&＃xfffd;ڲ&＃xfffd;ʱ&＃xfffd;&＃xfffd;&＃xfffd;
//Parameters
//enc_pkt.pts &＃61; (double)(framecnt*calc_duration)*(double)(av_q2d(time_base_q)) / (double)(av_q2d(time_base));
enc_pkt.pts&＃61; av_rescale_q(framecnt*calc_duration, time_base_q, time_base);
enc_pkt.dts&＃61; enc_pkt.pts;
enc_pkt.duration&＃61; av_rescale_q(calc_duration, time_base_q, time_base);//(double)(calc_duration)*(double)(av_q2d(time_base_q)) / (double)(av_q2d(time_base));
enc_pkt.pos&＃61; -1;
//Delay
int64_tpts_time &＃61; av_rescale_q(enc_pkt.dts, time_base, time_base_q);
int64_tnow_time &＃61; av_gettime() - start_time;
if(pts_time > now_time)
av_usleep(pts_time - now_time);
ret &＃61; av_interleaved_write_frame(ofmt_ctx, &enc_pkt);
av_free_packet(&enc_pkt);
}
return0;
}
JNIEXPORTjintJNICALLJava_csupport_lyjq_com_csupport_FlvActivity_flush
(JNIEnv*env,jobjectobj)
{
intret;
intgot_frame;
AVPacketenc_pkt;
if(!(ofmt_ctx->streams[0]->codec->codec->capabilities&
CODEC_CAP_DELAY))
return0;
while(1) {
enc_pkt.data&＃61;NULL;
enc_pkt.size&＃61;0;
av_init_packet(&enc_pkt);
ret &＃61; avcodec_encode_video2(ofmt_ctx->streams[0]->codec, &enc_pkt,
NULL, &got_frame);
if(ret <0)
break;
if(!got_frame){
ret &＃61;0;
break;
}
LOGI("Flush Encoder: Succeed to encode 1 frame!\tsize:%5d\n", enc_pkt.size);
//Write PTS
AVRationaltime_base &＃61; ofmt_ctx->streams[0]->time_base;//{ 1, 1000 };
AVRationalr_framerate1 &＃61; {60,2};
AVRationaltime_base_q &＃61; {1,AV_TIME_BASE};
//Duration between 2 frames (us)
int64_tcalc_duration &＃61; (double)(AV_TIME_BASE)*(1/ av_q2d(r_framerate1));//&＃xfffd;ڲ&＃xfffd;ʱ&＃xfffd;&＃xfffd;&＃xfffd;
//Parameters
enc_pkt.pts&＃61; av_rescale_q(framecnt*calc_duration, time_base_q, time_base);
enc_pkt.dts&＃61; enc_pkt.pts;
enc_pkt.duration&＃61; av_rescale_q(calc_duration, time_base_q, time_base);
//ת&＃xfffd;&＃xfffd;PTS/DTS&＃xfffd;&＃xfffd;Convert PTS/DTS&＃xfffd;&＃xfffd;
enc_pkt.pos&＃61; -1;
framecnt&＃43;&＃43;;
ofmt_ctx->duration&＃61; enc_pkt.duration* framecnt;
/* mux encoded frame */
ret &＃61; av_interleaved_write_frame(ofmt_ctx, &enc_pkt);
if(ret <0)
break;
}
//Write file trailer
av_write_trailer(ofmt_ctx);
return0;
}
JNIEXPORTjintJNICALLJava_csupport_lyjq_com_csupport_FlvActivity_close
(JNIEnv*env,jobjectobj)
{
if(video_st)
avcodec_close(video_st->codec);
avio_close(ofmt_ctx->pb);
avformat_free_context(ofmt_ctx);
return0;
}
效果图&＃xff0c;生成的flv文件在sd卡根目录下
Happy ending
京公网安备 11010802041100号