推流器

一、功能说明

获取pc端的摄像头流数据 + 展示直播效果 + 推流到rtmp服务器

二、代码实现

/**

* 推流器

* @param devicePath 摄像头的地址。可以是摄像头rtsp地址，也可以是设备号码，本机摄像头是0

* @param outputPath 接收路径

* @param v_rs 帧率

* @throws Exception

* @throws org.bytedeco.javacv.FrameRecorder.Exception

* @throws InterruptedException

*/

public static void recordPush(String outputPath,int v_rs) throws Exception, org.bytedeco.javacv.FrameRecorder.Exception, InterruptedException {

Loader.load(opencv_objdetect.class);

//创建采集器

OpenCVFrameGrabber grabber = new OpenCVFrameGrabber(0); //本地摄像头默认为0

//开启采集器

try {

grabber.start();

} catch (Exception e) {

try {

grabber.restart(); //一次重启尝试

} catch (Exception e2) {

throw e;

}

}

OpenCVFrameConverter.ToIplImage converter = new OpenCVFrameConverter.ToIplImage(); //转换器

Frame grabframe = grabber.grab(); //获取一帧

IplImage grabbedImage = null;

if (grabframe!=null) {

grabbedImage = converter.convert(grabframe); //将这一帧转换为IplImage

}

//创建录制器

FrameRecorder recorder;

recorder = FrameRecorder.createDefault(outputPath, 1280, 720); //输出路径，画面高，画面宽

recorder.setVideoCodec(avcodec.AV_CODEC_ID_H264); //设置编码格式

recorder.setFormat("flv");

recorder.setFrameRate(v_rs);

recorder.setGopSize(v_rs);

//开启录制器

try {

recorder.start();

} catch (java.lang.Exception e) {

System.out.println("recorder开启失败");

System.out.println(recorder);

try {

if (recorder != null) { //尝试重启录制器

recorder.stop();

recorder.start();

}

} catch (java.lang.Exception e1) {

// TODO Auto-generated catch block

e.printStackTrace();

}

}

//直播效果展示窗口

CanvasFrame frame = new CanvasFrame("直播效果",CanvasFrame.getDefaultGamma() / grabber.getGamma());

frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);

frame.setAlwaysOnTop(true);

//推流

while(frame.isVisible() && (grabframe=grabber.grab()) != null) {

frame.showImage(grabframe); //展示直播效果

grabbedImage = converter.convert(grabframe);

Frame rotatedFrame = converter.convert(grabbedImage);

if (rotatedFrame != null) {

recorder.record(rotatedFrame);

}

Thread.sleep(50); //50毫秒/帧

}

}

三、测试推流器

public static void main(String[] args) throws Exception, org.bytedeco.javacv.FrameRecorder.Exception, InterruptedException {

//设置rtmp服务器地址

String outputPath = "rtmp://192.168.1.48:1935/live/stream";

recordPush(outputPath, 25);

}

如果出现“直播效果”的swing窗口，并能够播放摄像头画面，则说明推流器成功。