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

---

FixSizes.h

Go to the documentation of this file.

#ifndef Impala_Core_ImageSet_FixSizes_h
#define Impala_Core_ImageSet_FixSizes_h

#include "Core/ImageSet/Reporter.h"
#include "Core/ImageSet/MakeImageSet.h"
#include "Core/Array/MakeFromData.h"
#include "Core/Array/ArrayListDelete.h"
#include "Core/Array/WriteRaw.h"
#include "Core/Array/Scale.h"
#include "Core/Array/ImageArchiveMemory.h"

namespace Impala
{
namespace Core
{
namespace ImageSet
{


class FixSizes : public Listener
{
public:

    FixSizes(Reporter* reporter, CmdOptions& options)
    {
        mReporter = reporter;
        mReporter->SetReportArray(false); // we will cause a "memory leak"
        mMaxWidth = 1000;
        mMaxHeight = 1000;
        if(options.GetNrArg() > 3)
        {
            mMaxWidth = atol(options.GetArg(2));
            mMaxHeight = atol(options.GetArg(3));
        }
        ILOG_INFO(" resizing images to "<< mMaxWidth <<" by "<< mMaxHeight);
        mSplit = options.GetNrArg() > 4;
        ILOG_INFO((mSplit?"":"not ") << "using split archives");

        mFileNameAddition = options.GetString("addToName");
        if(mFileNameAddition != "")
            ILOG_INFO(mFileNameAddition << " will be added to filenames");
    }

    virtual
    ~FixSizes()
    {
        Array::ArrayListDelete(&mImList);
    }

    void
    AddOriginal(ImageSet* is, int fileId)
    {
        Array::Array2dScalarUInt8* im = is->GetImageData(fileId);
        mImList.push_back(im);
        // ILOG_DEBUG("added "<< fileId <<" as is.");
    }

    void
    AddScaled(int fileId, Array::Array2dVec3UInt8* im)
    {
        // first scale image
        double scale = 1.;
        while(scale*im->CW() > mMaxWidth || scale*im->CH() > mMaxHeight)
            scale *= 0.5;
        Array::Array2dVec3UInt8* scaled = 0;
        Array::Scale(scaled, im, scale, scale, Geometry::NEAREST, true);
        // then compress the scaled image
        size_t bufSize = mMaxHeight*mMaxWidth;
        char* buf = new char[bufSize];
        size_t jpgSize=0;
        WriteJpgToMemory(scaled, buf, bufSize, &jpgSize);
        Array::Array2dScalarUInt8* jpg =
            Array::MakeFromData<Array::Array2dScalarUInt8>((UInt8*)buf, jpgSize, 1);
        delete buf;
        // now store it
        mImList.push_back(jpg);
        ILOG_DEBUG("fixed im "<< fileId <<" to "<< scaled->CW() <<" by "<<
                  scaled->CH());
        delete scaled;
    }
    
    virtual void
    HandleNewFile(ImageSet* is, int fileId, Array::Array2dVec3UInt8* im)
    {

        if(im->CW() <= mMaxWidth && im->CH() <= mMaxHeight)
            AddOriginal(is, fileId);
        else
            AddScaled(fileId, im);
    }

    virtual void
    HandleDoneDir(ImageSet* is, int dirId)
    {
        if (!mSplit)
            return;
        String arName = "images" + mFileNameAddition + ".raw";
#ifndef REPOSITORY_USED // Here comes the deprecated stuff
        int fileId = is->GetFirstFileId(dirId);
        String fName = is->GetFilePathImageArchive(fileId, arName, true, false);
        ILOG_INFO("writing archive [" << fName << "]");
        Array::WriteRawListVar(mImList, fName, is->GetDatabase(), true, true);
#else // REPOSITORY_USED
        String container = is->GetContainerDir(dirId);
        Persistency::ImageArchiveLocator loc(is->GetLocator(), false,
                                             container, arName, 0);
        Array::ImageArchiveMemory wrapper(&mImList);
        Persistency::ImageArchiveRepository().Add(loc, &wrapper);
#endif // REPOSITORY_USED
        Array::ArrayListDelete(&mImList);
    }

    virtual void
    HandleDoneWalk(ImageSet* is)
    {
        if (mSplit)
            return;

        String arName = is->GetSetNameBase() + mFileNameAddition + ".raw";
#ifndef REPOSITORY_USED // Here comes the deprecated stuff
        String fName = is->GetFilePathImageArchive(arName, true, false);
        ILOG_INFO("writing archive [" << fName << "]");
        Array::WriteRawListVar(mImList, fName, is->GetDatabase(), true, true);
#else // REPOSITORY_USED
        Persistency::ImageArchiveLocator loc(is->GetLocator(), false,
                                             "", arName, 0);
        Array::ImageArchiveMemory wrapper(&mImList);
        Persistency::ImageArchiveRepository().Add(loc, &wrapper);
#endif // REPOSITORY_USED
    }

private:

    String mFileNameAddition;
    int mMaxWidth;
    int mMaxHeight;
    bool mSplit;
    Reporter* mReporter;
    std::vector<Array::Array2dScalarUInt8*> mImList;
    ILOG_VAR_DEC;
};

ILOG_VAR_INIT(FixSizes, Impala.Core.ImageSet);

} // namespace ImageSet
} // namespace Core
} // namespace Impala

#endif

---