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

---

KeyframesRepositoryInMonet.h

Go to the documentation of this file.

#ifndef Impala_Persistency_KeyframesRepositoryInMonet_h
#define Impala_Persistency_KeyframesRepositoryInMonet_h

#include "Persistency/SegmentationRepository.h"
#include "Persistency/RepositoryInMonetDB.h"
#include "Core/VideoSet/Keyframes.h"
#include "Persistency/KeyframesLocator.h"

namespace Impala
{
namespace Persistency
{


class KeyframesRepositoryInMonet
{
public:

    typedef Link::Monet::Connection Connection;
    typedef Core::VideoSet::Keyframes Keyframes;
    typedef Core::VideoSet::Segmentation Segmentation;
    typedef Core::VideoSet::VideoSet VideoSet;

    KeyframesRepositoryInMonet()
    {
    }

    Keyframes*
    Get(KeyframesLocator loc, VideoSet* vidSet)
    {
        Connection* conn = RepMonet().GetConnection(loc);

        String vidSetBase = FileNameBase(loc.GetDataSet());
        String query =
            "select f.quid, fr.fragment_start, fr.fragment_name \
             from   fragment fr, file f, video_set vs, video_sets vss   \
             where  vss.set_name = '" + vidSetBase + "' and \
                    vs.video_sets_id = vss.video_sets_id and \
                    f.file_id = vs.file_id and \
                    fr.media_id = f.media_id and \
                    fr.keyframe = true \
             order by f.quid, fr.fragment_start;";

        MapiHdl hdl = conn->QueryPartStart(query);
        if (hdl == 0)
            return 0;

        Quid* quid = 0;
        int quidSize = 0;
        if (!conn->QueryPartFetchULL(hdl, 0, quid, quidSize))
            return 0;

        int* start = 0;
        int startSize = 0;
        if (!conn->QueryPartFetchInt(hdl, 1, start, startSize))
            return 0;

        std::vector<String> name;
        if (!conn->QueryPartFetchString(hdl, 2, name))
            return 0;

        conn->QueryPartEnd(hdl);

        SegmentationLocator segLoc(loc.GetProtocolAndHost(), loc.GetDataSet(),
                                   "segmentation");
        Segmentation* seg = SegmentationRepository().Get(segLoc, vidSet);
        Keyframes* keys = new Keyframes(vidSet, "");
        for (int i=0 ; i<quidSize ; i++)
        {
            int vidId = QuidId(quid[i]);
            int shotId = seg->GetShotId(vidId, start[i]);
            keys->Add(vidId, shotId, start[i], name[i]);
        }
        keys->UpdateGroups();
        delete seg;

        delete quid;
        delete start;
        return keys;
    }

    void
    Add(const KeyframesLocator& loc, Keyframes* keys)
    {
        Connection* conn = RepMonet().GetConnection(loc);

        VideoSet* vidSet = keys->GetVideoSet();
        conn->Query("delete from i_bulk_fragment;", false, false);
        Timer timer;
        for (int v=0 ; v<keys->GetNrVideos() ; v++)
        {
            ILOG_INFO("v = " << v);
            Quid vidQuid = vidSet->GetQuidVideo(v, true);
            int first = keys->GetFirstKeyframeVideo(v);
            int nr = keys->GetNrKeyframesVideo(v);
            String q = "copy " + MakeString(nr) +
                " records into i_bulk_fragment from stdin using delimiters ' ';\n";
            for (int k=first ; k<first+nr ; k++)
            {
                //int shot = keys->GetShotId(k); // cannot store in schema
                int frame = keys->GetFrameNr(k);
                String rkf = (keys->IsRKF(k)) ? "true" : "false";
                q += MakeString(vidQuid) + " " + MakeString(frame) + " 1 "
                    + keys->GetName(k) + " true " + rkf + "\n";
            }
            conn->Query(q, false, false);
        }
        ILOG_INFO("Did bulk in " << timer.SplitTime());
        conn->Query("insert into fragment (media_id, fragment_start, \
                                           fragment_length, fragment_name,  \
                                           keyframe, representative) \
                     select m.media_id, bf.fragment_start, bf.fragment_length, \
                            bf.fragment_name, bf.keyframe, bf.representative \
                     from i_bulk_fragment bf, media m \
                     where bf.quid = m.quid \
                     order by bf.quid, bf.fragment_start;", false, false);
        ILOG_INFO("Did insert select at " << timer.SplitTime());
        conn->Query("delete from i_bulk_fragment;", false, false);
    }

private:

    RepositoryInMonetDB&
    RepMonet()
    {
        return RepositoryInMonetDB::GetInstance();
    }

    ILOG_VAR_DEC;
};

ILOG_VAR_INIT(KeyframesRepositoryInMonet, Impala.Persistency);

} // namespace Persistency
} // namespace Impala

#endif

---