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_dtrsen_work( int matrix_order, char job, char compq,
00038 const lapack_logical* select, lapack_int n,
00039 double* t, lapack_int ldt, double* q,
00040 lapack_int ldq, double* wr, double* wi,
00041 lapack_int* m, double* s, double* sep,
00042 double* work, lapack_int lwork,
00043 lapack_int* iwork, lapack_int liwork )
00044 {
00045 lapack_int info = 0;
00046 if( matrix_order == LAPACK_COL_MAJOR ) {
00047
00048 LAPACK_dtrsen( &job, &compq, select, &n, t, &ldt, q, &ldq, wr, wi, m, s,
00049 sep, work, &lwork, iwork, &liwork, &info );
00050 if( info < 0 ) {
00051 info = info - 1;
00052 }
00053 } else if( matrix_order == LAPACK_ROW_MAJOR ) {
00054 lapack_int ldq_t = MAX(1,n);
00055 lapack_int ldt_t = MAX(1,n);
00056 double* t_t = NULL;
00057 double* q_t = NULL;
00058
00059 if( ldq < n ) {
00060 info = -9;
00061 LAPACKE_xerbla( "LAPACKE_dtrsen_work", info );
00062 return info;
00063 }
00064 if( ldt < n ) {
00065 info = -7;
00066 LAPACKE_xerbla( "LAPACKE_dtrsen_work", info );
00067 return info;
00068 }
00069
00070 t_t = (double*)LAPACKE_malloc( sizeof(double) * ldt_t * MAX(1,n) );
00071 if( t_t == NULL ) {
00072 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00073 goto exit_level_0;
00074 }
00075
00076 LAPACKE_dge_trans( matrix_order, n, n, t, ldt, t_t, ldt_t );
00077
00078 if( liwork == -1 || lwork == -1 ) {
00079 LAPACK_dtrsen( &job, &compq, select, &n, t_t, &ldt_t, q, &ldq_t, wr,
00080 wi, m, s, sep, work, &lwork, iwork, &liwork, &info );
00081 LAPACKE_free( t_t );
00082 return (info < 0) ? (info - 1) : info;
00083 }
00084
00085 if( LAPACKE_lsame( compq, 'v' ) ) {
00086 q_t = (double*)LAPACKE_malloc( sizeof(double) * ldq_t * MAX(1,n) );
00087 if( q_t == NULL ) {
00088 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00089 goto exit_level_1;
00090 }
00091 }
00092
00093 if( LAPACKE_lsame( compq, 'v' ) ) {
00094 LAPACKE_dge_trans( matrix_order, n, n, q, ldq, q_t, ldq_t );
00095 }
00096
00097 LAPACK_dtrsen( &job, &compq, select, &n, t_t, &ldt_t, q_t, &ldq_t, wr,
00098 wi, m, s, sep, work, &lwork, iwork, &liwork, &info );
00099 if( info < 0 ) {
00100 info = info - 1;
00101 }
00102
00103 LAPACKE_dge_trans( LAPACK_COL_MAJOR, n, n, t_t, ldt_t, t, ldt );
00104 if( LAPACKE_lsame( compq, 'v' ) ) {
00105 LAPACKE_dge_trans( LAPACK_COL_MAJOR, n, n, q_t, ldq_t, q, ldq );
00106 }
00107
00108 if( LAPACKE_lsame( compq, 'v' ) ) {
00109 LAPACKE_free( q_t );
00110 }
00111 exit_level_1:
00112 LAPACKE_free( t_t );
00113 exit_level_0:
00114 if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
00115 LAPACKE_xerbla( "LAPACKE_dtrsen_work", info );
00116 }
00117 } else {
00118 info = -1;
00119 LAPACKE_xerbla( "LAPACKE_dtrsen_work", info );
00120 }
00121 return info;
00122 }