OLD | NEW |
| (Empty) |
1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "content/common/partial_circular_buffer.h" | |
6 | |
7 #include <algorithm> | |
8 | |
9 #include "base/logging.h" | |
10 | |
11 namespace content { | |
12 | |
13 namespace { | |
14 | |
15 inline uint32 Min3(uint32 a, uint32 b, uint32 c) { | |
16 return std::min(a, std::min(b, c)); | |
17 } | |
18 | |
19 } // namespace | |
20 | |
21 PartialCircularBuffer::PartialCircularBuffer(void* buffer, | |
22 uint32 buffer_size) | |
23 : buffer_data_(reinterpret_cast<BufferData*>(buffer)), | |
24 memory_buffer_size_(buffer_size), | |
25 data_size_(0), | |
26 position_(0), | |
27 total_read_(0) { | |
28 uint32 header_size = | |
29 buffer_data_->data - reinterpret_cast<uint8*>(buffer_data_); | |
30 data_size_ = memory_buffer_size_ - header_size; | |
31 | |
32 DCHECK(buffer_data_); | |
33 DCHECK_GE(memory_buffer_size_, header_size); | |
34 DCHECK_LE(buffer_data_->total_written, data_size_); | |
35 DCHECK_LT(buffer_data_->wrap_position, data_size_); | |
36 DCHECK_LT(buffer_data_->end_position, data_size_); | |
37 } | |
38 | |
39 PartialCircularBuffer::PartialCircularBuffer(void* buffer, | |
40 uint32 buffer_size, | |
41 uint32 wrap_position) | |
42 : buffer_data_(reinterpret_cast<BufferData*>(buffer)), | |
43 memory_buffer_size_(buffer_size), | |
44 data_size_(0), | |
45 position_(0), | |
46 total_read_(0) { | |
47 uint32 header_size = | |
48 buffer_data_->data - reinterpret_cast<uint8*>(buffer_data_); | |
49 data_size_ = memory_buffer_size_ - header_size; | |
50 | |
51 DCHECK(buffer_data_); | |
52 DCHECK_GE(memory_buffer_size_, header_size); | |
53 DCHECK_LT(wrap_position, data_size_); | |
54 | |
55 buffer_data_->total_written = 0; | |
56 buffer_data_->wrap_position = wrap_position; | |
57 buffer_data_->end_position = 0; | |
58 } | |
59 | |
60 uint32 PartialCircularBuffer::Read(void* buffer, uint32 buffer_size) { | |
61 DCHECK(buffer_data_); | |
62 if (total_read_ >= buffer_data_->total_written) | |
63 return 0; | |
64 | |
65 uint8* buffer_uint8 = reinterpret_cast<uint8*>(buffer); | |
66 uint32 read = 0; | |
67 | |
68 // Read from beginning part. | |
69 if (position_ < buffer_data_->wrap_position) { | |
70 uint32 to_wrap_pos = buffer_data_->wrap_position - position_; | |
71 uint32 to_eow = buffer_data_->total_written - total_read_; | |
72 uint32 to_read = Min3(buffer_size, to_wrap_pos, to_eow); | |
73 memcpy(buffer_uint8, buffer_data_->data + position_, to_read); | |
74 position_ += to_read; | |
75 total_read_ += to_read; | |
76 read += to_read; | |
77 if (position_ == buffer_data_->wrap_position && | |
78 buffer_data_->total_written == data_size_) { | |
79 // We've read all the beginning part, set the position to the middle part. | |
80 // (The second condition above checks if the wrapping part is filled, i.e. | |
81 // writing has wrapped.) | |
82 position_ = buffer_data_->end_position; | |
83 } | |
84 if (read >= buffer_size) { | |
85 DCHECK_EQ(read, buffer_size); | |
86 return read; | |
87 } | |
88 if (read >= to_eow) { | |
89 DCHECK_EQ(read, to_eow); | |
90 DCHECK_EQ(total_read_, buffer_data_->total_written); | |
91 return read; | |
92 } | |
93 } | |
94 | |
95 // Read from middle part. | |
96 DCHECK_GE(position_, buffer_data_->wrap_position); | |
97 if (position_ >= buffer_data_->end_position) { | |
98 uint32 remaining_buffer_size = buffer_size - read; | |
99 uint32 to_eof = data_size_ - position_; | |
100 uint32 to_eow = buffer_data_->total_written - total_read_; | |
101 uint32 to_read = Min3(remaining_buffer_size, to_eof, to_eow); | |
102 memcpy(buffer_uint8 + read, buffer_data_->data + position_, to_read); | |
103 position_ += to_read; | |
104 total_read_ += to_read; | |
105 read += to_read; | |
106 if (position_ == data_size_) { | |
107 // We've read all the middle part, set position to the end part. | |
108 position_ = buffer_data_->wrap_position; | |
109 } | |
110 if (read >= buffer_size) { | |
111 DCHECK_EQ(read, buffer_size); | |
112 return read; | |
113 } | |
114 if (total_read_ >= buffer_data_->total_written) { | |
115 DCHECK_EQ(total_read_, buffer_data_->total_written); | |
116 return read; | |
117 } | |
118 } | |
119 | |
120 // Read from end part. | |
121 DCHECK_GE(position_, buffer_data_->wrap_position); | |
122 DCHECK_LT(position_, buffer_data_->end_position); | |
123 uint32 remaining_buffer_size = buffer_size - read; | |
124 uint32 to_eob = buffer_data_->end_position - position_; | |
125 uint32 to_eow = buffer_data_->total_written - total_read_; | |
126 uint32 to_read = Min3(remaining_buffer_size, to_eob, to_eow); | |
127 memcpy(buffer_uint8 + read, buffer_data_->data + position_, to_read); | |
128 position_ += to_read; | |
129 total_read_ += to_read; | |
130 read += to_read; | |
131 DCHECK_LE(read, buffer_size); | |
132 DCHECK_LE(total_read_, buffer_data_->total_written); | |
133 return read; | |
134 } | |
135 | |
136 void PartialCircularBuffer::Write(const void* buffer, uint32 buffer_size) { | |
137 DCHECK(buffer_data_); | |
138 uint32 position_before_write = position_; | |
139 | |
140 uint32 to_eof = data_size_ - position_; | |
141 uint32 to_write = std::min(buffer_size, to_eof); | |
142 DoWrite(buffer_data_->data + position_, buffer, to_write); | |
143 if (position_ >= data_size_) { | |
144 DCHECK_EQ(position_, data_size_); | |
145 position_ = buffer_data_->wrap_position; | |
146 } | |
147 | |
148 if (to_write < buffer_size) { | |
149 uint32 remainder_to_write = buffer_size - to_write; | |
150 DCHECK_LT(position_, position_before_write); | |
151 DCHECK_LE(position_ + remainder_to_write, position_before_write); | |
152 DoWrite(buffer_data_->data + position_, | |
153 reinterpret_cast<const uint8*>(buffer) + to_write, | |
154 remainder_to_write); | |
155 } | |
156 } | |
157 | |
158 void PartialCircularBuffer::DoWrite(void* dest, const void* src, uint32 num) { | |
159 memcpy(dest, src, num); | |
160 position_ += num; | |
161 buffer_data_->total_written = | |
162 std::min(buffer_data_->total_written + num, data_size_); | |
163 buffer_data_->end_position = position_; | |
164 } | |
165 | |
166 } // namespace content | |
OLD | NEW |