<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0"
         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    <parent>
        <artifactId>javacv-tutorials</artifactId>
        <groupId>com.bolingcavalry</groupId>
        <version>1.0-SNAPSHOT</version>
    </parent>
    <modelVersion>4.0.0</modelVersion>
    <groupId>com.bolingcavalry</groupId>
    <version>1.0-SNAPSHOT</version>
    <artifactId>simple-grab-push</artifactId>
    <packaging>jar</packaging>

    <properties>
        <!-- javacpp當(dāng)前版本 -->
        <javacpp.version>1.5.6</javacpp.version>
    </properties>

    <dependencies>
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
        </dependency>
        <dependency>
            <groupId>ch.qos.logback</groupId>
            <artifactId>logback-classic</artifactId>
            <version>1.2.3</version>
        </dependency>
        <dependency>
            <groupId>org.apache.logging.log4j</groupId>
            <artifactId>log4j-to-slf4j</artifactId>
            <version>2.13.3</version>
        </dependency>

        <!-- javacv相關(guān)依賴，一個(gè)就夠了 -->
        <dependency>
            <groupId>org.bytedeco</groupId>
            <artifactId>javacv-platform</artifactId>
            <version>${javacpp.version}</version>
        </dependency>
    </dependencies>
</project>

package com.bolingcavalry.grabpush.camera;

import lombok.Getter;
import lombok.extern.slf4j.Slf4j;
import org.bytedeco.ffmpeg.global.avutil;
import org.bytedeco.javacv.*;
import org.bytedeco.opencv.global.opencv_imgproc;
import org.bytedeco.opencv.opencv_core.Mat;
import org.bytedeco.opencv.opencv_core.Scalar;

import java.text.SimpleDateFormat;
import java.util.Date;

/**
 * @author will
 * @email zq2599@gmail.com
 * @date 2021/11/19 8:07 上午
 * @description 攝像頭應(yīng)用的基礎(chǔ)類，這里面定義了拉流和推流的基本流程，子類只需實(shí)現(xiàn)具體的業(yè)務(wù)方法即可
 */
@Slf4j
public abstract class AbstractCameraApplication {

    /**
     * 攝像頭序號(hào)，如果只有一個(gè)攝像頭，那就是0
     */
    protected static final int CAMERA_INDEX = 0;

    /**
     * 幀抓取器
     */
    protected FrameGrabber grabber;

    /**
     * 輸出幀率
     */
    @Getter
    private final double frameRate = 30;

    /**
     * 攝像頭視頻的寬
     */
    @Getter
    private final int cameraImageWidth = 1280;

    /**
     * 攝像頭視頻的高
     */
    @Getter
    private final int cameraImageHeight = 720;

    /**
     * 轉(zhuǎn)換器
     */
    private final OpenCVFrameConverter.ToIplImage openCVConverter = new OpenCVFrameConverter.ToIplImage();

    /**
     * 實(shí)例化、初始化輸出操作相關(guān)的資源
     */
    protected abstract void initOutput() throws Exception;

    /**
     * 輸出
     */
    protected abstract void output(Frame frame) throws Exception;

    /**
     * 釋放輸出操作相關(guān)的資源
     */
    protected abstract void releaseOutputResource() throws Exception;

    /**
     * 兩幀之間的間隔時(shí)間
     * @return
     */
    protected int getInterval() {
        // 假設(shè)一秒鐘15幀，那么兩幀間隔就是(1000/15)毫秒
        return (int)(1000/ frameRate);
    }

    /**
     * 實(shí)例化幀抓取器，默認(rèn)OpenCVFrameGrabber對(duì)象，
     * 子類可按需要自行覆蓋
     * @throws FFmpegFrameGrabber.Exception
     */
    protected void instanceGrabber() throws FrameGrabber.Exception {
        grabber = new OpenCVFrameGrabber(CAMERA_INDEX);
    }

    /**
     * 用幀抓取器抓取一幀，默認(rèn)調(diào)用grab()方法，
     * 子類可以按需求自行覆蓋
     * @return
     */
    protected Frame grabFrame() throws FrameGrabber.Exception {
        return grabber.grab();
    }

