ak/rxing
1
0
Fork 0
mirror of https://github.com/starovoid/rxing.git synced 2026-07-25 20:02:34 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
9a8df1931b4f70d119f078294987d5f6ca7da2e9
rxing/src/qrcode/cpp_port/test/QREncoderTest.cpp
Henry Schimke 8a1d0b3969 Beginning port of tests for qrcode from c++
2023-04-05 12:26:34 -05:00

658 lines
25 KiB
C++
Raw Blame History

/*
* Copyright 2008 ZXing authors
*/
// SPDX-License-Identifier: Apache-2.0
#include "BitArray.h"
#include "BitArrayUtility.h"
#include "BitMatrixIO.h"
#include "CharacterSet.h"
#include "TextDecoder.h"
#include "Utf.h"
#include "qrcode/QREncoder.h"
#include "qrcode/QRCodecMode.h"
#include "qrcode/QREncodeResult.h"
#include "qrcode/QRErrorCorrectionLevel.h"
#include "gtest/gtest.h"
namespace ZXing {
namespace QRCode {
int GetAlphanumericCode(int code);
CodecMode ChooseMode(const std::wstring& content, CharacterSet encoding);
void AppendModeInfo(CodecMode mode, BitArray& bits);
void AppendLengthInfo(int numLetters, const Version& version, CodecMode mode, BitArray& bits);
void AppendNumericBytes(const std::wstring& content, BitArray& bits);
void AppendAlphanumericBytes(const std::wstring& content, BitArray& bits);
void Append8BitBytes(const std::wstring& content, CharacterSet encoding, BitArray& bits);
void AppendKanjiBytes(const std::wstring& content, BitArray& bits);
void AppendBytes(const std::wstring& content, CodecMode mode, CharacterSet encoding, BitArray& bits);
void TerminateBits(int numDataBytes, BitArray& bits);
void GetNumDataBytesAndNumECBytesForBlockID(int numTotalBytes, int numDataBytes, int numRSBlocks, int blockID, int& numDataBytesInBlock, int& numECBytesInBlock);
void GenerateECBytes(const ByteArray& dataBytes, int numEcBytesInBlock, ByteArray& ecBytes);
BitArray InterleaveWithECBytes(const BitArray& bits, int numTotalBytes, int numDataBytes, int numRSBlocks);
}
}
using namespace ZXing;
using namespace ZXing::QRCode;
using namespace ZXing::Utility;
namespace {
std::wstring ShiftJISString(const std::vector<uint8_t>& bytes)
{
std::string str;
TextDecoder::Append(str, bytes.data(), bytes.size(), CharacterSet::Shift_JIS);
return FromUtf8(str);
}
std::string RemoveSpace(std::string s)
{
s.erase(std::remove(s.begin(), s.end(), ' '), s.end());
return s;
}
}
TEST(QREncoderTest, GetAlphanumericCode)
{
// The first ten code points are numbers.
for (int i = 0; i < 10; ++i) {
EXPECT_EQ(i, GetAlphanumericCode('0' + i));
}
// The next 26 code points are capital alphabet letters.
for (int i = 10; i < 36; ++i) {
EXPECT_EQ(i, GetAlphanumericCode('A' + i - 10));
}
// Others are symbol letters
EXPECT_EQ(36, GetAlphanumericCode(' '));
EXPECT_EQ(37, GetAlphanumericCode('$'));
EXPECT_EQ(38, GetAlphanumericCode('%'));
EXPECT_EQ(39, GetAlphanumericCode('*'));
EXPECT_EQ(40, GetAlphanumericCode('+'));
EXPECT_EQ(41, GetAlphanumericCode('-'));
EXPECT_EQ(42, GetAlphanumericCode('.'));
EXPECT_EQ(43, GetAlphanumericCode('/'));
EXPECT_EQ(44, GetAlphanumericCode(':'));
// Should return -1 for other letters;
EXPECT_EQ(-1, GetAlphanumericCode('a'));
EXPECT_EQ(-1, GetAlphanumericCode('#'));
EXPECT_EQ(-1, GetAlphanumericCode('\0'));
}
TEST(QREncoderTest, ChooseMode)
{
// Numeric mode.
EXPECT_EQ(CodecMode::NUMERIC, ChooseMode(L"0", CharacterSet::Unknown));
EXPECT_EQ(CodecMode::NUMERIC, ChooseMode(L"0123456789", CharacterSet::Unknown));
// Alphanumeric mode.
EXPECT_EQ(CodecMode::ALPHANUMERIC, ChooseMode(L"A", CharacterSet::Unknown));
EXPECT_EQ(CodecMode::ALPHANUMERIC,
ChooseMode(L"0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ $%*+-./:", CharacterSet::Unknown));
// 8-bit byte mode.
EXPECT_EQ(CodecMode::BYTE, ChooseMode(L"a", CharacterSet::Unknown));
EXPECT_EQ(CodecMode::BYTE, ChooseMode(L"#", CharacterSet::Unknown));
EXPECT_EQ(CodecMode::BYTE, ChooseMode(L"", CharacterSet::Unknown));
// Kanji mode. We used to use MODE_KANJI for these, but we stopped
// doing that as we cannot distinguish Shift_JIS from other encodings
// from data bytes alone. See also comments in qrcode_encoder.h.
// AIUE in Hiragana in Shift_JIS
EXPECT_EQ(CodecMode::BYTE,
ChooseMode(ShiftJISString({0x8, 0xa, 0x8, 0xa, 0x8, 0xa, 0x8, 0xa6}), CharacterSet::Unknown));
// Nihon in Kanji in Shift_JIS.
EXPECT_EQ(CodecMode::BYTE, ChooseMode(ShiftJISString({0x9, 0xf, 0x9, 0x7b}), CharacterSet::Unknown));
// Sou-Utsu-Byou in Kanji in Shift_JIS.
EXPECT_EQ(CodecMode::BYTE, ChooseMode(ShiftJISString({0xe, 0x4, 0x9, 0x5, 0x9, 0x61}), CharacterSet::Unknown));
}
TEST(QREncoderTest, Encode)
{
auto qrCode = Encode(L"ABCDEF", ErrorCorrectionLevel::High, CharacterSet::Unknown, 0, false, -1);
EXPECT_EQ(qrCode.mode, CodecMode::ALPHANUMERIC);
EXPECT_EQ(qrCode.ecLevel, ErrorCorrectionLevel::High);
ASSERT_NE(qrCode.version, nullptr);
EXPECT_EQ(qrCode.version->versionNumber(), 1);
EXPECT_EQ(qrCode.maskPattern, 4);
EXPECT_EQ(ToString(qrCode.matrix),
"X X X X X X X X X X X X X X X X \n"
"X X X X X X X \n"
"X X X X X X X X X X \n"
"X X X X X X X X X X X X \n"
"X X X X X X X X X X X X \n"
"X X X X X X X \n"
"X X X X X X X X X X X X X X X X X \n"
" X X \n"
" X X X X X X X X X X \n"
" X X X X X X X X X X X X \n"
"X X X X X X X X X X \n"
"X X X X X X X X X \n"
" X X X X X X X X X X X X X \n"
" X X X X X X \n"
"X X X X X X X X X X X X X \n"
"X X X X X X X X X X \n"
"X X X X X X X X X X X \n"
"X X X X X X X X X X X \n"
"X X X X X X X X X \n"
"X X X X X X X \n"
"X X X X X X X X X X X X \n");
}
TEST(QREncoderTest, EncodeWithVersion)
{
auto qrCode = Encode(L"ABCDEF", ErrorCorrectionLevel::High, CharacterSet::Unknown, 7, false, -1);
ASSERT_NE(qrCode.version, nullptr);
EXPECT_EQ(qrCode.version->versionNumber(), 7);
}
TEST(QREncoderTest, EncodeWithVersionTooSmall)
{
EXPECT_THROW(
Encode(L"THISMESSAGEISTOOLONGFORAQRCODEVERSION3", ErrorCorrectionLevel::High, CharacterSet::Unknown, 3, false, -1)
, std::invalid_argument);
}
TEST(QREncoderTest, SimpleUTF8ECI)
{
auto qrCode = Encode(L"hello", ErrorCorrectionLevel::High, CharacterSet::UTF8, 0, false, -1);
EXPECT_EQ(qrCode.mode, CodecMode::BYTE);
EXPECT_EQ(qrCode.ecLevel, ErrorCorrectionLevel::High);
ASSERT_NE(qrCode.version, nullptr);
EXPECT_EQ(qrCode.version->versionNumber(), 1);
EXPECT_EQ(qrCode.maskPattern, 6);
EXPECT_EQ(ToString(qrCode.matrix), // break the line comment
"X X X X X X X X X X X X X X X X \n"
"X X X X X X \n"
"X X X X X X X X X X X X X \n"
"X X X X X X X X X X X X \n"
"X X X X X X X X X X X X \n"
"X X X X X \n"
"X X X X X X X X X X X X X X X X X \n"
" X X X X \n"
" X X X X X X X \n"
" X X X X X X X X X \n"
"X X X X X X X X X X X \n"
" X X X X X X \n"
" X X X X X X X X X X X X X X X \n"
" X X X X X X X X X X X X \n"
"X X X X X X X X X X \n"
"X X X X X X \n"
"X X X X X X X X X \n"
"X X X X X X X X X X X X X \n"
"X X X X X X X X X X X X \n"
"X X X X X X \n"
"X X X X X X X X X X X \n");
}
TEST(QREncoderTest, SimpleBINARYECI)
{
auto qrCode = Encode(L"\u00E9", ErrorCorrectionLevel::High, CharacterSet::BINARY, 0, false, -1);
EXPECT_EQ(qrCode.mode, CodecMode::BYTE);
EXPECT_EQ(qrCode.ecLevel, ErrorCorrectionLevel::High);
ASSERT_NE(qrCode.version, nullptr);
EXPECT_EQ(qrCode.version->versionNumber(), 1);
EXPECT_EQ(qrCode.maskPattern, 6);
EXPECT_EQ(ToString(qrCode.matrix),
"X X X X X X X X X X X X X X X X X X \n"
"X X X X X \n"
"X X X X X X X X X X X X X \n"
"X X X X X X X X X X X X X X X \n"
"X X X X X X X X X X X \n"
"X X X X X X \n"
"X X X X X X X X X X X X X X X X X \n"
" X X X \n"
" X X X X X X X \n"
"X X X X X X X X \n"
"X X X X X X X X X X X X X X X \n"
"X X X X X X X X X X X X \n"
" X X X X X X X X X X \n"
" X X X X X X X X \n"
"X X X X X X X X X X X X X X X \n"
"X X X X X X X \n"
"X X X X X X X X X X X X X \n"
"X X X X X X X X X X X X X \n"
"X X X X X X X X X X \n"
"X X X X X X X \n"
"X X X X X X X X X X X X X \n");
}
TEST(QREncoderTest, EncodeKanjiMode)
{
auto qrCode = Encode(L"\u65e5\u672c", ErrorCorrectionLevel::Medium, CharacterSet::Shift_JIS, 0, false, -1);
EXPECT_EQ(qrCode.mode, CodecMode::KANJI);
EXPECT_EQ(qrCode.ecLevel, ErrorCorrectionLevel::Medium);
ASSERT_NE(qrCode.version, nullptr);
EXPECT_EQ(qrCode.version->versionNumber(), 1);
EXPECT_EQ(qrCode.maskPattern, 0);
EXPECT_EQ(ToString(qrCode.matrix),
"X X X X X X X X X X X X X X X X \n"
"X X X X X X \n"
"X X X X X X X X X X X X X X \n"
"X X X X X X X X X X X \n"
"X X X X X X X X X X X X X X X \n"
"X X X X X X X \n"
"X X X X X X X X X X X X X X X X X \n"
" X \n"
"X X X X X X X X \n"
"X X X X X X X X X X \n"
" X X X X X X X X X X X X \n"
"X X X X X X X X X X X \n"
" X X X X X X X X X X X X \n"
" X X X X X \n"
"X X X X X X X X X X X \n"
"X X X X X X X \n"
"X X X X X X X X X X X \n"
"X X X X X X X X X X \n"
"X X X X X X X X X X X X X \n"
"X X X X X X X X X \n"
"X X X X X X X X X X X X X X \n");
}
TEST(QREncoderTest, EncodeShiftjisNumeric)
{
auto qrCode = Encode(L"0123", ErrorCorrectionLevel::Medium, CharacterSet::Shift_JIS, 0, false, -1);
EXPECT_EQ(qrCode.mode, CodecMode::NUMERIC);
EXPECT_EQ(qrCode.ecLevel, ErrorCorrectionLevel::Medium);
ASSERT_NE(qrCode.version, nullptr);
EXPECT_EQ(qrCode.version->versionNumber(), 1);
EXPECT_EQ(qrCode.maskPattern, 2);
EXPECT_EQ(ToString(qrCode.matrix),
"X X X X X X X X X X X X X X X X X \n"
"X X X X X X \n"
"X X X X X X X X X X X \n"
"X X X X X X X X X X X X X X \n"
"X X X X X X X X X X X X X X \n"
"X X X X X X X \n"
"X X X X X X X X X X X X X X X X X \n"
" X X X X X \n"
"X X X X X X X X X X X X X X \n"
"X X X X X X X X \n"
" X X X X X X X X X X X X X X \n"
"X X X X X X X X X \n"
" X X X X X X X X \n"
" X X X X X X X \n"
"X X X X X X X X X X X \n"
"X X X X X X X X X X \n"
"X X X X X X X X X X \n"
"X X X X X X X X X X X \n"
"X X X X X X X X X X X X \n"
"X X X X X X X \n"
"X X X X X X X X X X X X X X \n");
}
TEST(QREncoderTest, EncodeGS1)
{
auto qrCode = Encode(L"100001%11171218", ErrorCorrectionLevel::High, CharacterSet::Unknown, 0, true, -1);
EXPECT_EQ(qrCode.mode, CodecMode::ALPHANUMERIC);
EXPECT_EQ(qrCode.ecLevel, ErrorCorrectionLevel::High);
ASSERT_NE(qrCode.version, nullptr);
EXPECT_EQ(qrCode.version->versionNumber(), 2);
EXPECT_EQ(qrCode.maskPattern, 4);
EXPECT_EQ(ToString(qrCode.matrix),
"X X X X X X X X X X X X X X X X X X X X \n"
"X X X X X X X X \n"
"X X X X X X X X X X X X X X \n"
"X X X X X X X X X X X X X X \n"
"X X X X X X X X X X X X X X \n"
"X X X X X X X X X X \n"
"X X X X X X X X X X X X X X X X X X X \n"
" X X X X X X \n"
" X X X X X X X X X X X \n"
" X X X X X X X X X X X X X X X \n"
" X X X X X X X X X X X X X X \n"
"X X X X X X X X X X X X X X X X X \n"
" X X X X X X X X X X X X \n"
"X X X X X X X X X X X \n"
" X X X X X X X X X X X X X X X X \n"
" X X X X X X X X X X \n"
"X X X X X X X X X X X X X X \n"
" X X X X X X X X \n"
"X X X X X X X X X X X X X X X \n"
"X X X X X X X X X X \n"
"X X X X X X X X X X X X X X X \n"
"X X X X X X X X X X X \n"
"X X X X X X X X X X X X X X \n"
"X X X X X X X X X X X X \n"
"X X X X X X X X X X X X X X \n");
}
TEST(QREncoderTest, EncodeGS1ModeHeaderWithECI)
{
auto qrCode = Encode(L"hello", ErrorCorrectionLevel::High, CharacterSet::UTF8, 0, true, -1);
EXPECT_EQ(qrCode.mode, CodecMode::BYTE);
EXPECT_EQ(qrCode.ecLevel, ErrorCorrectionLevel::High);
ASSERT_NE(qrCode.version, nullptr);
EXPECT_EQ(qrCode.version->versionNumber(), 1);
EXPECT_EQ(qrCode.maskPattern, 5);
EXPECT_EQ(ToString(qrCode.matrix),
"X X X X X X X X X X X X X X X X X \n"
"X X X X X X \n"
"X X X X X X X X X X X X X \n"
"X X X X X X X X X X X X \n"
"X X X X X X X X X X X X \n"
"X X X X X X X X \n"
"X X X X X X X X X X X X X X X X X \n"
" X X X X \n"
" X X X X X X X X \n"
" X X X X X X X X X X X X X X \n"
" X X X X X X X X X X X \n"
"X X X X X X X X X X X X \n"
"X X X X X X X X X X \n"
" X X X X X X X X \n"
"X X X X X X X X X X X X \n"
"X X X X X X X X \n"
"X X X X X X X X X X \n"
"X X X X X X X X X X X X \n"
"X X X X X X X X X X X \n"
"X X X X X X X X X \n"
"X X X X X X X X X X X X X X \n");
}
TEST(QREncoderTest, AppendModeInfo)
{
BitArray bits;
AppendModeInfo(CodecMode::NUMERIC, bits);
EXPECT_EQ(ToString(bits), "...X");
}
TEST(QREncoderTest, AppendLengthInfo)
{
BitArray bits;
AppendLengthInfo(1, // 1 letter (1/1).
*Version::FromNumber(1), CodecMode::NUMERIC, bits);
EXPECT_EQ(ToString(bits), RemoveSpace("........ .X")); // 10 bits.
bits = BitArray();
AppendLengthInfo(2, // 2 letters (2/1).
*Version::FromNumber(10), CodecMode::ALPHANUMERIC, bits);
EXPECT_EQ(ToString(bits), RemoveSpace("........ .X.")); // 11 bits.
bits = BitArray();
AppendLengthInfo(255, // 255 letter (255/1).
*Version::FromNumber(27), CodecMode::BYTE, bits);
EXPECT_EQ(ToString(bits), RemoveSpace("........ XXXXXXXX")); // 16 bits.
bits = BitArray();
AppendLengthInfo(512, // 512 letters (1024/2).
*Version::FromNumber(40), CodecMode::KANJI, bits);
EXPECT_EQ(ToString(bits), RemoveSpace("..X..... ....")); // 12 bits.
}
TEST(QREncoderTest, AppendNumericBytes)
{
// 1 = 01 = 0001 in 4 bits.
BitArray bits;
AppendNumericBytes(L"1", bits);
EXPECT_EQ(ToString(bits), RemoveSpace("...X"));
// 12 = 0xc = 0001100 in 7 bits.
bits = BitArray();
AppendNumericBytes(L"12", bits);
EXPECT_EQ(ToString(bits), RemoveSpace("...XX.."));
// 123 = 0x7b = 0001111011 in 10 bits.
bits = BitArray();
AppendNumericBytes(L"123", bits);
EXPECT_EQ(ToString(bits), RemoveSpace("...XXXX. XX"));
// 1234 = "123" + "4" = 0001111011 + 0100
bits = BitArray();
AppendNumericBytes(L"1234", bits);
EXPECT_EQ(ToString(bits), RemoveSpace("...XXXX. XX.X.."));
// Empty.
bits = BitArray();
AppendNumericBytes(L"", bits);
EXPECT_EQ(ToString(bits), RemoveSpace(""));
}
TEST(QREncoderTest, AppendAlphanumericBytes)
{
// A = 10 = 0xa = 001010 in 6 bits
BitArray bits;
AppendAlphanumericBytes(L"A", bits);
EXPECT_EQ(ToString(bits), RemoveSpace("..X.X."));
// AB = 10 * 45 + 11 = 461 = 0x1cd = 00111001101 in 11 bits
bits = BitArray();
AppendAlphanumericBytes(L"AB", bits);
EXPECT_EQ(ToString(bits), RemoveSpace("..XXX..X X.X"));
// ABC = "AB" + "C" = 00111001101 + 001100
bits = BitArray();
AppendAlphanumericBytes(L"ABC", bits);
EXPECT_EQ(ToString(bits), RemoveSpace("..XXX..X X.X..XX. ."));
// Empty.
bits = BitArray();
AppendAlphanumericBytes(L"", bits);
EXPECT_EQ(ToString(bits), RemoveSpace(""));
// Invalid data.
EXPECT_THROW(AppendAlphanumericBytes(L"abc", bits), std::invalid_argument);
}
TEST(QREncoderTest, Append8BitBytes)
{
// 0x61, 0x62, 0x63
BitArray bits;
Append8BitBytes(L"abc", CharacterSet::Unknown, bits);
EXPECT_EQ(ToString(bits), RemoveSpace(".XX....X .XX...X. .XX...XX"));
// Empty.
bits = BitArray();
Append8BitBytes(L"", CharacterSet::Unknown, bits);
EXPECT_EQ(ToString(bits), RemoveSpace(""));
}
// Numbers are from page 21 of JISX0510:2004
TEST(QREncoderTest, AppendKanjiBytes)
{
BitArray bits;
AppendKanjiBytes(ShiftJISString({ 0x93, 0x5f }), bits);
EXPECT_EQ(ToString(bits), RemoveSpace(".XX.XX.. XXXXX"));
AppendKanjiBytes(ShiftJISString({ 0xe4, 0xaa }), bits);
EXPECT_EQ(ToString(bits), RemoveSpace(".XX.XX.. XXXXXXX. X.X.X.X. X."));
}
TEST(QREncoderTest, AppendBytes)
{
// Should use appendNumericBytes.
// 1 = 01 = 0001 in 4 bits.
BitArray bits;
AppendBytes(L"1", CodecMode::NUMERIC, CharacterSet::Unknown, bits);
EXPECT_EQ(ToString(bits), RemoveSpace("...X"));
// Should use appendAlphanumericBytes.
// A = 10 = 0xa = 001010 in 6 bits
bits = BitArray();
AppendBytes(L"A", CodecMode::ALPHANUMERIC, CharacterSet::Unknown, bits);
EXPECT_EQ(ToString(bits), RemoveSpace("..X.X."));
// Lower letters such as 'a' cannot be encoded in MODE_ALPHANUMERIC.
bits = BitArray();
EXPECT_THROW(AppendBytes(L"a", CodecMode::ALPHANUMERIC, CharacterSet::Unknown, bits), std::invalid_argument);
// Should use append8BitBytes.
// 0x61, 0x62, 0x63
bits = BitArray();
AppendBytes(L"abc", CodecMode::BYTE, CharacterSet::Unknown, bits);
EXPECT_EQ(ToString(bits), RemoveSpace(".XX....X .XX...X. .XX...XX"));
// Anything can be encoded in QRCode.MODE_8BIT_BYTE.
AppendBytes(L"\0", CodecMode::BYTE, CharacterSet::Unknown, bits);
// Should use appendKanjiBytes.
// 0x93, 0x5f
bits = BitArray();
AppendBytes(ShiftJISString({0x93, 0x5f}), CodecMode::KANJI, CharacterSet::Unknown, bits);
EXPECT_EQ(ToString(bits), RemoveSpace(".XX.XX.. XXXXX"));
}
TEST(QREncoderTest, TerminateBits)
{
BitArray v;
TerminateBits(0, v);
EXPECT_EQ(ToString(v), RemoveSpace(""));
v = BitArray();
TerminateBits(1, v);
EXPECT_EQ(ToString(v), RemoveSpace("........"));
v = BitArray();
v.appendBits(0, 3);
TerminateBits(1, v);
EXPECT_EQ(ToString(v), RemoveSpace("........"));
v = BitArray();
v.appendBits(0, 5);
TerminateBits(1, v);
EXPECT_EQ(ToString(v), RemoveSpace("........"));
v = BitArray();
v.appendBits(0, 8);
TerminateBits(1, v);
EXPECT_EQ(ToString(v), RemoveSpace("........"));
v = BitArray();
TerminateBits(2, v);
EXPECT_EQ(ToString(v), RemoveSpace("........ XXX.XX.."));
v = BitArray();
v.appendBits(0, 1);
TerminateBits(3, v);
EXPECT_EQ(ToString(v), RemoveSpace("........ XXX.XX.. ...X...X"));
}
TEST(QREncoderTest, GetNumDataBytesAndNumECBytesForBlockID)
{
int numDataBytes;
int numEcBytes;
// Version 1-H.
GetNumDataBytesAndNumECBytesForBlockID(26, 9, 1, 0, numDataBytes, numEcBytes);
EXPECT_EQ(9, numDataBytes);
EXPECT_EQ(17, numEcBytes);
// Version 3-H. 2 blocks.
GetNumDataBytesAndNumECBytesForBlockID(70, 26, 2, 0, numDataBytes, numEcBytes);
EXPECT_EQ(13, numDataBytes);
EXPECT_EQ(22, numEcBytes);
GetNumDataBytesAndNumECBytesForBlockID(70, 26, 2, 1, numDataBytes, numEcBytes);
EXPECT_EQ(13, numDataBytes);
EXPECT_EQ(22, numEcBytes);
// Version 7-H. (4 + 1) blocks.
GetNumDataBytesAndNumECBytesForBlockID(196, 66, 5, 0, numDataBytes, numEcBytes);
EXPECT_EQ(13, numDataBytes);
EXPECT_EQ(26, numEcBytes);
GetNumDataBytesAndNumECBytesForBlockID(196, 66, 5, 4, numDataBytes, numEcBytes);
EXPECT_EQ(14, numDataBytes);
EXPECT_EQ(26, numEcBytes);
// Version 40-H. (20 + 61) blocks.
GetNumDataBytesAndNumECBytesForBlockID(3706, 1276, 81, 0, numDataBytes, numEcBytes);
EXPECT_EQ(15, numDataBytes);
EXPECT_EQ(30, numEcBytes);
GetNumDataBytesAndNumECBytesForBlockID(3706, 1276, 81, 20, numDataBytes, numEcBytes);
EXPECT_EQ(16, numDataBytes);
EXPECT_EQ(30, numEcBytes);
GetNumDataBytesAndNumECBytesForBlockID(3706, 1276, 81, 80, numDataBytes, numEcBytes);
EXPECT_EQ(16, numDataBytes);
EXPECT_EQ(30, numEcBytes);
}
// Numbers are from http://www.swetake.com/qr/qr3.html and
// http://www.swetake.com/qr/qr9.html
TEST(QREncoderTest, GenerateECBytes)
{
ByteArray ecBytes;
GenerateECBytes({ 32, 65, 205, 69, 41, 220, 46, 128, 236 }, 17, ecBytes);
ByteArray expected = { 42, 159, 74, 221, 244, 169, 239, 150, 138, 70, 237, 85, 224, 96, 74, 219, 61 };
EXPECT_EQ(ecBytes, expected);
GenerateECBytes({ 67, 70, 22, 38, 54, 70, 86, 102, 118, 134, 150, 166, 182, 198, 214 }, 18, ecBytes);
expected = { 175, 80, 155, 64, 178, 45, 214, 233, 65, 209, 12, 155, 117, 31, 140, 214, 27, 187 };
EXPECT_EQ(ecBytes, expected);
// High-order zero coefficient case.
GenerateECBytes({ 32, 49, 205, 69, 42, 20, 0, 236, 17 }, 17, ecBytes);
expected = { 0, 3, 130, 179, 194, 0, 55, 211, 110, 79, 98, 72, 170, 96, 211, 137, 213 };
EXPECT_EQ(ecBytes, expected);
}
TEST(QREncoderTest, InterleaveWithECBytes)
{
BitArray in;
for (int dataByte : {32, 65, 205, 69, 41, 220, 46, 128, 236})
in.appendBits(dataByte, 8);
BitArray out = InterleaveWithECBytes(in, 26, 9, 1);
std::vector<uint8_t> expected = {
32, 65, 205, 69, 41, 220, 46, 128, 236,
42, 159, 74, 221, 244, 169, 239, 150, 138, 70, 237, 85, 224, 96, 74, 219, 61,
}; // Error correction bytes in second block
ASSERT_EQ(Size(expected), out.sizeInBytes());
EXPECT_EQ(expected, out.toBytes());
// Numbers are from http://www.swetake.com/qr/qr8.html
in = BitArray();
for (int dataByte :
{67, 70, 22, 38, 54, 70, 86, 102, 118, 134, 150, 166, 182, 198, 214, 230, 247, 7, 23, 39, 55,
71, 87, 103, 119, 135, 151, 166, 22, 38, 54, 70, 86, 102, 118, 134, 150, 166, 182, 198, 214, 230,
247, 7, 23, 39, 55, 71, 87, 103, 119, 135, 151, 160, 236, 17, 236, 17, 236, 17, 236, 17})
in.appendBits(dataByte, 8);
out = InterleaveWithECBytes(in, 134, 62, 4);
expected = {
67, 230, 54, 55, 70, 247, 70, 71, 22, 7, 86, 87, 38, 23, 102, 103, 54, 39, 118, 119, 70,
55, 134, 135, 86, 71, 150, 151, 102, 87, 166, 160, 118, 103, 182, 236, 134, 119, 198, 17, 150, 135,
214, 236, 166, 151, 230, 17, 182, 166, 247, 236, 198, 22, 7, 17, 214, 38, 23, 236, 39, 17,
175, 155, 245, 236, 80, 146, 56, 74, 155, 165, 133, 142, 64, 183, 132, 13, 178, 54, 132, 108, 45,
113, 53, 50, 214, 98, 193, 152, 233, 147, 50, 71, 65, 190, 82, 51, 209, 199, 171, 54, 12, 112,
57, 113, 155, 117, 211, 164, 117, 30, 158, 225, 31, 190, 242, 38, 140, 61, 179, 154, 214, 138, 147,
87, 27, 96, 77, 47, 187, 49, 156, 214,
}; // Error correction bytes in second block
EXPECT_EQ(Size(expected), out.sizeInBytes());
EXPECT_EQ(expected, out.toBytes());
}
TEST(QREncoderTest, BugInBitVectorNumBytes)
{
// There was a bug in BitVector.sizeInBytes() that caused it to return a
// smaller-by-one value (ex. 1465 instead of 1466) if the number of bits
// in the vector is not 8-bit aligned. In QRCodeEncoder::InitQRCode(),
// BitVector::sizeInBytes() is used for finding the smallest QR Code
// version that can fit the given data. Hence there were corner cases
// where we chose a wrong QR Code version that cannot fit the given
// data. Note that the issue did not occur with MODE_8BIT_BYTE, as the
// bits in the bit vector are always 8-bit aligned.
//
// Before the bug was fixed, the following test didn't pass, because:
//
// - MODE_NUMERIC is chosen as all bytes in the data are '0'
// - The 3518-byte numeric data needs 1466 bytes
// - 3518 / 3 * 10 + 7 = 11727 bits = 1465.875 bytes
// - 3 numeric bytes are encoded in 10 bits, hence the first
// 3516 bytes are encoded in 3516 / 3 * 10 = 11720 bits.
// - 2 numeric bytes can be encoded in 7 bits, hence the last
// 2 bytes are encoded in 7 bits.
// - The version 27 QR Code with the EC level L has 1468 bytes for data.
// - 1828 - 360 = 1468
// - In InitQRCode(), 3 bytes are reserved for a header. Hence 1465 bytes
// (1468 -3) are left for data.
// - Because of the bug in BitVector::sizeInBytes(), InitQRCode() determines
// the given data can fit in 1465 bytes, despite it needs 1466 bytes.
// - Hence QRCodeEncoder.encode() failed and returned false.
// - To be precise, it needs 11727 + 4 (getMode info) + 14 (length info) =
// 11745 bits = 1468.125 bytes are needed (i.e. cannot fit in 1468
// bytes).
Encode(std::wstring(3518, L'0'), ErrorCorrectionLevel::Low, CharacterSet::Unknown, 0, false, -1);
}
Reference in New Issue View Git Blame Copy Permalink