NapCatQQ/packages/napcat-qrcode/vendor/QRCode/index.ts

// Copyright (c) 2009 Kazuhiko Arase
import QR8bitByte from './QR8bitByte';
import { getLengthInBits, getLostPoint, getPatternPosition, getBCHTypeNumber, getBCHTypeInfo, getMask, getErrorCorrectPolynomial } from './QRUtil';
import QRPolynomial from './QRPolynomial';
import { getRSBlocks, QRRSBlock as RSBlock } from './QRRSBlock';
import QRBitBuffer from './QRBitBuffer';

class QRCode {
  typeNumber: number;
  errorCorrectLevel: number;
  modules: (boolean | null)[][] | null;
  moduleCount: number;
  dataCache: number[] | null;
  dataList: QR8bitByte[];

  static PAD0 = 0xEC;
  static PAD1 = 0x11;

  constructor (typeNumber: number, errorCorrectLevel: number) {
    this.typeNumber = typeNumber;
    this.errorCorrectLevel = errorCorrectLevel;
    this.modules = null;
    this.moduleCount = 0;
    this.dataCache = null;
    this.dataList = [];
  }

  addData (data: string): void {
    const newData = new QR8bitByte(data);
    this.dataList.push(newData);
    this.dataCache = null;
  }

  isDark (row: number, col: number): boolean {
    if (row < 0 || this.moduleCount <= row || col < 0 || this.moduleCount <= col) {
      throw new Error(`${row},${col}`);
    }
    if (this.modules === null || this.modules[row] === null || this.modules[row]![col] === null) {
      throw new Error(`Module at (${row},${col}) is null`);
    }
    return this.modules[row]![col]!;
  }

  getModuleCount (): number {
    return this.moduleCount;
  }

  make (): void {
    if (this.typeNumber < 1) {
      let typeNumber = 1;
      for (typeNumber = 1; typeNumber < 40; typeNumber++) {
        const rsBlocks = getRSBlocks(typeNumber, this.errorCorrectLevel);

        const buffer = new QRBitBuffer();
        let totalDataCount = 0;
        for (let i = 0; i < rsBlocks.length; i++) {
          totalDataCount += rsBlocks[i]!.dataCount;
        }

        for (let x = 0; x < this.dataList.length; x++) {
          const data = this.dataList[x];
          buffer.put(data!.mode, 4);
          buffer.put(data!.getLength(), getLengthInBits(data!.mode, typeNumber));
          data!.write(buffer);
        }
        if (buffer.getLengthInBits() <= totalDataCount * 8) { break; }
      }
      this.typeNumber = typeNumber;
    }
    this.makeImpl(false, this.getBestMaskPattern());
  }

  makeImpl (test: boolean, maskPattern: number): void {
    this.moduleCount = this.typeNumber * 4 + 17;
    this.modules = new Array(this.moduleCount);

    for (let row = 0; row < this.moduleCount; row++) {
      this.modules[row] = new Array(this.moduleCount);
      for (let col = 0; col < this.moduleCount; col++) {
        this.modules[row]![col] = null;
      }
    }

    this.setupPositionProbePattern(0, 0);
    this.setupPositionProbePattern(this.moduleCount - 7, 0);
    this.setupPositionProbePattern(0, this.moduleCount - 7);
    this.setupPositionAdjustPattern();
    this.setupTimingPattern();
    this.setupTypeInfo(test, maskPattern);

    if (this.typeNumber >= 7) {
      this.setupTypeNumber(test);
    }

    if (this.dataCache === null) {
      this.dataCache = QRCode.createData(this.typeNumber, this.errorCorrectLevel, this.dataList);
    }

    this.mapData(this.dataCache, maskPattern);
  }

  setupPositionProbePattern (row: number, col: number): void {
    for (let r = -1; r <= 7; r++) {
      if (row + r <= -1 || this.moduleCount <= row + r) continue;

      for (let c = -1; c <= 7; c++) {
        if (col + c <= -1 || this.moduleCount <= col + c) continue;

        if ((r >= 0 && r <= 6 && (c === 0 || c === 6)) ||
          (c >= 0 && c <= 6 && (r === 0 || r === 6)) ||
          (r >= 2 && r <= 4 && c >= 2 && c <= 4)) {
          this.modules![row + r]![col + c] = true;
        } else {
          this.modules![row + r]![col + c] = false;
        }
      }
    }
  }

