smallvec_type.hpp

Go to the documentation of this file.

/* $Id: smallvec_type.hpp 13781 2008-07-22 14:17:29Z peter1138 $ */

#ifndef SMALLVEC_TYPE_HPP
#define SMALLVEC_TYPE_HPP

#include "alloc_func.hpp"
#include "math_func.hpp"

template <typename T, uint S>
class SmallVector {
protected:
  T *data;       
  uint items;    
  uint capacity; 

public:
  SmallVector() : data(NULL), items(0), capacity(0) { }

  ~SmallVector()
  {
    free(this->data);
  }

  FORCEINLINE void Clear()
  {
    /* In fact we just reset the item counter avoiding the need to
     * probably reallocate the same amount of memory the list was
     * previously using. */
    this->items = 0;
  }

  FORCEINLINE void Compact()
  {
    uint capacity = Align(this->items, S);
    if (capacity >= this->capacity) return;

    this->capacity = capacity;
    this->data = ReallocT(this->data, this->capacity);
  }

  FORCEINLINE T *Append()
  {
    if (this->items == this->capacity) {
      this->capacity += S;
      this->data = ReallocT(this->data, this->capacity);
    }

    return &this->data[this->items++];
  }

  FORCEINLINE uint Length() const
  {
    return this->items;
  }

  FORCEINLINE const T *Begin() const
  {
    return this->data;
  }

  FORCEINLINE T *Begin()
  {
    return this->data;
  }

  FORCEINLINE const T *End() const
  {
    return &this->data[this->items];
  }

  FORCEINLINE T *End()
  {
    return &this->data[this->items];
  }

  FORCEINLINE const T *Get(uint index) const
  {
    return &this->data[index];
  }

  FORCEINLINE T *Get(uint index)
  {
    return &this->data[index];
  }

  FORCEINLINE const T &operator[](uint index) const
  {
    return this->data[index];
  }

  FORCEINLINE T &operator[](uint index)
  {
    return this->data[index];
  }
};


template <typename T, uint S>
class AutoFreeSmallVector : public SmallVector<T, S> {
public:
  ~AutoFreeSmallVector()
  {
    this->Clear();
  }

  FORCEINLINE void Clear()
  {
    for (uint i = 0; i < this->items; i++) {
      free(this->data[i]);
    }

    this->items = 0;
  }
};

#endif /* SMALLVEC_TYPE_HPP */