在xcode上开发
//param splitSeconds 为视频分割的时长
+ (BOOL)executeSplit:(unsignedint)splitSeconds {
AVFormatContext*ifmtCtx,*ofmtCtx;
charconst *inputFileName, *outputFileName;
int video_index;
inputFileName = [[[NSBundle mainBundle] pathForResource:@"test" ofType:@"mp4"] UTF8String];
outputFileName ="/Users/ubaby/Library/Containers/bylh.testFFmpegOS/Data/Documents/test0.mp4";
AVPacketreadPkt, splitKeyPacket;
intret;
av_register_all();
if((ret =avformat_open_input(&ifmtCtx, inputFileName,0,0)) <0) {
return false;
}
if((ret =avformat_find_stream_info(ifmtCtx,0)) <0) {
return false;
}
for(inti =0; i < ifmtCtx->nb_streams; i++) {
AVStream*in_stream = ifmtCtx->streams[i];
if(in_stream->codec->codec_type==AVMEDIA_TYPE_VIDEO){
video_index = i;
}
}
intden = ifmtCtx->streams[video_index]->r_frame_rate.den;
intnum = ifmtCtx->streams[video_index]->r_frame_rate.num;
floatfps = (float)num / den;
unsignedintsplitVideoSize = fps*splitSeconds;
charconst *save_name = outputFileName;
charconst *temp_name = save_name;
avformat_alloc_output_context2(&ofmtCtx, NULL, NULL, temp_name);
if(!ofmtCtx) {
return false;
}
if (![testVideo writeVideoHeader:ifmtCtx out_filename:ofmtCtx out_filename:temp_name])
{
return false;
}
NSMutableArray *vecKeyFramePos = [NSMutableArray arrayWithCapacity:1];
uint64_tframe_index =0;
uint64_tkeyFrame_index =0;
intframeCount =0;
//读取分割点附近的关键帧位置
while(1)
{
++frame_index;
ret =av_read_frame(ifmtCtx, &readPkt);
if(ret <0)
{
break;
}
//过滤，只处理视频流
if(readPkt.stream_index== video_index){
++frameCount;
if(readPkt.flags&AV_PKT_FLAG_KEY)
{
keyFrame_index = frame_index;
}
if(frameCount==splitVideoSize)
{
// vecKeyFramePos.push_back(keyFrame_index);
[vecKeyFramePosaddObject:[NSNumbernumberWithUnsignedLongLong:keyFrame_index]];
frameCount =0;
}
}
av_packet_unref(&readPkt);
}
avformat_close_input(&ifmtCtx);
ifmtCtx =NULL;
//为了重新获取avformatcontext
if((ret =avformat_open_input(&ifmtCtx, inputFileName,0,0)) <0) {
return-1;
}
if((ret =avformat_find_stream_info(ifmtCtx,0)) <0) {
return-1;
}
intnumber =0;
av_init_packet(&splitKeyPacket);
splitKeyPacket.data=NULL;
splitKeyPacket.size=0;
//时长对应的帧数超过视频的总视频帧数，则拷贝完整视频
if(!vecKeyFramePos.count){
[vecKeyFramePosaddObject:[NSNumbernumberWithUnsignedLongLong:frame_index]];
}
keyFrame_index = [[vecKeyFramePosfirstObject]unsignedLongLongValue];
NSUIntegerkeyFrameIter =1;
frame_index =0;
int64_tlastPts =0;
int64_tlastDts =0;
int64_tprePts =0;
int64_tpreDts =0;
while(1)
{
++frame_index;
ret =av_read_frame(ifmtCtx, &readPkt);
if(ret <0)
{
break;
}
av_packet_rescale_ts(&readPkt, ifmtCtx->streams[readPkt.stream_index]->time_base, ofmtCtx->streams[readPkt.stream_index]->time_base);
prePts = readPkt.pts;
preDts = readPkt.dts;
readPkt.pts-= lastPts;
readPkt.dts-= lastDts;
if(readPkt.pts< readPkt.dts)
{
readPkt.pts= readPkt.dts+1;
}
//为分割点处的关键帧要进行拷贝
if(readPkt.flags&AV_PKT_FLAG_KEY&&frame_index == keyFrame_index)
{
av_copy_packet(&splitKeyPacket, &readPkt);
}
else{
ret =av_interleaved_write_frame(ofmtCtx, &readPkt);
if(ret <0) {
//break;
}
}
if(frame_index == keyFrame_index)
{
lastDts = preDts;
lastPts = prePts;
if(keyFrameIter < vecKeyFramePos.count)
{
keyFrame_index = [vecKeyFramePos[keyFrameIter]unsignedLongLongValue];
keyFrameIter++;
}
av_write_trailer(ofmtCtx);
avio_close(ofmtCtx->pb);
avformat_free_context(ofmtCtx);
++number;
NSString *tempStr = [NSString stringWithFormat:@"/Users/ubaby/Library/Containers/bylh.testFFmpegOS/Data/Documents/test%d.mp4",number];
NSLog(@"===========%@",tempStr);
temp_name = [tempStrUTF8String];
avformat_alloc_output_context2(&ofmtCtx,NULL,NULL, temp_name);
if(!ofmtCtx) {
returnfalse;
}
if(![testVideowriteVideoHeader:ifmtCtxout_filename:ofmtCtxout_filename:temp_name])
{
returnfalse;
}
splitKeyPacket.pts=0;
splitKeyPacket.dts=0;
//把上一个分片处的关键帧写入到下一个分片的起始处，保证下一个分片的开头为I帧
ret =av_interleaved_write_frame(ofmtCtx, &splitKeyPacket);
}
av_packet_unref(&readPkt);
}
av_packet_unref(&splitKeyPacket);
av_write_trailer(ofmtCtx);
avformat_close_input(&ifmtCtx);
avio_close(ofmtCtx->pb);
avformat_free_context(ofmtCtx);
return true;
}
+ (BOOL)writeVideoHeader:(AVFormatContext*)ifmtCtx out_filename:(AVFormatContext*)ofmtCtx out_filename:(charconst*)out_filename
{
AVOutputFormat *ofmt = NULL;
intret;
int video_index;
ofmt = ofmtCtx->oformat;
for(inti =0; i < ifmtCtx->nb_streams; i++) {
//∏˘æ› ‰»Î¡˜¥¥Ω® ‰≥ˆ¡˜£®Create output AVStream according to input AVStream£©
AVStream*in_stream = ifmtCtx->streams[i];
if(in_stream->codec->codec_type==AVMEDIA_TYPE_VIDEO){
video_index = i;
}
AVStream*out_stream =avformat_new_stream(ofmtCtx, in_stream->codec->codec);
if(!out_stream) {
ret =AVERROR_UNKNOWN;
returnfalse;
}
//∏¥÷∆AVCodecContextµƒ…Ë÷√£®Copy the settings of AVCodecContext£©
ret =avcodec_copy_context(out_stream->codec, in_stream->codec);
if(ret <0) {
returnfalse;
}
out_stream->codec->codec_tag=0;
if(ofmtCtx->oformat->flags&AVFMT_GLOBALHEADER)
out_stream->codec->flags|=AV_CODEC_FLAG_GLOBAL_HEADER;
}
if(!(ofmt->flags&AVFMT_NOFILE)) {
ret =avio_open(&ofmtCtx->pb, out_filename,AVIO_FLAG_WRITE);
if(ret <0) {
returnfalse;
}
}
ret =avformat_write_header(ofmtCtx,NULL);
if(ret <0){
return false;
}
return true;
}
网友评论