Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(244)

Side by Side Diff: core/src/fxcodec/jbig2/JBig2_BitStream.cpp

Issue 1800523005: Move core/src/ up to core/. (Closed) Base URL: https://pdfium.googlesource.com/pdfium.git@master
Patch Set: Created 4 years, 9 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « core/src/fxcodec/jbig2/JBig2_BitStream.h ('k') | core/src/fxcodec/jbig2/JBig2_Context.h » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
(Empty)
1 // Copyright 2015 PDFium 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 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
6
7 #include "core/src/fxcodec/jbig2/JBig2_BitStream.h"
8
9 #include <algorithm>
10
11 #include "core/include/fpdfapi/cpdf_stream.h"
12
13 CJBig2_BitStream::CJBig2_BitStream(CPDF_StreamAcc* pSrcStream)
14 : m_pBuf(pSrcStream->GetData()),
15 m_dwLength(pSrcStream->GetSize()),
16 m_dwByteIdx(0),
17 m_dwBitIdx(0),
18 m_dwObjNum(pSrcStream->GetStream() ? pSrcStream->GetStream()->GetObjNum()
19 : 0) {
20 if (m_dwLength > 256 * 1024 * 1024) {
21 m_dwLength = 0;
22 m_pBuf = nullptr;
23 }
24 }
25
26 CJBig2_BitStream::~CJBig2_BitStream() {
27 }
28
29 int32_t CJBig2_BitStream::readNBits(FX_DWORD dwBits, FX_DWORD* dwResult) {
30 FX_DWORD dwBitPos = getBitPos();
31 if (dwBitPos > LengthInBits())
32 return -1;
33
34 *dwResult = 0;
35 if (dwBitPos + dwBits <= LengthInBits())
36 dwBitPos = dwBits;
37 else
38 dwBitPos = LengthInBits() - dwBitPos;
39
40 for (; dwBitPos > 0; --dwBitPos) {
41 *dwResult =
42 (*dwResult << 1) | ((m_pBuf[m_dwByteIdx] >> (7 - m_dwBitIdx)) & 0x01);
43 AdvanceBit();
44 }
45 return 0;
46 }
47
48 int32_t CJBig2_BitStream::readNBits(FX_DWORD dwBits, int32_t* nResult) {
49 FX_DWORD dwBitPos = getBitPos();
50 if (dwBitPos > LengthInBits())
51 return -1;
52
53 *nResult = 0;
54 if (dwBitPos + dwBits <= LengthInBits())
55 dwBitPos = dwBits;
56 else
57 dwBitPos = LengthInBits() - dwBitPos;
58
59 for (; dwBitPos > 0; --dwBitPos) {
60 *nResult =
61 (*nResult << 1) | ((m_pBuf[m_dwByteIdx] >> (7 - m_dwBitIdx)) & 0x01);
62 AdvanceBit();
63 }
64 return 0;
65 }
66
67 int32_t CJBig2_BitStream::read1Bit(FX_DWORD* dwResult) {
68 if (!IsInBound())
69 return -1;
70
71 *dwResult = (m_pBuf[m_dwByteIdx] >> (7 - m_dwBitIdx)) & 0x01;
72 AdvanceBit();
73 return 0;
74 }
75
76 int32_t CJBig2_BitStream::read1Bit(FX_BOOL* bResult) {
77 if (!IsInBound())
78 return -1;
79
80 *bResult = (m_pBuf[m_dwByteIdx] >> (7 - m_dwBitIdx)) & 0x01;
81 AdvanceBit();
82 return 0;
83 }
84
85 int32_t CJBig2_BitStream::read1Byte(uint8_t* cResult) {
86 if (!IsInBound())
87 return -1;
88
89 *cResult = m_pBuf[m_dwByteIdx];
90 ++m_dwByteIdx;
91 return 0;
92 }
93
94 int32_t CJBig2_BitStream::readInteger(FX_DWORD* dwResult) {
95 if (m_dwByteIdx + 3 >= m_dwLength)
96 return -1;
97
98 *dwResult = (m_pBuf[m_dwByteIdx] << 24) | (m_pBuf[m_dwByteIdx + 1] << 16) |
99 (m_pBuf[m_dwByteIdx + 2] << 8) | m_pBuf[m_dwByteIdx + 3];
100 m_dwByteIdx += 4;
101 return 0;
102 }
103
104 int32_t CJBig2_BitStream::readShortInteger(FX_WORD* dwResult) {
105 if (m_dwByteIdx + 1 >= m_dwLength)
106 return -1;
107
108 *dwResult = (m_pBuf[m_dwByteIdx] << 8) | m_pBuf[m_dwByteIdx + 1];
109 m_dwByteIdx += 2;
110 return 0;
111 }
112
113 void CJBig2_BitStream::alignByte() {
114 if (m_dwBitIdx != 0) {
115 ++m_dwByteIdx;
116 m_dwBitIdx = 0;
117 }
118 }
119
120 uint8_t CJBig2_BitStream::getCurByte() const {
121 return IsInBound() ? m_pBuf[m_dwByteIdx] : 0;
122 }
123
124 void CJBig2_BitStream::incByteIdx() {
125 if (IsInBound())
126 ++m_dwByteIdx;
127 }
128
129 uint8_t CJBig2_BitStream::getCurByte_arith() const {
130 return IsInBound() ? m_pBuf[m_dwByteIdx] : 0xFF;
131 }
132
133 uint8_t CJBig2_BitStream::getNextByte_arith() const {
134 return m_dwByteIdx + 1 < m_dwLength ? m_pBuf[m_dwByteIdx + 1] : 0xFF;
135 }
136
137 FX_DWORD CJBig2_BitStream::getOffset() const {
138 return m_dwByteIdx;
139 }
140
141 void CJBig2_BitStream::setOffset(FX_DWORD dwOffset) {
142 m_dwByteIdx = std::min(dwOffset, m_dwLength);
143 }
144
145 FX_DWORD CJBig2_BitStream::getBitPos() const {
146 return (m_dwByteIdx << 3) + m_dwBitIdx;
147 }
148
149 void CJBig2_BitStream::setBitPos(FX_DWORD dwBitPos) {
150 m_dwByteIdx = dwBitPos >> 3;
151 m_dwBitIdx = dwBitPos & 7;
152 }
153
154 const uint8_t* CJBig2_BitStream::getBuf() const {
155 return m_pBuf;
156 }
157
158 const uint8_t* CJBig2_BitStream::getPointer() const {
159 return m_pBuf + m_dwByteIdx;
160 }
161
162 void CJBig2_BitStream::offset(FX_DWORD dwOffset) {
163 m_dwByteIdx += dwOffset;
164 }
165
166 FX_DWORD CJBig2_BitStream::getByteLeft() const {
167 return m_dwLength - m_dwByteIdx;
168 }
169
170 void CJBig2_BitStream::AdvanceBit() {
171 if (m_dwBitIdx == 7) {
172 ++m_dwByteIdx;
173 m_dwBitIdx = 0;
174 } else {
175 ++m_dwBitIdx;
176 }
177 }
178
179 bool CJBig2_BitStream::IsInBound() const {
180 return m_dwByteIdx < m_dwLength;
181 }
182
183 FX_DWORD CJBig2_BitStream::LengthInBits() const {
184 return m_dwLength << 3;
185 }
186
187 FX_DWORD CJBig2_BitStream::getObjNum() const {
188 return m_dwObjNum;
189 }
OLDNEW
« no previous file with comments | « core/src/fxcodec/jbig2/JBig2_BitStream.h ('k') | core/src/fxcodec/jbig2/JBig2_Context.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698