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

---

LabelSet.h

Go to the documentation of this file.

#ifndef Impala_Core_Feature_LabelSet_h
#define Impala_Core_Feature_LabelSet_h

#include "Util/Database.h"
#include "Util/SimpleMap.h"

namespace Impala
{
namespace Core
{
namespace Feature
{


class LabelSet
{
public:

    LabelSet()
    {
        mId = 0;
    }

    int
    GetNrLabels()
    {
        return mMap.Size();
    }

    Quid
    GetQuid(String label)
    {
        Quid quid = 0;
        mMap.GetIdx(label, quid);
        return quid;
    }

    String
    GetString(Quid quid)
    {
        String res = "unknown";
        mMap.Get(quid, res);
        return res;
    }

    Quid
    Add(String label)
    {
        Quid quid = GetQuid(label);
        if (quid == 0)
        {
            quid = MakeQuidLabel(mId++);
            mMap.Add(quid, label);
        }
        return quid;
    }

#ifndef REPOSITORY_USED // Here comes the deprecated stuff
    void
    Load(String fileName, Util::Database* db)
    {
        Util::IOBuffer* buf = db->GetIOBuffer(fileName, true, true, "");
        if (buf)
        {
            int nr = 0;
            while (buf->Available())
            {
                Quid q;
                buf->NativeTypeRead(&q);
                String label = buf->ReadLine();
                mMap.Add(q, label);
                nr++;
            }
            mId += nr;
        }
        delete buf;
    }

    void
    Save(String fileName, Util::Database* db)
    {
        Util::IOBuffer* buf = db->GetIOBuffer(fileName, false, false, "tmp");
        if (buf)
        {
            std::vector<Quid> quids = mMap.GetAllIdx();
            std::vector<String> labels = mMap.GetAll();
            for (int i=0 ; i<quids.size() ; i++)
            {
                buf->NativeTypeWrite(quids[i]);
                buf->Puts(labels[i]);
            }
        }
        delete buf;
    }
#endif // REPOSITORY_USED

    void
    Read(Util::IOBuffer* buf)
    {
        if (buf)
        {
            int nr = 0;
            while (buf->Available())
            {
                Quid q;
                buf->NativeTypeRead(&q);
                String label = buf->ReadLine();
                mMap.Add(q, label);
                nr++;
            }
            mId += nr;
        }
    }

    void
    Write(Util::IOBuffer* buf)
    {
        if (buf)
        {
            std::vector<Quid> quids = mMap.GetAllIdx();
            std::vector<String> labels = mMap.GetAll();
            for (int i=0 ; i<quids.size() ; i++)
            {
                buf->NativeTypeWrite(quids[i]);
                buf->Puts(labels[i]);
            }
        }
    }

private:

    int mId;
    Util::SimpleMap<Quid, String> mMap;

};

} // namespace Feature
} // namespace Core
} // namespace Impala

#endif

---