  getBestMaskPattern (): number {
    let minLostPoint = 0;
    let pattern = 0;

    for (let i = 0; i < 8; i++) {
      this.makeImpl(true, i);
      const lostPoint = getLostPoint(this);

      if (i === 0 || minLostPoint > lostPoint) {
        minLostPoint = lostPoint;
        pattern = i;
      }
    }

    return pattern;
  }

  createMovieClip (target_mc: { createEmptyMovieClip: (name: string, depth: number) => { beginFill: (color: number, alpha: number) => void; moveTo: (x: number, y: number) => void; lineTo: (x: number, y: number) => void; endFill: () => void; }; }, instance_name: string, depth: number): { beginFill: (color: number, alpha: number) => void; moveTo: (x: number, y: number) => void; lineTo: (x: number, y: number) => void; endFill: () => void; } {
    const qr_mc = target_mc.createEmptyMovieClip(instance_name, depth);
    const cs = 1;

    this.make();

    for (let row = 0; row < this.modules!.length; row++) {
      const y = row * cs;

      for (let col = 0; col < this.modules![row]!.length; col++) {
        const x = col * cs;
        const dark = this.modules![row]![col];

        if (dark) {
          qr_mc.beginFill(0, 100);
          qr_mc.moveTo(x, y);
          qr_mc.lineTo(x + cs, y);
          qr_mc.lineTo(x + cs, y + cs);
          qr_mc.lineTo(x, y + cs);
          qr_mc.endFill();
        }
      }
    }

    return qr_mc;
  }

  setupTimingPattern (): void {
    for (let r = 8; r < this.moduleCount - 8; r++) {
      if (this.modules![r]![6] !== null) {
        continue;
      }
      this.modules![r]![6] = (r % 2 === 0);
    }

    for (let c = 8; c < this.moduleCount - 8; c++) {
      if (this.modules![6]![c] !== null) {
        continue;
      }
      this.modules![6]![c] = (c % 2 === 0);
    }
  }

  setupPositionAdjustPattern (): void {
    const pos = getPatternPosition(this.typeNumber);

    for (let i = 0; i < pos.length; i++) {
      for (let j = 0; j < pos.length; j++) {
        const row = pos[i];
        const col = pos[j];

        if (this.modules![row!]![col!] !== null) {
          continue;
        }

        for (let r = -2; r <= 2; r++) {
          for (let c = -2; c <= 2; c++) {
            if (Math.abs(r) === 2 ||
              Math.abs(c) === 2 ||
              (r === 0 && c === 0)) {
              this.modules![row! + r]![col! + c] = true;
            } else {
              this.modules![row! + r]![col! + c] = false;
            }
          }
        }
      }
    }
  }

  setupTypeNumber (test: boolean): void {
    const bits = getBCHTypeNumber(this.typeNumber);
    let mod: boolean;

    for (let i = 0; i < 18; i++) {
      mod = (!test && ((bits >> i) & 1) === 1);
      this.modules![Math.floor(i / 3)]![i % 3 + this.moduleCount - 8 - 3] = mod;
    }

    for (let x = 0; x < 18; x++) {
      mod = (!test && ((bits >> x) & 1) === 1);
      this.modules![x % 3 + this.moduleCount - 8 - 3]![Math.floor(x / 3)] = mod;
    }
  }

  setupTypeInfo (test: boolean, maskPattern: number): void {
    const data = (this.errorCorrectLevel << 3) | maskPattern;
    const bits = getBCHTypeInfo(data);
    let mod: boolean;

    // vertical
    for (let v = 0; v < 15; v++) {
      mod = (!test && ((bits >> v) & 1) === 1);

      if (v < 6) {
        this.modules![v]![8] = mod;
      } else if (v < 8) {
        this.modules![v + 1]![8] = mod;
      } else {
        this.modules![this.moduleCount - 15 + v]![8] = mod;
      }
    }

    // horizontal
    for (let h = 0; h < 15; h++) {
      mod = (!test && ((bits >> h) & 1) === 1);

      if (h < 8) {
        this.modules![8]![this.moduleCount - h - 1] = mod;
      } else if (h < 9) {
        this.modules![8]![15 - h - 1 + 1] = mod;
      } else {
        this.modules![8]![15 - h - 1] = mod;
      }
    }

    // fixed module
    this.modules![this.moduleCount - 8]![8] = (!test);
  }

