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

---

AppControlSrc.h

Go to the documentation of this file.

#ifndef Impala_Visualization_AppControlSrc_h
#define Impala_Visualization_AppControlSrc_h

#include "Persistency/ImageRepository.h"
#include "Basis/Timer.h"
#include "Basis/CmdOptions.h"
#include "Util/Database.h"
#include "Basis/FileName.h"
#include "Core/Array/WritePng.h"
#include "Core/Stream/RgbDataSrc.h"
#include "Visualization/AppControl.h"

namespace Impala
{
namespace Visualization
{


class AppControlSrc : public AppControl
{
public:
    typedef Core::Stream::RgbDataSrc RgbDataSrc;

    AppControlSrc(int id) : AppControl(id, false), mTimer(1)
    {
        CmdOptions& options = CmdOptions::GetInstance();
        mSrc = 0;
        mSrc2 = 0;
        mDone = false;
        mHadJump = false;
        mStartFrame = options.GetInt("startFrame");
        mMarkedFrameBegin = -1;
        mMarkedFrameEnd = -1;
        mDoContinuous = ! options.GetBool("step");
        mStepSize = options.GetInt("stepSize");
        mDoLoop = options.GetBool("loop");
        mDoOneStep = true;
        mFramesDone = 0;
        mDelay = options.GetInt("delay");
        mDidSrcError = false;
    }

    virtual void
    HandleActivate()
    {
        HandleNewFrame();
        HandleCycleSrc();
    }

    void
    SetSrc(RgbDataSrc* src, int startFrame = -1, // -1 indicates default
           RgbDataSrc* src2 = 0)
    {
        mSrc = src;
        mSrc2 = src2;
        //mDidSrcError = false;
        if (!mSrc)
            return;
        //mTimer.Start();
        FpsReset();
        if (startFrame == -1)
            startFrame = mStartFrame;
        if ((startFrame != 0) || (mDelay != 0))
        {
            mSrc->GotoFrame(startFrame);
            if (mSrc2)
                mSrc2->GotoFrame(startFrame + mDelay);
        }
        else
        {
            if (mSrc->FrameNr() == -1) // -1 indicates not started yet
                mSrc->NextFrame();
            if (mSrc2 && (mSrc2->FrameNr() == -1))
                mSrc2->NextFrame();
        }
        CheckJumpFrame();
    }

    RgbDataSrc*
    GetSrc()
    {
        return mSrc;
    }

    RgbDataSrc*
    GetSrc2()
    {
        return mSrc2;
    }

    void
    Jump(int frameNr)
    {
        if (!mSrc || frameNr < 0)
            return;
        if (mSrc2)
            frameNr -= mDelay; // hack : mSrc2 is reference point
        mSrc->GotoFrame(frameNr);
        if (mSrc2)
            mSrc2->GotoFrame(frameNr + mDelay);
        CheckJumpFrame();
    }

    void
    SetDoLoop(bool flag)
    {
        mDoLoop = flag;
    }

    void
    SetDoContinuous(bool flag)
    {
        mDoContinuous = flag;
    }

    bool GetDoContinuous()
    {
        return mDoContinuous;
    }

    int
    GetMarkedFrameBegin()
    {
        return mMarkedFrameBegin;
    }

    void
    SetMarkedFrameBegin(int frameNr)
    {
        mMarkedFrameBegin = frameNr;
    }

    int
    GetMarkedFrameEnd()
    {
        return mMarkedFrameEnd;
    }

    void
    SetMarkedFrameEnd(int frameNr)
    {
        mMarkedFrameEnd = frameNr;
    }

    void
    FpsReset()
    {
        mTimer.Start();
        mFramesDone = 0;
    }

    std::string
    GetFpsString()
    {
        if (!mSrc)
            return std::string("no src");
        double timeVal = mTimer.SplitTime();
        double fps = (double) mFramesDone / timeVal;
        std::ostringstream fpsStream;
        fpsStream << "FrameNr = " << mSrc->FrameNr()
                  << " (" << mSrc->FrameType() << ")";
        if (mSrc2)
            fpsStream << ", FrameNr2 = " << mSrc2->FrameNr()
                      << " (" << mSrc->FrameType() << ")";
        fpsStream << ", did " << mFramesDone << " frames in " << timeVal
                  << " sec = " << fps << " fps" << std::ends;
        return std::string(fpsStream.str());
    }

    // specialization of base class

    virtual void
    ComputeCycle(double elapsedTime)
    {
        if (! mSrc)
        {
            if (! mDidSrcError)
            {
                ILOG_ERROR("AppControlSrc::ComputeCycle : no src");
                mDidSrcError = true;
            }
            return;
        }
        if (mDoContinuous || mDoOneStep)
        {
            if (! mHadJump)  // already moved to current frame
            {
                mSrc->NextFrame(mStepSize);
                if (mSrc2)
                    mSrc2->NextFrame(mStepSize);
            }
            mHadJump = false;
            HandleNewFrame();
            mFramesDone++;
            mDoOneStep = false;
        }
        HandleCycleSrc();
    }

