zhetrd_2stage function

Implementation

void zhetrd_2stage(
  final String VECT,
  final String UPLO,
  final int N,
  final Matrix<Complex> A_,
  final int LDA,
  final Array<double> D_,
  final Array<double> E_,
  final Array<Complex> TAU_,
  final Array<Complex> HOUS2_,
  final int LHOUS2,
  final Array<Complex> WORK_,
  final int LWORK,
  final Box<int> INFO,
) {
  final A = A_.having(ld: LDA);
  final TAU = TAU_.having();
  final HOUS2 = HOUS2_.having();
  final WORK = WORK_.having();
  final D = D_.having();
  final E = E_.having();
  bool LQUERY, UPPER;
  int KD, IB, LWMIN, LHMIN, LWRK, LDAB, WPOS, ABPOS;

  // Test the input parameters

  INFO.value = 0;
  // WANTQ = lsame(VECT, 'V');
  UPPER = lsame(UPLO, 'U');
  LQUERY = (LWORK == -1) || (LHOUS2 == -1);

  // Determine the block size, the workspace size and the hous size.

  KD = ilaenv2stage(1, 'ZHETRD_2STAGE', VECT, N, -1, -1, -1);
  IB = ilaenv2stage(2, 'ZHETRD_2STAGE', VECT, N, KD, -1, -1);
  if (N == 0) {
    LHMIN = 1;
    LWMIN = 1;
  } else {
    LHMIN = ilaenv2stage(3, 'ZHETRD_2STAGE', VECT, N, KD, IB, -1);
    LWMIN = ilaenv2stage(4, 'ZHETRD_2STAGE', VECT, N, KD, IB, -1);
  }

  if (!lsame(VECT, 'N')) {
    INFO.value = -1;
  } else if (!UPPER && !lsame(UPLO, 'L')) {
    INFO.value = -2;
  } else if (N < 0) {
    INFO.value = -3;
  } else if (LDA < max(1, N)) {
    INFO.value = -5;
  } else if (LHOUS2 < LHMIN && !LQUERY) {
    INFO.value = -10;
  } else if (LWORK < LWMIN && !LQUERY) {
    INFO.value = -12;
  }

  if (INFO.value == 0) {
    HOUS2[1] = LHMIN.toComplex();
    WORK[1] = LWMIN.toComplex();
  }

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

  // Quick return if possible

  if (N == 0) {
    WORK[1] = Complex.one;
    return;
  }

  // Determine pointer position

  LDAB = KD + 1;
  LWRK = LWORK - LDAB * N;
  ABPOS = 1;
  WPOS = ABPOS + LDAB * N;
  zhetrd_he2hb(UPLO, N, KD, A, LDA, WORK(ABPOS).asMatrix(LDAB), LDAB, TAU,
      WORK(WPOS), LWRK, INFO);
  if (INFO.value != 0) {
    xerbla('ZHETRD_HE2HB', -INFO.value);
    return;
  }
  zhetrd_hb2st('Y', VECT, UPLO, N, KD, WORK(ABPOS).asMatrix(LDAB), LDAB, D, E,
      HOUS2, LHOUS2, WORK(WPOS), LWRK, INFO);
  if (INFO.value != 0) {
    xerbla('ZHETRD_HB2ST', -INFO.value);
    return;
  }

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