common/symtab/template.h

Go to the documentation of this file.

//
//      aegis - project change supervisor
//      Copyright (C) 2004-2008 Peter Miller
//
//      This program is free software; you can redistribute it and/or modify
//      it under the terms of the GNU General Public License as published by
//      the Free Software Foundation; either version 3 of the License, or
//      (at your option) any later version.
//
//      This program is distributed in the hope that it will be useful,
//      but WITHOUT ANY WARRANTY; without even the implied warranty of
//      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//      GNU General Public License for more details.
//
//      You should have received a copy of the GNU General Public License
//      along with this program. If not, see
//      <http://www.gnu.org/licenses/>.
//

#ifndef COMMON_SYMTAB_TEMPLATE_H
#define COMMON_SYMTAB_TEMPLATE_H

#include <common/nstring.h>
#include <common/nstring/list.h>
#include <common/symtab.h>

template <class value_type_t>
class symtab
{
public:
    ~symtab()
    {
        if (stp)
        {
            delete stp;
            stp = 0;
        }
    }

    symtab() :
        stp(0)
    {
    }

    symtab(const symtab &arg) :
        stp(0)
    {
        copy(arg);
    }

    symtab &operator=(const symtab &arg)
    {
        if (this != &arg)
        {
            clear();
            copy(arg);
        }
        return *this;
    }

    void
    clear()
    {
        if (stp)
            stp->clear();
    }

    value_type_t
    get(const nstring &key)
        const
    {
        value_type_t *vp = query(key);
        return (vp ? *vp : value_type_t());
    }

    nstring
    query_fuzzy(const nstring &key)
        const
    {
        return (stp ? stp->query_fuzzy(key) : nstring());
    }

    value_type_t *
    query(string_ty *key)
        const
    {
        if (!stp)
            return 0;
        return (value_type_t *)stp->query(key);
    }

    value_type_t *
    query(const nstring &key)
        const
    {
        if (!stp)
            return 0;
        return (value_type_t *)stp->query(key);
    }

    value_type_t *
    query(const nstring_list &key_list)
        const
    {
        if (!stp)
            return 0;
        return (value_type_t *)stp->query(key_list);
    }

    void
    assign(string_ty *key, value_type_t *value)
    {
        if (!stp)
            stp = new symtab_ty(5);
        stp->assign(key, (void *)value);
    }

    void
    assign(const nstring &key, value_type_t *value)
    {
        if (!stp)
            stp = new symtab_ty(5);
        stp->assign(key, (void *)value);
    }

    void
    assign(const nstring &key, const value_type_t &value)
    {
        // I hope they called set_reaper already.
        assign(key, new value_type_t(value));
    }

    void
    remove(const nstring &key)
    {
        if (stp)
            stp->remove(key);
    }

    void
    dump(const char *caption)
        const
    {
        if (stp)
            stp->dump(caption);
    }

    void
    set_reaper()
    {
        if (!stp)
            stp = new symtab_ty(5);
        stp->set_reap(reaper);
    }

    bool
    empty()
        const
    {
        return (!stp || stp->empty());
    }

    size_t
    size()
        const
    {
        return (stp ? stp->size() : 0);
    }

    void
    keys(nstring_list &result)
        const
    {
        if (stp)
            stp->keys(result);
        else
            result.clear();
    }

private:
    symtab_ty *stp;

    static void reaper(void *p) { delete (value_type_t *)p; }

    void
    copy(const symtab &arg)
    {
        nstring_list names;
        arg.keys(names);
        for (size_t j = 0; j < names.size(); ++j)
        {
            value_type_t *p = arg.query(names[j]);
            if (p)
                assign(names[j], *p);
        }
    }
};

#endif // COMMON_SYMTAB_TEMPLATE_H