zungr2 function

Implementation

void zungr2(
  final int M,
  final int N,
  final int K,
  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();
  int I, II, J, L;

  // Test the input arguments

  INFO.value = 0;
  if (M < 0) {
    INFO.value = -1;
  } else if (N < M) {
    INFO.value = -2;
  } else if (K < 0 || K > M) {
    INFO.value = -3;
  } else if (LDA < max(1, M)) {
    INFO.value = -5;
  }
  if (INFO.value != 0) {
    xerbla('ZUNGR2', -INFO.value);
    return;
  }

  // Quick return if possible

  if (M <= 0) return;

  if (K < M) {
    // Initialise rows 1:m-k to rows of the unit matrix

    for (J = 1; J <= N; J++) {
      for (L = 1; L <= M - K; L++) {
        A[L][J] = Complex.zero;
      }
      if (J > N - M && J <= N - K) A[M - N + J][J] = Complex.one;
    }
  }

  for (I = 1; I <= K; I++) {
    II = M - K + I;

    // Apply H(i)**H to A(1:m-k+i,1:n-k+i) from the right

    zlacgv(N - M + II - 1, A(II, 1).asArray(), LDA);
    A[II][N - M + II] = Complex.one;
    zlarf('Right', II - 1, N - M + II, A(II, 1).asArray(), LDA,
        TAU[I].conjugate(), A, LDA, WORK);
    zscal(N - M + II - 1, -TAU[I], A(II, 1).asArray(), LDA);
    zlacgv(N - M + II - 1, A(II, 1).asArray(), LDA);
    A[II][N - M + II] = Complex.one - TAU[I].conjugate();

    // Set A(m-k+i,n-k+i+1:n) to zero

    for (L = N - M + II + 1; L <= N; L++) {
      A[II][L] = Complex.zero;
    }
  }
}