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

---

ComputeKernelData.h

Go to the documentation of this file.

#ifndef Impala_Core_VideoSet_ComputeKernelData_h
#define Impala_Core_VideoSet_ComputeKernelData_h

#include "Core/VideoSet/Reporter.h"
#include "Core/Training/PrecomputeTask.h"

namespace Impala
{
namespace Core
{
namespace VideoSet
{


class ComputeKernelData : public Listener
{
public:

    typedef Persistency::KernelMatrixLocator KernelMatrixLocator;

    ComputeKernelData(Reporter* reporter, CmdOptions& options)
    {
        if (options.GetNrArg() < 5)
        {
            ILOG_ERROR("Missing argument");
            return;
        }
        String name2 = options.GetArg(2);
        mDevelSet = Database::MakeRawDataSet(name2, true);
        mModel = options.GetArg(3);
        mKernelMatrixName = options.GetArg(4);
        mFeatureIndexCat = options.GetString("featureIndexCat");
        KernelMatrixLocator loc(mDevelSet->GetLocator(), true, "", "",
                                mModel, mKernelMatrixName, "");
        loc.SetFeatureIndexCat(mFeatureIndexCat);
        loc.SetStartNode(0);
        loc.SetNodeCount(Link::Mpi::NrProcs());
        loc.SetDoParts(0);
        mDA = Persistency::DistributedAccessRepository().Get(loc);
    }

    virtual
    ~ComputeKernelData()
    {
        delete mDA;
        delete mDevelSet;
    }

    virtual void
    HandleNewWalk(VideoSet* vs, String walkType)
    {
        mWalkType = walkType;
    }

    virtual void
    HandleDoneFile(VideoSet* vs, int fileId, Stream::RgbDataSrc* src)
    {
        Training::PrecomputeTask::ComputeTestKernel
            (vs->GetLocator(), mWalkType, vs->GetContainer(fileId), mDevelSet,
             mDA, mFeatureIndexCat, mModel, mKernelMatrixName);
    }

private:

    Database::RawDataSet*      mDevelSet;
    String                     mWalkType;
    String                     mFeatureIndexCat;
    String                     mModel;
    String                     mKernelMatrixName;
    Matrix::DistributedAccess* mDA;

    ILOG_VAR_DEC;
};

ILOG_VAR_INIT(ComputeKernelData, Impala.Core.VideoSet);

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

#endif

---