/ cpp / QrCodeGeneratorWorker.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  }