zunmrz function

Implementation

void zunmrz(
  final String SIDE,
  final String TRANS,
  final int M,
  final int N,
  final int K,
  final int L,
  final Matrix<Complex> A_,
  final int LDA,
  final Array<Complex> TAU_,
  final Matrix<Complex> C_,
  final int LDC,
  final Array<Complex> WORK_,
  final int LWORK,
  final Box<int> INFO,
) {
  final A = A_.having(ld: LDA);
  final C = C_.having(ld: LDC);
  final TAU = TAU_.having();
  final WORK = WORK_.having();
  const NBMAX = 64, LDT = NBMAX + 1, TSIZE = LDT * NBMAX;
  bool LEFT, LQUERY, NOTRAN;
  String TRANST;
  int I,
      I1,
      I2,
      I3,
      IB,
      IC = 0,
      IWT,
      JA,
      JC = 0,
      LDWORK,
      LWKOPT = 0,
      MI = 0,
      NB,
      NBMIN,
      NI = 0,
      NQ,
      NW;
  final IINFO = Box(0);

  // Test the input arguments

  INFO.value = 0;
  LEFT = lsame(SIDE, 'L');
  NOTRAN = lsame(TRANS, 'N');
  LQUERY = (LWORK == -1);

  // NQ is the order of Q and NW is the minimum dimension of WORK

  if (LEFT) {
    NQ = M;
    NW = max(1, N);
  } else {
    NQ = N;
    NW = max(1, M);
  }
  if (!LEFT && !lsame(SIDE, 'R')) {
    INFO.value = -1;
  } else if (!NOTRAN && !lsame(TRANS, 'C')) {
    INFO.value = -2;
  } else if (M < 0) {
    INFO.value = -3;
  } else if (N < 0) {
    INFO.value = -4;
  } else if (K < 0 || K > NQ) {
    INFO.value = -5;
  } else if (L < 0 || (LEFT && (L > M)) || (!LEFT && (L > N))) {
    INFO.value = -6;
  } else if (LDA < max(1, K)) {
    INFO.value = -8;
  } else if (LDC < max(1, M)) {
    INFO.value = -11;
  } else if (LWORK < max(1, NW) && !LQUERY) {
    INFO.value = -13;
  }

  if (INFO.value == 0) {
    // Compute the workspace requirements

    if (M == 0 || N == 0) {
      LWKOPT = 1;
    } else {
      NB = min(NBMAX, ilaenv(1, 'ZUNMRQ', SIDE + TRANS, M, N, K, -1));
      LWKOPT = NW * NB + TSIZE;
    }
    WORK[1] = LWKOPT.toComplex();
  }

  if (INFO.value != 0) {
    xerbla('ZUNMRZ', -INFO.value);
    return;
  } else if (LQUERY) {
    return;
  }

  // Quick return if possible

  if (M == 0 || N == 0) {
    return;
  }

  // Determine the block size.  NB may be at most NBMAX, where NBMAX
  // is used to define the local array T.

  NB = min(NBMAX, ilaenv(1, 'ZUNMRQ', SIDE + TRANS, M, N, K, -1));
  NBMIN = 2;
  LDWORK = NW;
  if (NB > 1 && NB < K) {
    if (LWORK < LWKOPT) {
      NB = (LWORK - TSIZE) ~/ LDWORK;
      NBMIN = max(2, ilaenv(2, 'ZUNMRQ', SIDE + TRANS, M, N, K, -1));
    }
  }

  if (NB < NBMIN || NB >= K) {
    // Use unblocked code

    zunmr3(SIDE, TRANS, M, N, K, L, A, LDA, TAU, C, LDC, WORK, IINFO);
  } else {
    // Use blocked code

    IWT = 1 + NW * NB;
    if ((LEFT && !NOTRAN) || (!LEFT && NOTRAN)) {
      I1 = 1;
      I2 = K;
      I3 = NB;
    } else {
      I1 = ((K - 1) ~/ NB) * NB + 1;
      I2 = 1;
      I3 = -NB;
    }

    if (LEFT) {
      NI = N;
      JC = 1;
      JA = M - L + 1;
    } else {
      MI = M;
      IC = 1;
      JA = N - L + 1;
    }

    if (NOTRAN) {
      TRANST = 'C';
    } else {
      TRANST = 'N';
    }

    for (I = I1; I3 < 0 ? I >= I2 : I <= I2; I += I3) {
      IB = min(NB, K - I + 1);

      // Form the triangular factor of the block reflector
      // H = H(i+ib-1) . . . H(i+1) H(i)

      zlarzt('Backward', 'Rowwise', L, IB, A(I, JA), LDA, TAU(I),
          WORK(IWT).asMatrix(LDT), LDT);

      if (LEFT) {
        // H or H**H is applied to C(i:m,1:n)

        MI = M - I + 1;
        IC = I;
      } else {
        // H or H**H is applied to C(1:m,i:n)

        NI = N - I + 1;
        JC = I;
      }

      // Apply H or H**H

      zlarzb(
          SIDE,
          TRANST,
          'Backward',
          'Rowwise',
          MI,
          NI,
          IB,
          L,
          A(I, JA),
          LDA,
          WORK(IWT).asMatrix(LDT),
          LDT,
          C(IC, JC),
          LDC,
          WORK.asMatrix(LDWORK),
          LDWORK);
    }
  }

  WORK[1] = LWKOPT.toComplex();
}