C++封装opencv读取视频与摄像头的库

来源：互联网发布：java dbconnection 编辑：程序博客网时间：2024/05/17 16:02

GetImage.h

#ifndef _GET_IMAGE_H_#define _GET_IMAGE_H_#include "opencv2/highgui.hpp"using namespace cv;#define FILENAMELEN 1000typedef enum enGetImgMode{    GET_IMG_MODE_RECORD,    GET_IMG_MODE_CAM}GET_IMG_MODE_E;typedef enum enGetImgWorkMode{    GET_IMG_WORK_MODE_ORC,              //获取原图    GET_IMG_WORK_MODE_RESIZE_YUV420,    //获取缩放后的YUV420图    GET_IMG_WORK_MODE_RESIZE_GRAY       //获取缩放后灰度图}GET_IMG_WORK_MODE_E;class GetImage{public:    GetImage();    GetImage(GET_IMG_MODE_E mode, GET_IMG_WORK_MODE_E workMode, int width, int height, char* pVideoPath);    ~GetImage();    int Work(Mat** ppstImage);public:    GET_IMG_MODE_E m_Mode;    GET_IMG_WORK_MODE_E m_WorkMode;    char m_VideoPath[FILENAMELEN];    int m_Width;    int m_Height;    VideoCapture m_Capture;    Mat m_OrcImg;    Mat m_ScaleImg;    Mat m_ScaleGrayImg;};#endif

GetImage.cpp

#include "GetImage.h"#include "opencv2/video.hpp"GetImage::GetImage(){    m_Mode = GET_IMG_MODE_CAM;    m_WorkMode = GET_IMG_WORK_MODE_ORC;    memset(&m_VideoPath[0], 0, sizeof(m_VideoPath[0])*FILENAMELEN);    m_Width = 1280;    m_Height = 720;    m_Capture.open(0);    m_ScaleImg.create(m_Width, m_Height, CV_8UC3);    m_ScaleGrayImg.create(m_Width, m_Height, CV_8UC1);}GetImage::GetImage(GET_IMG_MODE_E mode, GET_IMG_WORK_MODE_E workMode, int width, int height, char* pVideoPath){    m_Mode = mode;    m_WorkMode = workMode;    m_Width = width;    m_Height = height;    if (m_Mode == GET_IMG_MODE_RECORD)    {        memcpy(&m_VideoPath[0], pVideoPath, strlen(pVideoPath));        m_VideoPath[strlen(pVideoPath)] = '\0';        m_Capture.open(m_VideoPath);    }    else if (m_Mode == GET_IMG_MODE_CAM)    {        m_Capture.open(0);    }    m_ScaleImg.create(m_Width, m_Height, CV_8UC3);    m_ScaleGrayImg.create(m_Width, m_Height, CV_8UC1);}GetImage::~GetImage(){    m_Capture.release();}int GetImage::Work(Mat** ppstImage){    m_Capture >> m_OrcImg;    if (m_OrcImg.empty())    {        (*ppstImage) = NULL;        return 0;    }    //图像缩放    if (m_WorkMode == GET_IMG_WORK_MODE_ORC)    {        *ppstImage = &m_OrcImg;        return 0;    }    else if (m_WorkMode == GET_IMG_WORK_MODE_RESIZE_YUV420)    {        resize(m_OrcImg, m_ScaleImg, Size(m_Width, m_Height), (0, 0), (0, 0), INTER_AREA);        *ppstImage = &m_ScaleImg;        return 0;    }    else if (m_WorkMode == GET_IMG_WORK_MODE_RESIZE_GRAY)    {        resize(m_OrcImg, m_ScaleImg, Size(m_Width, m_Height), (0, 0), (0, 0), INTER_AREA);        //转换为灰度图像        cvtColor(m_ScaleImg, m_ScaleGrayImg, CV_BGR2GRAY);        *ppstImage = &m_ScaleGrayImg;        return 0;    }    return 0;}

阅读全文

0 0