Set.h

Go to the documentation of this file.

/*
 * 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.
 *
 * Written (W) 2009 Soeren Sonnenburg
 * Copyright (C) 2009 Fraunhofer Institute FIRST and Max-Planck-Society
 */

#ifndef _SET_H_
#define _SET_H_

#include <shogun/lib/common.h>
#include <shogun/mathematics/Math.h>
#include <shogun/base/DynArray.h>
#include <shogun/base/SGObject.h>

namespace shogun
{
template <class T> class CSet : public CSGObject
{
    public:
        CSet(bool traceable=true)
        {
            array = new DynArray<T>(1024, traceable);
        }

        ~CSet()
        {
            delete array;
        }

        inline void add(T e)
        {
            if (!contains(e))
                array->append_element(e);
        }

        inline void remove(T e)
        {
            int32_t idx=array->find_element(e);
            if (idx>=0)
                array->delete_element(idx);
        }

        inline bool contains(T e)
        {
            int32_t idx=array->find_element(e);
            return (idx!=-1);
        }

        inline int32_t index_of(T e)
        {
            return array->find_element(e);
        }

        inline int32_t get_num_elements() const
        {
            return array->get_num_elements();
        }

        inline T get_element(int32_t index) const
        {
            return array->get_element(index);
        }

        inline T* get_element_ptr(int32_t index)
        {
            return array->get_element_ptr(index);
        }

        inline T operator[](int32_t index) const
        {
            return array->get_element(index);
        }

        inline T* get_array()
        {
            return array->get_array();
        }

        inline virtual const char* get_name() const { return "Set"; }

    protected:
        DynArray<T>* array;
};
}
#endif //_SET_H_