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

---

mainShowImSet.cpp

Go to the documentation of this file.

//345678901234567890123456789012345678901234567890123456789012345678901234567890
#include "OglGui/Tabs.h"
#include "OglGui/SplashScreen.h"
// since we are not using libraries:
#include "OglGui/OglLib.cpp"

#include "Visualization/Window.h"
#include "Visualization/ShowImSetGuiMain.h"
#include "Visualization/AnnotationTableGui.h"
#include "Visualization/ImageSetIdxGridScroller.h"
#include "Core/ImageSet/MakeIxsDocument.h"
#include "Core/Table/SimilarityTableSet.h"

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


namespace Impala {
namespace Application {

using namespace Visualization;
//using namespace Core::ImageSet;

class WindowShowImSet : public Window,
                        public OglGui::TabsListener
{
public:

    const static int TAB_SET = 0;
    const static int TAB_MAIN = 1;
    const static int TAB_ALL = 2;
    const static int TAB_ANNOTATIONS = 3;
    const static int TAB_CONCEPTS = 4;
    const static int TAB_CONCEPTSSIM = 5;
    const static int TAB_CONCEPTSTRUTH = 6;

    WindowShowImSet(int width, int height) : Window(0, 0, width, height, true)
    {
        CmdOptions& options = CmdOptions::GetInstance();
        mViewScale = options.GetDouble("viewScale");
        ILOG_INFO("loading data");
        InitData();
        ILOG_INFO("building GUI");
        View::SetTags(selectableTag | showBgTag | showBorderTag |
                      showBorderTag | showTitleTag);
        BuildGUI(width, height);
    }

    // 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();
        String imSetName = options.GetString("imageSet");
        mIxsDoc = Core::ImageSet::MakeIxsDocument(imSetName);
        //mSegDoc = Core::VideoSet::MakeSegmentationDocument(videoSetName);
        
        Core::ImageSet::ImageSet* imSetThumb = mIxsDoc->GetImageSet();
        imSetThumb->SetImageSrc(options.GetBool("imArchive"),
                                options.GetBool("imFileArchive"),
                                options.GetBool("imSplitArchive"));
        if (imSetThumb)
        {
            Core::Array::Array2dVec3UInt8* im = imSetThumb->GetImage(0);
            mThumbWidth = im->CW();
            mThumbHeight = im->CH();
            delete im;
        }

        String conceptSetName = options.GetString("conceptSet");
        mConceptRanking = 0;
        if (!conceptSetName.empty())
        {
            String model = options.GetString("conceptModel");
            String feature = options.GetString("conceptFeature");
            ILOG_INFO("reading concept similarity set");
            mConceptRanking = Core::Table::SimilarityTableSet::MakeFromFile
                (mIxsDoc->GetImageSet(), conceptSetName, model, feature);
        }
    }

