296 lines
12 KiB
C++
296 lines
12 KiB
C++
/*
|
|
* QR Code generator library (C++)
|
|
*
|
|
* Copyright (c) Project Nayuki
|
|
* https://www.nayuki.io/page/qr-code-generator-library
|
|
*
|
|
* (MIT License)
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy of
|
|
* this software and associated documentation files (the "Software"), to deal in
|
|
* the Software without restriction, including without limitation the rights to
|
|
* use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
|
|
* the Software, and to permit persons to whom the Software is furnished to do so,
|
|
* subject to the following conditions:
|
|
* - The above copyright notice and this permission notice shall be included in
|
|
* all copies or substantial portions of the Software.
|
|
* - The Software is provided "as is", without warranty of any kind, express or
|
|
* implied, including but not limited to the warranties of merchantability,
|
|
* fitness for a particular purpose and noninfringement. In no event shall the
|
|
* authors or copyright holders be liable for any claim, damages or other
|
|
* liability, whether in an action of contract, tort or otherwise, arising from,
|
|
* out of or in connection with the Software or the use or other dealings in the
|
|
* Software.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <cstdint>
|
|
#include <string>
|
|
#include <vector>
|
|
#include "QrSegment.hpp"
|
|
|
|
|
|
namespace qrcodegen {
|
|
|
|
/*
|
|
* Represents an immutable square grid of black and white cells for a QR Code symbol, and
|
|
* provides static functions to create a QR Code from user-supplied textual or binary data.
|
|
* This class covers the QR Code model 2 specification, supporting all versions (sizes)
|
|
* from 1 to 40, all 4 error correction levels, and only 3 character encoding modes.
|
|
*/
|
|
class QrCode final {
|
|
|
|
/*---- Public helper enumeration ----*/
|
|
|
|
/*
|
|
* Represents the error correction level used in a QR Code symbol.
|
|
*/
|
|
public: class Ecc final {
|
|
// Constants declared in ascending order of error protection.
|
|
public: const static Ecc LOW, MEDIUM, QUARTILE, HIGH;
|
|
|
|
// Fields.
|
|
public: const int ordinal; // (Public) In the range 0 to 3 (unsigned 2-bit integer).
|
|
public: const int formatBits; // (Package-private) In the range 0 to 3 (unsigned 2-bit integer).
|
|
|
|
// Constructor.
|
|
private: Ecc(int ord, int fb);
|
|
};
|
|
|
|
|
|
|
|
/*---- Public static factory functions ----*/
|
|
|
|
/*
|
|
* Returns a QR Code symbol representing the given Unicode text string at the given error correction level.
|
|
* As a conservative upper bound, this function is guaranteed to succeed for strings that have 738 or fewer Unicode
|
|
* code points (not UTF-16 code units). The smallest possible QR Code version is automatically chosen for the output.
|
|
* The ECC level of the result may be higher than the ecl argument if it can be done without increasing the version.
|
|
*/
|
|
public: static QrCode encodeText(const char *text, const Ecc &ecl);
|
|
|
|
|
|
/*
|
|
* Returns a QR Code symbol representing the given binary data string at the given error correction level.
|
|
* This function always encodes using the binary segment mode, not any text mode. The maximum number of
|
|
* bytes allowed is 2953. The smallest possible QR Code version is automatically chosen for the output.
|
|
* The ECC level of the result may be higher than the ecl argument if it can be done without increasing the version.
|
|
*/
|
|
public: static QrCode encodeBinary(const std::vector<uint8_t> &data, const Ecc &ecl);
|
|
|
|
|
|
/*
|
|
* Returns a QR Code symbol representing the given data segments with the given encoding parameters.
|
|
* The smallest possible QR Code version within the given range is automatically chosen for the output.
|
|
* This function allows the user to create a custom sequence of segments that switches
|
|
* between modes (such as alphanumeric and binary) to encode text more efficiently.
|
|
* This function is considered to be lower level than simply encoding text or binary data.
|
|
*/
|
|
public: static QrCode encodeSegments(const std::vector<QrSegment> &segs, const Ecc &ecl,
|
|
int minVersion=1, int maxVersion=40, int mask=-1, bool boostEcl=true); // All optional parameters
|
|
|
|
|
|
|
|
/*---- Instance fields ----*/
|
|
|
|
// Public immutable scalar parameters
|
|
|
|
/* This QR Code symbol's version number, which is always between 1 and 40 (inclusive). */
|
|
public: const int version;
|
|
|
|
/* The width and height of this QR Code symbol, measured in modules.
|
|
* Always equal to version × 4 + 17, in the range 21 to 177. */
|
|
public: const int size;
|
|
|
|
/* The error correction level used in this QR Code symbol. */
|
|
public: const Ecc &errorCorrectionLevel;
|
|
|
|
/* The mask pattern used in this QR Code symbol, in the range 0 to 7 (i.e. unsigned 3-bit integer).
|
|
* Note that even if a constructor was called with automatic masking requested
|
|
* (mask = -1), the resulting object will still have a mask value between 0 and 7. */
|
|
private: int mask;
|
|
|
|
// Private grids of modules/pixels (conceptually immutable)
|
|
private: std::vector<std::vector<bool>> modules; // The modules of this QR Code symbol (false = white, true = black)
|
|
private: std::vector<std::vector<bool>> isFunction; // Indicates function modules that are not subjected to masking
|
|
|
|
|
|
|
|
/*---- Constructors ----*/
|
|
|
|
/*
|
|
* Creates a new QR Code symbol with the given version number, error correction level, binary data array,
|
|
* and mask number. This is a cumbersome low-level constructor that should not be invoked directly by the user.
|
|
* To go one level up, see the encodeSegments() function.
|
|
*/
|
|
public: QrCode(int ver, const Ecc &ecl, const std::vector<uint8_t> &dataCodewords, int mask);
|
|
|
|
|
|
/*
|
|
* Creates a new QR Code symbol based on the given existing object, but with a potentially
|
|
* different mask pattern. The version, error correction level, codewords, etc. of the newly
|
|
* created object are all identical to the argument object; only the mask may differ.
|
|
*/
|
|
public: QrCode(const QrCode &qr, int mask);
|
|
|
|
|
|
|
|
/*---- Public instance methods ----*/
|
|
|
|
public: int getMask() const;
|
|
|
|
|
|
/*
|
|
* Returns the color of the module (pixel) at the given coordinates, which is either 0 for white or 1 for black. The top
|
|
* left corner has the coordinates (x=0, y=0). If the given coordinates are out of bounds, then 0 (white) is returned.
|
|
*/
|
|
public: int getModule(int x, int y) const;
|
|
|
|
|
|
/*
|
|
* Based on the given number of border modules to add as padding, this returns a
|
|
* string whose contents represents an SVG XML file that depicts this QR Code symbol.
|
|
* Note that Unix newlines (\n) are always used, regardless of the platform.
|
|
*/
|
|
public: std::string toSvgString(int border) const;
|
|
|
|
|
|
|
|
/*---- Private helper methods for constructor: Drawing function modules ----*/
|
|
|
|
private: void drawFunctionPatterns();
|
|
|
|
|
|
// Draws two copies of the format bits (with its own error correction code)
|
|
// based on the given mask and this object's error correction level field.
|
|
private: void drawFormatBits(int mask);
|
|
|
|
|
|
// Draws two copies of the version bits (with its own error correction code),
|
|
// based on this object's version field (which only has an effect for 7 <= version <= 40).
|
|
private: void drawVersion();
|
|
|
|
|
|
// Draws a 9*9 finder pattern including the border separator, with the center module at (x, y).
|
|
private: void drawFinderPattern(int x, int y);
|
|
|
|
|
|
// Draws a 5*5 alignment pattern, with the center module at (x, y).
|
|
private: void drawAlignmentPattern(int x, int y);
|
|
|
|
|
|
// Sets the color of a module and marks it as a function module.
|
|
// Only used by the constructor. Coordinates must be in range.
|
|
private: void setFunctionModule(int x, int y, bool isBlack);
|
|
|
|
|
|
/*---- Private helper methods for constructor: Codewords and masking ----*/
|
|
|
|
// Returns a new byte string representing the given data with the appropriate error correction
|
|
// codewords appended to it, based on this object's version and error correction level.
|
|
private: std::vector<uint8_t> appendErrorCorrection(const std::vector<uint8_t> &data) const;
|
|
|
|
|
|
// Draws the given sequence of 8-bit codewords (data and error correction) onto the entire
|
|
// data area of this QR Code symbol. Function modules need to be marked off before this is called.
|
|
private: void drawCodewords(const std::vector<uint8_t> &data);
|
|
|
|
|
|
// XORs the data modules in this QR Code with the given mask pattern. Due to XOR's mathematical
|
|
// properties, calling applyMask(m) twice with the same value is equivalent to no change at all.
|
|
// This means it is possible to apply a mask, undo it, and try another mask. Note that a final
|
|
// well-formed QR Code symbol needs exactly one mask applied (not zero, not two, etc.).
|
|
private: void applyMask(int mask);
|
|
|
|
|
|
// A messy helper function for the constructors. This QR Code must be in an unmasked state when this
|
|
// method is called. The given argument is the requested mask, which is -1 for auto or 0 to 7 for fixed.
|
|
// This method applies and returns the actual mask chosen, from 0 to 7.
|
|
private: int handleConstructorMasking(int mask);
|
|
|
|
|
|
// Calculates and returns the penalty score based on state of this QR Code's current modules.
|
|
// This is used by the automatic mask choice algorithm to find the mask pattern that yields the lowest score.
|
|
private: int getPenaltyScore() const;
|
|
|
|
|
|
|
|
/*---- Private static helper functions ----*/
|
|
|
|
// Returns a set of positions of the alignment patterns in ascending order. These positions are
|
|
// used on both the x and y axes. Each value in the resulting array is in the range [0, 177).
|
|
// This stateless pure function could be implemented as table of 40 variable-length lists of unsigned bytes.
|
|
private: static std::vector<int> getAlignmentPatternPositions(int ver);
|
|
|
|
|
|
// Returns the number of raw data modules (bits) available at the given version number.
|
|
// These data modules are used for both user data codewords and error correction codewords.
|
|
// This stateless pure function could be implemented as a 40-entry lookup table.
|
|
private: static int getNumRawDataModules(int ver);
|
|
|
|
|
|
// Returns the number of 8-bit data (i.e. not error correction) codewords contained in any
|
|
// QR Code of the given version number and error correction level, with remainder bits discarded.
|
|
// This stateless pure function could be implemented as a (40*4)-cell lookup table.
|
|
private: static int getNumDataCodewords(int ver, const Ecc &ecl);
|
|
|
|
|
|
/*---- Private tables of constants ----*/
|
|
|
|
// For use in getPenaltyScore(), when evaluating which mask is best.
|
|
private: static const int PENALTY_N1;
|
|
private: static const int PENALTY_N2;
|
|
private: static const int PENALTY_N3;
|
|
private: static const int PENALTY_N4;
|
|
|
|
private: static const int16_t NUM_ERROR_CORRECTION_CODEWORDS[4][41];
|
|
private: static const int8_t NUM_ERROR_CORRECTION_BLOCKS[4][41];
|
|
|
|
|
|
|
|
/*---- Private helper class ----*/
|
|
|
|
/*
|
|
* Computes the Reed-Solomon error correction codewords for a sequence of data codewords
|
|
* at a given degree. Objects are immutable, and the state only depends on the degree.
|
|
* This class exists because the divisor polynomial does not need to be recalculated for every input.
|
|
*/
|
|
private: class ReedSolomonGenerator final {
|
|
|
|
/*-- Immutable field --*/
|
|
|
|
// Coefficients of the divisor polynomial, stored from highest to lowest power, excluding the leading term which
|
|
// is always 1. For example the polynomial x^3 + 255x^2 + 8x + 93 is stored as the uint8 array {255, 8, 93}.
|
|
private: std::vector<uint8_t> coefficients;
|
|
|
|
|
|
/*-- Constructor --*/
|
|
|
|
/*
|
|
* Creates a Reed-Solomon ECC generator for the given degree. This could be implemented
|
|
* as a lookup table over all possible parameter values, instead of as an algorithm.
|
|
*/
|
|
public: ReedSolomonGenerator(int degree);
|
|
|
|
|
|
/*-- Method --*/
|
|
|
|
/*
|
|
* Computes and returns the Reed-Solomon error correction codewords for the given sequence of data codewords.
|
|
* The returned object is always a new byte array. This method does not alter this object's state (because it is immutable).
|
|
*/
|
|
public: std::vector<uint8_t> getRemainder(const std::vector<uint8_t> &data) const;
|
|
|
|
|
|
/*-- Static function --*/
|
|
|
|
// Returns the product of the two given field elements modulo GF(2^8/0x11D). The arguments and result
|
|
// are unsigned 8-bit integers. This could be implemented as a lookup table of 256*256 entries of uint8.
|
|
private: static uint8_t multiply(uint8_t x, uint8_t y);
|
|
|
|
};
|
|
|
|
};
|
|
|
|
}
|