zgeqr2 function

Implementation

void zgeqr2(
  final int M,
  final int N,
  final Matrix<Complex> A_,
  final int LDA,
  final Array<Complex> TAU_,
  final Array<Complex> WORK_,
  final Box<int> INFO,
) {
  final A = A_.having(ld: LDA);
  final TAU = TAU_.having();
  final WORK = WORK_.having();

  // Test the input arguments

  INFO.value = 0;
  if (M < 0) {
    INFO.value = -1;
  } else if (N < 0) {
    INFO.value = -2;
  } else if (LDA < max(1, M)) {
    INFO.value = -4;
  }
  if (INFO.value != 0) {
    xerbla('ZGEQR2', -INFO.value);
    return;
  }

  final K = min(M, N);

  for (var I = 1; I <= K; I++) {
    // Generate elementary reflector H(i) to annihilate A(i+1:m,i)

    zlarfg(
        M - I + 1, A.box(I, I), A(min(I + 1, M), I).asArray(), 1, TAU.box(I));
    if (I < N) {
      // Apply H(i)**H to A(i:m,i+1:n) from the left

      final ALPHA = A[I][I];
      A[I][I] = Complex.one;
      zlarf('Left', M - I + 1, N - I, A(I, I).asArray(), 1, TAU[I].conjugate(),
          A(I, I + 1), LDA, WORK);
      A[I][I] = ALPHA;
    }
  }
}