package org.springblade.jfpt.rvideo.controller;
import org.bytedeco.javacpp.avcodec;
import org.bytedeco.javacv.*;

import java.io.BufferedReader;
import java.io.File;
import java.io.IOException;
import java.io.InputStreamReader;
public class RecordVideoThread implements Runnable {
	public String streamURL;// 流地址（测试可以用obs推流）
	public String filePath;// 文件路径
	public Long timesSec = 100L;// 停止录制时长 0为不限制时长
	public String fileFormat = "mp4";//录制的文件格式
	public boolean isAudio = false;//是否录制声音

	public void setStreamURL(String streamURL) {
		this.streamURL = streamURL;
	}

	public void setFilePath(String filePath) {
		this.filePath = filePath;
	}

	@Override
	public void run() {
		// 获取视频源
		FFmpegFrameGrabber grabber = new FFmpegFrameGrabber(streamURL);
		FFmpegFrameRecorder recorder = null;
		try {
			grabber.start();
			Frame frame = grabber.grabFrame();
			createFile();
			//"rtmp://192.168.1.225:1935/live/home"
			// 流媒体输出地址，分辨率（长，高），是否录制音频（0:不录制/1:录制）
			recorder = new FFmpegFrameRecorder(filePath, 1280, 720, isAudio ? 1 : 0);
			recorder.setVideoCodec(avcodec.AV_CODEC_ID_H264);// 直播流格式
			recorder.setFormat(fileFormat);// 录制的视频格式
			recorder.setFrameRate(25);// 帧数
			//百度翻译的比特率，默认400000，但是我400000贼模糊，调成800000比较合适
			recorder.setVideoBitrate(800000);
			recorder.start();
			System.out.println("推流开始");
//            new Thread(new Runnable() {
//                @Override
//                public void run() {
//                    RecordVideoThread thread = new RecordVideoThread();
//                    thread.pushVideoAsRTSP(1, null);
//                }
//            }).start();
			// 计算结束时间
			//long endTime = System.currentTimeMillis() + timesSec * 1000;
			// 如果没有到录制结束时间并且获取到了下一帧则继续录制
			while (frame != null) {
				System.out.println("开始");
				recorder.record(frame);//录制
				frame = grabber.grabFrame();//获取下一帧
				System.out.println("结束");
			}
			recorder.record(frame);

		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			//停止录制
			if (null != grabber) {
				try {
					grabber.stop();
				} catch (FrameGrabber.Exception e) {
					e.printStackTrace();
				}
			}
			if (recorder != null) {
				try {
					recorder.stop();
				} catch (FrameRecorder.Exception e) {
					e.printStackTrace();
				}
			}
			System.out.println("录制完成，录制时长：" + timesSec + "秒");
		}
	}

	private void createFile() {
		File outFile = new File(filePath);
		if (filePath.isEmpty() || !outFile.exists() || outFile.isFile()) {
			try {
				outFile.createNewFile();
			} catch (IOException e) {
				// TODO Auto-generated catch block
				e.printStackTrace();
			}
		}

	}
}