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

---

ImageRepositoryInFile.h

Go to the documentation of this file.

#ifndef Impala_Persistency_ImageRepositoryInFile_h
#define Impala_Persistency_ImageRepositoryInFile_h

#include "Persistency/RepositoryInFileSystem.h"
#include "Core/Array/Arrays.h"
#include "Persistency/ImageLocator.h"
#include "Core/Array/ReadFile.h"
#include "Core/Array/WriteFile.h"

namespace Impala
{
namespace Persistency
{


class ImageRepositoryInFile
{
public:

    typedef Core::Array::Array2dVec3UInt8 Array2dVec3UInt8;
    typedef Core::Array::Array2dScalarUInt8 Array2dScalarUInt8;

    ImageRepositoryInFile()
    {
    }

    Array2dVec3UInt8*
    Get(const ImageLocator& loc)
    {
        String dir = GetDir(loc);
        File file = RepFS().GetFile(loc, dir, loc.GetName(), false, false);
        if (!file.Valid())
            return 0;

        Array2dVec3UInt8* res = 0;
        Core::Array::ReadFile(res, file);
        return res;
    }

    Array2dScalarUInt8*
    GetImageData(const ImageLocator& loc)
    {
        String dir = GetDir(loc);
        File file = RepFS().GetFile(loc, dir, loc.GetName(), false, false);
        if (!file.Valid())
            return 0;

        Util::IOBuffer* ioBuf = file.GetReadBuffer();
        Int64 bufSize = ioBuf->Size();
        Array2dScalarUInt8* bufIm =
            Core::Array::ArrayCreate<Array2dScalarUInt8>(bufSize, 1);
        Int64 nrRead = ioBuf->Read(bufIm->CPB(), bufSize);
        if (nrRead != bufSize)
            ILOG_ERROR("[GetImageData] Read failed");
        delete ioBuf;
        return bufIm;
    }

    void
    Add(const ImageLocator& loc, Array2dVec3UInt8* image)
    {
        String dir = GetDir(loc);
        File file = RepFS().GetFile(loc, dir, loc.GetName(), true, false);
        Core::Array::WriteFile(image, file);
    }

private:

    RepositoryInFileSystem&
    RepFS()
    {
        return RepositoryInFileSystem::GetInstance();
    }

    String
    GetDir(const ImageLocator& loc)
    {
        String dir = "";
        if (FileNameExt(loc.GetDataSet()) == "txt")
            dir = "ImageData";
        dir = FileNameConcat(dir, loc.GetContainer());
        return dir;
    }

    ILOG_VAR_DEC;
};

ILOG_VAR_INIT(ImageRepositoryInFile, Impala.Persistency);

} // namespace Persistency
} // namespace Impala

#endif

---