import org.bytedeco.ffmpeg.avcodec.AVPacket;
import org.bytedeco.ffmpeg.global.avcodec;
import org.bytedeco.javacv.CanvasFrame;
import org.bytedeco.javacv.FFmpegFrameGrabber;
import org.bytedeco.javacv.FFmpegFrameRecorder;
import org.bytedeco.javacv.Frame;
import org.bytedeco.javacv.FrameRecorder.Exception;


/**
	 * 视频分割
	 * @param input 可以是动态图片(apng,gif等等)，视频文件（mp4,flv,avi等等）,流媒体地址（http-flv,rtmp，rtsp等等）
	 * @param output 视频名称或者名称模板
	 * @param frameRate 帧率
	 * @param segmentTime 单个分片时长（单位：秒）
	 */
	public static void recordSegment(String input,String output,Integer segmentTime,Integer frameRate) throws Exception, org.bytedeco.javacv.FrameGrabber.Exception{
		FFmpegFrameGrabber grabber = new FFmpegFrameGrabber(input);
		
		grabber.start();
		
		FFmpegFrameRecorder recorder =new FFmpegFrameRecorder(output,grabber.getImageWidth(),grabber.getImageHeight(),0);
		
		recorder.setFormat("segment");
		
		recorder.setOption("segment_time", segmentTime.toString());
		
		//生成模式：live（实时生成）,cache（边缓存边生成，只支持m3u8清单文件缓存）
		recorder.setOption("segment_list_flags", "live");
		//强制锁定切片时长
		recorder.setOption("segment_atclocktime", "1");
		
//		if(frameRate==null) {
//			frameRate=25;
//		}
//		recorder.setFrameRate(frameRate);//设置帧率
		//因为我们是直播，如果需要保证最小延迟，gop最好设置成帧率相同或者帧率*2
		//一个gop表示关键帧间隔，假设25帧/秒视频，gop是50，则每隔两秒有一个关键帧，播放器必须加载到关键帧才能够开始解码播放，也就是说这个直播流最多有2秒延迟
//		recorder.setGopSize(frameRate*frameRate);//设置gop
//		recorder.setVideoQuality(1.0); //视频质量
//		recorder.setVideoBitrate(10*1024);//码率,10kb/s

//		recorder.setVideoCodecName("h264");//设置视频编码
		recorder.setVideoCodec(avcodec.AV_CODEC_ID_H264);//这种方式也可以
//		recorder.setAudioCodecName("aac");//设置音频编码，这种方式设置音频编码也可以
//		recorder.setAudioCodec(avcodec.AV_CODEC_ID_AAC);//设置音频编码
		
		recorder.start(grabber.getFormatContext());
		
//		CanvasFrame canvas = new CanvasFrame("视频预览");// 新建一个窗口
//		canvas.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
		Frame frame = null;
		
		// 只抓取图像画面
//		for (;(frame = grabber.grabImage()) != null;) {
//			try {
//				//显示画面
				canvas.showImage(frame);
//				//录制/推流
//				recorder.record(frame);
//				
//			} catch (org.bytedeco.javacv.FrameRecorder.Exception e) {
//				e.printStackTrace();
//			}
//		}
		
		AVPacket packet = null;
		// 解封装/解复用
		for (; (packet = grabber.grabPacket()) != null;) {
			// 封装/复用
			recorder.recordPacket(packet);
		}
		
		recorder.close();//close包含stop和release方法。录制文件必须保证最后执行stop()方法，才能保证文件头写入完整，否则文件损坏。
		grabber.close();//close包含stop和release方法
//		canvas.dispose();
	}