Set.h

Go to the documentation of this file.

#ifndef _SET_H_
#define _SET_H_

#include <stdlib.h>
#include <string.h>

template <class TYPE>
class Set {
        public:
                Set(const unsigned int s = 0);
                virtual ~Set();
                TYPE *getArray() const;
                TYPE &operator [] (const unsigned int i) const;
                unsigned int getCount() const;
                unsigned int add(const TYPE&);
                void remove(const unsigned int i);
                void removeObject(const TYPE&);
                void clear();
                void reset();

        private:
                unsigned int size, count;
                TYPE *list;
};

/********************************************************************
 * Counstructor
 */
template <class TYPE>
Set<TYPE>::Set(const unsigned int s) {
        count = 0;
        size = s;
        if (size > 0)
                list = (TYPE *)malloc(size*sizeof(TYPE));
        else
                list = NULL;
}

/********************************************************************
 * Destructor
 */
template <class TYPE>
Set<TYPE>::~Set() {
        free(list);
}

/********************************************************************
 * Get array
 */
template <class TYPE>
TYPE *Set<TYPE>::getArray() const {
        return list;
}

/********************************************************************
 * [] Operator
 */
template <class TYPE>
TYPE &Set<TYPE>::operator [] (const unsigned int i) const {
        return list[i];
}

/********************************************************************
 * Get count
 */
template <class TYPE>
unsigned int Set<TYPE>::getCount() const {
        return count;
}

/********************************************************************
 * Add
 */
template <class TYPE>
unsigned int Set<TYPE>::add(const TYPE& obj) {
        if (count >= size) {
                if (size)
                        size += size;
                else
                        size = 8;

                list = (TYPE *)realloc(list, size*sizeof(TYPE));
        }
        list[count] = obj;
        return count++;
}

/********************************************************************
 * Remove
 */
template <class TYPE>
void Set<TYPE>::remove(const unsigned int i) {
        if (i < count) {
                count--;
                list[i] = list[count];
        }
}

/********************************************************************
 * Remove object
 */
template <class TYPE>
void Set<TYPE>::removeObject(const TYPE& obj) {
        for (unsigned int i=0; i<count; i++) {
                if (obj == list[i]) {
                        remove(i);
                        return;
                }
        }
}

/********************************************************************
 * Clear
 */
template <class TYPE>
void Set<TYPE>::clear() {
        count = 0;
}

/********************************************************************
 * Reset
 */
template <class TYPE>
void Set<TYPE>::reset() {
        free(list);
        list = NULL;
        count = size = 0;
}

#endif

---