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_stgevc_work( int matrix_order, char side, char howmny,
00038 const lapack_logical* select, lapack_int n,
00039 const float* s, lapack_int lds, const float* p,
00040 lapack_int ldp, float* vl, lapack_int ldvl,
00041 float* vr, lapack_int ldvr, lapack_int mm,
00042 lapack_int* m, float* work )
00043 {
00044 lapack_int info = 0;
00045 if( matrix_order == LAPACK_COL_MAJOR ) {
00046
00047 LAPACK_stgevc( &side, &howmny, select, &n, s, &lds, p, &ldp, vl, &ldvl,
00048 vr, &ldvr, &mm, m, work, &info );
00049 if( info < 0 ) {
00050 info = info - 1;
00051 }
00052 } else if( matrix_order == LAPACK_ROW_MAJOR ) {
00053 lapack_int ldp_t = MAX(1,n);
00054 lapack_int lds_t = MAX(1,n);
00055 lapack_int ldvl_t = MAX(1,n);
00056 lapack_int ldvr_t = MAX(1,n);
00057 float* s_t = NULL;
00058 float* p_t = NULL;
00059 float* vl_t = NULL;
00060 float* vr_t = NULL;
00061
00062 if( ldp < n ) {
00063 info = -9;
00064 LAPACKE_xerbla( "LAPACKE_stgevc_work", info );
00065 return info;
00066 }
00067 if( lds < n ) {
00068 info = -7;
00069 LAPACKE_xerbla( "LAPACKE_stgevc_work", info );
00070 return info;
00071 }
00072 if( ldvl < mm ) {
00073 info = -11;
00074 LAPACKE_xerbla( "LAPACKE_stgevc_work", info );
00075 return info;
00076 }
00077 if( ldvr < mm ) {
00078 info = -13;
00079 LAPACKE_xerbla( "LAPACKE_stgevc_work", info );
00080 return info;
00081 }
00082
00083 s_t = (float*)LAPACKE_malloc( sizeof(float) * lds_t * MAX(1,n) );
00084 if( s_t == NULL ) {
00085 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00086 goto exit_level_0;
00087 }
00088 p_t = (float*)LAPACKE_malloc( sizeof(float) * ldp_t * MAX(1,n) );
00089 if( p_t == NULL ) {
00090 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00091 goto exit_level_1;
00092 }
00093 if( LAPACKE_lsame( side, 'b' ) || LAPACKE_lsame( side, 'l' ) ) {
00094 vl_t = (float*)LAPACKE_malloc( sizeof(float) * ldvl_t * MAX(1,mm) );
00095 if( vl_t == NULL ) {
00096 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00097 goto exit_level_2;
00098 }
00099 }
00100 if( LAPACKE_lsame( side, 'b' ) || LAPACKE_lsame( side, 'r' ) ) {
00101 vr_t = (float*)LAPACKE_malloc( sizeof(float) * ldvr_t * MAX(1,mm) );
00102 if( vr_t == NULL ) {
00103 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00104 goto exit_level_3;
00105 }
00106 }
00107
00108 LAPACKE_sge_trans( matrix_order, n, n, s, lds, s_t, lds_t );
00109 LAPACKE_sge_trans( matrix_order, n, n, p, ldp, p_t, ldp_t );
00110 if( ( LAPACKE_lsame( side, 'l' ) || LAPACKE_lsame( side, 'b' ) ) &&
00111 LAPACKE_lsame( howmny, 'b' ) ) {
00112 LAPACKE_sge_trans( matrix_order, n, mm, vl, ldvl, vl_t, ldvl_t );
00113 }
00114 if( ( LAPACKE_lsame( side, 'r' ) || LAPACKE_lsame( side, 'b' ) ) &&
00115 LAPACKE_lsame( howmny, 'b' ) ) {
00116 LAPACKE_sge_trans( matrix_order, n, mm, vr, ldvr, vr_t, ldvr_t );
00117 }
00118
00119 LAPACK_stgevc( &side, &howmny, select, &n, s_t, &lds_t, p_t, &ldp_t,
00120 vl_t, &ldvl_t, vr_t, &ldvr_t, &mm, m, work, &info );
00121 if( info < 0 ) {
00122 info = info - 1;
00123 }
00124
00125 if( LAPACKE_lsame( side, 'b' ) || LAPACKE_lsame( side, 'l' ) ) {
00126 LAPACKE_sge_trans( LAPACK_COL_MAJOR, n, mm, vl_t, ldvl_t, vl,
00127 ldvl );
00128 }
00129 if( LAPACKE_lsame( side, 'b' ) || LAPACKE_lsame( side, 'r' ) ) {
00130 LAPACKE_sge_trans( LAPACK_COL_MAJOR, n, mm, vr_t, ldvr_t, vr,
00131 ldvr );
00132 }
00133
00134 if( LAPACKE_lsame( side, 'b' ) || LAPACKE_lsame( side, 'r' ) ) {
00135 LAPACKE_free( vr_t );
00136 }
00137 exit_level_3:
00138 if( LAPACKE_lsame( side, 'b' ) || LAPACKE_lsame( side, 'l' ) ) {
00139 LAPACKE_free( vl_t );
00140 }
00141 exit_level_2:
00142 LAPACKE_free( p_t );
00143 exit_level_1:
00144 LAPACKE_free( s_t );
00145 exit_level_0:
00146 if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
00147 LAPACKE_xerbla( "LAPACKE_stgevc_work", info );
00148 }
00149 } else {
00150 info = -1;
00151 LAPACKE_xerbla( "LAPACKE_stgevc_work", info );
00152 }
00153 return info;
00154 }