1
0
mirror of https://github.com/opencv/opencv.git synced 2026-07-29 07:13:02 +04:00

Merge pull request #24299 from dkurt:qrcode_decode

In-house QR codes decoding #24299

### Pull Request Readiness Checklist

QR codes decoding pipeline without 3rdparty dependency (Quirc library). Implemented according to standard https://github.com/yansikeim/QR-Code/blob/master/ISO%20IEC%2018004%202015%20Standard.pdf

**Merge with extra**: https://github.com/opencv/opencv_extra/pull/1124

resolves https://github.com/opencv/opencv/issues/24225
resolves https://github.com/opencv/opencv/issues/17290
resolves https://github.com/opencv/opencv/issues/24318 https://github.com/opencv/opencv/issues/24346

Resources:
* https://en.wikiversity.org/wiki/Reed%E2%80%93Solomon_codes_for_coders
* https://en.wikipedia.org/wiki/Berlekamp%E2%80%93Massey_algorithm

```
Geometric mean (ms)

                                   Name of Test                                      quirc   new2      new2   
                                                                                                        vs    
                                                                                                      quirc   
                                                                                                    (x-factor)
decode::Perf_Objdetect_Not_QRCode::("chessboard", 640x480)                           9.151   9.157     1.00   
decode::Perf_Objdetect_Not_QRCode::("chessboard", 1280x720)                         21.609  21.609     1.00   
decode::Perf_Objdetect_Not_QRCode::("chessboard", 1920x1080)                        42.088  41.924     1.00   
decode::Perf_Objdetect_Not_QRCode::("chessboard", 3840x2160)                        169.737 169.050    1.00   
decode::Perf_Objdetect_Not_QRCode::("random", 640x480)                               8.552   8.611     0.99   
decode::Perf_Objdetect_Not_QRCode::("random", 1280x720)                             21.264  21.581     0.99   
decode::Perf_Objdetect_Not_QRCode::("random", 1920x1080)                            42.415  43.468     0.98   
decode::Perf_Objdetect_Not_QRCode::("random", 3840x2160)                            175.003 174.294    1.00   
decode::Perf_Objdetect_Not_QRCode::("zero", 640x480)                                 8.528   8.421     1.01   
decode::Perf_Objdetect_Not_QRCode::("zero", 1280x720)                               21.548  21.209     1.02   
decode::Perf_Objdetect_Not_QRCode::("zero", 1920x1080)                              42.581  42.529     1.00   
decode::Perf_Objdetect_Not_QRCode::("zero", 3840x2160)                              176.231 174.410    1.01   
decode::Perf_Objdetect_QRCode::"kanji.jpg"                                           6.105   6.072     1.01   
decode::Perf_Objdetect_QRCode::"link_github_ocv.jpg"                                 6.069   6.076     1.00   
decode::Perf_Objdetect_QRCode::"link_ocv.jpg"                                        6.143   6.240     0.98   
decode::Perf_Objdetect_QRCode::"link_wiki_cv.jpg"                                    6.369   6.420     0.99   
decode::Perf_Objdetect_QRCode::"russian.jpg"                                         6.558   6.549     1.00   
decode::Perf_Objdetect_QRCode::"version_1_down.jpg"                                  5.634   5.621     1.00   
decode::Perf_Objdetect_QRCode::"version_1_left.jpg"                                  5.560   5.609     0.99   
decode::Perf_Objdetect_QRCode::"version_1_right.jpg"                                 5.539   5.631     0.98   
decode::Perf_Objdetect_QRCode::"version_1_top.jpg"                                   5.622   5.566     1.01   
decode::Perf_Objdetect_QRCode::"version_1_up.jpg"                                    5.569   5.534     1.01   
decode::Perf_Objdetect_QRCode::"version_5_down.jpg"                                  6.514   6.436     1.01   
decode::Perf_Objdetect_QRCode::"version_5_left.jpg"                                  6.668   6.479     1.03   
decode::Perf_Objdetect_QRCode::"version_5_top.jpg"                                   6.481   6.484     1.00   
decode::Perf_Objdetect_QRCode::"version_5_up.jpg"                                    7.011   6.513     1.08   
decodeMulti::Perf_Objdetect_QRCode_Multi::("2_qrcodes.png", "aruco_based")          14.885  15.089     0.99   
decodeMulti::Perf_Objdetect_QRCode_Multi::("2_qrcodes.png", "contours_based")       14.896  14.906     1.00   
decodeMulti::Perf_Objdetect_QRCode_Multi::("3_close_qrcodes.png", "aruco_based")     6.661   6.663     1.00   
decodeMulti::Perf_Objdetect_QRCode_Multi::("3_close_qrcodes.png", "contours_based")  6.614   6.592     1.00   
decodeMulti::Perf_Objdetect_QRCode_Multi::("3_qrcodes.png", "aruco_based")          14.814  14.592     1.02   
decodeMulti::Perf_Objdetect_QRCode_Multi::("3_qrcodes.png", "contours_based")       15.245  15.135     1.01   
decodeMulti::Perf_Objdetect_QRCode_Multi::("4_qrcodes.png", "aruco_based")          10.923  10.881     1.00   
decodeMulti::Perf_Objdetect_QRCode_Multi::("4_qrcodes.png", "contours_based")       10.680  10.128     1.05   
decodeMulti::Perf_Objdetect_QRCode_Multi::("5_qrcodes.png", "contours_based")       11.788  11.576     1.02   
decodeMulti::Perf_Objdetect_QRCode_Multi::("6_qrcodes.png", "aruco_based")          25.887  25.979     1.00   
decodeMulti::Perf_Objdetect_QRCode_Multi::("6_qrcodes.png", "contours_based")       26.183  25.627     1.02   
decodeMulti::Perf_Objdetect_QRCode_Multi::("7_qrcodes.png", "aruco_based")          32.786  32.253     1.02   
decodeMulti::Perf_Objdetect_QRCode_Multi::("7_qrcodes.png", "contours_based")       24.290  24.435     0.99   
decodeMulti::Perf_Objdetect_QRCode_Multi::("8_close_qrcodes.png", "aruco_based")    89.696  89.247     1.01   
decodeMulti::Perf_Objdetect_QRCode_Multi::("8_close_qrcodes.png", "contours_based") 89.872  89.600     1.00
```
This commit is contained in:
Dmitry Kurtaev
2023-11-24 11:35:36 +03:00
committed by GitHub
parent 332748dd55
commit d296d29a1c
10 changed files with 656 additions and 151 deletions
+537 -34
View File
@@ -6,6 +6,8 @@
#include "precomp.hpp"
#include "qrcode_encoder_table.inl.hpp"
#include "graphical_code_detector_impl.hpp"
namespace cv
{
using std::vector;
@@ -19,6 +21,7 @@ const uint8_t INVALID_REGION_VALUE = 110;
static void decToBin(const int dec_number, const int total_bits, std::vector<uint8_t> &bin_number);
static uint8_t gfPow(uint8_t x, int power);
static uint8_t gfMul(const uint8_t x, const uint8_t y);
static uint8_t gfDiv(const uint8_t x, const uint8_t y);
static void gfPolyMul(const vector<uint8_t> &p, const vector<uint8_t> &q, vector<uint8_t> &product);
static void gfPolyDiv(const vector<uint8_t> &dividend, const vector<uint8_t> &divisor, const int ecc_num, vector<uint8_t> &quotient);
static void polyGenerator(const int n, vector<uint8_t> &result);
@@ -51,6 +54,13 @@ static uint8_t gfMul(const uint8_t x, const uint8_t y)
return gf_exp[(gf_log[x] + gf_log[y]) % 255];
}
static uint8_t gfDiv(const uint8_t x, const uint8_t y)
{
if (x == 0 || y == 0)
return 0;
return gf_exp[(gf_log[x] + 255 - gf_log[y]) % 255];
}
static void gfPolyMul(const vector<uint8_t> &p, const vector<uint8_t> &q, vector<uint8_t> &product)
{
int len_p = (int)p.size();
@@ -141,6 +151,8 @@ static int mapSymbol(char c)
return -1;
}
static void maskData(const Mat& original, const int mask_type_num, Mat &masked);
QRCodeEncoder::QRCodeEncoder()
{
// nothing
@@ -221,7 +233,6 @@ protected:
void formatGenerate(const int mask_type_num, vector<uint8_t> &format_array);
void versionInfoGenerate(const int version_level_num, vector<uint8_t> &version_array);
void fillReserved(const vector<uint8_t> &format_array, Mat &masked);
void maskData(const int mask_type_num, Mat &masked);
void findAutoMaskType();
bool estimateVersion(const int input_length, EncodeMode mode, vector<int> &possible_version);
int versionAuto(const std::string &input_str);
@@ -387,36 +398,10 @@ void QRCodeEncoderImpl::generateQR(const std::string &input)
void QRCodeEncoderImpl::formatGenerate(const int mask_type_num, vector<uint8_t> &format_array)
{
const int mask_bits_num = 3;
const int level_bits_num = 2;
std::vector<uint8_t> mask_type_bin(mask_bits_num);
std::vector<uint8_t> ec_level_bin(level_bits_num);
decToBin(mask_type_num, mask_bits_num, mask_type_bin);
decToBin(eccLevelToCode(ecc_level), level_bits_num, ec_level_bin);
std::vector<uint8_t> format_bits;
hconcat(ec_level_bin, mask_type_bin, format_bits);
std::reverse(format_bits.begin(), format_bits.end());
const int ecc_info_bits = 10;
std::vector<uint8_t> shift(ecc_info_bits, 0);
std::vector<uint8_t> polynomial;
hconcat(shift, format_bits, polynomial);
const int generator_len = 11;
const uint8_t generator_arr[generator_len] = {1, 1, 1, 0, 1, 1, 0, 0, 1, 0, 1};
std::vector<uint8_t> format_generator (generator_arr, generator_arr + sizeof(generator_arr) / sizeof(generator_arr[0]));
vector<uint8_t> ecc_code;
gfPolyDiv(polynomial, format_generator, ecc_info_bits, ecc_code);
hconcat(ecc_code, format_bits, format_array);
const uint8_t mask_arr[MAX_FORMAT_LENGTH] = {0, 1, 0, 0, 1, 0, 0, 0, 0, 0, 1, 0, 1, 0, 1};
std::vector<uint8_t> system_mask (mask_arr, mask_arr + sizeof(mask_arr) / sizeof(mask_arr[0]));
for(int i = 0; i < MAX_FORMAT_LENGTH; i++)
{
format_array[i] ^= system_mask[i];
int idx = (eccLevelToCode(ecc_level) << 3) | mask_type_num;
format_array.resize(MAX_FORMAT_LENGTH);
for (int i = 0; i < MAX_FORMAT_LENGTH; ++i) {
format_array[i] = (formatInfoLUT[idx] >> i) & 1;
}
}
@@ -850,7 +835,7 @@ void QRCodeEncoderImpl::findAutoMaskType()
{
Mat test_result = masked_data.clone();
vector<uint8_t> test_format = format;
maskData(cur_type, test_result);
maskData(original, cur_type, test_result);
formatGenerate(cur_type, test_format);
fillReserved(test_format, test_result);
int continued_num = 0;
@@ -962,8 +947,9 @@ void QRCodeEncoderImpl::findAutoMaskType()
mask_type = best_index;
}
void QRCodeEncoderImpl::maskData(const int mask_type_num, Mat& masked)
void maskData(const Mat& original, const int mask_type_num, Mat& masked)
{
int version_size = original.rows;
for (int i = 0; i < version_size; i++)
{
for (int j = 0; j < version_size; j++)
@@ -1267,7 +1253,7 @@ void QRCodeEncoderImpl::structureFinalMessage()
writeReservedArea();
writeData();
findAutoMaskType();
maskData(mask_type, masked_data);
maskData(original, mask_type, masked_data);
formatGenerate(mask_type, format);
versionInfoGenerate(version_level, version_reserved);
fillReserved(format, masked_data);
@@ -1323,4 +1309,521 @@ Ptr<QRCodeEncoder> QRCodeEncoder::create(const QRCodeEncoder::Params& parameters
return makePtr<QRCodeEncoderImpl>(parameters);
}
class QRCodeDecoderImpl : public QRCodeDecoder {
public:
bool decode(const Mat& straight, String& decoded_info) CV_OVERRIDE;
private:
QRCodeEncoder::CorrectionLevel level;
int version;
struct Bitstream {
int next(int bits) {
CV_Assert(idx < data.size());
int val = 0;
while (bits >= actualBits) {
val |= data[idx++] << (bits - actualBits);
bits -= actualBits;
actualBits = 8;
}
if (bits) {
val |= data[idx] >> (actualBits - bits);
actualBits -= bits;
data[idx] &= 255 >> (8 - actualBits);
}
return val;
}
bool empty() {
return idx >= data.size();
}
std::vector<uint8_t> data;
int actualBits = 8;
size_t idx = 0;
} bitstream;
bool run(const Mat& straight, String& decoded_info);
bool decodeFormatInfo(const Mat& straight, int& mask);
bool correctFormatInfo(uint16_t& format_info);
void extractCodewords(Mat& source, std::vector<uint8_t>& codewords);
bool errorCorrection(std::vector<uint8_t>& codewords);
bool errorCorrectionBlock(std::vector<uint8_t>& codewords);
void decodeSymbols(String& result);
void decodeNumeric(String& result);
void decodeAlpha(String& result);
void decodeByte(String& result);
void decodeECI(String& result);
void decodeKanji(String& result);
};
QRCodeDecoder::~QRCodeDecoder()
{
// nothing
}
Ptr<QRCodeDecoder> QRCodeDecoder::create() {
return makePtr<QRCodeDecoderImpl>();
}
bool QRCodeDecoderImpl::decode(const Mat& _straight, String& decoded_info) {
Mat straight = ~_straight; // Invert modules
bool decoded = run(straight, decoded_info);
if (!decoded) {
cv::transpose(straight, straight);
decoded = run(straight, decoded_info);
}
return decoded;
}
// Unmask format info bits and apply error correction
bool QRCodeDecoderImpl::correctFormatInfo(uint16_t& format_info) {
static const uint16_t mask_pattern = 0b101010000010010;
cv::Hamming hd;
for (int i = 0; i < 32; ++i) {
// Compute Hamming distance
int distance = hd(reinterpret_cast<const unsigned char*>(&formatInfoLUT[i]),
reinterpret_cast<const unsigned char*>(&format_info), 2);
// Up to 3 bit errors might be corrected.
// So if distance is less or equal than 3 - we found a correct format info.
if (distance <= 3) {
format_info = formatInfoLUT[i] ^ mask_pattern;
return true;
}
}
return false;
}
bool QRCodeDecoderImpl::decodeFormatInfo(const Mat& straight, int& mask) {
// Read left-top format info
uint16_t format_info = 0;
for (int i = 0; i < 6; ++i)
format_info |= (straight.at<uint8_t>(i, 8) & 1) << i;
format_info |= (straight.at<uint8_t>(7, 8) & 1) << 6;
format_info |= (straight.at<uint8_t>(8, 8) & 1) << 7;
format_info |= (straight.at<uint8_t>(8, 7) & 1) << 8;
for (int i = 9; i < 15; ++i)
format_info |= (straight.at<uint8_t>(8, 14 - i) & 1) << i;
bool correct = correctFormatInfo(format_info);
// Format information 15bit sequence appears twice.
// Try extract format info from different position.
uint16_t format_info_dup = 0;
for (int i = 0; i < 8; ++i)
format_info_dup |= (straight.at<uint8_t>(8, straight.cols - 1 - i) & 1) << i;
for (int i = 0; i < 7; ++i)
format_info_dup |= (straight.at<uint8_t>(straight.rows - 7 + i, 8) & 1) << (i + 8);
if (correctFormatInfo(format_info_dup)) {
// Both strings must be the same
if (correct && format_info != format_info_dup)
return false;
format_info = format_info_dup;
} else {
if (!correct)
return false;
}
switch((format_info >> 13) & 0b11) {
case 0: level = QRCodeEncoder::CorrectionLevel::CORRECT_LEVEL_M; break;
case 1: level = QRCodeEncoder::CorrectionLevel::CORRECT_LEVEL_L; break;
case 2: level = QRCodeEncoder::CorrectionLevel::CORRECT_LEVEL_H; break;
case 3: level = QRCodeEncoder::CorrectionLevel::CORRECT_LEVEL_Q; break;
};
mask = (format_info >> 10) & 0b111;
return true;
}
bool QRCodeDecoderImpl::run(const Mat& straight, String& decoded_info) {
CV_Assert(straight.rows == straight.cols);
version = (straight.rows - 21) / 4 + 1;
decoded_info = "";
mode = static_cast<QRCodeEncoder::EncodeMode>(0);
eci = static_cast<QRCodeEncoder::ECIEncodings>(0);
// Decode format info
int maskPattern;
bool decoded = decodeFormatInfo(straight, maskPattern);
if (!decoded) {
return false;
}
// Generate data mask
Mat masked = straight.clone();
maskData(straight, maskPattern, masked);
extractCodewords(masked, bitstream.data);
if (!errorCorrection(bitstream.data)) {
return false;
}
decodeSymbols(decoded_info);
return true;
}
bool QRCodeDecoderImpl::errorCorrection(std::vector<uint8_t>& codewords) {
CV_CheckEQ((int)codewords.size(), version_info_database[version].total_codewords,
"Number of codewords");
int numBlocks = version_info_database[version].ecc[level].num_blocks_in_G1 +
version_info_database[version].ecc[level].num_blocks_in_G2;
if (numBlocks == 1) {
return errorCorrectionBlock(codewords);
}
size_t numData = 0;
std::vector<int> blockSizes;
blockSizes.reserve(numBlocks);
for (int i = 0; i < version_info_database[version].ecc[level].num_blocks_in_G1; ++i) {
blockSizes.push_back(version_info_database[version].ecc[level].data_codewords_in_G1);
numData += blockSizes.back();
}
for (int i = 0; i < version_info_database[version].ecc[level].num_blocks_in_G2; ++i) {
blockSizes.push_back(version_info_database[version].ecc[level].data_codewords_in_G2);
numData += blockSizes.back();
}
// TODO: parallel_for
std::vector<std::vector<uint8_t>> blocks(numBlocks);
int minBlockSize = *std::min_element(blockSizes.begin(), blockSizes.end());
size_t offset = 0;
for (int i = 0; i < minBlockSize; ++i) {
for (int j = 0; j < numBlocks; ++j) {
blocks[j].push_back(codewords[offset++]);
}
}
// Put remaining data codewords
for (int j = 0; j < numBlocks; ++j) {
CV_Assert(blockSizes[j] == minBlockSize || blockSizes[j] == minBlockSize + 1);
if (blockSizes[j] > minBlockSize)
blocks[j].push_back(codewords[offset++]);
}
// Copy error correction codewords
int numEcc = version_info_database[version].ecc[level].ecc_codewords;
for (int i = 0; i < numEcc; ++i) {
for (int j = 0; j < numBlocks; ++j) {
blocks[j].push_back(codewords[offset++]);
}
}
parallel_for_(Range(0, numBlocks), [&](const Range& r) {
for (int i = r.start; i < r.end; ++i) {
if (!errorCorrectionBlock(blocks[i])) {
blocks[i].clear();
return;
}
}
});
// Collect blocks back after error correction. Trim error correction codewords.
codewords.resize(numData);
offset = 0;
for (size_t i = 0; i < blocks.size(); ++i) {
if (blocks[i].empty())
return false;
std::copy(blocks[i].begin(), blocks[i].end(), codewords.begin() + offset);
offset += blocks[i].size();
}
return true;
}
bool QRCodeDecoderImpl::errorCorrectionBlock(std::vector<uint8_t>& codewords) {
size_t numEcc = version_info_database[version].ecc[level].ecc_codewords;
size_t numSyndromes = numEcc;
// According to the ISO there is a formula for a number of the syndromes.
// However several tests don't pass the error correction step because of less number of syndromes:
// 1M: qrcodes/detection/lots/image001.jpg from BoofCV (8 syndromes by formula, 10 needed)
// 1L: Objdetect_QRCode_Multi.regression/13 (4 syndromes by formula, 6 needed)
// 2L: qrcodes/detection/brightness/image011.jpg from BoofCV (8 syndromes by formula, 10 needed)
if (numSyndromes % 2 == 1)
numSyndromes -= 1;
// Compute syndromes
bool hasError = false;
std::vector<uint8_t> syndromes(numSyndromes, codewords[0]);
for (size_t i = 0; i < syndromes.size(); ++i) {
for (size_t j = 1; j < codewords.size(); ++j) {
syndromes[i] = gfMul(syndromes[i], gfPow(2, static_cast<int>(i))) ^ codewords[j];
}
hasError |= syndromes[i] != 0;
}
if (!hasError) {
// Trim error correction codewords
codewords.resize(codewords.size() - numEcc);
return true;
}
// Run BerlekampMassey algorithm to find error positions (coefficients of locator poly)
size_t L = 0; // number of assumed errors
size_t m = 1; // shift value (between C and B)
uint8_t b = 1; // discrepancy from last L update
std::vector<uint8_t> C(numSyndromes, 0); // Error locator polynomial
std::vector<uint8_t> B(numSyndromes, 0); // A copy of error locator from previos L update
C[0] = B[0] = 1;
for (size_t i = 0; i < numSyndromes; ++i) {
CV_Assert(m + L - 1 < C.size()); // m >= 1 on any iteration
uint8_t discrepancy = syndromes[i];
for (size_t j = 1; j <= L; ++j) {
discrepancy ^= gfMul(C[j], syndromes[i - j]);
}
if (discrepancy == 0) {
m += 1;
} else {
std::vector<uint8_t> C_copy = C;
uint8_t inv_b = gfDiv(1, b);
uint8_t tmp = gfMul(discrepancy, inv_b);
for (size_t j = 0; j < L; ++j) {
C[m + j] ^= gfMul(tmp, B[j]);
}
if (2 * L <= i) {
L = i + 1 - L;
B = C_copy;
b = discrepancy;
m = 1;
} else {
m += 1;
}
}
}
// There is an error at i-th position if i is a root of locator poly
std::vector<size_t> errLocs;
errLocs.reserve(L);
for (size_t i = 0; i < codewords.size(); ++i) {
uint8_t val = 1;
uint8_t pos = gfPow(2, static_cast<int>(i));
for (size_t j = 1; j <= L; ++j) {
val = gfMul(val, pos) ^ C[j];
}
if (val == 0) {
errLocs.push_back(static_cast<int>(codewords.size() - 1 - i));
}
}
// Number of assumed errors does not match number of error locations
if (errLocs.size() != L)
return false;
// Forney algorithm for error correction using syndromes and known error locations
std::vector<uint8_t> errEval;
gfPolyMul(C, syndromes, errEval);
for (size_t i = 0; i < errLocs.size(); ++i) {
uint8_t numenator = 0, denominator = 0;
uint8_t X = gfPow(2, static_cast<int>(codewords.size() - 1 - errLocs[i]));
uint8_t inv_X = gfDiv(1, X);
for (size_t j = 0; j < L; ++j) {
numenator = gfMul(numenator, inv_X) ^ errEval[L - 1 - j];
}
// Compute demoninator as a product of (1-X_i * X_k) for i != k
// TODO: optimize, there is a dubplicated compute
denominator = 1;
for (size_t j = 0; j < errLocs.size(); ++j) {
if (i == j)
continue;
uint8_t Xj = gfPow(2, static_cast<int>(codewords.size() - 1 - errLocs[j]));
denominator = gfMul(denominator, 1 ^ gfMul(inv_X, Xj));
}
uint8_t errValue = gfDiv(numenator, denominator);
codewords[errLocs[i]] ^= errValue;
}
// Trim error correction codewords
codewords.resize(codewords.size() - numEcc);
return true;
}
void QRCodeDecoderImpl::extractCodewords(Mat& source, std::vector<uint8_t>& codewords) {
const VersionInfo& version_info = version_info_database[version];
// Mask alignment markers
std::vector<int> alignCenters;
alignCenters.reserve(MAX_ALIGNMENT);
for (int i = 0; i < MAX_ALIGNMENT && version_info.alignment_pattern[i]; i++)
alignCenters.push_back(version_info.alignment_pattern[i]);
for (size_t i = 0; i < alignCenters.size(); i++)
{
for (size_t j = 0; j < alignCenters.size(); j++)
{
if ((i == alignCenters.size() - 1 && j == 0) || (i == 0 && j == 0) ||
(j == alignCenters.size() - 1 && i == 0))
continue;
int x = alignCenters[i];
int y = alignCenters[j];
Mat area = source({x - 2, x + 3}, {y - 2, y + 3});
area.setTo(INVALID_REGION_VALUE);
}
}
// Mask detection markers
source.rowRange(0, 9).colRange(source.cols - 8, source.cols).setTo(INVALID_REGION_VALUE);
source.rowRange(0, 9).colRange(0, 9).setTo(INVALID_REGION_VALUE);
source.colRange(0, 9).rowRange(source.rows - 8, source.rows).setTo(INVALID_REGION_VALUE);
// Mask Version Information blocks
if (version >= 7) {
source.rowRange(0, 6).colRange(source.cols - 12, source.cols - 9).setTo(INVALID_REGION_VALUE);
source.colRange(0, 6).rowRange(source.rows - 12, source.rows - 9).setTo(INVALID_REGION_VALUE);
}
// Mask timing pattern
source.row(6) = INVALID_REGION_VALUE;
std::vector<uint8_t> bits;
bits.reserve(source.total() - source.cols);
bool moveUpwards = true;
for (auto& data : {source.colRange(7, source.cols), source.colRange(0, 6)}) {
for (int i = data.cols / 2 - 1; i >= 0; --i) {
Mat col0 = data.col(i * 2);
Mat col1 = data.col(i * 2 + 1);
for (int j = 0; j < data.rows; ++j) {
if (moveUpwards) {
bits.push_back(col1.at<uint8_t>(data.rows - 1 - j));
bits.push_back(col0.at<uint8_t>(data.rows - 1 - j));
} else {
bits.push_back(col1.at<uint8_t>(j));
bits.push_back(col0.at<uint8_t>(j));
}
}
moveUpwards = !moveUpwards;
}
}
// Combine bits to codewords
size_t numCodewords = version_info.total_codewords;
codewords.resize(numCodewords);
size_t offset = 0;
for (size_t i = 0; i < numCodewords; ++i) {
codewords[i] = 0;
for (size_t j = 0; j < 8; ++j) {
while (bits[offset] == INVALID_REGION_VALUE) {
offset += 1;
CV_Assert(offset < bits.size());
}
codewords[i] |= (bits[offset] & 1) << (7 - j);
offset += 1;
}
}
}
void QRCodeDecoderImpl::decodeSymbols(String& result) {
CV_Assert(!bitstream.empty());
// Decode depends on the mode
result = "";
while (!bitstream.empty()) {
// Determine mode
auto currMode = static_cast<QRCodeEncoder::EncodeMode>(bitstream.next(4));
if (this->mode == 0) {
mode = currMode;
}
if (currMode == 0 || bitstream.empty())
return;
if (currMode == QRCodeEncoder::EncodeMode::MODE_NUMERIC)
decodeNumeric(result);
else if (currMode == QRCodeEncoder::EncodeMode::MODE_ALPHANUMERIC)
decodeAlpha(result);
else if (currMode == QRCodeEncoder::EncodeMode::MODE_BYTE)
decodeByte(result);
else if (currMode == QRCodeEncoder::EncodeMode::MODE_ECI)
decodeECI(result);
else if (currMode == QRCodeEncoder::EncodeMode::MODE_KANJI)
decodeKanji(result);
else
CV_Error(Error::StsNotImplemented, format("mode %d", currMode));
}
}
void QRCodeDecoderImpl::decodeNumeric(String& result) {
int numDigits = bitstream.next(version <= 9 ? 10 : (version <= 26 ? 12 : 14));
for (int i = 0; i < numDigits / 3; ++i) {
int triple = bitstream.next(10);
result += static_cast<char>('0' + triple / 100);
result += static_cast<char>('0' + (triple / 10) % 10);
result += static_cast<char>('0' + triple % 10);
}
int remainingDigits = numDigits % 3;
if (remainingDigits) {
int triple = bitstream.next(remainingDigits == 1 ? 4 : 7);
if (remainingDigits == 2)
result += '0' + (triple / 10) % 10;
result += '0' + triple % 10;
}
}
void QRCodeDecoderImpl::decodeAlpha(String& result) {
static const char map[] = {'0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J',
'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T',
'U', 'V', 'W', 'X', 'Y', 'Z', ' ', '$', '%', '*',
'+', '-', '.', '/', ':'};
int num = bitstream.next(version <= 9 ? 9 : (version <= 26 ? 11 : 13));
for (int i = 0; i < num / 2; ++i) {
int tuple = bitstream.next(11);
result += map[tuple / 45];
result += map[tuple % 45];
}
if (num % 2) {
int value = bitstream.next(6);
result += map[value];
}
}
void QRCodeDecoderImpl::decodeByte(String& result) {
int num = bitstream.next(version <= 9 ? 8 : 16);
for (int i = 0; i < num; ++i) {
result += static_cast<char>(bitstream.next(8));
}
}
void QRCodeDecoderImpl::decodeECI(String& result) {
int eciAssignValue = bitstream.next(8);
for (int i = 0; i < 8; ++i) {
if (eciAssignValue & 1 << (7 - i))
eciAssignValue |= bitstream.next(8) << (i + 1) * 8;
else
break;
}
if (this->eci == 0) {
this->eci = static_cast<QRCodeEncoder::ECIEncodings>(eciAssignValue);
}
decodeSymbols(result);
}
void QRCodeDecoderImpl::decodeKanji(String& result) {
int num = bitstream.next(version <= 9 ? 8 : (version <= 26 ? 10 : 12));
for (int i = 0; i < num; ++i) {
int data = bitstream.next(13);
int high_byte = data / 0xC0;
int low_byte = data - high_byte * 0xC0;
int symbol = (high_byte << 8) + low_byte;
if (0 <= symbol && symbol <= 0x9FFC - 0x8140) {
symbol += 0x8140;
} else if (0xE040 - 0xC140 <= symbol && symbol <= 0xEBBF - 0xC140) {
symbol += 0xC140;
}
result += (symbol >> 8) & 0xff;
result += symbol & 0xff;
}
}
}