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

---

mainAnnoVidSet.cpp

Go to the documentation of this file.

#include "Visualization/SegmentationDocumentGuiAnno.h"
#include "OglGui/Tabs.h"
#include "OglGui/OglLib.cpp"
#include "Core/VideoSet/MakeSegmentationDocument.h"
#include "Visualization/SegmentationDocumentGuiStills.h"

// since we are not using libraries:
#include "Link/ImpalaLib.cpp"


namespace Impala {
namespace Application {

using namespace Visualization;

class WindowAnnoVidSet :    public Window,
                            public OglGui::ButtonListener,
                            public OglGui::TabsListener
{
public:

    const static int TAB_ANNO = 0;
    const static int TAB_STILLS = 1;

        WindowAnnoVidSet(int width, int height)
                : Window(0, 0, width, height, true)
    {
        SetControlId(-10);
        CmdOptions& options = CmdOptions::GetInstance();
        mViewScale = options.GetDouble("viewScale");
        ILOG_INFO("loading data");
        InitData();
        ILOG_INFO("building GUI");
        BuildGUI(width, height);
                ILOG_INFO("STARTING MAIN LOOP");
        AppController::Instance().MainLoop();
    }

    // specialization of base classes

    virtual void
    TabSelectionEvent(OglGui::Tabs* src, int tabNr, void* listenerData,
                      int tabControlData)
    {
        AppController::Instance().SetCurrentControl(tabControlData);
    }

    virtual void
    DisplayFunc()
    {
        static bool initialized = false;

        if (!initialized)
        {
            initialized = true;
            ScaleChildren(0, true);
            SetReposChildren(false, true);
        }
        Window::DisplayFunc();
    }

private:

    // initialization

    void
    InitData()
    {
        CmdOptions& options = CmdOptions::GetInstance();
        std::string videoSetName = options.GetString("videoSet");
        printf("VideoSetName = %s\n", videoSetName.c_str());
        mSegDoc = Core::VideoSet::MakeSegmentationDocument(videoSetName);
    }

    void BuildGUI(int wndWidth, int wndHeight)
    {
        CmdOptions& options = CmdOptions::GetInstance();
        //SetBackground(0xffd2c799);
        SetBackground(oglGUI_BG);
        mTabs = new OglGui::Tabs(this, wndWidth, wndHeight);
        mTabs->SetTabsListener(this);
        OglGui::Window* tab;
        AppController& ctr = AppController::Instance();

        typedef Core::VideoSet::SegmentationDocument SegmentationDocument;
        tab = mTabs->CreateTab("Anno", -1, TAB_ANNO);
        std::string conceptSet = options.GetString("conceptSet");
        int maxAnno = options.GetInt("maxAnnoPerVideo");
        mSegDocAnno = new SegmentationDocumentGuiAnno
            (tab, tab->W(), tab->H(), mSegDoc, conceptSet, maxAnno, TAB_ANNO);
        ctr.AddControl((AppControlDoc*) mSegDocAnno);
// RvB: To start with the directory selected
mSegDocAnno->StringSelectionEvent(NULL,0,(void*)1);

        if (mSegDoc->GetImSetStills())
        {
            Core::ImageSet::ImageSet* imSet = mSegDoc->GetImSetKeyframes(false);
            Core::Array::Array2dVec3UInt8* im = imSet->GetImage(0);
            int thumbWidth = im->CW() / 2;
            int thumbHeight = im->CH() / 2;
            delete im;
            Window* wnd2 = new Window(wndWidth+6, 0, wndWidth, wndHeight);
            wnd2->SetControlId(-11);
            wnd2->SetBackground(0xffd2c799);
            mSegDocStills = new SegmentationDocumentGuiStills
                (wnd2, wndWidth, wndHeight, 4, thumbWidth,
                 thumbHeight, mViewScale, true, true, mSegDoc, TAB_STILLS);
            mSegDocAnno->AddDocListener(mSegDocStills);
            oglSys.SetAlwaysDraw(mSegDocStills->GetOGLWND(), 1);
        }

        mTabs->SetActiveTab(0);
    }

    double                                mViewScale;
    Core::VideoSet::SegmentationDocument* mSegDoc;

    OglGui::Tabs*                     mTabs;
    SegmentationDocumentGuiAnno*      mSegDocAnno;
    SegmentationDocumentGuiStills*    mSegDocStills;

    ILOG_VAR_DEC;
};

ILOG_VAR_INIT(WindowAnnoVidSet, Application);


char GetKeyBinding(std::string name, std::string dflt)
{
    CmdOptions& options = CmdOptions::GetInstance();
        std::string str = options.GetString(name, dflt);
        return str[0];
}

int
mainAnnoVidSet(int argc, char* argv[])
{
    OglInit(&argc, &argv[0]);

    CmdOptions& options = CmdOptions::GetInstance();

    options.Initialise(true, false, true);
    options.SetDefault("wndWidth", "1272");
    options.SetDefault("wndHeight", "955");
    if (! options.ParseArgs(argc, argv, "", 0))
        return 1;
/*
        std::vector<char> keyBinding;
        keyBinding.push_back(GetKeyBinding("forward", "0"));
        keyBinding.push_back(GetKeyBinding("backward", "9"));
        keyBinding.push_back(GetKeyBinding("positive", "1"));
        keyBinding.push_back(GetKeyBinding("skip", "2"));
        keyBinding.push_back(GetKeyBinding("negative", "3"));
        keyBinding.push_back(GetKeyBinding("shotpositive", "6"));
        keyBinding.push_back(GetKeyBinding("shotnegative", "7"));
        keyBinding.push_back(GetKeyBinding("togglePlay", " "));
*/
    ILOG_VAR(Application.mainAnnoVidSet);

    int reqWndWidth = options.GetInt("wndWidth");
    int reqWndHeight = options.GetInt("wndHeight");
    WindowAnnoVidSet* oglWnd =
                new WindowAnnoVidSet(reqWndWidth, reqWndHeight);
    if (! oglWnd->Valid())
    {
        ILOG_ERROR("WindowAnnoVidSet failed");
        return 1;
    }
    return 0;
}

} // namespace Application
} // namespace Impala

int
main(int argc, char* argv[])
{
    return Impala::Application::mainAnnoVidSet(argc, argv);
}

---