專案實戰:Qt+Ffmpeg+OpenCV相機程式(開啟攝像頭、支援多種攝像頭、解析度調整、翻轉、旋轉、亮度調整、拍照、錄影、回放圖片、回放錄影)

紅胖子(紅模仿)發表於2020-09-09

若該文為原創文章,未經允許不得轉載
原博主部落格地址:https://blog.csdn.net/qq21497936
原博主部落格導航:https://blog.csdn.net/qq21497936/article/details/102478062
本文章部落格地址:https://blog.csdn.net/qq21497936/article/details/108489004

 

需求

  嵌入式windows裝置上的相機程式。
  開啟攝像頭,相容多種攝像頭,攝像頭解析度切換(攝像頭管理)。
  對影像進行翻轉、旋轉、亮度調整(影像基本演算法管理)
  對調整後的影像進行拍照、延時拍照。
  對調整後的影像進行錄影(編碼錄製)。
  對照片和錄影進行回看(圖片瀏覽器、視訊播放器)
  長時間執行穩定。

 

Demo

  在這裡插入圖片描述
  在這裡插入圖片描述
  在這裡插入圖片描述
  在這裡插入圖片描述
  在這裡插入圖片描述
  在這裡插入圖片描述
  在這裡插入圖片描述

體驗下載地址

  CSDN:https://download.csdn.net/download/qq21497936/12827160
  QQ群:1047134658(點選“檔案”搜尋“camera”,群內與博文同步更新)

 

原理

  使用ffmpeg處理攝像頭、使用OpenCV處理錄影和播放;

 

相關部落格

 

v1.5.0功能

  • 開啟攝像頭,相容多種攝像頭,攝像頭解析度切換(攝像頭管理)。
  • 對影像進行翻轉、旋轉、亮度調整(影像基本演算法管理)
  • 對調整後的影像進行拍照、延時拍照。
  • 對調整後的影像進行錄影(編碼錄製)。
  • 對照片和錄影進行回看(圖片瀏覽器、視訊播放器)
 

Demo核心程式碼

FfmpegCameraManager.h:攝像頭管理類

#ifndef FFMPEGCAMERAMANAGER_H
#define FFMPEGCAMERAMANAGER_H

/************************************************************\
 * 控制元件名稱: FfmpegCameraManager, ffmpeg管理類(用於攝像頭操作)
 * 控制元件描述:
 *          1.開啟攝像頭
 *          2.支援動態切換解析度
 * 作者:紅模仿    聯絡方式:QQ21497936
 * 部落格地址:https://blog.csdn.net/qq21497936
 *       日期                版本               描述
 *    2018年09年14日     v1.0.0         ffmpeg模組封裝空類
 *    2020年09年05日     v1.1.0         ffmpeg開啟攝像頭,支援的動態解析度切換
 *    2020年09年08日     v1.2.0         相容各種攝像頭,解決記憶體溢位bug,對最高幀率做了支援範圍內的限制
 *                                     限制幀率一般為25fps(除非最大小於25fps或者最小大於25fps)
\************************************************************/

#include <QObject>
#include <QString>
#include <QDebug>
#include <QTimer>
#include <QThread>
#include <QImage>
#include <QProcess>
#include <QMessageBox>
#include <QDateTime>

extern "C" {
    #include "libavcodec/avcodec.h"
    #include "libavformat/avformat.h"
    #include "libswscale/swscale.h"
    #include "libavdevice/avdevice.h"
    #include "libavformat/version.h"
    #include "libavutil/time.h"
    #include "libavutil/mathematics.h"
    #include "libavformat/avformat.h"
    #include "libswscale/swscale.h"
    #include "libswresample/swresample.h"
    #include "errno.h"
    #include "error.h"
}

#define LOG qDebug()<<__FILE__<<__LINE__<<QDateTime::currentDateTime().toString("hh:mm:ss:zzz")

class FfmpegCameraManager : public QObject
{
    Q_OBJECT
public:

public:
    explicit FfmpegCameraManager(QObject *parent = nullptr);

signals:
    void signal_captureOneFrame(QImage image);

public:
    static QString getAvcodecConfiguration();

public:
    bool init();
    bool openUsbCamera();
    QString getUsbCameraName();
    QList<QString> getUsbCameraInfo();
    int getCurrentFps();
    int getCurrentSizeFpsIndex();

    QList<QSize> getListSize() const;

public slots:
    void slot_start();
    void slot_stop();
    void slot_setSizeFps(int index);

protected slots:
    void slot_captureOneFrame();

signals:

public slots:


private:
    static bool _init;

    AVFormatContext *_pAVFormatContext;         // 全域性上下文
    AVInputFormat *_pAVInputFormat;
    AVDictionary* _pAVDictionary;               // 開啟編碼器的配置

    AVCodecContext *_pAVCodecContextForAudio;   // 音訊解碼器上下文
    AVCodecContext *_pAVCodecContextForVideo;   // 視訊解碼器上下文(不帶音訊)

    AVCodec * _pAVCodecForAudio;                // 音訊解碼器
    AVCodec * _pAVCodecForVideo;                // 視訊解碼器(不帶音訊)