    virtual void
    KeyEvent(OglGui::Window* src, int c, int state)
    {
        if (!GetActive())
            return;

        if (c == 'x')
        {
            ExportFrame();
        }

        if (c == 's')
        {
            mDoOneStep = true;
            mStepSize = abs(mStepSize);
        }
        if (c == 'S')
        {
            mDoOneStep = true;
            mStepSize = - abs(mStepSize);
        }
        if (c == 'a')
            HandleDelayUpdate(1);
        if (c == 'A')
            HandleDelayUpdate(-1);
        if (c == 'c')
        {
            mDoContinuous = true;
            FpsReset();
        }
        if (c == 'C')
            mDoContinuous = false;
        if (c == 'd')
            Jump(mMarkedFrameBegin);
        if (c == 'D')
            mMarkedFrameBegin = mSrc->FrameNr();
        if (c == 'e')
            Jump(mMarkedFrameEnd);
        if (c == 'E')
            mMarkedFrameEnd = mSrc->FrameNr();
        if (c == 'Q')
        {
            ILOG_INFO("Instant exit button pressed.");
            mDone = true;
            HandleDestroyEvent();
        }
        if (c == ' ')
            mDoContinuous=!mDoContinuous ;
        if (c == '<')
            Jump(mSrc->FrameNr()-1000);
        if (c == '>')
            Jump(mSrc->FrameNr()+1000);
        if (c == '[')
            Jump(mSrc->FrameNr()-100);
        if (c == ']')
            Jump(mSrc->FrameNr()+100);
        if (c == '}' )
            Jump(mSrc->FrameNr()+10);
        if (c == '{' )
            Jump(mSrc->FrameNr()-10);
    }

    virtual bool
    Done()
    {
        //if (!mSrc || mDone)
        if(mDone)
            HandleDestroyEvent();

        if (mDone)
            return true;
        if (!mSrc)
            return false;
        if (mSrc->TheEnd())
        {
            if (mDoLoop)
            {
                mSrc->Reset();
                FpsReset();
                CheckJumpFrame();
            }
            else
            {
                mDone = true;
            }
        }
        if (mDone)
            HandleDestroyEvent();
        return mDone;
    }

    // interface to derived classes

    virtual void
    HandleNewFrame() = 0;
    
    virtual void
    HandleDestroyEvent()
    {
    }

    virtual void
    HandleCycleSrc()
    {
    }

private:

    void
    CheckJumpFrame()
    {
        int nStartRetry = 3;
        while ((nStartRetry > 0) && (mSrc->DataPtr() == 0))
        {
            ILOG_WARN("Skipping empty frame");
            nStartRetry--;
            mSrc->NextFrame();
            if (mSrc2)
                mSrc2->NextFrame();
        }
        mHadJump = true;
        mDoOneStep = true; // to make sure the frame is processed
    }

    void
    HandleDelayUpdate(int inc)
    {
        if (!mSrc2)
            return;
        mDelay += inc;
        mSrc2->NextFrame(inc);
        mHadJump = true;
        mDoOneStep = true; // to make sure the frame is processed
    }

    bool
    ExportFrame()
    {
        const int frameNr = mSrc->FrameNr();
        if (!mSrc || frameNr < 0)
        {
            ILOG_ERROR("No valid source or no current frame set (" << mSrc <<
                ", " << frameNr << ")");
            return false;
        }

        Core::Array::Array2dVec3UInt8* im = 
            Core::Array::ArrayCreate<Core::Array::Array2dVec3UInt8>
            (mSrc->FrameWidth(), mSrc->FrameHeight(), 0, 0, mSrc->DataPtr(), true);
        if (!im)
        {
            ILOG_ERROR("Couldn't load image for frame nr " << frameNr);
            return false;
        }

        String fileName = FileNameTail(mSrc->GetName());
        fileName += "_frame" + MakeString(frameNr) + ".png";
#ifndef REPOSITORY_USED // Here comes the deprecated stuff
        Core::Array::WritePng(im, fileName, new Util::Database());
#else // REPOSITORY_USED
        Persistency::ImageLocator loc("file:", ".", "", fileName);
        Persistency::ImageRepository().Add(loc, im);
#endif // REPOSITORY_USED
        delete im;
        return true;
    }

    RgbDataSrc* mSrc;
    bool        mDone;
    bool        mHadJump;
    int         mStartFrame;
    int         mMarkedFrameBegin;
    int         mMarkedFrameEnd;
    bool        mDoContinuous;
    bool        mDoOneStep;
    int         mStepSize;
    bool        mDoLoop;
    int         mFramesDone;
    Timer       mTimer;
    bool        mDidSrcError;

    RgbDataSrc* mSrc2;
    int         mDelay;

    ILOG_VAR_DEC;
};

ILOG_VAR_INIT(AppControlSrc, Visualization);


} // namespace Visualization
} // namespace Impala

#endif

---