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

---

SvmRepositoryInFile.h

Go to the documentation of this file.

#ifndef Impala_Persistency_SvmRepositoryInFile_h
#define Impala_Persistency_SvmRepositoryInFile_h

#include "Core/Training/Svm.h"
#include "Persistency/RepositoryInFileSystem.h"
#include "Persistency/ModelLocator.h"

namespace Impala
{
namespace Persistency
{


class SvmRepositoryInFile
{
public:

    typedef Core::Training::Svm Svm;

    SvmRepositoryInFile()
    {
    }

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

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

        svm_model* model = 0;
        if (loc.GetProtocol() == "file")
        {
            String path = file.GetPath();
            model = svm_load_model(path.c_str());
        }
        else
        {
            FileLocator tmpLoc("file", "direct", FileNameTmp());
            File tmpFile = RepFS().GetFile(tmpLoc, true, false);
            tmpFile.CopyFrom(file);
            String path = tmpFile.GetPath();
            model = svm_load_model(path.c_str());
            tmpFile.Delete();
        }
        if (!model)
        {
            ILOG_ERROR("[Get] Unable to load " << loc);
            return 0;
        }
        Svm* res = new Svm();
        res->SetModel(model);
        return res;
    }

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

        if (loc.GetProtocol() == "file")
        {
            String path = file.GetPath();
            if (svm_save_model(path.c_str(), svm->GetModel()) == -1)
                ILOG_ERROR("[Add] Unable to save " << loc);
        }
        else
        {
            FileLocator tmpLoc("file", "direct", FileNameTmp());
            File tmpFile = RepFS().GetFile(tmpLoc, true, false);
            String path = tmpFile.GetPath();
            if (svm_save_model(path.c_str(), svm->GetModel()) == -1)
                ILOG_ERROR("[Add] Unable to save " << loc);
            file.CopyFrom(tmpFile);
            tmpFile.Delete();
        }
    }

    void
    Delete(const ModelLocator& loc)
    {
        if (loc.GetConceptSet() == "ALL")
        {
            String dir = GetDir(loc, true);
            RepFS().DeleteDir(loc, dir);
        }
        else
        {
            String dir = GetDir(loc, false);
            File file = RepFS().GetFile(loc, dir, loc.GetConcept() + ".model",
                                        true, true);
            file.Delete();
        }
    }

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());
        return dir;
    }

    ILOG_VAR_DEC;
};

ILOG_VAR_INIT(SvmRepositoryInFile, Impala.Persistency);

} // namespace Persistency
} // namespace Impala

#endif

---