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

---

RawImageDataset.h

Go to the documentation of this file.

#ifndef Impala_Application_Videolympics_RawImageDataset_h
#define Impala_Application_Videolympics_RawImageDataset_h

#include "Basis/ILog.h"

#include "Core/Array/Arrays.h"
#include "Core/Array/ReadFile.h"
#include "Core/Array/ReadImage.h"
#include "Core/Array/Set.h"
#include "Core/Array/ArrayListDelete.h"

namespace Impala {
namespace Application {
namespace Videolympics {

using namespace Core::Array;

class RawImageDataset
{
public:
    RawImageDataset(std::string fileName)
    {
        ILOG_DEBUG("reading dataset " << fileName << "...");
        mIOBuffer = new Util::IOBufferFile(fileName, true, false);
        ReadRawListVarIndex<Array2dScalarUInt8>(mFileOffset, mIOBuffer);
        ILOG_DEBUG("imageset loaded. " << NrImages() << " images available.");
    }

    int
    NrImages()
    {
        if (mIOBuffer)
            return mFileOffset.size();
        return mImages.size();
    }

    void
    RequestNext(int inc)
    {
        int newNr = mCurNr + inc;
        if (newNr < 0)
            newNr = 0;
        if (newNr >= NrImages())
            newNr = NrImages() - 1;
        if (newNr != mCurNr)
        {
            mCurNr = newNr;
        }
    }

    Array2dVec3UInt8*
    GetImage(int nr)
    {
        if (nr < 0 || nr > NrImages())
        {
            ILOG_WARN("request for image nr " << nr << ", only " << NrImages() << " in dataset.");
            return 0;
        }

        char buf[200];
        int width;
        int height;
        mIOBuffer->SetPosition(mFileOffset[nr]);
        mIOBuffer->Read(buf, 20);
        sscanf(buf, "%d %d ", &width, &height);
        size_t blockSize = width*height;
        char* data = new char[blockSize];
        size_t nrRead = mIOBuffer->Read(data, blockSize);
        if (nrRead != blockSize)
        {
            ILOG_ERROR("GetImage: read " << nrRead
                      << " bytes instead of " << blockSize);
        }

        Array2dVec3UInt8* image = 0;

        ReadImageFromMemory(image, data, blockSize);
        delete data;

        return image;
    }

private:

    std::vector<Array2dVec3UInt8*> mImages;
    int                  mCurNr;
  
    // for file-based jpg/png archive
    Util::IOBuffer*      mIOBuffer;
    std::vector<Int64>   mFileOffset;

    ILOG_VAR_DEC;

};
ILOG_VAR_INIT(RawImageDataset, Application.Videolympics);

}
}
}

#endif


/*class RawImageDataset
{
public:
    RawImageDataset(std::string fileName, std::string dispMode = "jpg")
    {
        ILOG_DEBUG("reading dataset " << fileName << "...");
        mDispMode = dispMode;
        mIOBuffer = new Util::IOBufferFile(fileName, true, false);
        ReadRawListVarIndex<Array2dScalarUInt8>(mFileOffset, mIOBuffer);
        mFileIm = 0;
        UpdateFileImage(0);

        mCurNr = -1;
        RequestNext(1);

        ILOG_DEBUG("imageset loaded. " << NrImages() << " images available.");
    }

    int
    NrImages()
    {
        if (mIOBuffer)
            return mFileOffset.size();
        return mImages.size();
    }

    int H() { return mFileIm->CH(); }
    int W() { return mFileIm->CW(); }

    void
    RequestNext(int inc)
    {
        int newNr = mCurNr + inc;
        if (newNr < 0)
            newNr = 0;
        if (newNr >= NrImages())
            newNr = NrImages() - 1;
        if (newNr != mCurNr)
        {
            mCurNr = newNr;
        }
    }

    Array2dVec3UInt8*
    GetImage(int nr)
    {
        if (nr < 0 || nr > NrImages())
        {
            ILOG_WARN("request for image nr " << nr << ", only " << NrImages() << " in dataset.");
            return 0;
        }
        UpdateFileImage(nr);
        return mFileIm;
    }

private:
    void
    UpdateFileImage(int nr)
    {
        char buf[200];
        int width;
        int height;
        mIOBuffer->SetPosition(mFileOffset[nr]);
        mIOBuffer->Read(buf, 20);
        sscanf(buf, "%d %d ", &width, &height);
        size_t blockSize = width*height;
        char* data = new char[blockSize];
        size_t nrRead = mIOBuffer->Read(data, blockSize);
        if (nrRead != blockSize)
            std::cout << "UpdateFileImage: read " << nrRead
                      << " bytes instead of " << blockSize << std::endl;
        if (mFileIm)
            delete mFileIm;
        mFileIm = 0;
        if (mDispMode == std::string("jpg"))
            ReadJpgFromMemory(mFileIm, data, blockSize);
        else
            ReadPngFromMemory(mFileIm, data, blockSize);
        delete data;
    }

    std::vector<Array2dVec3UInt8*> mImages;
    int                  mCurNr;
    std::string          mDispMode;

    // for file-based jpg/png archive
    Util::IOBuffer*      mIOBuffer;
    std::vector<Int64>   mFileOffset;
    Array2dVec3UInt8*    mFileIm;

*/

---