|| || || || || ||

---

Grid.h

Go to the documentation of this file.

#ifndef Impala_Core_Feature_Grid_h
#define Impala_Core_Feature_Grid_h

#include "Basis/String.h"
#include "Core/Database/RawDataSet.h"
#include "Core/Feature/FeatureDefinition.h"
#include "Core/VideoSet/VideoSet.h"
#include "Core/Array/WritePng.h"

// comment this line when commit to impala
//#define __BITMAP_FUNC__

#ifdef __BITMAP_FUNC__
#include "Core/Feature/Bitmap.h"
#endif

#ifndef BYTE
#define BYTE unsigned char
#endif

#ifndef BOOL
#define BOOL bool
#endif

#ifndef UINT
#define UINT unsigned int
#endif


namespace Impala
{
namespace Core
{
namespace Feature
{


class Grid
{
public:

    Grid()
    {
    }

    Grid(int nRow, int nCol)
    {
        mRow = nRow;
        mCol = nCol;
    }
    
    ~Grid()
    {
    }

    String
    GetName() const
    {
        return mName;
    }

    // have verified for 2x2, 4x4, 2x4, 4x2, 
    // not yet: 3x3
    void
    RetrievalGridBuffers(BYTE* im, BYTE** buf_list, int nWidth, int nHeight, Core::VideoSet::VideoSet* vs = NULL)
    {
        String fileName;

#ifdef __BITMAP_FUNC__
        //Core::Feature::Bitmap bitmap;
        //bitmap.SaveRgb2BitmapFile("rgb_00.bmp", im, nWidth, nHeight);

        Array::Array2dVec3UInt8* buf = Array::ArrayCreate<Array::Array2dVec3UInt8>
                    (nWidth, nHeight, 0, 0, im, true);
        fileName = "rgb_00.png";
        Core::Array::WritePng(buf, fileName, vs->GetDatabase());
#endif

        //--------------
        //   1   |   2
        //--------------
        //   3   |   4
        //--------------
        for (int x=0; x<mRow; x++)
        {
            for (int y=0; y<mCol; y++)
            {
                int nFrameWidth = nWidth;
                int nFrameHeight = nHeight;

                int nGridWidth = nWidth/mCol;
                int nGridHeight = nHeight/mRow;

                int idx = y+x*mCol;

                BYTE* cur_buf = buf_list[idx];
                // process each row of the buffer
                for (int j=0; j<nGridHeight; j++)
                {
                    int nCurSize = nGridWidth*3*sizeof(BYTE);

                    //memcpy(cur_buf + j*nGridWidth*3, im + j*nFrameWidth*3, nCurSize);
                    BYTE* dst = cur_buf + j*nGridWidth*3;
                    //BYTE* src = im + j*nFrameWidth*3 + (nGridWidth*3*y+x*nGridWidth*nGridHeight*3*2);
                    int nFrameRowOffset1 = (nGridHeight * x) * nFrameWidth * 3; // three entry points
                    int nFrameRowOffset2 = j * nFrameWidth * 3;                 // within one grid row
                    int nFrameColOffset = nGridWidth * y * 3;                   // in column dimension   
                    
                    int nOffset = nFrameRowOffset1 + nFrameRowOffset2 + nFrameColOffset;
                    BYTE* src = im + nOffset;
                    memcpy(dst, src, nCurSize);
                }

                //verificatiion of the buffers
#ifdef __BITMAP_FUNC__
                String idxStr;
                char index[128];
                itoa(idx+1,index,10);
                idxStr = index;
                fileName = "rgb_grid_0" + idxStr + ".bmp";

                //Core::Feature::Bitmap bitmap;
                //bitmap.SaveRgb2BitmapFile(fileName.c_str(), cur_buf, nGridWidth, nGridHeight);

                Array::Array2dVec3UInt8* sc = Array::ArrayCreate<Array::Array2dVec3UInt8>
                    (nGridWidth, nGridHeight, 0, 0, cur_buf, true);
                fileName = "rgb_grid_0" + idxStr + ".png";
                Core::Array::WritePng(sc, fileName, vs->GetDatabase());
#endif
            }
        }

    }


private:

    String                mName;
    int                   mRow;
    int                   mCol;

};

} // namespace Feature
} // namespace Core
} // namespace Impala

#endif

---