pip/lib/main/containers/pivector.h

/*! \file pivector.h
 * \brief Dynamic array of any type
 *
 * This file declares PIVector
*/
/*
	PIP - Platform Independent Primitives
	Dynamic array of any type
	Ivan Pelipenko peri4ko@yandex.ru, Andrey Bychkov work.a.b@yandex.ru

	This program is free software: you can redistribute it and/or modify
	it under the terms of the GNU Lesser 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 Lesser General Public License for more details.

	You should have received a copy of the GNU Lesser General Public License
	along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

#ifndef PIVECTOR_H
#define PIVECTOR_H

#include "picontainers.h"


template <typename T>
class PIVector {
public:
	inline PIVector(): piv_data(0), piv_size(0), piv_rsize(0) {
		PIINTROSPECTION_CONTAINER_NEW(T, sizeof(T))
	}
	inline PIVector(const T * data, size_t size): piv_data(0), piv_size(0), piv_rsize(0) {
		PIINTROSPECTION_CONTAINER_NEW(T, sizeof(T))
		alloc(size);
		newT(piv_data, data, piv_size);
	}
	inline PIVector(const PIVector<T> & other): piv_data(0), piv_size(0), piv_rsize(0) {
		PIINTROSPECTION_CONTAINER_NEW(T, sizeof(T))
		alloc(other.piv_size);
		newT(piv_data, other.piv_data, piv_size);
	}
	inline PIVector(size_t piv_size, const T & f = T()): piv_data(0), piv_size(0), piv_rsize(0) {
		PIINTROSPECTION_CONTAINER_NEW(T, sizeof(T))
		resize(piv_size, f);
	}
	inline virtual ~PIVector() {
		PIINTROSPECTION_CONTAINER_DELETE(T)
		PIINTROSPECTION_CONTAINER_FREE(T, (piv_rsize))
		deleteT(piv_data, piv_size);
		dealloc();
		_reset();
	}

	inline PIVector<T> & operator =(const PIVector<T> & other) {
		if (this == &other) return *this;
		clear();
		deleteT(piv_data, piv_size);
		alloc(other.piv_size);
		newT(piv_data, other.piv_data, piv_size);
		return *this;
	}

	typedef T value_type;

	class iterator {
		friend class PIVector<T>;
	private:
		inline iterator(PIVector<T> * v, size_t p): parent(v), pos(p) {}
		PIVector<T> * parent;
		size_t pos;
	public:
		inline iterator(): parent(0), pos(0) {}
		inline T & operator *() {return (*parent)[pos];}
		inline const T & operator *() const {return (*parent)[pos];}
		inline void operator ++() {++pos;}
		inline void operator ++(int) {++pos;}
		inline void operator --() {--pos;}
		inline void operator --(int) {--pos;}
		inline bool operator ==(const iterator & it) const {return (pos == it.pos);}
		inline bool operator !=(const iterator & it) const {return (pos != it.pos);}
	};

	class const_iterator {
		friend class PIVector<T>;
	private:
		inline const_iterator(const PIVector<T> * v, size_t p): parent(v), pos(p) {}
		const PIVector<T> * parent;
		size_t pos;
	public:
		inline const_iterator(): parent(0), pos(0) {}
		inline const T & operator *() const {return (*parent)[pos];}
		inline void operator ++() {++pos;}
		inline void operator ++(int) {++pos;}
		inline void operator --() {--pos;}
		inline void operator --(int) {--pos;}
		inline bool operator ==(const const_iterator & it) const {return (pos == it.pos);}
		inline bool operator !=(const const_iterator & it) const {return (pos != it.pos);}
	};

	class reverse_iterator {
		friend class PIVector<T>;
	private:
		inline reverse_iterator(PIVector<T> * v, size_t p): parent(v), pos(p) {}
		PIVector<T> * parent;
		size_t pos;
	public:
		inline reverse_iterator(): parent(0), pos(0) {}
		inline T & operator *() {return (*parent)[pos];}
		inline const T & operator *() const {return (*parent)[pos];}
		inline void operator ++() {--pos;}
		inline void operator ++(int) {--pos;}
		inline void operator --() {++pos;}
		inline void operator --(int) {++pos;}
		inline bool operator ==(const reverse_iterator & it) const {return (pos == it.pos);}
		inline bool operator !=(const reverse_iterator & it) const {return (pos != it.pos);}
	};

	class const_reverse_iterator {
		friend class PIVector<T>;
	private:
		inline const_reverse_iterator(const PIVector<T> * v, size_t p): parent(v), pos(p) {}
		const PIVector<T> * parent;
		size_t pos;
	public:
		inline const_reverse_iterator(): parent(0), pos(0) {}
		inline const T & operator *() const {return (*parent)[pos];}
		inline void operator ++() {--pos;}
		inline void operator ++(int) {--pos;}
		inline void operator --() {++pos;}
		inline void operator --(int) {++pos;}
		inline bool operator ==(const const_reverse_iterator & it) const {return (pos == it.pos);}
		inline bool operator !=(const const_reverse_iterator & it) const {return (pos != it.pos);}
	};

	inline iterator begin() {return iterator(this, 0);}
	inline iterator end() {return iterator(this, piv_size);}
	inline const_iterator begin() const {return const_iterator(this, 0);}
	inline const_iterator end() const {return const_iterator(this, piv_size);}
	inline reverse_iterator rbegin() {return reverse_iterator(this, piv_size - 1);}
	inline reverse_iterator rend() {return reverse_iterator(this, -1);}
	inline const_reverse_iterator rbegin() const {return const_reverse_iterator(this, piv_size - 1);}
	inline const_reverse_iterator rend() const {return const_reverse_iterator(this, -1);}

	inline size_t size() const {return piv_size;}
	inline ssize_t size_s() const {return piv_size;}
	inline size_t length() const {return piv_size;}
	inline size_t capacity() const {return piv_rsize;}
	inline bool isEmpty() const {return (piv_size == 0);}

	inline T & operator [](size_t index) {return piv_data[index];}
	inline T & at(size_t index) {return piv_data[index];}
	inline const T & operator [](size_t index) const {return piv_data[index];}
	inline const T & at(size_t index) const {return piv_data[index];}
	inline T & back() {return piv_data[piv_size - 1];}
	inline const T & back() const {return piv_data[piv_size - 1];}
	inline T & front() {return piv_data[0];}
	inline const T & front() const {return piv_data[0];}
	inline bool operator ==(const PIVector<T> & t) const {
		if (piv_size != t.piv_size)
			return false;
		for (size_t i = 0; i < piv_size; ++i)
			if (t[i] != piv_data[i])
				return false;
		return true;
	}
	inline bool operator !=(const PIVector<T> & t) const {return !(*this == t);}
	inline bool contains(const T & v) const {
		for (size_t i = 0; i < piv_size; ++i)
			if (v == piv_data[i])
				return true;
		return false;
	}
	inline int etries(const T & v) const {
		int ec = 0;
		for (size_t i = 0; i < piv_size; ++i)
			if (v == piv_data[i]) ++ec;
		return ec;
	}
	inline ssize_t indexOf(const T & v) const {
		for (size_t i = 0; i < piv_size; ++i)
			if (v == piv_data[i])
				return i;
		return -1;
	}
	inline ssize_t lastIndexOf(const T & v) const {
		for (ssize_t i = piv_size - 1; i >= 0; --i)
			if (v == piv_data[i])
				return i;
		return -1;
	}

	inline T * data(size_t index = 0) {return &(piv_data[index]);}
	inline const T * data(size_t index = 0) const {return &(piv_data[index]);}

	inline PIVector<T> & clear() {resize(0); return *this;}
	inline PIVector<T> & fill(const T & f = T()) {
		deleteT(piv_data, piv_size);
		PIINTROSPECTION_CONTAINER_USED(T, piv_size)
		for (size_t i = 0; i < piv_size; ++i)
			elementNew(piv_data + i, f);
		return *this;
	}
	inline PIVector<T> & assign(const T & f = T()) {return fill(f);}
	inline PIVector<T> & assign(size_t new_size, const T & f) {
		resize(new_size);
		return fill(f);
	}

	inline PIVector<T> & resize(size_t new_size, const T & f = T()) {
		if (new_size < piv_size) {
			T * de = &(piv_data[new_size]);
			deleteT(de, piv_size - new_size);
			piv_size = new_size;
		}
		if (new_size > piv_size) {
			size_t os = piv_size;
			alloc(new_size);
			PIINTROSPECTION_CONTAINER_USED(T, (new_size-os))
			for (size_t i = os; i < new_size; ++i)
				elementNew(piv_data + i, f);
		}
		return *this;
	}
	inline PIVector<T> & _resizeRaw(size_t new_size) {
		piCout << "Error, \"resizeRaw()\" only allowed for simple type declared with __PIVECTOR_SIMPLE_TYPE__ macro!";
		assert(0);
		return *this;
	}
	inline void _copyRaw(T * dst, const T * src, size_t size) {
		newT(dst, src, size);
	}

	inline PIVector<T> & reserve(size_t new_size) {
		if (new_size <= piv_rsize) return *this;
		size_t os = piv_size;
		alloc(new_size);
		piv_size = os;
		return *this;
	}

	inline PIVector<T> & insert(size_t index, const T & v = T()) {
		alloc(piv_size + 1);
		if (index < piv_size - 1) {
			size_t os = piv_size - index - 1;
			memmove((void*)(&(piv_data[index + 1])), (const void*)(&(piv_data[index])), os * sizeof(T));
		}
		PIINTROSPECTION_CONTAINER_USED(T, 1)
		elementNew(piv_data + index, v);
		return *this;
	}
	inline PIVector<T> & insert(size_t index, const PIVector<T> & other) {
		if (other.isEmpty()) return *this;
		assert(&other != this);
		ssize_t os = piv_size - index;
		alloc(piv_size + other.piv_size);
		if (os > 0)
			memmove((void*)(&(piv_data[index + other.piv_size])), (const void*)(&(piv_data[index])), os * sizeof(T));
		newT(piv_data + index, other.piv_data, other.piv_size);
		return *this;
	}

	inline PIVector<T> & remove(size_t index, size_t count = 1) {
		if (count == 0) return *this;
		if (index + count >= piv_size) {
			resize(index);
			return *this;
		}
		size_t os = piv_size - index - count;
		deleteT(&(piv_data[index]), count);
		memmove((void*)(&(piv_data[index])), (const void*)(&(piv_data[index + count])), os * sizeof(T));
		piv_size -= count;
		return *this;
	}

	inline void swap(PIVector<T> & other) {
		piSwap<T*>(piv_data, other.piv_data);
		piSwap<size_t>(piv_size, other.piv_size);
		piSwap<size_t>(piv_rsize, other.piv_rsize);
	}

	typedef int (*CompareFunc)(const T * , const T * );
	static int compare_func(const T * t0, const T * t1) {return (*t0) < (*t1) ? -1 : ((*t0) == (*t1) ? 0 : 1);}
	inline PIVector<T> & sort(CompareFunc compare = compare_func) {
		piqsort(piv_data, piv_size, sizeof(T), (int(*)(const void * , const void * ))compare);
		return *this;
	}

	inline PIVector<T> & enlarge(llong piv_size) {
		llong ns = size_s() + piv_size;
		if (ns <= 0) clear();
		else resize(size_t(ns));
		return *this;
	}

	inline PIVector<T> & removeOne(const T & v) {
		for (size_t i = 0; i < piv_size; ++i)
			if (piv_data[i] == v) {
				remove(i);
				return *this;
			}
		return *this;
	}
	inline PIVector<T> & removeAll(const T & v) {
		for (ssize_t i = 0; i < ssize_t(piv_size); ++i)
			if (piv_data[i] == v) {
				remove(i);
				--i;
			}
		return *this;
	}

	inline PIVector<T> & push_back(const T & v) {
		alloc(piv_size + 1);
		PIINTROSPECTION_CONTAINER_USED(T, 1);
		elementNew(piv_data + piv_size - 1, v);
		return *this;
	}
	inline PIVector<T> & append(const T & v) {return push_back(v);}
	inline PIVector<T> & append(const PIVector<T> & other) {
		assert(&other != this);
		size_t ps = piv_size;
		alloc(piv_size + other.piv_size);
		newT(piv_data + ps, other.piv_data, other.piv_size);
		return *this;
	}
	inline PIVector<T> & operator <<(const T & v) {return push_back(v);}
	inline PIVector<T> & operator <<(const PIVector<T> & other) {return append(other);}

	inline PIVector<T> & push_front(const T & v) {insert(0, v); return *this;}
	inline PIVector<T> & prepend(const T & v) {return push_front(v);}

	inline PIVector<T> & pop_back() {
		if (piv_size == 0)
			return *this;
		resize(piv_size - 1);
		return *this;
	}
	inline PIVector<T> & pop_front() {
		if (piv_size == 0)
			return *this;
		remove(0);
		return *this;
	}

	inline T take_back() {T t(back()); pop_back(); return t;}
	inline T take_front() {T t(front()); pop_front(); return t;}

	template <typename ST>
	PIVector<ST> toType() const {
		PIVector<ST> ret(piv_size);
		for (uint i = 0; i < piv_size; ++i)
			ret[i] = ST(piv_data[i]);
		return ret;
	}

#ifdef PIP_CXX11_SUPPORT
	const PIVector<T> & forEach(std::function<void(const T &)> f) const {
		for (uint i = 0; i < piv_size; ++i)
			f(piv_data[i]);
		return *this;
	}
	PIVector<T> copyForEach(std::function<T(const T &)> f) const {
		PIVector<T> ret; ret.reserve(piv_size);
		for (uint i = 0; i < piv_size; ++i)
			ret << f(piv_data[i]);
		return ret;
	}
	PIVector<T> & forEachInplace(std::function<T(const T &)> f) {
		for (uint i = 0; i < piv_size; ++i)
			piv_data[i] = f(piv_data[i]);
		return *this;
	}
	template <typename ST>
	PIVector<ST> toType(std::function<ST(const T &)> f) const {
		PIVector<ST> ret; ret.reserve(piv_size);
		for (uint i = 0; i < piv_size; ++i)
			ret << f(piv_data[i]);
		return ret;
	}
#endif

private:
	inline void _reset() {piv_size = piv_rsize = 0; piv_data = 0;}
	inline size_t asize(size_t s) {
		if (s == 0) return 0;
		if (piv_rsize + piv_rsize >= s && piv_rsize < s)
			return piv_rsize + piv_rsize;
		ssize_t t = 0, s_ = s - 1;
		while (s_ >> t) ++t;
		return (1 << t);
	}
	inline void newT(T * dst, const T * src, size_t s) {
		PIINTROSPECTION_CONTAINER_USED(T, s)
		for (size_t i = 0; i < s; ++i)
			elementNew(dst + i, src[i]);
	}
	inline void deleteT(T * d, size_t sz) {
		PIINTROSPECTION_CONTAINER_UNUSED(T, sz)
		if ((uchar*)d != 0) {
			for (size_t i = 0; i < sz; ++i)
				elementDelete(d[i]);
		}
	}
	inline void elementNew(T * to, const T & from) {new(to)T(from);}
	inline void elementDelete(T & from) {from.~T();}
	inline void dealloc() {
		if ((uchar*)piv_data != 0) free((uchar*)piv_data);
		piv_data = 0;
	}
	inline void alloc(size_t new_size) {
		if (new_size <= piv_rsize) {
			piv_size = new_size;
			return;
		}
		piv_size = new_size;
		size_t as = asize(new_size);
		if (as == piv_rsize) return;
		PIINTROSPECTION_CONTAINER_ALLOC(T, (as-piv_rsize))
		T * p_d = (T*)(realloc((void*)(piv_data), as*sizeof(T)));
		assert(p_d);
		piv_data = p_d;
		piv_rsize = as;
	}

	T * piv_data;
	size_t piv_size, piv_rsize;
};


#define __PIVECTOR_SIMPLE_TYPE__(T) \
	template<> inline void PIVector<T>::newT(T * dst, const T * src, size_t s) {PIINTROSPECTION_CONTAINER_USED(T, s); memcpy((void*)(dst), (const void*)(src), s * sizeof(T));} \
	template<> inline void PIVector<T>::deleteT(T *, size_t sz) {PIINTROSPECTION_CONTAINER_UNUSED(T, sz);} \
	template<> inline void PIVector<T>::elementNew(T * to, const T & from) {(*to) = from;} \
	template<> inline void PIVector<T>::elementDelete(T &) {;} \
	template<> inline PIVector<T> & PIVector<T>::_resizeRaw(size_t new_size) { \
		if (new_size > piv_size) { \
			PIINTROSPECTION_CONTAINER_USED(T, (new_size-piv_size)); \
		} \
		if (new_size < piv_size) { \
			PIINTROSPECTION_CONTAINER_UNUSED(T, (piv_size-new_size)); \
		} \
		alloc(new_size); \
		return *this; \
	} \
	template<> inline PIVector<T> & PIVector<T>::clear() {PIINTROSPECTION_CONTAINER_UNUSED(T, piv_size); piv_size = 0; return *this;} \
	template<> inline PIVector<T> & PIVector<T>::assign(size_t new_size, const T & f) {_resizeRaw(new_size); return fill(f);}


__PIVECTOR_SIMPLE_TYPE__(bool)
__PIVECTOR_SIMPLE_TYPE__(char)
__PIVECTOR_SIMPLE_TYPE__(uchar)
__PIVECTOR_SIMPLE_TYPE__(short)
__PIVECTOR_SIMPLE_TYPE__(ushort)
__PIVECTOR_SIMPLE_TYPE__(int)
__PIVECTOR_SIMPLE_TYPE__(uint)
__PIVECTOR_SIMPLE_TYPE__(long)
__PIVECTOR_SIMPLE_TYPE__(ulong)
__PIVECTOR_SIMPLE_TYPE__(llong)
__PIVECTOR_SIMPLE_TYPE__(ullong)
__PIVECTOR_SIMPLE_TYPE__(float)
__PIVECTOR_SIMPLE_TYPE__(double)
__PIVECTOR_SIMPLE_TYPE__(ldouble)


#ifdef PIP_STD_IOSTREAM
template<typename T>
inline std::ostream & operator <<(std::ostream & s, const PIVector<T> & v) {s << "{"; for (size_t i = 0; i < v.size(); ++i) {s << v[i]; if (i < v.size() - 1) s << ", ";} s << "}"; return s;}
#endif

template<typename T>
inline PICout operator <<(PICout s, const PIVector<T> & v) {
	s.space();
	s.setControl(0, true);
	s << "{";
	for (size_t i = 0; i < v.size(); ++i) {
		s << v[i];
		if (i < v.size() - 1)
			s << ", ";
	}
	s << "}";
	s.restoreControl();
	return s;
}


#endif // PIVECTOR_H