OLD | NEW |
| (Empty) |
1 // Copyright 2009 Google Inc. | |
2 // | |
3 // Licensed under the Apache License, Version 2.0 (the "License"); | |
4 // you may not use this file except in compliance with the License. | |
5 // You may obtain a copy of the License at | |
6 // | |
7 // http://www.apache.org/licenses/LICENSE-2.0 | |
8 // | |
9 // Unless required by applicable law or agreed to in writing, software | |
10 // distributed under the License is distributed on an "AS IS" BASIS, | |
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
12 // See the License for the specific language governing permissions and | |
13 // limitations under the License. | |
14 // ======================================================================== | |
15 // | |
16 // Derived from RangeCoder.h and RangeCoderBit.h in the LZMA SDK | |
17 | |
18 #include "omaha/mi_exe_stub/x86_encoder/range_encoder.h" | |
19 | |
20 #include "base/basictypes.h" | |
21 | |
22 namespace omaha { | |
23 | |
24 RangeEncoder::RangeEncoder(std::string* output) | |
25 : output_(output), | |
26 cache_(0), | |
27 cache_size_(1), | |
28 low_(0), | |
29 range_(0xFFFFFFFF) { | |
30 } | |
31 | |
32 void RangeEncoder::Encode(uint32 start, uint32 size, uint32 total) { | |
33 low_ += start * (range_ /= total); | |
34 range_ *= size; | |
35 while (range_ < kTopValue) { | |
36 range_ <<= 8; | |
37 ShiftLow(); | |
38 } | |
39 } | |
40 | |
41 void RangeEncoder::ShiftLow() { | |
42 if (static_cast<uint32>(low_) < static_cast<uint32>(0xFF000000) || | |
43 static_cast<int>(low_ >> 32) != 0) { | |
44 uint8 temp = cache_; | |
45 do { | |
46 *output_ += static_cast<uint8>(temp + static_cast<uint8>(low_ >> 32)); | |
47 temp = 0xFF; | |
48 } while (--cache_size_ != 0); | |
49 cache_ = static_cast<uint8>(static_cast<uint32>(low_) >> 24); | |
50 } | |
51 cache_size_++; | |
52 low_ = static_cast<uint32>(low_) << 8; | |
53 } | |
54 | |
55 void RangeEncoder::Flush() { | |
56 for (int i = 0; i < 5; ++i) { | |
57 ShiftLow(); | |
58 } | |
59 } | |
60 | |
61 } // namespace | |
OLD | NEW |