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

---

SimilarityTableSetRepositoryInFile.h

Go to the documentation of this file.

#ifndef Impala_Persistency_SimilarityTableSetRepositoryInFile_h
#define Impala_Persistency_SimilarityTableSetRepositoryInFile_h

#include "Persistency/RepositoryInFileSystem.h"
#include "Core/Table/SimilarityTableSet.h"
#include "Persistency/SimilarityTableSetLocator.h"

namespace Impala
{
namespace Persistency
{


class SimilarityTableSetRepositoryInFile
{
public:

    typedef Core::Table::SimilarityTableSet SimilarityTableSet;

    SimilarityTableSetRepositoryInFile()
    {
    }

    bool
    Exists(const SimilarityTableSetLocator& loc)
    {
        String dir = GetDir(loc, false);
        File file = RepFS().GetFile(loc, dir, "names.txt", false, true);
        return file.Valid();
    }

    SimilarityTableSet*
    Get(const SimilarityTableSetLocator& loc)
    {
        String dir = GetDir(loc, false);
        File file = RepFS().GetFile(loc, dir, "names.txt", false, false);
        if (!file.Valid())
            return 0;
        std::vector<String> names;
        file.ReadStrings(std::back_inserter(names));
        SimilarityTableSet* res = new SimilarityTableSet(names, 0);
        res->SetDescription(loc.GetFeature());

        LoadQuids(res, loc, dir);
        LoadSims(res, loc, dir);
        LoadRanks(res, loc, dir);
        return res;
    }

    void
    Add(const SimilarityTableSetLocator& loc, SimilarityTableSet* simSet)
    {
        String dir = GetDir(loc, false);
        Save(simSet, loc, dir);
    }

    void
    Delete(const SimilarityTableSetLocator& loc)
    {
        if (loc.GetConceptSet() == "ALL")
        {
            String dir = GetDir(loc, true);
            RepFS().DeleteDir(loc, dir);
        }
        else
        {
            String dir = GetDir(loc, false);
            RepFS().DeleteDir(loc, dir);
        }
    }

private:

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

    String
    GetDir(const SimilarityTableSetLocator& loc, bool forAll)
    {
        String dir = loc.GetInIndexString();
        if ((!loc.GetInIndex()) && (loc.GetWalkType() != "Image"))
            dir = FileNameConcat(dir, loc.GetWalkType());

        if (forAll)
            return dir;

        dir = FileNameConcat(dir, loc.GetConceptSet());
        dir = FileNameConcat(dir, loc.GetModel());
        dir = FileNameConcat(dir, loc.GetFeature());
        if (!loc.GetInIndex())
            dir = FileNameConcat(dir, loc.GetContainer());
        return dir;
    }

    void
    LoadQuids(SimilarityTableSet* res, const SimilarityTableSetLocator& loc,
              CString dir)
    {
        File file = RepFS().GetFile(loc, dir, "all_quids.tab", false, false);
        Read(res->GetQuidTable(), file);
    }

    void
    LoadSim(SimilarityTableSet* res, int tableIdx,
            const SimilarityTableSetLocator& loc, CString dir)
    {
        String name = res->GetName(tableIdx) + "_sim.tab";
        File file = RepFS().GetFile(loc, dir, name, false, false);
        Read(res->GetSimTable(tableIdx), file);
    }

    void
    LoadSims(SimilarityTableSet* res, const SimilarityTableSetLocator& loc,
             CString dir)
    {
        for (int i=0 ; i<res->NrTables() ; i++)
            LoadSim(res, i, loc, dir);
    }

    void
    LoadRank(SimilarityTableSet* res, int tableIdx,
             const SimilarityTableSetLocator& loc, CString dir)
    {
        String name = res->GetName(tableIdx) + "_rank.tab";
        File file = RepFS().GetFile(loc, dir, name, false, false);
        Read(res->GetRankTable(tableIdx), file);
    }

    void
    LoadRanks(SimilarityTableSet* res, const SimilarityTableSetLocator& loc,
              CString dir)
    {
        for (int i=0 ; i<res->NrTables() ; i++)
            LoadRank(res, i, loc, dir);
    }

    void
    Save(SimilarityTableSet* simSet, const SimilarityTableSetLocator& loc,
         CString dir)
    {
        SaveNames(simSet, loc, dir);
        SaveQuids(simSet, loc, dir);
        SaveSims(simSet, loc, dir);
        SaveRanks(simSet, loc, dir);
    }

    void
    SaveNames(SimilarityTableSet* simSet, const SimilarityTableSetLocator& loc,
              CString dir)
    {
        File file = RepFS().GetFile(loc, dir, "names.txt", true, false);
        std::vector<String> names = simSet->GetNames();
        file.WriteStrings(names);
    }

    void
    SaveQuids(SimilarityTableSet* simSet, const SimilarityTableSetLocator& loc,
              CString dir)
    {
        File file = RepFS().GetFile(loc, dir, "all_quids.tab", true, false);
        Write(simSet->GetQuidTable(), file);
    }

    void
    SaveSim(SimilarityTableSet* simSet, int tableIdx,
            const SimilarityTableSetLocator& loc, CString dir)
    {
        String name = simSet->GetName(tableIdx) + "_sim.tab";
        File file = RepFS().GetFile(loc, dir, name, true, false);
        Write(simSet->GetSimTable(tableIdx), file);
    }

    void
    SaveSims(SimilarityTableSet* simSet, const SimilarityTableSetLocator& loc,
             CString dir)
    {
        for (int i=0 ; i<simSet->NrTables() ; i++)
            SaveSim(simSet, i, loc, dir);
    }

    void
    SaveRank(SimilarityTableSet* simSet, int tableIdx,
             const SimilarityTableSetLocator& loc, CString dir)
    {
        String name = simSet->GetName(tableIdx) + "_rank.tab";
        File file = RepFS().GetFile(loc, dir, name, true, false);
        Write(simSet->GetRankTable(tableIdx), file);
    }

    void
    SaveRanks(SimilarityTableSet* simSet, const SimilarityTableSetLocator& loc,
              CString dir)
    {
        for (int i=0 ; i<simSet->NrTables() ; i++)
            SaveRank(simSet, i, loc, dir);
    }

    ILOG_VAR_DEC;
};

ILOG_VAR_INIT(SimilarityTableSetRepositoryInFile, Impala.Persistency);

} // namespace Persistency
} // namespace Impala

#endif

---