00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034 #include "lapacke.h"
00035 #include "lapacke_utils.h"
00036
00037 lapack_int LAPACKE_ztgexc_work( int matrix_order, lapack_logical wantq,
00038 lapack_logical wantz, lapack_int n,
00039 lapack_complex_double* a, lapack_int lda,
00040 lapack_complex_double* b, lapack_int ldb,
00041 lapack_complex_double* q, lapack_int ldq,
00042 lapack_complex_double* z, lapack_int ldz,
00043 lapack_int ifst, lapack_int ilst )
00044 {
00045 lapack_int info = 0;
00046 if( matrix_order == LAPACK_COL_MAJOR ) {
00047
00048 LAPACK_ztgexc( &wantq, &wantz, &n, a, &lda, b, &ldb, q, &ldq, z, &ldz,
00049 &ifst, &ilst, &info );
00050 if( info < 0 ) {
00051 info = info - 1;
00052 }
00053 } else if( matrix_order == LAPACK_ROW_MAJOR ) {
00054 lapack_int lda_t = MAX(1,n);
00055 lapack_int ldb_t = MAX(1,n);
00056 lapack_int ldq_t = MAX(1,n);
00057 lapack_int ldz_t = MAX(1,n);
00058 lapack_complex_double* a_t = NULL;
00059 lapack_complex_double* b_t = NULL;
00060 lapack_complex_double* q_t = NULL;
00061 lapack_complex_double* z_t = NULL;
00062
00063 if( lda < n ) {
00064 info = -6;
00065 LAPACKE_xerbla( "LAPACKE_ztgexc_work", info );
00066 return info;
00067 }
00068 if( ldb < n ) {
00069 info = -8;
00070 LAPACKE_xerbla( "LAPACKE_ztgexc_work", info );
00071 return info;
00072 }
00073 if( ldq < n ) {
00074 info = -10;
00075 LAPACKE_xerbla( "LAPACKE_ztgexc_work", info );
00076 return info;
00077 }
00078 if( ldz < n ) {
00079 info = -12;
00080 LAPACKE_xerbla( "LAPACKE_ztgexc_work", info );
00081 return info;
00082 }
00083
00084 a_t = (lapack_complex_double*)
00085 LAPACKE_malloc( sizeof(lapack_complex_double) * lda_t * MAX(1,n) );
00086 if( a_t == NULL ) {
00087 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00088 goto exit_level_0;
00089 }
00090 b_t = (lapack_complex_double*)
00091 LAPACKE_malloc( sizeof(lapack_complex_double) * ldb_t * MAX(1,n) );
00092 if( b_t == NULL ) {
00093 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00094 goto exit_level_1;
00095 }
00096 if( wantq ) {
00097 q_t = (lapack_complex_double*)
00098 LAPACKE_malloc( sizeof(lapack_complex_double) *
00099 ldq_t * MAX(1,n) );
00100 if( q_t == NULL ) {
00101 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00102 goto exit_level_2;
00103 }
00104 }
00105 if( wantz ) {
00106 z_t = (lapack_complex_double*)
00107 LAPACKE_malloc( sizeof(lapack_complex_double) *
00108 ldz_t * MAX(1,n) );
00109 if( z_t == NULL ) {
00110 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00111 goto exit_level_3;
00112 }
00113 }
00114
00115 LAPACKE_zge_trans( matrix_order, n, n, a, lda, a_t, lda_t );
00116 LAPACKE_zge_trans( matrix_order, n, n, b, ldb, b_t, ldb_t );
00117 if( wantq ) {
00118 LAPACKE_zge_trans( matrix_order, n, n, q, ldq, q_t, ldq_t );
00119 }
00120 if( wantz ) {
00121 LAPACKE_zge_trans( matrix_order, n, n, z, ldz, z_t, ldz_t );
00122 }
00123
00124 LAPACK_ztgexc( &wantq, &wantz, &n, a_t, &lda_t, b_t, &ldb_t, q_t,
00125 &ldq_t, z_t, &ldz_t, &ifst, &ilst, &info );
00126 if( info < 0 ) {
00127 info = info - 1;
00128 }
00129
00130 LAPACKE_zge_trans( LAPACK_COL_MAJOR, n, n, a_t, lda_t, a, lda );
00131 LAPACKE_zge_trans( LAPACK_COL_MAJOR, n, n, b_t, ldb_t, b, ldb );
00132 if( wantq ) {
00133 LAPACKE_zge_trans( LAPACK_COL_MAJOR, n, n, q_t, ldq_t, q, ldq );
00134 }
00135 if( wantz ) {
00136 LAPACKE_zge_trans( LAPACK_COL_MAJOR, n, n, z_t, ldz_t, z, ldz );
00137 }
00138
00139 if( wantz ) {
00140 LAPACKE_free( z_t );
00141 }
00142 exit_level_3:
00143 if( wantq ) {
00144 LAPACKE_free( q_t );
00145 }
00146 exit_level_2:
00147 LAPACKE_free( b_t );
00148 exit_level_1:
00149 LAPACKE_free( a_t );
00150 exit_level_0:
00151 if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
00152 LAPACKE_xerbla( "LAPACKE_ztgexc_work", info );
00153 }
00154 } else {
00155 info = -1;
00156 LAPACKE_xerbla( "LAPACKE_ztgexc_work", info );
00157 }
00158 return info;
00159 }