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

---

FikSvmRepositoryInFile.h

Go to the documentation of this file.

#ifndef Impala_Persistency_FikSvmRepositoryInFile_h
#define Impala_Persistency_FikSvmRepositoryInFile_h

#include "Core/Training/FikSvm.h"
#include "Persistency/RepositoryInFileSystem.h"
#include "Persistency/ModelLocator.h"
#include "Core/Array/ReadRaw.h"
#include "Core/Array/WriteRaw.h"

namespace Impala
{
namespace Persistency
{


class FikSvmRepositoryInFile
{
public:

    typedef Core::Training::FikSvm FikSvm;
    typedef Core::Feature::WeightedFeatureList WeightedFeatureList;

    FikSvmRepositoryInFile()
    {
    }

    bool
    Exists(const ModelLocator& loc)
    {
        String dir = GetDir(loc, false);
        File file = RepFS().GetFile(loc, dir, loc.GetConcept() + ".props",
                                    false, true);
        return file.Valid();
    }

    FikSvm*
    Get(const ModelLocator& loc)
    {
        String dir = GetDir(loc, false);
        File file = RepFS().GetFile(loc, dir, loc.GetConcept() + ".props",
                                    false, false);
        if (!file.Valid())
            return 0;

        Util::PropertySet ps;
        Read(&ps, file);

        File featFile = RepFS().GetFile(loc, dir, loc.GetConcept() + ".features",
                                        false, false);
        WeightedFeatureList features;
        Read(&features, featFile);

        FikSvm* svm = new FikSvm(features, ps.GetInt("nrBins"),
                                 ps.GetInt("nrSV"), ps.GetDouble("rho"),
                                 ps.GetDouble("probA"), ps.GetDouble("probB"),
                                 ps.GetInt("probIndex"));

        for (int i=0 ; i<svm->GetNrFeatures() ; i++)
        {
            File rawFile = RepFS().GetFile(loc, dir, MakeString(i) + ".raw",
                                           false, false);
            Util::IOBuffer* buf = rawFile.GetReadBuffer();
            Core::Matrix::Mat* h = 0;
            ReadRaw(h, buf);
            Core::Matrix::Mat* a = 0;
            ReadRaw(a, buf);
            Core::Matrix::Mat* b = 0;
            ReadRaw(b, buf);
            svm->ImportSvm(h, a, b);
            delete b;
            delete a;
            delete h;
            delete buf;
        }
        return svm;
    }

    void
    Add(const ModelLocator& loc, FikSvm* svm)
    {
        String dir = GetDir(loc, false);
        File file = RepFS().GetFile(loc, dir, loc.GetConcept() + ".props",
                                    true, false);
        if (!file.Valid())
            return;

        Util::PropertySet ps;
        ps.Add("nrBins", svm->GetNrBins());
        ps.Add("nrSV", svm->GetNrSV());
        ps.Add("rho", svm->GetRho());
        ps.Add("probA", svm->GetProbA());
        ps.Add("probB", svm->GetProbB());
        ps.Add("probIndex", svm->GetProbIndex());
        Write(&ps, file);

        File featFile = RepFS().GetFile(loc, dir, loc.GetConcept() + ".features",
                                        true, false);
        Write(svm->GetFeatureList(), featFile);

        for (int i=0 ; i<svm->GetNrFeatures() ; i++)
        {
            File rawFile = RepFS().GetFile(loc, dir, MakeString(i) + ".raw",
                                           true, false);
            Util::IOBuffer* buf = rawFile.GetWriteBuffer();
            Core::Matrix::Mat* mat = svm->GetH(i);
            WriteRaw(mat, buf, 1);
            delete mat;
            mat = svm->GetAMat(i);
            WriteRaw(mat, buf, 1);
            delete mat;
            mat = svm->GetBMat(i);
            WriteRaw(mat, buf, 1);
            delete mat;
            delete buf;
        }
    }

private:

    RepositoryInFileSystem&
    RepFS()
    {
        return RepositoryInFileSystem::GetInstance();
    }

    String
    GetDir(const ModelLocator& loc, bool forAll)
    {
        String dir = "ConceptModels";

        if (forAll)
            return dir;

        dir = FileNameConcat(dir, loc.GetConceptSet());
        dir = FileNameConcat(dir, loc.GetModel());
        dir = FileNameConcat(dir, loc.GetFeature());
        dir = FileNameConcat(dir, loc.GetConcept());
        return dir;
    }

    ILOG_VAR_DEC;
};

ILOG_VAR_INIT(FikSvmRepositoryInFile, Impala.Persistency);

} // namespace Persistency
} // namespace Impala

#endif

---