blob: bd7ca80571cbadca829f28a8cc1289affeed11cc [file] [log] [blame]
// Copyright 2018 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROMECAST_MEDIA_BASE_ALIGNED_BUFFER_H_
#define CHROMECAST_MEDIA_BASE_ALIGNED_BUFFER_H_
#include <algorithm>
#include <memory>
#include "base/memory/aligned_memory.h"
namespace {
const int kAlignment = 16;
} // namespace
namespace chromecast {
namespace media {
// Convenient class for 16-byte aligned buffers.
template <typename T>
class AlignedBuffer {
public:
AlignedBuffer() : size_(0), data_(nullptr) {}
explicit AlignedBuffer(size_t size)
: size_(size),
data_(static_cast<T*>(
base::AlignedAlloc(size_ * sizeof(T), kAlignment))) {}
AlignedBuffer(int size, const T& val) : AlignedBuffer(size) {}
// Copy constructor
AlignedBuffer(const AlignedBuffer& other)
: size_(other.size()),
data_(static_cast<T*>(
base::AlignedAlloc(size_ * sizeof(T), kAlignment))) {
std::memcpy(data_.get(), other.data(), size_ * sizeof(T));
}
~AlignedBuffer() = default;
// Assignment operator
void operator=(const AlignedBuffer& other) {
size_ = other.size();
data_.reset(
static_cast<T*>(base::AlignedAlloc(size_ * sizeof(T), kAlignment)));
std::memcpy(data_.get(), other.data(), size_ * sizeof(T));
}
// Move operator
AlignedBuffer& operator=(AlignedBuffer&& other) {
size_ = other.size();
data_ = std::move(other.data_);
return *this;
}
void resize(size_t new_size) {
std::unique_ptr<T, base::AlignedFreeDeleter> new_data(
static_cast<T*>(base::AlignedAlloc(new_size * sizeof(T), kAlignment)));
size_t size_to_copy = std::min(new_size, size_);
std::memcpy(new_data.get(), data_.get(), size_to_copy * sizeof(T));
size_ = new_size;
data_ = std::move(new_data);
}
void assign(size_t n, const T& val) {
size_ = n;
data_.reset(
static_cast<T*>(base::AlignedAlloc(size_ * sizeof(T), kAlignment)));
std::fill_n(data_.get(), size_, val);
}
// Returns a pointer to the underlying data.
T* data() { return data_.get(); }
const T* data() const { return data_.get(); }
T& operator[](size_t i) { return data()[i]; }
const T& operator[](size_t i) const { return data()[i]; }
// Returns number of elements.
size_t size() const { return size_; }
bool empty() const { return size_ == 0; }
private:
size_t size_;
std::unique_ptr<T, base::AlignedFreeDeleter> data_;
};
} // namespace media
} // namespace chromecast
#endif // CHROMECAST_MEDIA_BASE_ALIGNED_BUFFER_H_