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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
// Copyright 2017 PDFium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
// Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
#include "core/fxcrt/cfx_blockbuffer.h"
#include <algorithm>
#include <utility>
#include "third_party/base/stl_util.h"
namespace {
const int kAllocStep = 1024 * 1024;
} // namespace
CFX_BlockBuffer::CFX_BlockBuffer()
: m_iDataLength(0), m_iBufferSize(0), m_iStartPosition(0) {}
CFX_BlockBuffer::~CFX_BlockBuffer() {}
int32_t CFX_BlockBuffer::GetAllocStep() const {
return kAllocStep;
}
std::pair<wchar_t*, int32_t> CFX_BlockBuffer::GetAvailableBlock() {
if (m_BlockArray.empty())
return {nullptr, 0};
int32_t iRealIndex = m_iStartPosition + m_iDataLength;
if (iRealIndex == m_iBufferSize) {
m_BlockArray.emplace_back(FX_Alloc(wchar_t, kAllocStep));
m_iBufferSize += kAllocStep;
return {m_BlockArray.back().get(), 0};
}
return {m_BlockArray[iRealIndex / kAllocStep].get(), iRealIndex % kAllocStep};
}
bool CFX_BlockBuffer::InitBuffer() {
m_BlockArray.clear();
m_BlockArray.emplace_back(FX_Alloc(wchar_t, kAllocStep));
m_iBufferSize = kAllocStep;
return true;
}
void CFX_BlockBuffer::SetTextChar(int32_t iIndex, wchar_t ch) {
if (iIndex < 0)
return;
int32_t iRealIndex = m_iStartPosition + iIndex;
int32_t iBlockIndex = iRealIndex / kAllocStep;
int32_t iInnerIndex = iRealIndex % kAllocStep;
int32_t iBlockSize = pdfium::CollectionSize<int32_t>(m_BlockArray);
if (iBlockIndex >= iBlockSize) {
int32_t iNewBlocks = iBlockIndex - iBlockSize + 1;
do {
m_BlockArray.emplace_back(FX_Alloc(wchar_t, kAllocStep));
m_iBufferSize += kAllocStep;
} while (--iNewBlocks);
}
wchar_t* pTextData = m_BlockArray[iBlockIndex].get();
pTextData[iInnerIndex] = ch;
m_iDataLength = std::max(m_iDataLength, iIndex + 1);
}
int32_t CFX_BlockBuffer::DeleteTextChars(int32_t iCount) {
if (iCount <= 0)
return m_iDataLength;
if (iCount >= m_iDataLength) {
Reset(false);
return 0;
}
m_iDataLength -= iCount;
return m_iDataLength;
}
CFX_WideString CFX_BlockBuffer::GetTextData(int32_t iStart,
int32_t iLength) const {
int32_t iMaybeDataLength = m_iBufferSize - 1 - m_iStartPosition;
if (iStart < 0 || iStart > iMaybeDataLength)
return CFX_WideString();
if (iLength == -1 || iLength > iMaybeDataLength)
iLength = iMaybeDataLength;
if (iLength <= 0)
return CFX_WideString();
CFX_WideString wsTextData;
wchar_t* pBuf = wsTextData.GetBuffer(iLength);
if (!pBuf)
return CFX_WideString();
int32_t iStartBlock = 0;
int32_t iStartInner = 0;
std::tie(iStartBlock, iStartInner) = TextDataIndex2BufIndex(iStart);
int32_t iEndBlock = 0;
int32_t iEndInner = 0;
std::tie(iEndBlock, iEndInner) = TextDataIndex2BufIndex(iStart + iLength);
int32_t iPointer = 0;
for (int32_t i = iStartBlock; i <= iEndBlock; i++) {
int32_t iBufferPointer = 0;
int32_t iCopyLength = kAllocStep;
if (i == iStartBlock) {
iCopyLength -= iStartInner;
iBufferPointer = iStartInner;
}
if (i == iEndBlock)
iCopyLength -= ((kAllocStep - 1) - iEndInner);
wchar_t* pBlockBuf = m_BlockArray[i].get();
memcpy(pBuf + iPointer, pBlockBuf + iBufferPointer,
iCopyLength * sizeof(wchar_t));
iPointer += iCopyLength;
}
wsTextData.ReleaseBuffer(iLength);
return wsTextData;
}
std::pair<int32_t, int32_t> CFX_BlockBuffer::TextDataIndex2BufIndex(
const int32_t iIndex) const {
ASSERT(iIndex >= 0);
int32_t iRealIndex = m_iStartPosition + iIndex;
return {iRealIndex / kAllocStep, iRealIndex % kAllocStep};
}
|
