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

---

SegmentationRepositoryInMonet.h

Go to the documentation of this file.

#ifndef Impala_Persistency_SegmentationRepositoryInMonet_h
#define Impala_Persistency_SegmentationRepositoryInMonet_h

#include "Persistency/RepositoryInMonetDB.h"
#include "Core/VideoSet/Segmentation.h"
#include "Persistency/SegmentationLocator.h"

namespace Impala
{
namespace Persistency
{


class SegmentationRepositoryInMonet
{
public:

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

    SegmentationRepositoryInMonet()
    {
    }

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

        String vidSetBase = FileNameBase(loc.GetDataSet());
        String query =
            "select f.quid, fr.fragment_start, fr.fragment_length, \
                    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.fragment_length > 1 and \
                    fr.keyframe = false \
             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;

        int* length = 0;
        int lengthSize = 0;
        if (!conn->QueryPartFetchInt(hdl, 2, length, lengthSize))
            return 0;

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

        conn->QueryPartEnd(hdl);

        Segmentation* seg = new Segmentation(vidSet, "");
        for (int i=0 ; i<quidSize ; i++)
        {
            int id = QuidId(quid[i]);
            int end = start[i] + length[i] - 1;
            seg->Add(id, start[i], end, name[i]);
        }
        seg->UpdateGroups();

        delete quid;
        delete start;
        delete length;
        return seg;
    }

    void
    Add(const SegmentationLocator& loc, Segmentation* seg)
    {
        Connection* conn = RepMonet().GetConnection(loc);

        VideoSet* vidSet = seg->GetVideoSet();
        conn->Query("delete from i_bulk_fragment;", false, false);
        Timer timer;
        for (int v=0 ; v<seg->GetNrVideos() ; v++)
        {
            ILOG_INFO("v = " << v);
            Quid vidQuid = vidSet->GetQuidVideo(v, true);
            int first = seg->GetFirstShotVideo(v);
            int nr = seg->GetNrShotsVideo(v);
            String q = "copy " + MakeString(nr) +
              " records into i_bulk_fragment from stdin using delimiters ' ';\n";
            for (int s=first ; s<first+nr ; s++)
            {
                int start = seg->GetStart(s);
                int len = seg->GetEnd(s) - start + 1;
                q += MakeString(vidQuid) + " " + MakeString(start) + " "
                    + MakeString(len) + " " + seg->GetName(s)
                    + " false false\n";
            }
            conn->Query(q, false, false);

            nr = seg->GetNrFramesVideo(v);
            q = "copy " + MakeString(nr) +
              " records into i_bulk_fragment from stdin using delimiters ' ';\n";
            for (int f=0 ; f<nr ; f++)
            {
                q += MakeString(vidQuid) + " " + MakeString(f) + " 1 frame"
                    + MakeString(f) + " false false\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(SegmentationRepositoryInMonet, Impala.Persistency);

} // namespace Persistency
} // namespace Impala

#endif

---