blob: 40ece26020b7ee6f8166b6d45bb2f7e5220835e6 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
/*
* Copyright 2020 The libgav1 Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef LIBGAV1_SRC_UTILS_DYNAMIC_BUFFER_H_
#define LIBGAV1_SRC_UTILS_DYNAMIC_BUFFER_H_
#include <memory>
#include <new>
#include "src/utils/memory.h"
namespace libgav1 {
template <typename T>
class DynamicBuffer {
public:
T* get() { return buffer_.get(); }
const T* get() const { return buffer_.get(); }
// Resizes the buffer so that it can hold at least |size| elements. Existing
// contents will be destroyed when resizing to a larger size.
//
// Returns true on success. If Resize() returns false, then subsequent calls
// to get() will return nullptr.
bool Resize(size_t size) {
if (size <= size_) return true;
buffer_.reset(new (std::nothrow) T[size]);
if (buffer_ == nullptr) {
size_ = 0;
return false;
}
size_ = size;
return true;
}
size_t size() const { return size_; }
private:
std::unique_ptr<T[]> buffer_;
size_t size_ = 0;
};
template <typename T, int alignment>
class AlignedDynamicBuffer {
public:
T* get() { return buffer_.get(); }
// Resizes the buffer so that it can hold at least |size| elements. Existing
// contents will be destroyed when resizing to a larger size.
//
// Returns true on success. If Resize() returns false, then subsequent calls
// to get() will return nullptr.
bool Resize(size_t size) {
if (size <= size_) return true;
buffer_ = MakeAlignedUniquePtr<T>(alignment, size);
if (buffer_ == nullptr) {
size_ = 0;
return false;
}
size_ = size;
return true;
}
private:
AlignedUniquePtr<T> buffer_;
size_t size_ = 0;
};
} // namespace libgav1
#endif // LIBGAV1_SRC_UTILS_DYNAMIC_BUFFER_H_
|
