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

---

mainVdiff.cpp

Go to the documentation of this file.

#include "Basis/CmdOptions.h"
#include "Core/Stream/RgbDataSrcFactory.h"
#include "Core/Array/MakeFromData.h"
#include "Core/Array/RGB2Intensity.h"
#include "Core/Array/GetRgbPixels.h"
#include "Core/Array/Trait/BpoEqual.h"
#include "Visualization/VideoNav.h"
#include "Visualization/ImageStrip.h"
#include "Core/Array/Pattern/PatBinaryPixOp.h"
#include "Core/Array/Trait/BpoTestEqual.h"
#include "Visualization/AppControlSrc.h"

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

namespace Impala
{
namespace Application
{


using namespace Visualization;
using namespace Core::Array;
using namespace Core::Stream;

class WindowVdiff : public Window, public AppControlSrc,
                    public VideoNavListener, public ImagesListener
{
public:

    WindowVdiff(RgbDataSrc* src1, RgbDataSrc* src2) :
        Window(100, 0, SuggestWndWidth(2, src1->FrameWidth() + 2),
               SuggestWndHeight(1.5, src1->FrameHeight()) + 75, true),
        AppControlSrc(10)
    {
        CmdOptions& options = CmdOptions::GetInstance();
        SetSrc(src1, -1, src2); // tell AppControlSrc about src's
        AppController::Instance().AddControl((AppControlSrc*) this, true);

        mViewScale = options.GetDouble("viewScale");
        mDiffBuf = "";
        mVideoNav1 = new VideoNav(this, GetSrc(), mViewScale, false);
        mVideoNav1->SetVideoNavListener(this);
        mVideoNav2 = new VideoNav(this, GetSrc2(), mViewScale);
        mVideoNav2->SetVideoNavListener(this);
        mDiffWnd = new ImageStrip(this, src1->FrameWidth(), src1->FrameHeight(),
                                  0.5 * mViewScale, 3);
        mDiffWnd->SetImagesListener(this); // RvB: No ImagesListener defined here
        AppController::Instance().MainLoop();
    }

    void
    DoProcessing()
    {
        if ((!GetSrc()->DataPtr() || !GetSrc2()->DataPtr()) ||
            (GetSrc()->FrameWidth() != GetSrc2()->FrameWidth()) ||
            (GetSrc()->FrameHeight() != GetSrc2()->FrameHeight()))
        {
            return;
        }

        bool verbose = false;
        ArraySystem& aSys = ArraySystem::Instance();
        aSys.MarkMemoryUsage(verbose);

        Array2dVec3UInt8* wrap1 = ArrayCreate<Array2dVec3UInt8>
            (GetSrc()->FrameWidth(), GetSrc()->FrameHeight(), 0, 0,
             GetSrc()->DataPtr(), true);
        Array2dVec3UInt8* wrap2 = ArrayCreate<Array2dVec3UInt8>
            (GetSrc2()->FrameWidth(), GetSrc2()->FrameHeight(), 0, 0,
             GetSrc2()->DataPtr(), true);

        double eps = 0.00001;
        Trait::BpoTestEqual
            <Array2dScalarReal64, Array2dVec3UInt8, Array2dVec3UInt8> bpo(eps);
        Array2dScalarReal64* res = 0;
        Pattern::PatBinaryPixOp(res, wrap1, wrap2, bpo);

        mDiffBuf = "";
        std::ostringstream diffStream;
        if (bpo.mNrDiff > 0)
        {
            diffStream << ", found " << bpo.mNrDiff << " differences, in "
                << bpo.mNr << " pixels, sum diffs : " << bpo.mSumDiff 
                << " avg diff : " << bpo.mSumDiff / bpo.mNrDiff 
                << " avg src1 : " << bpo.mSumSrc1 / bpo.mNr
                << " avg src2 : " << bpo.mSumSrc2 / bpo.mNr << std::ends;
        }
        else
        {
            diffStream << ", frames are equal" << std::ends;
        }
        mDiffBuf = diffStream.str();

        mDiffWnd->RemoveImages();
        mDiffWnd->AddImage(res, "Stretch", "", false);

        delete wrap1;
        delete wrap2;
        delete res;
        aSys.CheckMemoryUsageSinceMark(verbose);
        if (verbose)
            std::cout << std::endl;
    }

    // specialization of base classes

    void
    HandleCycleSrc()
    {
        mOglWnd->updateScene = 1;
        SetStatusStr((char*) GetFpsString().c_str());
    }

    void
    HandleNewFrame()
    {
        mVideoNav1->UpdateView(0, GetSrc()->DataPtr(), GetSrc()->FrameWidth(),
                               GetSrc()->FrameHeight(), mViewScale);
        mVideoNav2->UpdateView(0, GetSrc2()->DataPtr(), GetSrc2()->FrameWidth(),
                               GetSrc2()->FrameHeight(), mViewScale);
        DoProcessing();
    }

    void
    NavFrameEvent(VideoNav* src, int requestedFrame, void* listenerData)
    {
        Jump(requestedFrame);
    }

private:

    double      mViewScale;
    String      mDiffBuf;
    VideoNav*   mVideoNav1;
    VideoNav*   mVideoNav2;
    ImageStrip* mDiffWnd;

};

int
mainVdiff(int argc, char* argv[])
{
    CmdOptions& options = CmdOptions::GetInstance();
    options.Initialise(true, true, true);
    options.AddOption(0, "src2", "type", "", "dx|avi|mpeg2|mpeg2es");
    if (! options.ParseArgs(argc, argv, "filename1 filename2", 2))
        return 1;

    ILOG_VAR(Application.mainVdiff);

    String srcName1 = options.GetArg(0);
    String srcName2 = options.GetArg(1);

    RgbDataSrcFactory& factory = RgbDataSrcFactory::Instance();
    RgbDataSrc* src1 = factory.Construct(srcName1, options.GetString("src"));
    if (! src1->Valid())
    {
        ILOG_ERROR("RgbDataSrc failed");
        return 0;
    }
    ILOG_INFO("src1: " << src1->LastFrame()+1 << " frames of "
              << src1->FrameWidth() << " x " << src1->FrameHeight());

    RgbDataSrc* src2 = factory.Construct(srcName2, options.GetString("src2"));
    if (! src2->Valid())
    {
        ILOG_ERROR("RgbDataSrc failed");
        return 0;
    }
    ILOG_INFO("src2: " << src2->LastFrame()+1 << " frames of "
              << src2->FrameWidth() << " x " << src2->FrameHeight());

    WindowVdiff* oglWnd = new WindowVdiff(src1, src2);
    if (! oglWnd->Valid())
    {
        ILOG_ERROR("WindowVdiff failed");
        return 0;
    }
    return 1;
}

} // namespace Application
} // namespace Impala

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

---