| // Copyright 2014 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 |
| // Original code is licensed as follows: |
| /* |
| * Copyright 2008 ZXing authors |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #include "fxbarcode/qrcode/BC_QRCoderEncoder.h" |
| |
| #include <algorithm> |
| #include <memory> |
| #include <utility> |
| |
| #include "fxbarcode/BC_UtilCodingConvert.h" |
| #include "fxbarcode/common/BC_CommonByteArray.h" |
| #include "fxbarcode/common/BC_CommonByteMatrix.h" |
| #include "fxbarcode/common/reedsolomon/BC_ReedSolomon.h" |
| #include "fxbarcode/common/reedsolomon/BC_ReedSolomonGF256.h" |
| #include "fxbarcode/qrcode/BC_QRCoder.h" |
| #include "fxbarcode/qrcode/BC_QRCoderBitVector.h" |
| #include "fxbarcode/qrcode/BC_QRCoderBlockPair.h" |
| #include "fxbarcode/qrcode/BC_QRCoderECBlocks.h" |
| #include "fxbarcode/qrcode/BC_QRCoderMaskUtil.h" |
| #include "fxbarcode/qrcode/BC_QRCoderMatrixUtil.h" |
| #include "fxbarcode/qrcode/BC_QRCoderMode.h" |
| #include "fxbarcode/qrcode/BC_QRCoderVersion.h" |
| #include "third_party/base/ptr_util.h" |
| |
| using ModeStringPair = std::pair<CBC_QRCoderMode*, ByteString>; |
| |
| namespace { |
| |
| // This is a mapping for an ASCII table, starting at an index of 32. |
| const int8_t g_alphaNumericTable[] = { |
| 36, -1, -1, -1, 37, 38, -1, -1, -1, -1, 39, 40, -1, 41, 42, 43, // 32-47 |
| 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 44, -1, -1, -1, -1, -1, // 48-63 |
| -1, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, // 64-79 |
| 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35}; |
| |
| int32_t GetAlphaNumericCode(int32_t code) { |
| if (code < 32) |
| return -1; |
| size_t code_index = static_cast<size_t>(code - 32); |
| if (code_index >= FX_ArraySize(g_alphaNumericTable)) |
| return -1; |
| return g_alphaNumericTable[code_index]; |
| } |
| |
| void AppendNumericBytes(const ByteString& content, |
| CBC_QRCoderBitVector* bits, |
| int32_t& e) { |
| int32_t length = content.GetLength(); |
| int32_t i = 0; |
| while (i < length) { |
| int32_t num1 = content[i] - '0'; |
| if (i + 2 < length) { |
| int32_t num2 = content[i + 1] - '0'; |
| int32_t num3 = content[i + 2] - '0'; |
| bits->AppendBits(num1 * 100 + num2 * 10 + num3, 10); |
| i += 3; |
| } else if (i + 1 < length) { |
| int32_t num2 = content[i + 1] - '0'; |
| bits->AppendBits(num1 * 10 + num2, 7); |
| i += 2; |
| } else { |
| bits->AppendBits(num1, 4); |
| i++; |
| } |
| } |
| } |
| |
| void AppendAlphaNumericBytes(const ByteString& content, |
| CBC_QRCoderBitVector* bits, |
| int32_t& e) { |
| int32_t length = content.GetLength(); |
| int32_t i = 0; |
| while (i < length) { |
| int32_t code1 = GetAlphaNumericCode(content[i]); |
| if (code1 == -1) { |
| e = BCExceptionInvalidateCharacter; |
| return; |
| } |
| if (i + 1 < length) { |
| int32_t code2 = GetAlphaNumericCode(content[i + 1]); |
| if (code2 == -1) { |
| e = BCExceptionInvalidateCharacter; |
| return; |
| } |
| bits->AppendBits(code1 * 45 + code2, 11); |
| i += 2; |
| } else { |
| bits->AppendBits(code1, 6); |
| i++; |
| } |
| } |
| } |
| |
| void AppendGBKBytes(const ByteString& content, |
| CBC_QRCoderBitVector* bits, |
| int32_t& e) { |
| int32_t length = content.GetLength(); |
| uint32_t value = 0; |
| for (int32_t i = 0; i < length; i += 2) { |
| value = (uint32_t)(content[i] << 8 | content[i + 1]); |
| if (value <= 0xAAFE && value >= 0xA1A1) { |
| value -= 0xA1A1; |
| } else if (value <= 0xFAFE && value >= 0xB0A1) { |
| value -= 0xA6A1; |
| } else { |
| e = BCExceptionInvalidateCharacter; |
| return; |
| } |
| value = (uint32_t)((value >> 8) * 0x60) + (uint32_t)(value & 0xff); |
| bits->AppendBits(value, 13); |
| } |
| } |
| |
| void Append8BitBytes(const ByteString& content, |
| CBC_QRCoderBitVector* bits, |
| ByteString encoding, |
| int32_t& e) { |
| for (size_t i = 0; i < content.GetLength(); i++) |
| bits->AppendBits(content[i], 8); |
| } |
| |
| void AppendKanjiBytes(const ByteString& content, |
| CBC_QRCoderBitVector* bits, |
| int32_t& e) { |
| std::vector<uint8_t> bytes; |
| uint32_t value = 0; |
| for (size_t i = 0; i < bytes.size(); i += 2) { |
| value = (uint32_t)((content[i] << 8) | content[i + 1]); |
| if (value <= 0x9ffc && value >= 0x8140) { |
| value -= 0x8140; |
| } else if (value <= 0xebbf && value >= 0xe040) { |
| value -= 0xc140; |
| } else { |
| e = BCExceptionInvalidateCharacter; |
| return; |
| } |
| value = (uint32_t)((value >> 8) * 0xc0) + (uint32_t)(value & 0xff); |
| bits->AppendBits(value, 13); |
| } |
| } |
| |
| void AppendModeInfo(CBC_QRCoderMode* mode, CBC_QRCoderBitVector* bits) { |
| bits->AppendBits(mode->GetBits(), 4); |
| if (mode == CBC_QRCoderMode::sGBK) |
| bits->AppendBits(1, 4); |
| } |
| |
| bool AppendLengthInfo(int32_t numLetters, |
| int32_t version, |
| CBC_QRCoderMode* mode, |
| CBC_QRCoderBitVector* bits) { |
| int32_t e = BCExceptionNO; |
| const auto* qcv = CBC_QRCoderVersion::GetVersionForNumber(version); |
| if (!qcv) |
| return false; |
| int32_t numBits = mode->GetCharacterCountBits(qcv->GetVersionNumber(), e); |
| if (e != BCExceptionNO) |
| return false; |
| if (numBits > ((1 << numBits) - 1)) |
| return true; |
| |
| if (mode == CBC_QRCoderMode::sGBK) |
| bits->AppendBits(numLetters / 2, numBits); |
| bits->AppendBits(numLetters, numBits); |
| return true; |
| } |
| |
| void AppendBytes(const ByteString& content, |
| CBC_QRCoderMode* mode, |
| CBC_QRCoderBitVector* bits, |
| ByteString encoding, |
| int32_t& e) { |
| if (mode == CBC_QRCoderMode::sNUMERIC) |
| AppendNumericBytes(content, bits, e); |
| else if (mode == CBC_QRCoderMode::sALPHANUMERIC) |
| AppendAlphaNumericBytes(content, bits, e); |
| else if (mode == CBC_QRCoderMode::sBYTE) |
| Append8BitBytes(content, bits, encoding, e); |
| else if (mode == CBC_QRCoderMode::sKANJI) |
| AppendKanjiBytes(content, bits, e); |
| else if (mode == CBC_QRCoderMode::sGBK) |
| AppendGBKBytes(content, bits, e); |
| else |
| e = BCExceptionUnsupportedMode; |
| } |
| |
| bool InitQRCode(int32_t numInputBytes, |
| const CBC_QRCoderErrorCorrectionLevel* ecLevel, |
| CBC_QRCoderMode* mode, |
| CBC_QRCoder* qrCode) { |
| qrCode->SetECLevel(ecLevel); |
| qrCode->SetMode(mode); |
| for (int32_t i = 1; i <= CBC_QRCoderVersion::kMaxVersion; ++i) { |
| const auto* version = CBC_QRCoderVersion::GetVersionForNumber(i); |
| int32_t numBytes = version->GetTotalCodeWords(); |
| const auto* ecBlocks = version->GetECBlocksForLevel(*ecLevel); |
| int32_t numEcBytes = ecBlocks->GetTotalECCodeWords(); |
| int32_t numRSBlocks = ecBlocks->GetNumBlocks(); |
| int32_t numDataBytes = numBytes - numEcBytes; |
| if (numDataBytes >= numInputBytes + 3) { |
| qrCode->SetVersion(i); |
| qrCode->SetNumTotalBytes(numBytes); |
| qrCode->SetNumDataBytes(numDataBytes); |
| qrCode->SetNumRSBlocks(numRSBlocks); |
| qrCode->SetNumECBytes(numEcBytes); |
| qrCode->SetMatrixWidth(version->GetDimensionForVersion()); |
| return true; |
| } |
| } |
| return false; |
| } |
| |
| std::unique_ptr<CBC_CommonByteArray> GenerateECBytes( |
| CBC_CommonByteArray* dataBytes, |
| int32_t numEcBytesInBlock) { |
| int32_t numDataBytes = dataBytes->Size(); |
| std::vector<int32_t> toEncode(numDataBytes + numEcBytesInBlock); |
| for (int32_t i = 0; i < numDataBytes; ++i) |
| toEncode[i] = dataBytes->At(i); |
| CBC_ReedSolomonEncoder encode(CBC_ReedSolomonGF256::QRCodeField); |
| encode.Init(); |
| if (!encode.Encode(&toEncode, numEcBytesInBlock)) |
| return nullptr; |
| auto ecBytes = pdfium::MakeUnique<CBC_CommonByteArray>(numEcBytesInBlock); |
| for (int32_t i = 0; i < numEcBytesInBlock; ++i) |
| ecBytes->Set(i, toEncode[numDataBytes + i]); |
| return ecBytes; |
| } |
| |
| int32_t GetSpanByVersion(CBC_QRCoderMode* modeFirst, |
| CBC_QRCoderMode* modeSecond, |
| int32_t versionNum, |
| int32_t& e) { |
| if (versionNum == 0) |
| return 0; |
| |
| if (modeFirst == CBC_QRCoderMode::sALPHANUMERIC && |
| modeSecond == CBC_QRCoderMode::sBYTE) { |
| if (versionNum >= 1 && versionNum <= 9) |
| return 11; |
| if (versionNum >= 10 && versionNum <= 26) |
| return 15; |
| if (versionNum >= 27 && versionNum <= CBC_QRCoderVersion::kMaxVersion) |
| return 16; |
| e = BCExceptionNoSuchVersion; |
| return 0; |
| } |
| if (modeSecond == CBC_QRCoderMode::sALPHANUMERIC && |
| modeFirst == CBC_QRCoderMode::sNUMERIC) { |
| if (versionNum >= 1 && versionNum <= 9) |
| return 13; |
| if (versionNum >= 10 && versionNum <= 26) |
| return 15; |
| if (versionNum >= 27 && versionNum <= CBC_QRCoderVersion::kMaxVersion) |
| return 17; |
| e = BCExceptionNoSuchVersion; |
| return 0; |
| } |
| if (modeSecond == CBC_QRCoderMode::sBYTE && |
| modeFirst == CBC_QRCoderMode::sNUMERIC) { |
| if (versionNum >= 1 && versionNum <= 9) |
| return 6; |
| if (versionNum >= 10 && versionNum <= 26) |
| return 8; |
| if (versionNum >= 27 && versionNum <= CBC_QRCoderVersion::kMaxVersion) |
| return 9; |
| e = BCExceptionNoSuchVersion; |
| return 0; |
| } |
| return -1; |
| } |
| |
| int32_t CalculateMaskPenalty(CBC_CommonByteMatrix* matrix) { |
| return CBC_QRCoderMaskUtil::ApplyMaskPenaltyRule1(matrix) + |
| CBC_QRCoderMaskUtil::ApplyMaskPenaltyRule2(matrix) + |
| CBC_QRCoderMaskUtil::ApplyMaskPenaltyRule3(matrix) + |
| CBC_QRCoderMaskUtil::ApplyMaskPenaltyRule4(matrix); |
| } |
| |
| int32_t ChooseMaskPattern(CBC_QRCoderBitVector* bits, |
| const CBC_QRCoderErrorCorrectionLevel* ecLevel, |
| int32_t version, |
| CBC_CommonByteMatrix* matrix, |
| int32_t& e) { |
| int32_t minPenalty = 65535; |
| int32_t bestMaskPattern = -1; |
| for (int32_t maskPattern = 0; maskPattern < CBC_QRCoder::kNumMaskPatterns; |
| maskPattern++) { |
| CBC_QRCoderMatrixUtil::BuildMatrix(bits, ecLevel, version, maskPattern, |
| matrix, e); |
| if (e != BCExceptionNO) |
| return 0; |
| int32_t penalty = CalculateMaskPenalty(matrix); |
| if (penalty < minPenalty) { |
| minPenalty = penalty; |
| bestMaskPattern = maskPattern; |
| } |
| } |
| return bestMaskPattern; |
| } |
| |
| void GetNumDataBytesAndNumECBytesForBlockID(int32_t numTotalBytes, |
| int32_t numDataBytes, |
| int32_t numRSBlocks, |
| int32_t blockID, |
| int32_t& numDataBytesInBlock, |
| int32_t& numECBytesInBlock) { |
| if (blockID >= numRSBlocks) |
| return; |
| |
| int32_t numRsBlocksInGroup2 = numTotalBytes % numRSBlocks; |
| int32_t numRsBlocksInGroup1 = numRSBlocks - numRsBlocksInGroup2; |
| int32_t numTotalBytesInGroup1 = numTotalBytes / numRSBlocks; |
| int32_t numTotalBytesInGroup2 = numTotalBytesInGroup1 + 1; |
| int32_t numDataBytesInGroup1 = numDataBytes / numRSBlocks; |
| int32_t numDataBytesInGroup2 = numDataBytesInGroup1 + 1; |
| int32_t numEcBytesInGroup1 = numTotalBytesInGroup1 - numDataBytesInGroup1; |
| int32_t numEcBytesInGroup2 = numTotalBytesInGroup2 - numDataBytesInGroup2; |
| if (blockID < numRsBlocksInGroup1) { |
| numDataBytesInBlock = numDataBytesInGroup1; |
| numECBytesInBlock = numEcBytesInGroup1; |
| } else { |
| numDataBytesInBlock = numDataBytesInGroup2; |
| numECBytesInBlock = numEcBytesInGroup2; |
| } |
| } |
| |
| bool TerminateBits(int32_t numDataBytes, CBC_QRCoderBitVector* bits) { |
| size_t capacity = numDataBytes << 3; |
| if (bits->Size() > capacity) |
| return false; |
| |
| for (int32_t i = 0; i < 4 && bits->Size() < capacity; ++i) |
| bits->AppendBit(0); |
| |
| int32_t numBitsInLastByte = bits->Size() % 8; |
| if (numBitsInLastByte > 0) { |
| int32_t numPaddingBits = 8 - numBitsInLastByte; |
| for (int32_t j = 0; j < numPaddingBits; ++j) |
| bits->AppendBit(0); |
| } |
| |
| if (bits->Size() % 8 != 0) |
| return false; |
| |
| int32_t numPaddingBytes = numDataBytes - bits->sizeInBytes(); |
| for (int32_t k = 0; k < numPaddingBytes; ++k) |
| bits->AppendBits(k % 2 ? 0x11 : 0xec, 8); |
| return bits->Size() == capacity; |
| } |
| |
| void MergeString(std::vector<ModeStringPair>* result, |
| int32_t versionNum, |
| int32_t& e) { |
| size_t mergeNum = 0; |
| for (size_t i = 0; i + 1 < result->size(); i++) { |
| auto& element1 = (*result)[i]; |
| auto& element2 = (*result)[i + 1]; |
| if (element1.first == CBC_QRCoderMode::sALPHANUMERIC) { |
| int32_t tmp = GetSpanByVersion(CBC_QRCoderMode::sALPHANUMERIC, |
| CBC_QRCoderMode::sBYTE, versionNum, e); |
| if (e != BCExceptionNO) |
| return; |
| if (element2.first == CBC_QRCoderMode::sBYTE && tmp >= 0 && |
| element1.second.GetLength() < static_cast<size_t>(tmp)) { |
| element2.second = element1.second + element2.second; |
| result->erase(result->begin() + i); |
| i--; |
| mergeNum++; |
| } |
| } else if (element1.first == CBC_QRCoderMode::sBYTE) { |
| if (element2.first == CBC_QRCoderMode::sBYTE) { |
| element1.second += element2.second; |
| result->erase(result->begin() + i + 1); |
| i--; |
| mergeNum++; |
| } |
| } else if (element1.first == CBC_QRCoderMode::sNUMERIC) { |
| int32_t tmp = GetSpanByVersion(CBC_QRCoderMode::sNUMERIC, |
| CBC_QRCoderMode::sBYTE, versionNum, e); |
| if (e != BCExceptionNO) |
| return; |
| if (element2.first == CBC_QRCoderMode::sBYTE && tmp >= 0 && |
| element1.second.GetLength() < static_cast<size_t>(tmp)) { |
| element2.second = element1.second + element2.second; |
| result->erase(result->begin() + i); |
| i--; |
| mergeNum++; |
| } |
| tmp = GetSpanByVersion(CBC_QRCoderMode::sNUMERIC, |
| CBC_QRCoderMode::sALPHANUMERIC, versionNum, e); |
| if (e != BCExceptionNO) |
| return; |
| if (element2.first == CBC_QRCoderMode::sALPHANUMERIC && tmp >= 0 && |
| element1.second.GetLength() < static_cast<size_t>(tmp)) { |
| element2.second = element1.second + element2.second; |
| result->erase(result->begin() + i); |
| i--; |
| mergeNum++; |
| } |
| } |
| } |
| if (mergeNum != 0) |
| MergeString(result, versionNum, e); |
| } |
| |
| void SplitString(const ByteString& content, |
| std::vector<ModeStringPair>* result) { |
| size_t index = 0; |
| while (index < content.GetLength()) { |
| uint8_t c = static_cast<uint8_t>(content[index]); |
| if (!((c >= 0xA1 && c <= 0xAA) || (c >= 0xB0 && c <= 0xFA))) |
| break; |
| index += 2; |
| } |
| if (index) |
| result->push_back({CBC_QRCoderMode::sGBK, content.Left(index)}); |
| if (index >= content.GetLength()) |
| return; |
| |
| size_t flag = index; |
| while (GetAlphaNumericCode(content[index]) == -1 && |
| index < content.GetLength()) { |
| uint8_t c = static_cast<uint8_t>(content[index]); |
| if (((c >= 0xA1 && c <= 0xAA) || (c >= 0xB0 && c <= 0xFA))) |
| break; |
| #if _FX_PLATFORM_ == _FX_PLATFORM_WINDOWS_ |
| bool high = !!IsDBCSLeadByte(content[index]); |
| #else |
| bool high = content[index] > 127; |
| #endif |
| ++index; |
| if (high) |
| ++index; |
| } |
| if (index != flag) { |
| result->push_back( |
| {CBC_QRCoderMode::sBYTE, content.Mid(flag, index - flag)}); |
| } |
| flag = index; |
| if (index >= content.GetLength()) |
| return; |
| |
| while (index < content.GetLength() && isdigit(content[index])) |
| ++index; |
| |
| if (index != flag) { |
| result->push_back( |
| {CBC_QRCoderMode::sNUMERIC, content.Mid(flag, index - flag)}); |
| } |
| flag = index; |
| if (index >= content.GetLength()) |
| return; |
| |
| while (index < content.GetLength() && |
| GetAlphaNumericCode(content[index]) != -1) { |
| ++index; |
| } |
| if (index != flag) { |
| result->push_back( |
| {CBC_QRCoderMode::sALPHANUMERIC, content.Mid(flag, index - flag)}); |
| } |
| flag = index; |
| if (index < content.GetLength()) |
| SplitString(content.Right(content.GetLength() - index), result); |
| } |
| |
| CBC_QRCoderMode* ChooseMode(const ByteString& content, ByteString encoding) { |
| if (encoding.Compare("SHIFT_JIS") == 0) |
| return CBC_QRCoderMode::sKANJI; |
| |
| bool hasNumeric = false; |
| bool hasAlphaNumeric = false; |
| for (size_t i = 0; i < content.GetLength(); i++) { |
| if (isdigit(content[i])) { |
| hasNumeric = true; |
| } else if (GetAlphaNumericCode(content[i]) != -1) { |
| hasAlphaNumeric = true; |
| } else { |
| return CBC_QRCoderMode::sBYTE; |
| } |
| } |
| if (hasAlphaNumeric) |
| return CBC_QRCoderMode::sALPHANUMERIC; |
| if (hasNumeric) |
| return CBC_QRCoderMode::sNUMERIC; |
| return CBC_QRCoderMode::sBYTE; |
| } |
| |
| bool InterleaveWithECBytes(CBC_QRCoderBitVector* bits, |
| int32_t numTotalBytes, |
| int32_t numDataBytes, |
| int32_t numRSBlocks, |
| CBC_QRCoderBitVector* result) { |
| ASSERT(numTotalBytes >= 0); |
| ASSERT(numDataBytes >= 0); |
| if (bits->sizeInBytes() != static_cast<size_t>(numDataBytes)) |
| return false; |
| |
| int32_t dataBytesOffset = 0; |
| int32_t maxNumDataBytes = 0; |
| int32_t maxNumEcBytes = 0; |
| std::vector<CBC_QRCoderBlockPair> blocks(numRSBlocks); |
| for (int32_t i = 0; i < numRSBlocks; i++) { |
| int32_t numDataBytesInBlock; |
| int32_t numEcBytesInBlosk; |
| GetNumDataBytesAndNumECBytesForBlockID(numTotalBytes, numDataBytes, |
| numRSBlocks, i, numDataBytesInBlock, |
| numEcBytesInBlosk); |
| auto dataBytes = pdfium::MakeUnique<CBC_CommonByteArray>(); |
| dataBytes->Set(bits->GetArray(), dataBytesOffset, numDataBytesInBlock); |
| std::unique_ptr<CBC_CommonByteArray> ecBytes = |
| GenerateECBytes(dataBytes.get(), numEcBytesInBlosk); |
| if (!ecBytes) |
| return false; |
| |
| maxNumDataBytes = std::max(maxNumDataBytes, dataBytes->Size()); |
| maxNumEcBytes = std::max(maxNumEcBytes, ecBytes->Size()); |
| blocks[i].SetData(std::move(dataBytes), std::move(ecBytes)); |
| dataBytesOffset += numDataBytesInBlock; |
| } |
| if (numDataBytes != dataBytesOffset) |
| return false; |
| |
| for (int32_t x = 0; x < maxNumDataBytes; x++) { |
| for (size_t j = 0; j < blocks.size(); j++) { |
| const CBC_CommonByteArray* dataBytes = blocks[j].GetDataBytes(); |
| if (x < dataBytes->Size()) |
| result->AppendBits(dataBytes->At(x), 8); |
| } |
| } |
| for (int32_t y = 0; y < maxNumEcBytes; y++) { |
| for (size_t l = 0; l < blocks.size(); l++) { |
| const CBC_CommonByteArray* ecBytes = blocks[l].GetErrorCorrectionBytes(); |
| if (y < ecBytes->Size()) |
| result->AppendBits(ecBytes->At(y), 8); |
| } |
| } |
| return static_cast<size_t>(numTotalBytes) == result->sizeInBytes(); |
| } |
| |
| } // namespace |
| |
| CBC_QRCoderEncoder::CBC_QRCoderEncoder() {} |
| |
| CBC_QRCoderEncoder::~CBC_QRCoderEncoder() {} |
| |
| // static |
| bool CBC_QRCoderEncoder::Encode(const WideString& content, |
| const CBC_QRCoderErrorCorrectionLevel* ecLevel, |
| CBC_QRCoder* qrCode) { |
| ByteString encoding = "utf8"; |
| ByteString utf8Data; |
| CBC_UtilCodingConvert::UnicodeToUTF8(content, utf8Data); |
| CBC_QRCoderMode* mode = ChooseMode(utf8Data, encoding); |
| CBC_QRCoderBitVector dataBits; |
| int32_t e = BCExceptionNO; |
| AppendBytes(utf8Data, mode, &dataBits, encoding, e); |
| if (e != BCExceptionNO) |
| return false; |
| int32_t numInputBytes = dataBits.sizeInBytes(); |
| if (!InitQRCode(numInputBytes, ecLevel, mode, qrCode)) |
| return false; |
| CBC_QRCoderBitVector headerAndDataBits; |
| AppendModeInfo(mode, &headerAndDataBits); |
| int32_t numLetters = mode == CBC_QRCoderMode::sBYTE ? dataBits.sizeInBytes() |
| : content.GetLength(); |
| if (!AppendLengthInfo(numLetters, qrCode->GetVersion(), mode, |
| &headerAndDataBits)) { |
| return false; |
| } |
| headerAndDataBits.AppendBitVector(&dataBits); |
| if (!TerminateBits(qrCode->GetNumDataBytes(), &headerAndDataBits)) |
| return false; |
| CBC_QRCoderBitVector finalBits; |
| if (!InterleaveWithECBytes(&headerAndDataBits, qrCode->GetNumTotalBytes(), |
| qrCode->GetNumDataBytes(), |
| qrCode->GetNumRSBlocks(), &finalBits)) { |
| return false; |
| } |
| |
| auto matrix = pdfium::MakeUnique<CBC_CommonByteMatrix>( |
| qrCode->GetMatrixWidth(), qrCode->GetMatrixWidth()); |
| matrix->Init(); |
| int32_t maskPattern = ChooseMaskPattern( |
| &finalBits, qrCode->GetECLevel(), qrCode->GetVersion(), matrix.get(), e); |
| if (e != BCExceptionNO) |
| return false; |
| |
| qrCode->SetMaskPattern(maskPattern); |
| CBC_QRCoderMatrixUtil::BuildMatrix(&finalBits, qrCode->GetECLevel(), |
| qrCode->GetVersion(), |
| qrCode->GetMaskPattern(), matrix.get(), e); |
| if (e != BCExceptionNO) |
| return false; |
| |
| qrCode->SetMatrix(std::move(matrix)); |
| return qrCode->IsValid(); |
| } |
