examples/common/QRCode/repo/cpp/QrCodeGeneratorWorker.cpp

   1 /*
   2  * QR Code generator test worker (C++)
   3  *
   4  * This program reads data and encoding parameters from standard input and writes
   5  * QR Code bitmaps to standard output. The I/O format is one integer per line.
   6  * Run with no command line arguments. The program is intended for automated
   7  * batch testing of end-to-end functionality of this QR Code generator library.
   8  *
   9  * Copyright (c) Project Nayuki. (MIT License)
  10  * https://www.nayuki.io/page/qr-code-generator-library
  11  *
  12  * Permission is hereby granted, free of charge, to any person obtaining a copy of
  13  * this software and associated documentation files (the "Software"), to deal in
  14  * the Software without restriction, including without limitation the rights to
  15  * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
  16  * the Software, and to permit persons to whom the Software is furnished to do so,
  17  * subject to the following conditions:
  18  * - The above copyright notice and this permission notice shall be included in
  19  *   all copies or substantial portions of the Software.
  20  * - The Software is provided "as is", without warranty of any kind, express or
  21  *   implied, including but not limited to the warranties of merchantability,
  22  *   fitness for a particular purpose and noninfringement. In no event shall the
  23  *   authors or copyright holders be liable for any claim, damages or other
  24  *   liability, whether in an action of contract, tort or otherwise, arising from,
  25  *   out of or in connection with the Software or the use or other dealings in the
  26  *   Software.
  27  */
  28
  29 #include <cstddef>
  30 #include <cstdint>
  31 #include <cstdlib>
  32 #include <cstring>
  33 #include <iostream>
  34 #include <vector>
  35 #include "QrCode.hpp"
  36
  37 using qrcodegen::QrCode;
  38 using qrcodegen::QrSegment;
  39
  40
  41 static const std::vector<QrCode::Ecc> ECC_LEVELS{
  42         QrCode::Ecc::LOW,
  43         QrCode::Ecc::MEDIUM,
  44         QrCode::Ecc::QUARTILE,
  45         QrCode::Ecc::HIGH,
  46 };
  47
  48
  49 int main() {
  50         while (true) {
  51
  52                 // Read data length or exit
  53                 int length;
  54                 std::cin >> length;
  55                 if (length == -1)
  56                         break;
  57
  58                 // Read data bytes
  59                 bool isAscii = true;
  60                 std::vector<uint8_t> data;
  61                 for (int i = 0; i < length; i++) {
  62                         int b;
  63                         std::cin >> b;
  64                         data.push_back(static_cast<uint8_t>(b));
  65                         isAscii &= 0 < b && b < 128;
  66                 }
  67
  68                 // Read encoding parameters
  69                 int errCorLvl, minVersion, maxVersion, mask, boostEcl;
  70                 std::cin >> errCorLvl;
  71                 std::cin >> minVersion;
  72                 std::cin >> maxVersion;
  73                 std::cin >> mask;
  74                 std::cin >> boostEcl;
  75
  76                 // Make list of segments
  77                 std::vector<QrSegment> segs;
  78                 if (isAscii) {
  79                         std::vector<char> text(data.cbegin(), data.cend());
  80                         text.push_back('\0');
  81                         segs = QrSegment::makeSegments(text.data());
  82                 } else
  83                         segs.push_back(QrSegment::makeBytes(data));
  84
  85                 try {  // Try to make QR Code symbol
  86                         const QrCode qr = QrCode::encodeSegments(segs,
  87                                 ECC_LEVELS.at(static_cast<std::size_t>(errCorLvl)), minVersion, maxVersion, mask, boostEcl == 1);
  88                         // Print grid of modules
  89                         std::cout << qr.getVersion() << std::endl;
  90                         for (int y = 0; y < qr.getSize(); y++) {
  91                                 for (int x = 0; x < qr.getSize(); x++)
  92                                         std::cout << (qr.getModule(x, y) ? 1 : 0) << std::endl;
  93                         }
  94
  95                 } catch (const qrcodegen::data_too_long &ex) {
  96                         std::cout << -1 << std::endl;
  97                 }
  98                 std::cout << std::flush;
  99         }
 100         return EXIT_SUCCESS;
 101 }