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

---

Stills.h

Go to the documentation of this file.

#ifndef Impala_Core_VideoSet_Stills_h
#define Impala_Core_VideoSet_Stills_h

#include "Core/Table/MakeGroupIndex.h"
#include "Core/VideoSet/VideoSet.h"
//#include "Core/ImageSet/ImageSet.h"
#include "Core/Table/TableTem.h"
#include "Core/Table/Read.h"
#include "Core/Table/Write.h"
#include "Core/Table/RemoveRow.h"
#include "Core/Column/Find.h"

namespace Impala
{
namespace Core
{
namespace VideoSet
{


typedef Table::TableTem<Column::ColumnTem<Int32>,
                        Column::ColumnTem<Int32>,
                        Column::ColumnTem<Int32>,
                        Column::ColumnTem<String> > StillsBaseType;


class Stills : public StillsBaseType
{
    typedef Impala::Core::Table::GroupIndex GroupIndex;

public:

    Stills(VideoSet* vidSet, String name) : StillsBaseType(0)
    {
        mVidSet = vidSet;
        mName = name;
        mVideoGroups = 0;
        mShotGroups = 0;
        SetColName(1, "videoId");
        SetColName(2, "shotId");
        SetColName(3, "frameNr");
        SetColName(4, "name");
#ifndef REPOSITORY_USED // Here comes the deprecated stuff
        if (!mName.empty())
            Load();
#else // REPOSITORY_USED
        if (!mName.empty())
            ILOG_ERROR("Load stills via repository");
#endif // REPOSITORY_USED
    }

    // inquiry part

    int
    GetNrStills()
    {
        return Size();
    }

    int
    GetVideoId(int stillId)
    {
        return Get1(stillId);
    }

    int
    GetShotId(int stillId)
    {
        return Get2(stillId);
    }

    String
    GetName(int stillId)
    {
        return Get4(stillId);
    }

    int
    GetFrameNr(int stillId)
    {
        return Get3(stillId);
    }

    int
    GetFrameId(String name)
    {
        int rank = Find(GetColumn4(), name, 0, Size());
        if (rank == Size())
            return -1;
        return rank;
    }

    // frameNr's are not unique, so "segment" start and end are required
    int
    GetFrameId(int frameNr, int idxStart, int idxEnd)
    {
        int rank = Find(GetColumn3(), frameNr, idxStart, idxEnd);
        if (rank == idxEnd)
            return -1;
        return rank;
    }

    VideoSet*
    GetVideoSet()
    {
        return mVidSet;
    }

    // video and shot groups

    void
    UpdateGroups()
    {
        if (mVideoGroups == 0)
        {
            mVideoGroups = new GroupIndex(0);
            mVideoGroups->SetColName(1, "firstStillId");
            mVideoGroups->SetColName(2, "nrStills");
        }
        MakeGroupIndex(mVideoGroups, GetColumn1(), 0, Size());
        if (mShotGroups == 0)
        {
            mShotGroups = new GroupIndex(0);
            mShotGroups->SetColName(1, "firstStillId");
            mShotGroups->SetColName(2, "nrStills");
        }
        // todo?: statement below fails in case a video has only 1 shot
        MakeGroupIndex(mShotGroups, GetColumn2(), 0, Size());
    }

    int
    GetNrVideos()
    {
        return mVideoGroups->Size();
    }

    int
    GetFirstStillVideo(int videoId)
    {
        return mVideoGroups->Get1(videoId);
    }

    int
    GetNrStillsVideo(int videoId)
    {
        return mVideoGroups->Get2(videoId);
    }

    int
    GetFrameIdVideo(int videoId, int frameNr)
    {
        int start = GetFirstStillVideo(videoId);
        int nr = GetNrStillsVideo(videoId);
        return GetFrameId(frameNr, start, start+nr);
    }

    int
    GetNrShots()
    {
        return mShotGroups->Size();
    }

    int
    GetFirstStillShot(int shotId)
    {
        return mShotGroups->Get1(shotId);
    }

    int
    GetNrStillsShot(int shotId)
    {
        return mShotGroups->Get2(shotId);
    }

    // misc

    int
    Diff(Stills* arg)
    {
        if (GetNrStills() != arg->GetNrStills())
        {
            ILOG_ERROR("Diff: nrStills differs: " << GetNrStills() << " vs " <<
                       arg->GetNrStills());
            return 1;
        }
        int nDiff = 0;
        for (int i=0 ; i<GetNrStills() ; i++)
        {
            if (GetVideoId(i) != arg->GetVideoId(i))
            {
                ILOG_DEBUG("Id " << i << " differs " << GetVideoId(i) <<
                           " vs " << arg->GetVideoId(i));
                nDiff++;
            }
            else if (GetShotId(i) != arg->GetShotId(i))
            {
                ILOG_DEBUG("Shot " << i << " differs " << GetShotId(i) <<
                           " vs " << arg->GetShotId(i));
                nDiff++;
            }
            else if (GetFrameNr(i) != arg->GetFrameNr(i))
            {
                ILOG_DEBUG("Frame " << i << " differs " << GetFrameNr(i) <<
                           " vs " << arg->GetFrameNr(i));
                nDiff++;
            }
            else if (GetName(i) != arg->GetName(i))
            {
                ILOG_DEBUG("Name " << i << " differs " << GetName(i) <<
                           " vs " << arg->GetName(i));
                nDiff++;
            }
        }
        if (nDiff > 0)
            ILOG_ERROR("Found " << nDiff << " differences");
        return nDiff;
    }

#ifndef REPOSITORY_USED // Here comes the deprecated stuff
    // I/O part

    void
    Load()
    {
        String fName = mVidSet->GetFilePathVideoIndex("", mName + ".tab",
                                                      false, false);
        Read(this, fName, mVidSet->GetDatabase());
        UpdateGroups();
    }

    void
    Save(String name, bool binary)
    {
        if (!name.empty())
            mName = name;
        String fName = mVidSet->GetFilePathVideoIndex("", mName + ".tab",
                                                      true, false);
        Write(this, fName, mVidSet->GetDatabase(), binary);
    }
#endif // REPOSITORY_USED

    void
    DumpGroups()
    {
        mVideoGroups->Dump();
        mShotGroups->Dump();
    }

private:

    VideoSet*   mVidSet;
    String      mName;
    GroupIndex* mVideoGroups;
    GroupIndex* mShotGroups;

    ILOG_VAR_DEC;
};

ILOG_VAR_INIT(Stills, Impala.Core.VideoSet);

} // namespace VideoSet
} // namespace Core
} // namespace Impala

#endif

---