    int _streamIndexForAudio;                   // 音訊流序號
    int _streamIndexForVideo;                   // 視訊流序號

    SwrContext *_pSwrContextForAudio;           // 音訊轉換上下文

    bool _running;
    bool _first;
    bool _opened;
    uint8_t *_pOutBuffer;

    AVFrame * _pFrame;
    AVFrame * _pFrameRGB;
    AVPacket *_pAVPacket;
    SwsContext *_pSwsContext;

    int _videoIndex;

    QString _cameraDescription;

    QList<QSize> _listSize;
    QList<int> _listFps;
    QList<QString> _listSizeFpsInfo;
    int _currentSizeFpsIndex;
};

#endif // FfmpegCameraManager_H

FfmpegCameraManager.cpp:攝像頭管理類

...
bool FfmpegCameraManager::openUsbCamera()
{
    if(!_init)
    {
        LOG << "未初始化";
        return true;
    }
    _pAVInputFormat = av_find_input_format("dshow");
    if(!_pAVInputFormat)
    {
        LOG << "Failed to av_find_input_format";
        return false;
    }
    if(_cameraDescription == "")
    {
        LOG << "無攝像頭";
        return false;
    }
    QString cameraDescription = QString("video=%1").arg(_cameraDescription);
    if(_listSizeFpsInfo.size() == 0)
    {
        LOG << "未獲取到解析度和幀率";
        return false;
    }
    // 設定解析度
    av_dict_set(&_pAVDictionary,
                "video_size",
                QString("%1x%2").arg(_listSize.at(_currentSizeFpsIndex).width())
                                .arg(_listSize.at(_currentSizeFpsIndex).height()).toUtf8().data(),
                0);
    // 設定幀率
    int frame = _listFps.at(_currentSizeFpsIndex);

    av_dict_set(&_pAVDictionary,
                "framerate",
                QString("%1").arg(frame).toUtf8().data(),
                0);

    LOG << "開啟攝像頭:" << _cameraDescription
        << "解析度:" << _listSize.at(_currentSizeFpsIndex).width() << "x" << _listSize.at(_currentSizeFpsIndex).height()
        << "幀率:" << _listFps.at(_currentSizeFpsIndex);

    if(avformat_open_input(&_pAVFormatContext,
                           cameraDescription.toUtf8().data(),
                           _pAVInputFormat,
                           &_pAVDictionary) != 0)
    {
        LOG << "開啟攝像頭失敗";
        return false;
    }
    LOG << "開啟攝像頭成功";
    _first = true;
    _opened = true;
    QTimer::singleShot(0, this, SLOT(slot_captureOneFrame()));
    return true;
}
...

OpenCVManager.h:錄影與播放視訊類

#ifndef OPENCVMANAGER_H
#define OPENCVMANAGER_H

/************************************************************\
 * 控制元件名稱: OpenCVManager,OpenCV管理類
 * 控制元件描述:
 *          1.OpenCV操作支援
 *          2.支援錄影(.avi格式)
 * 作者:紅模仿    聯絡方式:QQ21497936
 * 部落格地址:https://blog.csdn.net/qq21497936
 *       日期             版本               描述
 *   2019年11月09日      v1.0.0   opencv拍照和錄影Demo
 *   2020年09月07日      v1.1.0   增加了單純錄影的介面
\************************************************************/

#include <QObject>
#include <QImage>
#include <QDateTime>
#include <QTimer>
// opencv
#include "opencv/highgui.h"
#include "opencv/cxcore.h"
#include "opencv2/core/core.hpp"
#include "opencv2/highgui/highgui.hpp"
#include "opencv2/opencv.hpp"

class OpenCVManager : public QObject
{
    Q_OBJECT
public:
    explicit OpenCVManager(QObject *parent = nullptr);
    ~OpenCVManager();

public:
    QString getWindowTitle() const;
    double getBrightness();
    double getContrast() const;
    double getSaturation() const;
    double getHue() const;
    double getGain() const;
    bool getShowProperty() const;
    double getExposure() const;
    int getRotate() const;
    bool getMirror() const;

public:
    void setBrightness(double value);
    void setContrast(double value);
    void setSaturation(double value);
    void setHue(double value);
    void setGain(double value);
    void setShowProperty(bool value);
    void setExposure(double value);
    void setRotate(int rotate);
    void setMirror(bool mirror);


signals:
    void signal_captureOneFrame(cv::Mat mat);   // 接收影像後丟擲訊號

public:
    bool startCapture(int usb, int width = 1280, int height = 720);
                                        // 開啟攝像頭, 0...
    bool startCapture(QString url, int width = 1280, int height = 720);
                                        // 開啟攝像頭, 網路攝像頭地址
    bool stopCapture();                 // 關閉攝像頭
    void startRecord(QString filePath); // 開始錄影(使用的是opencv開啟的攝像頭)
    void stopRecord();                  // 停止錄影(使用的是opencv開啟的攝像頭)

public slots:
    void slot_inputRecordImage(QImage image);
    void slot_stopRecordFormOut();

public:     // 單獨的一塊業務,使用的是開始錄影後,從類外面輸入QImage進行錄影
    void startRecordFromOut(QString filePath, int fps);
    void inputRecordImage(QImage image);
    void stopRecordFormOut();

public slots:
    bool start();                       // 開啟執行緒
    bool stop();                        // 關閉執行緒

protected slots:
    void slot_captrueFrame();           // 訊息迴圈獲取影像
    void slot_stopCapture();            // 當正在採集中時(>>時),關閉攝像頭會導致程式崩潰,所以採集與停止放一個執行緒中(訊息迴圈)

protected slots:
    void slot_startRecord(QString filePath);        // 錄影(使用的是opencv開啟的攝像頭)
    void slot_stopRecord();                         // 停止錄屏(使用的是opencv開啟的攝像頭)

public:
    static QImage cvMat2QImage(const cv::Mat &mat);
    static cv::Mat image2Mat(QImage image);    // Qimage 轉 cv::Mat
    static QImage mat2Image(cv::Mat mat);      // cv::Mat 轉 QImage

private:
    cv::VideoCapture *_pVideoCapture;   // 攝像頭例項
    cv::VideoWriter *_pVideoWrite;      // 錄影例項
    QString _recordFilePath;            // 錄製檔案路徑

