blob: c0485513ee27445cf62a37774d8d0c4090172a2b [file] [log] [blame]
//===-- tsan_vector.h -------------------------------------------*- C++ -*-===//
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
//
// This file is a part of ThreadSanitizer (TSan), a race detector.
//
//===----------------------------------------------------------------------===//
// Low-fat STL-like vector container.
#ifndef TSAN_VECTOR_H
#define TSAN_VECTOR_H
#include "tsan_defs.h"
#include "tsan_mman.h"
namespace __tsan {
template<typename T>
class Vector {
public:
explicit Vector(MBlockType typ)
: typ_(typ)
, begin_()
, end_()
, last_() {
}
~Vector() {
if (begin_)
internal_free(begin_);
}
void Reset() {
if (begin_)
internal_free(begin_);
begin_ = 0;
end_ = 0;
last_ = 0;
}
uptr Size() const {
return end_ - begin_;
}
T &operator[](uptr i) {
DCHECK_LT(i, end_ - begin_);
return begin_[i];
}
const T &operator[](uptr i) const {
DCHECK_LT(i, end_ - begin_);
return begin_[i];
}
T *PushBack() {
EnsureSize(Size() + 1);
T *p = &end_[-1];
internal_memset(p, 0, sizeof(*p));
return p;
}
T *PushBack(const T& v) {
EnsureSize(Size() + 1);
T *p = &end_[-1];
internal_memcpy(p, &v, sizeof(*p));
return p;
}
void PopBack() {
DCHECK_GT(end_, begin_);
end_--;
}
void Resize(uptr size) {
if (size == 0) {
end_ = begin_;
return;
}
uptr old_size = Size();
EnsureSize(size);
if (old_size < size) {
for (uptr i = old_size; i < size; i++)
internal_memset(&begin_[i], 0, sizeof(begin_[i]));
}
}
private:
const MBlockType typ_;
T *begin_;
T *end_;
T *last_;
void EnsureSize(uptr size) {
if (size <= Size())
return;
if (size <= (uptr)(last_ - begin_)) {
end_ = begin_ + size;
return;
}
uptr cap0 = last_ - begin_;
uptr cap = cap0 * 5 / 4; // 25% growth
if (cap == 0)
cap = 16;
if (cap < size)
cap = size;
T *p = (T*)internal_alloc(typ_, cap * sizeof(T));
if (cap0) {
internal_memcpy(p, begin_, cap0 * sizeof(T));
internal_free(begin_);
}
begin_ = p;
end_ = begin_ + size;
last_ = begin_ + cap;
}
Vector(const Vector&);
void operator=(const Vector&);
};
} // namespace __tsan
#endif // #ifndef TSAN_VECTOR_H