OLD | NEW |
| (Empty) |
1 /* | |
2 * Copyright (c) 2014 The WebRTC project authors. All Rights Reserved. | |
3 * | |
4 * Use of this source code is governed by a BSD-style license | |
5 * that can be found in the LICENSE file in the root of the source | |
6 * tree. An additional intellectual property rights grant can be found | |
7 * in the file PATENTS. All contributing project authors may | |
8 * be found in the AUTHORS file in the root of the source tree. | |
9 */ | |
10 | |
11 #ifndef WEBRTC_SYSTEM_WRAPPERS_INTERFACE_ALIGNED_ARRAY_ | |
12 #define WEBRTC_SYSTEM_WRAPPERS_INTERFACE_ALIGNED_ARRAY_ | |
13 | |
14 #include "webrtc/base/checks.h" | |
15 #include "webrtc/system_wrappers/interface/aligned_malloc.h" | |
16 | |
17 namespace webrtc { | |
18 | |
19 // Wrapper class for aligned arrays. Every row (and the first dimension) are | |
20 // aligned to the given byte alignment. | |
21 template<typename T> class AlignedArray { | |
22 public: | |
23 AlignedArray(int rows, size_t cols, int alignment) | |
24 : rows_(rows), | |
25 cols_(cols), | |
26 alignment_(alignment) { | |
27 RTC_CHECK_GT(alignment_, 0); | |
28 head_row_ = static_cast<T**>(AlignedMalloc(rows_ * sizeof(*head_row_), | |
29 alignment_)); | |
30 for (int i = 0; i < rows_; ++i) { | |
31 head_row_[i] = static_cast<T*>(AlignedMalloc(cols_ * sizeof(**head_row_), | |
32 alignment_)); | |
33 } | |
34 } | |
35 | |
36 ~AlignedArray() { | |
37 for (int i = 0; i < rows_; ++i) { | |
38 AlignedFree(head_row_[i]); | |
39 } | |
40 AlignedFree(head_row_); | |
41 } | |
42 | |
43 T* const* Array() { | |
44 return head_row_; | |
45 } | |
46 | |
47 const T* const* Array() const { | |
48 return head_row_; | |
49 } | |
50 | |
51 T* Row(int row) { | |
52 RTC_CHECK_LE(row, rows_); | |
53 return head_row_[row]; | |
54 } | |
55 | |
56 const T* Row(int row) const { | |
57 RTC_CHECK_LE(row, rows_); | |
58 return head_row_[row]; | |
59 } | |
60 | |
61 T& At(int row, size_t col) { | |
62 RTC_CHECK_LE(col, cols_); | |
63 return Row(row)[col]; | |
64 } | |
65 | |
66 const T& At(int row, size_t col) const { | |
67 RTC_CHECK_LE(col, cols_); | |
68 return Row(row)[col]; | |
69 } | |
70 | |
71 int rows() const { | |
72 return rows_; | |
73 } | |
74 | |
75 size_t cols() const { | |
76 return cols_; | |
77 } | |
78 | |
79 private: | |
80 int rows_; | |
81 size_t cols_; | |
82 int alignment_; | |
83 T** head_row_; | |
84 }; | |
85 | |
86 } // namespace webrtc | |
87 | |
88 #endif // WEBRTC_SYSTEM_WRAPPERS_INTERFACE_ALIGNED_ARRAY_ | |
OLD | NEW |