    void BuildGUI(int wndWidth, int wndHeight)
    {
        CmdOptions& options = CmdOptions::GetInstance();

        SetBorderType(-1);
        if (!GetBorderBackground())
            SetBorderBackground(oglGUI_BG);

        mTabs = new OglGui::Tabs(this, wndWidth, wndHeight);
        mTabs->SetTabsListener(this);

        OglGui::Window* tab = mTabs->CreateTab("Main", -1, TAB_MAIN);
        //int nrImOnRow = wndWidth / (mThumbWidth + 3);
        mThumbWidth = 176;
        mThumbHeight = 176;
        int nrImOnRow = options.GetInt("NrImOnRow", 5);
        int height = wndHeight;
        mGuiMain = new ShowImSetGuiMain(tab, tab->W(), tab->H(), nrImOnRow,
                                        true, mIxsDoc, TAB_MAIN);
        AppController& ctr = AppController::Instance();
        ctr.AddControl((AppControlDoc*) mGuiMain);

        tab = mTabs->CreateTab("All", 60, TAB_ALL);
        mImSetIdxGridScroller =
            new ImageSetIdxGridScroller(tab,0,0,tab->W(),tab->H());
        //mImSetIdxGridScroller->PageUpDownExtra(-2);
        mImSetIdxGridScroller->ConnectTo(tab);
        ImageSetIdxGrid* idxGrid = mImSetIdxGridScroller->ImSetIdxGrid();
        idxGrid->AddImageSetIdxs(mIxsDoc->GetImageSet());
        idxGrid->UpdateLayout(true);

        String conceptAnnotations = options.GetString("conceptAnnotations");
        if (!conceptAnnotations.empty())
        {
            tab = mTabs->CreateTab("Annotations", 100, TAB_ANNOTATIONS);
            mAnnotation = new AnnotationTableGui
                (tab, tab->W(), tab->H(), nrImOnRow, mThumbWidth, mThumbHeight,
                 mViewScale, mIxsDoc, conceptAnnotations, TAB_ANNOTATIONS);
            ctr.AddControl((AppControlDoc*) mAnnotation);
        }

        if (mConceptRanking)
        {
            tab = mTabs->CreateTab("Concepts", 82, TAB_CONCEPTS);
            mSimSetRank = new SimilarityTableSetRank
                (tab, tab->W(), tab->H(), nrImOnRow, mThumbWidth, mThumbHeight,
                 mViewScale, mIxsDoc, mConceptRanking, false,
                 conceptAnnotations, false, TAB_CONCEPTS);
            ctr.AddControl((AppControlDoc*) mSimSetRank);

            tab = mTabs->CreateTab("ConceptsSim", -1, TAB_CONCEPTSSIM);
            mSimSetRankSim = new SimilarityTableSetRank
                (tab, tab->W(), tab->H(), nrImOnRow, mThumbWidth, mThumbHeight,
                 mViewScale, mIxsDoc, mConceptRanking, true,
                 conceptAnnotations, false, TAB_CONCEPTSSIM);
            ctr.AddControl((AppControlDoc*) mSimSetRankSim);

            if (!conceptAnnotations.empty())
            {
                tab = mTabs->CreateTab("ConceptsTruth", 120, TAB_CONCEPTSTRUTH);
                mSimSetRankTruth = new SimilarityTableSetRank
                    (tab, tab->W(), tab->H(), nrImOnRow, mThumbWidth,
                     mThumbHeight, mViewScale, mIxsDoc, mConceptRanking, false,
                     conceptAnnotations, true, TAB_CONCEPTSTRUTH);
                ctr.AddControl((AppControlDoc*) mSimSetRankTruth);
            }
        }

        // RvB: To start with the directory selected
        mGuiMain->StringSelectionEvent(NULL,0,(void*)1);

        mTabs->SetActiveTab(0);
    }

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


    // Configuration
    int         mThumbWidth;
    int         mThumbHeight;
    ULONG       mBackGround;

    // DATA
    double                              mViewScale;

    Core::ImageSet::IxsDocument*        mIxsDoc;
    Core::Table::SimilarityTableSet*    mConceptRanking;

    ShowImSetGuiMain*                   mGuiMain;
    AnnotationTableGui*                 mAnnotation;
    SimilarityTableSetRank*             mSimSetRank;
    SimilarityTableSetRank*             mSimSetRankSim;
    SimilarityTableSetRank*             mSimSetRankTruth;

    ImageSetIdxGridScroller*            mImSetIdxGridScroller;

    // GUI
    OglGui::Tabs*                     mTabs;
    ILOG_VAR_DEC;
};

ILOG_VAR_INIT(WindowShowImSet, Application);


int
mainShowImSet(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");
    options.AddOption(0, "small", "", "0");
    options.AddOption(0, "noArchive", "", "0");
    options.AddOption(0, "imFileArchive", "", "0");
    //options.AddOption(0, "conceptSetSubDir", "featurename", "");
    if (! options.ParseArgs(argc, argv, "", 0))
        return 1;

    ILOG_VAR(Application.mainShowImSet);

    int reqWndWidth = options.GetInt("wndWidth");
    int reqWndHeight = options.GetInt("wndHeight");
    if (options.GetBool("small"))
    {
        reqWndWidth = 1020;
        reqWndHeight = 740;
    }

    OglGui::SplashScreen* splash;
    if (options.GetBool("splashScreen"))
    {
        splash = new OglGui::SplashScreen(0,0,reqWndWidth,reqWndHeight);
        OGLStartThread(splash->GetOGLWND());
    }

    WindowShowImSet* oglWnd = new WindowShowImSet(reqWndWidth, reqWndHeight);

    if (options.GetBool("splashScreen"))
        splash->Stop();

    if (! oglWnd->Valid())
    {
        ILOG_ERROR("WindowShowImSet failed");
        return 1;
    }

    ILOG_INFO("STARTING MAIN LOOP");
    AppController::Instance().MainLoop();

    return 0;
}

} // namespace Application
} // namespace Impala

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

---