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

---

VisSemTrain.h

Go to the documentation of this file.

#ifndef Impala_Core_VideoSet_VisSemTrain_h
#define Impala_Core_VideoSet_VisSemTrain_h

#include "Core/Feature/VisSem.h"
#include "Core/VideoSet/Reporter.h"

namespace Impala
{
namespace Core
{
namespace VideoSet
{


class VisSemTrain : public Listener
{
    typedef Array::Array2dScalarReal64 Array2dScalarReal64;

public:

    VisSemTrain(Reporter* reporter, String name, CmdOptions& options)
    {
        mReporter = reporter;
        mFeature = new Feature::VisSem(name, options);
    }

    virtual void
    HandleNewFile(VideoSet* vs, int fileId, Stream::RgbDataSrc* src,
                  int bookmarkNr, Geometry::VxRectangle vr)
    {
        mFeature->ClearAnnoAccu();
    }

    virtual void
    HandleDoneFile(VideoSet* vs, int fileId, Stream::RgbDataSrc* src,
                   int bookmarkNr, Geometry::VxRectangle vr)
    {
        mFeature->AddAccuToAnnoTable(bookmarkNr, vr.mAnno);
    }

    virtual void
    HandleNewFrame(VideoSet* vs, int fileId, Stream::RgbDataSrc* src,
                   Geometry::Rectangle rect, String anno)
    {
        Array::Array2dVec3UInt8* im = Array::ArrayCreate<Array::Array2dVec3UInt8>
            (src->FrameWidth(), src->FrameHeight(), 0, 0, src->DataPtr(), true);
        mFeature->AddImageToAccu(im, rect);
        delete im;
    }

    virtual void
    HandleDoneWalk(VideoSet* vs, int nrBookmarks)
    {
        mFeature->WriteProtoFeatureTableSet(vs);
    }

private:

    Reporter*        mReporter;
    Feature::VisSem* mFeature;

};

} // namespace VideoSet
} // namespace Core
} // namespace Impala

#endif

---