    /**
     * 初始化幀抓取器
     * @throws Exception
     */
    protected void initGrabber() throws Exception {
        // 實(shí)例化幀抓取器
        instanceGrabber();

        // 攝像頭有可能有多個(gè)分辨率，這里指定
        // 可以指定寬高，也可以不指定反而調(diào)用grabber.getImageWidth去獲取，
        grabber.setImageWidth(cameraImageWidth);
        grabber.setImageHeight(cameraImageHeight);

        // 開啟抓取器
        grabber.start();
    }

    /**
     * 預(yù)覽和輸出
     * @param grabSeconds 持續(xù)時(shí)長(zhǎng)
     * @throws Exception
     */
    private void grabAndOutput(int grabSeconds) throws Exception {
        // 添加水印時(shí)用到的時(shí)間工具
        SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");

        long endTime = System.currentTimeMillis() + 1000L *grabSeconds;

        // 兩幀輸出之間的間隔時(shí)間，默認(rèn)是1000除以幀率，子類可酌情修改
        int interVal = getInterval();

        // 水印在圖片上的位置
        org.bytedeco.opencv.opencv_core.Point point = new org.bytedeco.opencv.opencv_core.Point(15, 35);

        Frame captureFrame;
        Mat mat;

        // 超過指定時(shí)間就結(jié)束循環(huán)
        while (System.currentTimeMillis()<endTime) {
            // 取一幀
            captureFrame = grabFrame();

            if (null==captureFrame) {
                log.error("幀對(duì)象為空");
                break;
            }

            // 將幀對(duì)象轉(zhuǎn)為mat對(duì)象
            mat = openCVConverter.convertToMat(captureFrame);

            // 在圖片上添加水印，水印內(nèi)容是當(dāng)前時(shí)間，位置是左上角
            opencv_imgproc.putText(mat,
                    simpleDateFormat.format(new Date()),
                    point,
                    opencv_imgproc.CV_FONT_VECTOR0,
                    0.8,
                    new Scalar(0, 200, 255, 0),
                    1,
                    0,
                    false);

            // 子類輸出
            output(openCVConverter.convert(mat));

            // 適當(dāng)間隔，讓肉感感受不到閃屏即可
            if(interVal>0) {
                Thread.sleep(interVal);
            }
        }

        log.info("輸出結(jié)束");
    }

    /**
     * 釋放所有資源
     */
    private void safeRelease() {
        try {
            // 子類需要釋放的資源
            releaseOutputResource();
        } catch (Exception exception) {
            log.error("do releaseOutputResource error", exception);
        }

        if (null!=grabber) {
            try {
                grabber.close();
            } catch (Exception exception) {
                log.error("close grabber error", exception);
            }
        }
    }

    /**
     * 整合了所有初始化操作
     * @throws Exception
     */
    private void init() throws Exception {
        long startTime = System.currentTimeMillis();

        // 設(shè)置ffmepg日志級(jí)別
        avutil.av_log_set_level(avutil.AV_LOG_INFO);
        FFmpegLogCallback.set();

        // 實(shí)例化、初始化幀抓取器
        initGrabber();

        // 實(shí)例化、初始化輸出操作相關(guān)的資源，
        // 具體怎么輸出由子類決定，例如窗口預(yù)覽、存視頻文件等
        initOutput();

        log.info("初始化完成，耗時(shí)[{}]毫秒，幀率[{}]，圖像寬度[{}]，圖像高度[{}]",
                System.currentTimeMillis()-startTime,
                frameRate,
                cameraImageWidth,
                cameraImageHeight);
    }

    /**
     * 執(zhí)行抓取和輸出的操作
     */
    public void action(int grabSeconds) {
        try {
            // 初始化操作
            init();
            // 持續(xù)拉取和推送
            grabAndOutput(grabSeconds);
        } catch (Exception exception) {
            log.error("execute action error", exception);
        } finally {
            // 無論如何都要釋放資源
            safeRelease();
        }
    }
}

docker run -d --name nginx_rtmp -p 1935:1935 -p 18080:80 alfg/nginx-rtmp:v1.3.1

docker run -d --name nginx_rtmp -p 1935:1935 -p 18080:80 shamelesscookie/nginx-rtmp-ffmpeg:latest