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

---

TableDataSourceImageDir.h

Go to the documentation of this file.

#ifndef MediaTable_TableDataSourceImageDir_h
#define MediaTable_TableDataSourceImageDir_h

#include "TableDataSource.h"
#include "Core/Array/ReadFile.h"
#include "Core/Stream/RgbDataSrcRaw.h"
#include "Util/DirImageFileNames.h"

namespace Impala {
namespace Application {
namespace MediaTable {

class TableDataSourceImageDir : public TableDataSource
{
public:
    typedef Impala::Core::Stream::RgbDataSrcRaw         RgbDataSrcRaw;

    TableDataSourceImageDir(CString directory):
        TableDataSource()
    {
        Init();
        OpenDir(directory);

        // TODO: SetSortColumn("filename"); Make preferred?
    }

    ~TableDataSourceImageDir()
    {
        ClearFiles();
    }

    int GetTotalRows() {
        return mTotalRows;
    }

    void OpenDir(CString dirName)
    {
        ClearFiles();

        mDirName    = dirName;
        String ext  = FileNameExt(dirName,true);

        if (ext == "raw")
            mRgbDataSrcRaw = new RgbDataSrcRaw(dirName);
        else
            Util::DirImageFileNames(dirName, mFileNames);

        if (mRgbDataSrcRaw)
            mTotalRows = mRgbDataSrcRaw->LastFrame();
        else
            mTotalRows = mFileNames.size();
    }

    void ClearFiles()
    {
        if (mRgbDataSrcRaw)
        {
            delete mRgbDataSrcRaw;
            mRgbDataSrcRaw = 0;
        }
        for (unsigned int i=0 ; i<mFileNames.size() ; i++)
            delete mFileNames[i];
        mFileNames.clear();
    }

    /* get individual row characteristics ************************/
    virtual String
    GetTextDataByID(String column, int row)
    {
        if (column == "filename")
        {
            if (!mRgbDataSrcRaw)
                return *mFileNames[row];
            return String("raw:") + MakeString(row);
        }
        if (column == "id")
            return MakeString(row);
        return "N/A";
    }

    //virtual int
    //GetRowHeight(int row)
    //{
    //    return 40;
    //}

    virtual double
    GetNormalizedDataByID(String column, int row)
    {
        double v = row / (double)mTotalRows;
        if (column == "red")
            return v;
        if (column == "green")
            return 1.0 - v;
        if (column == "blue") {
            return (row % 10) / 10.0;
        }
        return 0.0;
    }

    virtual Array2dVec3UInt8*
    GetImageDataByID(String colum, int row)
    {
        Array2dVec3UInt8* ar = 0;
        if (mRgbDataSrcRaw)
        {
            mRgbDataSrcRaw->GotoFrame(row);
            int imW = mRgbDataSrcRaw->FrameWidth();
            int imH = mRgbDataSrcRaw->FrameHeight();
            ar = new Array2dVec3UInt8(imW, imH, 0, 0, 0, false);
            memcpy((void *)ar->CPB(0, 0), mRgbDataSrcRaw->DataPtr(), imW*imH*3);
        }
        else
        {
            String fName = mDirName + "/" + *mFileNames[row];
            Core::Array::ReadFile(ar, fName, &Util::Database::GetInstance());
            if (ar == 0)
                ILOG_WARN("Could not read file: " << fName);
        }
        return ar;
    }

private:

    void Init()
    {
        mRgbDataSrcRaw = 0;

        AddStaticColumn("id", TYPE_TEXT, 50);
        AddStaticColumn("image", TYPE_IMAGE, 120);
        AddStaticColumn("filename", TYPE_TEXT, 300);
        /*AddColumn("red", TYPE_NORMALIZEDVALUE, 50, 1);
        AddColumn("green", TYPE_NORMALIZEDVALUE, 50, 2);
        AddColumn("blue", TYPE_NORMALIZEDVALUE, 50, 3);
        AddColumn("red", TYPE_NORMALIZEDVALUE, 50, 1);
        AddColumn("green", TYPE_NORMALIZEDVALUE, 50, 2);
        AddColumn("blue", TYPE_NORMALIZEDVALUE, 50, 3);*/
    }

    String                      mDirName;
    RgbDataSrcRaw*              mRgbDataSrcRaw;
    std::vector<String*>        mFileNames;
    int                                         mTotalRows;

    ILOG_VAR_DEC;
};

ILOG_VAR_INIT(TableDataSourceImageDir, Application.MediaTable);

} // namespace MediaTable
} // namespace Application
} // namespace Impala

#endif // TableDataSourceImageDir_h

---