    bool _running;                      // 執行緒是否執行
    bool _showProperty;                 // 是否顯示屬性引數
    double _brightness;                 // 亮度
    double _contrast;                   // 對比度
    double _saturation;                 // 飽和度
    double _hue;                        // 色調
    double _gain;                       // 增益
    double _exposure;                   // 曝光度

    int _width;                         // 寬度
    int _height;                        // 高度
    bool _recording;                    // 標誌是否正在錄影
    bool _startRecording;

    int _rotate;                        // 旋轉度數
    bool _mirror;                       // 是否翻轉
    int _fps;                           // 幀率
    int _index;                         // 幀序號

private:
    cv::VideoWriter *_pVideoWriteForOut; // 錄影例項(從外部輸入影像,非從opencv開啟攝像頭)
    QString _recordFilePathForOut;       // 錄影檔案路徑(從外部輸入影像,非從opencv開啟攝像頭)

private:
    QString _windowTitle;

};

#endif // OPENCVMANAGER_H

OpenCVManager.h:錄影與播放視訊類

...
void OpenCVManager::inputRecordImage(QImage image)
{
    if(!_startRecording)
    {
        return;
    }
    cv::Mat mat = image2Mat(image);

    if(!_recording)
    {
        QString ext = _recordFilePath.mid(_recordFilePathForOut.lastIndexOf(".") + 1);
        int cvFourcc = 0;
        if(ext == "mpg")
        {
            cvFourcc = CV_FOURCC('D','I','V','X');
            qDebug() << __FILE__ << __LINE__<< ext << "DIVX" << cvFourcc;
        }else if(ext == "avi")
        {
            cvFourcc = CV_FOURCC('M','J','P','G');
            qDebug() << __FILE__ << __LINE__<< ext << "avi" << cvFourcc;
        }else if(ext == "mp4")
        {
            // mp4目前錄製不成功(可以生成檔案,但是開啟失敗)
            cvFourcc = CV_FOURCC('M','P','4','2');
            qDebug() << __FILE__ << __LINE__<< ext << "MP42" << cvFourcc;
        }
        qDebug() << __FILE__ << __LINE__ << mat.type() << mat.channels();
        _pVideoWriteForOut->open(_recordFilePath.toStdString(), cvFourcc, _fps, cv::Size(mat.cols, mat.rows));

        std::vector<cv::Mat> listMat;
        cv::split(mat, listMat);
        std::vector<cv::Mat> listMat2;
        // 由於opencv對avi中mat的限制大小隻能為0xFFFF,修改原始碼突破限制為0xFFFFFFFF後
        // 在錄影時,發現錄入的mat是正確的,錄製出來通道顏色變換了,需要手動對顏色通道進行修正
        // 注意:僅限avi使用mjpg編碼格式
        // 1 2 0 偏綠 
        // 0 1 2 偏藍
        // 0 2 1 偏綠
        // 1 2 3 嚴重不對
        // 2 0 1 偏藍
        // 2 1 0 偏藍
        listMat2.push_back(listMat.at(0));
        listMat2.push_back(listMat.at(1));
        listMat2.push_back(listMat.at(2));
        cv::merge(listMat2, mat);

        _pVideoWriteForOut->write(mat);
        _recording = true;
    }else{

        std::vector<cv::Mat> listMat;
        cv::split(mat, listMat);
        std::vector<cv::Mat> listMat2;
        listMat2.push_back(listMat.at(0));
        listMat2.push_back(listMat.at(1));
        listMat2.push_back(listMat.at(2));
        cv::merge(listMat2, mat);
        _pVideoWriteForOut->write(mat);
    }
}
...
 

若該文為原創文章,未經允許不得轉載
原博主部落格地址:https://blog.csdn.net/qq21497936
原博主部落格導航:https://blog.csdn.net/qq21497936/article/details/102478062
本文章部落格地址:https://blog.csdn.net/qq21497936/article/details/108489004

相關文章