  mapData (data: number[], maskPattern: number): void {
    let inc = -1;
    let row = this.moduleCount - 1;
    let bitIndex = 7;
    let byteIndex = 0;

    for (let col = this.moduleCount - 1; col > 0; col -= 2) {
      if (col === 6) col--;

      while (true) {
        for (let c = 0; c < 2; c++) {
          if (this.modules![row]![col - c] === null) {
            let dark = false;

            if (byteIndex < data.length) {
              dark = (((data[byteIndex] ?? 0) >>> bitIndex) & 1) === 1;
            }

            const mask = getMask(maskPattern, row, col - c);

            if (mask) {
              dark = !dark;
            }

            this.modules![row]![col - c] = dark;
            bitIndex--;

            if (bitIndex === -1) {
              byteIndex++;
              bitIndex = 7;
            }
          }
        }

        row += inc;

        if (row < 0 || this.moduleCount <= row) {
          row -= inc;
          inc = -inc;
          break;
        }
      }
    }
  }

  static createData (typeNumber: number, errorCorrectLevel: number, dataList: QR8bitByte[]): number[] {
    const rsBlocks = getRSBlocks(typeNumber, errorCorrectLevel);

    const buffer = new QRBitBuffer();

    for (let i = 0; i < dataList.length; i++) {
      const data = dataList[i];
      buffer.put(data!.mode, 4);
      buffer.put(data!.getLength(), getLengthInBits(data!.mode, typeNumber));
      data!.write(buffer);
    }

    // calc num max data.
    let totalDataCount = 0;
    for (let x = 0; x < rsBlocks.length; x++) {
      totalDataCount += rsBlocks[x]!.dataCount;
    }

    if (buffer.getLengthInBits() > totalDataCount * 8) {
      throw new Error(`code length overflow. (${buffer.getLengthInBits()} > ${totalDataCount * 8})`);
    }

    // end code
    if (buffer.getLengthInBits() + 4 <= totalDataCount * 8) {
      buffer.put(0, 4);
    }

    // padding
    while (buffer.getLengthInBits() % 8 !== 0) {
      buffer.putBit(false);
    }

    // padding
    while (true) {
      if (buffer.getLengthInBits() >= totalDataCount * 8) {
        break;
      }
      buffer.put(QRCode.PAD0, 8);

      if (buffer.getLengthInBits() >= totalDataCount * 8) {
        break;
      }
      buffer.put(QRCode.PAD1, 8);
    }

    return QRCode.createBytes(buffer, rsBlocks);
  }

  static createBytes (buffer: QRBitBuffer, rsBlocks: RSBlock[]): number[] {
    let offset = 0;

    let maxDcCount = 0;
    let maxEcCount = 0;

    const dcdata: number[][] = new Array(rsBlocks.length);
    const ecdata: number[][] = new Array(rsBlocks.length);

    for (let r = 0; r < rsBlocks.length; r++) {
      const dcCount = rsBlocks[r]!.dataCount;
      const ecCount = rsBlocks[r]!.totalCount - dcCount;

      maxDcCount = Math.max(maxDcCount, dcCount);
      maxEcCount = Math.max(maxEcCount, ecCount);

      dcdata[r] = new Array(dcCount);

      for (let i = 0; i < dcdata[r]!.length; i++) {
        dcdata[r]![i] = 0xff & buffer.buffer[i + offset]!;
      }
      offset += dcCount;

      const rsPoly = getErrorCorrectPolynomial(ecCount);
      const rawPoly = new QRPolynomial(dcdata[r]!, rsPoly.getLength() - 1);

      const modPoly = rawPoly.mod(rsPoly);
      ecdata[r] = new Array(rsPoly.getLength() - 1);
      for (let x = 0; x < ecdata[r]!.length; x++) {
        const modIndex = x + modPoly.getLength() - ecdata[r]!.length;
        ecdata[r]![x] = (modIndex >= 0) ? modPoly.get(modIndex) : 0;
      }
    }

    let totalCodeCount = 0;
    for (let y = 0; y < rsBlocks.length; y++) {
      totalCodeCount += rsBlocks[y]!.totalCount;
    }

    const data: number[] = new Array(totalCodeCount);
    let index = 0;

    for (let z = 0; z < maxDcCount; z++) {
      for (let s = 0; s < rsBlocks.length; s++) {
        if (z < dcdata[s]!.length) {
          data[index++] = dcdata[s]![z]!;
        }
      }
    }

    for (let xx = 0; xx < maxEcCount; xx++) {
      for (let t = 0; t < rsBlocks.length; t++) {
        if (xx < ecdata[t]!.length) {
          data[index++] = ecdata[t]![xx]!;
        }
      }
    }

    return data;
  }
}

export default QRCode;