图形编程框架-来自RMW

港控/mmm° 2021-06-24 15:58 205阅读 0赞

来自于南京理工大学的一位极好老师，如果对图像处理感兴趣，强烈推荐。该框架有如下好处：  
1、构造函数只是简单初始化，而不传递参数进行初始化。因为视频的帧处理时使用的类是相同的，重复声明类是没有必要的。可以利用已经生成的对象。因此用一个专门的可以主动调用的初始化函数来对参数进行初始化，以适应不同参数的图像。  
2、在初始化函数中申请内存，如果已申请内存大于要处理的图像，将不会重复申请，并且有专门的内存释放函数。在析构函数中主动调用，释放内存。  
3、可以调试某一帧图像。

// RmwImgProcFrame.h : header file
    //
    #ifndef RMW_IMG_PROC_FRAME_H
    #define RMW_IMG_PROC_FRAME_H
    
    class RmwImgProcFrame { 
    public:
        //构造/析构
        RmwImgProcFrame();
        ~RmwImgProcFrame();
        //初始化
        bool Initialize(bool isRGB,int width,int height);
        //执行
        bool DoNext(BYTE *pImgData);
        //结果
       BYTE *GetResImg();
        int GetResWidth();
        int GetResHeight();
    
    private:
        // 内存释放
        void Dump();
        // 调试
        void Debug();
    
    private:
        //初始化成功
        bool m_isInitOK;
        //图像属性
        bool m_isRGB;
        int m_width;
        int m_height;
        //内存
       BYTE *m_pRGBImg;
       BYTE *m_pGryImg;
       BYTE *m_pGrdImg;
       BYTE *m_pTmpImg;
       BYTE *m_pResImg;
        int m_memSize;
        //内部计数器
        int m_nFrameID;
    };
    #endif

////////////////////////////////////////////////////////////////////////////////////////////////////
    //
    // RmwImgProcFrame.cpp : implementation file
    //
    ////////////////////////////////////////////////////////////////////////////////////////////////////
    #include <windows.h>
    #include <stdio.h>
    #include <stdlib.h>
    #include <string.h>
    #include <math.h>
    #include <time.h>
    
    #include "RmwImgProcFrame.h"
    
    ////////////////////////////////////////////////////////////////////////////////////////////////////
    //
    // 构造/析构函数
    //
    ////////////////////////////////////////////////////////////////////////////////////////////////////
    RmwImgProcFrame::RmwImgProcFrame()
    {   //<在构造函数中把所有的成员变量赋初值>
        //是否初始化成功
        m_isInitOK=false;
        //图像属性
        m_isRGB=false;
        m_width=0;
        m_height=0;
        //内存空间
        m_pRGBImg=NULL;
        m_pGryImg=NULL;
        m_pGrdImg=NULL;
        m_pTmpImg=NULL;
        m_pResImg=NULL;
        m_memSize=0; //已申请的内存字节数
        //内部计数器
        m_nFrameID=0;
    }
    
    RmwImgProcFrame::~RmwImgProcFrame()
    {   //<在析构函数中释放所有的内存>
        Dump();
    }
    
    void RmwImgProcFrame::Dump()
    {   //<写一个专门的用作释放内存的函数,因为内存会有多次的申请与释放>
        if (m_pRGBImg) { delete m_pRGBImg; m_pRGBImg=NULL; }
        if (m_pGryImg) { delete m_pGryImg; m_pGryImg=NULL; }
        if (m_pGrdImg) { delete m_pGrdImg; m_pGrdImg=NULL; }
        if (m_pTmpImg) { delete m_pTmpImg; m_pTmpImg=NULL; }
        if (m_pResImg) { delete m_pResImg; m_pResImg=NULL; }
        m_memSize=0;
    }
    ////////////////////////////////////////////////////////////////////////////////////////////////////
    //
    // 初始化
    //
    ////////////////////////////////////////////////////////////////////////////////////////////////////
    bool RmwImgProcFrame::Initialize(bool isRGB,int width,int height)
    {   //<在初始化函数中对所有的成员变量赋初值>
        //<仅在初始化函数中申请内存,因为初始化函数不会被频繁调用>
        //<初始化函数可以被多次调用>
        //<内存申请时,若已经申请的内存大于本次需要的内存,则不再次申请>
    
        //step.1------图像属性--------------------------//
        m_isRGB=isRGB;
        m_width=width;
        m_height=height;
        //step.2------内存申请--------------------------//
        if (m_width*m_height>m_memSize)
        {
            //先释放
            Dump();
            //后申请
            m_pRGBImg=new BYTE [m_width*m_height*3]; //彩色图像
            m_pGryImg=new BYTE [m_width*m_height];  //灰度图像
            m_pGrdImg=new BYTE [m_width*m_height]; //梯度图像
            m_pTmpImg=new BYTE [m_width*m_height]; //临时图像
            m_pResImg=new BYTE [m_width*m_height]; //结果图像
            //记录申请的大小
            m_memSize=m_width*m_height;
        }
        //step.3------初始化成功标志--------------------//
        m_isInitOK=(m_pRGBImg && m_pGryImg && m_pGrdImg && m_pTmpImg && m_pResImg);
        return m_isInitOK;
    }
    ////////////////////////////////////////////////////////////////////////////////////////////////////
    //
    // 执行
    //
    ////////////////////////////////////////////////////////////////////////////////////////////////////
    bool RmwImgProcFrame::DoNext(BYTE *pImgData)
    {   
        if (!m_isInitOK) return false;
        //...
        //...
        //...
        //某个特定帧的调试,比如10
        if (m_nFrameID==10)
        {
            Debug();
        }
        //内部计数器累加
        m_nFrameID++;
        return true;
    }
    ////////////////////////////////////////////////////////////////////////////////////////////////////
    //
    // 结果
    //
    ////////////////////////////////////////////////////////////////////////////////////////////////////
    BYTE *RmwImgProcFrame::GetResImg()
    {   //返回结果图像
        return m_pResImg;
    }
    int RmwImgProcFrame::GetResWidth()
    {   //返回结果图像的宽度
        return m_width;
    }
    int RmwImgProcFrame::GetResHeight()
    {   //返回结果图像的高度
        return m_height;
    }
    ////////////////////////////////////////////////////////////////////////////////////////////////////
    //
    // 调试
    //
    ////////////////////////////////////////////////////////////////////////////////////////////////////
    void RmwImgProcFrame::Debug()
    {   //<本类的内部调试: 屏幕输出、文件输出等>
        return;
    }