FastVector.h

Go to the documentation of this file.

/* ----------------------------------------------------------------------------

 * GTSAM Copyright 2010, Georgia Tech Research Corporation,
 * Atlanta, Georgia 30332-0415
 * All Rights Reserved
 * Authors: Frank Dellaert, et al. (see THANKS for the full author list)

 * See LICENSE for the license information

 * -------------------------------------------------------------------------- */

#pragma once

#include <vector>
#include <boost/serialization/nvp.hpp>
#include <boost/serialization/vector.hpp>

#include <gtsam/base/FastList.h>
#include <gtsam/base/FastSet.h>

namespace gtsam {

template<typename VALUE>
class FastVector: public std::vector<VALUE, typename internal::FastDefaultVectorAllocator<VALUE>::type> {

public:

  typedef std::vector<VALUE, typename internal::FastDefaultVectorAllocator<VALUE>::type> Base;

  FastVector() {}

  explicit FastVector(size_t size) : Base(size) {}

  explicit FastVector(size_t size, const VALUE& initial) : Base(size, initial) {}

  template<typename INPUTITERATOR>
  explicit FastVector(INPUTITERATOR first, INPUTITERATOR last) {
    // This if statement works around a bug in boost pool allocator and/or
    // STL vector where if the size is zero, the pool allocator will allocate
    // huge amounts of memory.
    if(first != last)
      Base::assign(first, last);
  }

  FastVector(const FastList<VALUE>& x) {
    if(x.size() > 0)
      Base::assign(x.begin(), x.end());
  }

  FastVector(const FastSet<VALUE>& x) {
    if(x.size() > 0)
      Base::assign(x.begin(), x.end());
  }

  FastVector(const Base& x) : Base(x) {}

  template<typename ALLOCATOR>
  FastVector(const std::vector<VALUE, ALLOCATOR>& x)
  {
    // This if statement works around a bug in boost pool allocator and/or
    // STL vector where if the size is zero, the pool allocator will allocate
    // huge amounts of memory.
    if(x.size() > 0)
      Base::assign(x.begin(), x.end());
  }

  operator std::vector<VALUE>() const {
    return std::vector<VALUE>(this->begin(), this->end());
  }

private:
  friend class boost::serialization::access;
  template<class ARCHIVE>
  void serialize(ARCHIVE & ar, const unsigned int version) {
    ar & BOOST_SERIALIZATION_BASE_OBJECT_NVP(Base);
  }

};

}