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_dgeev_work( int matrix_order, char jobvl, char jobvr,
00038 lapack_int n, double* a, lapack_int lda,
00039 double* wr, double* wi, double* vl,
00040 lapack_int ldvl, double* vr, lapack_int ldvr,
00041 double* work, lapack_int lwork )
00042 {
00043 lapack_int info = 0;
00044 if( matrix_order == LAPACK_COL_MAJOR ) {
00045
00046 LAPACK_dgeev( &jobvl, &jobvr, &n, a, &lda, wr, wi, vl, &ldvl, vr, &ldvr,
00047 work, &lwork, &info );
00048 if( info < 0 ) {
00049 info = info - 1;
00050 }
00051 } else if( matrix_order == LAPACK_ROW_MAJOR ) {
00052 lapack_int lda_t = MAX(1,n);
00053 lapack_int ldvl_t = MAX(1,n);
00054 lapack_int ldvr_t = MAX(1,n);
00055 double* a_t = NULL;
00056 double* vl_t = NULL;
00057 double* vr_t = NULL;
00058
00059 if( lda < n ) {
00060 info = -6;
00061 LAPACKE_xerbla( "LAPACKE_dgeev_work", info );
00062 return info;
00063 }
00064 if( ldvl < n ) {
00065 info = -10;
00066 LAPACKE_xerbla( "LAPACKE_dgeev_work", info );
00067 return info;
00068 }
00069 if( ldvr < n ) {
00070 info = -12;
00071 LAPACKE_xerbla( "LAPACKE_dgeev_work", info );
00072 return info;
00073 }
00074
00075 if( lwork == -1 ) {
00076 LAPACK_dgeev( &jobvl, &jobvr, &n, a, &lda_t, wr, wi, vl, &ldvl_t,
00077 vr, &ldvr_t, work, &lwork, &info );
00078 return (info < 0) ? (info - 1) : info;
00079 }
00080
00081 a_t = (double*)LAPACKE_malloc( sizeof(double) * lda_t * MAX(1,n) );
00082 if( a_t == NULL ) {
00083 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00084 goto exit_level_0;
00085 }
00086 if( LAPACKE_lsame( jobvl, 'v' ) ) {
00087 vl_t = (double*)
00088 LAPACKE_malloc( sizeof(double) * ldvl_t * MAX(1,n) );
00089 if( vl_t == NULL ) {
00090 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00091 goto exit_level_1;
00092 }
00093 }
00094 if( LAPACKE_lsame( jobvr, 'v' ) ) {
00095 vr_t = (double*)
00096 LAPACKE_malloc( sizeof(double) * ldvr_t * MAX(1,n) );
00097 if( vr_t == NULL ) {
00098 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00099 goto exit_level_2;
00100 }
00101 }
00102
00103 LAPACKE_dge_trans( matrix_order, n, n, a, lda, a_t, lda_t );
00104
00105 LAPACK_dgeev( &jobvl, &jobvr, &n, a_t, &lda_t, wr, wi, vl_t, &ldvl_t,
00106 vr_t, &ldvr_t, work, &lwork, &info );
00107 if( info < 0 ) {
00108 info = info - 1;
00109 }
00110
00111 LAPACKE_dge_trans( LAPACK_COL_MAJOR, n, n, a_t, lda_t, a, lda );
00112 if( LAPACKE_lsame( jobvl, 'v' ) ) {
00113 LAPACKE_dge_trans( LAPACK_COL_MAJOR, n, n, vl_t, ldvl_t, vl, ldvl );
00114 }
00115 if( LAPACKE_lsame( jobvr, 'v' ) ) {
00116 LAPACKE_dge_trans( LAPACK_COL_MAJOR, n, n, vr_t, ldvr_t, vr, ldvr );
00117 }
00118
00119 if( LAPACKE_lsame( jobvr, 'v' ) ) {
00120 LAPACKE_free( vr_t );
00121 }
00122 exit_level_2:
00123 if( LAPACKE_lsame( jobvl, 'v' ) ) {
00124 LAPACKE_free( vl_t );
00125 }
00126 exit_level_1:
00127 LAPACKE_free( a_t );
00128 exit_level_0:
00129 if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
00130 LAPACKE_xerbla( "LAPACKE_dgeev_work", info );
00131 }
00132 } else {
00133 info = -1;
00134 LAPACKE_xerbla( "LAPACKE_dgeev_work", info );
00135 }
00136 return info;
00137 }