growable_vector.h

Go to the documentation of this file.

#ifndef GROWABLE_VECTOR_H
#define GROWABLE_VECTOR_H

#include <vector>
#include <algorithm>
#include <functional>

template <typename T> class growable_vector {
public:
    typedef std::vector<T> container_type;
    typedef typename container_type::size_type size_type;


    T operator[](size_type index) const {
        if (index >= container.size())
            return T();
        return container[index];
    }


    T & operator[](size_type index) {
        if (index >= container.size())
            container.resize(index + 1);
        return container[index];
    }


    growable_vector<T> & operator+=(growable_vector<T> const & rhs) {
        if (rhs.container.size() > container.size())
            container.resize(rhs.container.size());

        size_type min_size = min(container.size(), rhs.container.size());
        for (size_type i = 0 ; i < min_size; ++i)
            container[i] += rhs.container[i];

        return *this;
    }


    growable_vector<T> & operator-=(growable_vector<T> const & rhs) {
        if (rhs.container.size() > container.size())
            container.resize(rhs.container.size());

        size_type min_size = min(container.size(), rhs.container.size());
        for (size_type i = 0 ; i < min_size; ++i)
            container[i] -= rhs.container[i];

        return *this;
    }


    size_type size() const {
        return container.size();
    }


    void fill(size_type size, T const & value) {
        container.resize(size, value);
    }


    bool zero() const {
        return std::find_if(container.begin(), container.end(),
                                std::bind2nd(std::not_equal_to<T>(), T()))
                    == container.end();
    }

private:
    container_type container;
};

#endif // GROWABLE_VECTOR_H