148 lines
		
	
	
		
			6.4 KiB
		
	
	
	
		
			C
		
	
	
	
			
		
		
	
	
			148 lines
		
	
	
		
			6.4 KiB
		
	
	
	
		
			C
		
	
	
	
| /*****************************************************************************
 | |
|   Copyright (c) 2011, Intel Corp.
 | |
|   All rights reserved.
 | |
| 
 | |
|   Redistribution and use in source and binary forms, with or without
 | |
|   modification, are permitted provided that the following conditions are met:
 | |
| 
 | |
|     * Redistributions of source code must retain the above copyright notice,
 | |
|       this list of conditions and the following disclaimer.
 | |
|     * Redistributions in binary form must reproduce the above copyright
 | |
|       notice, this list of conditions and the following disclaimer in the
 | |
|       documentation and/or other materials provided with the distribution.
 | |
|     * Neither the name of Intel Corporation nor the names of its contributors
 | |
|       may be used to endorse or promote products derived from this software
 | |
|       without specific prior written permission.
 | |
| 
 | |
|   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 | |
|   AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 | |
|   IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 | |
|   ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
 | |
|   LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 | |
|   CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 | |
|   SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 | |
|   INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 | |
|   CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 | |
|   ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
 | |
|   THE POSSIBILITY OF SUCH DAMAGE.
 | |
| *****************************************************************************
 | |
| * Contents: Native middle-level C interface to LAPACK function zhsein
 | |
| * Author: Intel Corporation
 | |
| * Generated November, 2011
 | |
| *****************************************************************************/
 | |
| 
 | |
| #include "lapacke_utils.h"
 | |
| 
 | |
| lapack_int LAPACKE_zhsein_work( int matrix_order, char job, char eigsrc,
 | |
|                                 char initv, const lapack_logical* select,
 | |
|                                 lapack_int n, const lapack_complex_double* h,
 | |
|                                 lapack_int ldh, lapack_complex_double* w,
 | |
|                                 lapack_complex_double* vl, lapack_int ldvl,
 | |
|                                 lapack_complex_double* vr, lapack_int ldvr,
 | |
|                                 lapack_int mm, lapack_int* m,
 | |
|                                 lapack_complex_double* work, double* rwork,
 | |
|                                 lapack_int* ifaill, lapack_int* ifailr )
 | |
| {
 | |
|     lapack_int info = 0;
 | |
|     if( matrix_order == LAPACK_COL_MAJOR ) {
 | |
|         /* Call LAPACK function and adjust info */
 | |
|         LAPACK_zhsein( &job, &eigsrc, &initv, select, &n, h, &ldh, w, vl, &ldvl,
 | |
|                        vr, &ldvr, &mm, m, work, rwork, ifaill, ifailr, &info );
 | |
|         if( info < 0 ) {
 | |
|             info = info - 1;
 | |
|         }
 | |
|     } else if( matrix_order == LAPACK_ROW_MAJOR ) {
 | |
|         lapack_int ldh_t = MAX(1,n);
 | |
|         lapack_int ldvl_t = MAX(1,n);
 | |
|         lapack_int ldvr_t = MAX(1,n);
 | |
|         lapack_complex_double* h_t = NULL;
 | |
|         lapack_complex_double* vl_t = NULL;
 | |
|         lapack_complex_double* vr_t = NULL;
 | |
|         /* Check leading dimension(s) */
 | |
|         if( ldh < n ) {
 | |
|             info = -8;
 | |
|             LAPACKE_xerbla( "LAPACKE_zhsein_work", info );
 | |
|             return info;
 | |
|         }
 | |
|         if( ldvl < mm ) {
 | |
|             info = -11;
 | |
|             LAPACKE_xerbla( "LAPACKE_zhsein_work", info );
 | |
|             return info;
 | |
|         }
 | |
|         if( ldvr < mm ) {
 | |
|             info = -13;
 | |
|             LAPACKE_xerbla( "LAPACKE_zhsein_work", info );
 | |
|             return info;
 | |
|         }
 | |
|         /* Allocate memory for temporary array(s) */
 | |
|         h_t = (lapack_complex_double*)
 | |
|             LAPACKE_malloc( sizeof(lapack_complex_double) * ldh_t * MAX(1,n) );
 | |
|         if( h_t == NULL ) {
 | |
|             info = LAPACK_TRANSPOSE_MEMORY_ERROR;
 | |
|             goto exit_level_0;
 | |
|         }
 | |
|         if( LAPACKE_lsame( job, 'b' ) || LAPACKE_lsame( job, 'l' ) ) {
 | |
|             vl_t = (lapack_complex_double*)
 | |
|                 LAPACKE_malloc( sizeof(lapack_complex_double) *
 | |
|                                 ldvl_t * MAX(1,mm) );
 | |
|             if( vl_t == NULL ) {
 | |
|                 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
 | |
|                 goto exit_level_1;
 | |
|             }
 | |
|         }
 | |
|         if( LAPACKE_lsame( job, 'b' ) || LAPACKE_lsame( job, 'r' ) ) {
 | |
|             vr_t = (lapack_complex_double*)
 | |
|                 LAPACKE_malloc( sizeof(lapack_complex_double) *
 | |
|                                 ldvr_t * MAX(1,mm) );
 | |
|             if( vr_t == NULL ) {
 | |
|                 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
 | |
|                 goto exit_level_2;
 | |
|             }
 | |
|         }
 | |
|         /* Transpose input matrices */
 | |
|         LAPACKE_zge_trans( matrix_order, n, n, h, ldh, h_t, ldh_t );
 | |
|         if( ( LAPACKE_lsame( job, 'l' ) || LAPACKE_lsame( job, 'b' ) ) &&
 | |
|             LAPACKE_lsame( initv, 'v' ) ) {
 | |
|             LAPACKE_zge_trans( matrix_order, n, mm, vl, ldvl, vl_t, ldvl_t );
 | |
|         }
 | |
|         if( ( LAPACKE_lsame( job, 'r' ) || LAPACKE_lsame( job, 'b' ) ) &&
 | |
|             LAPACKE_lsame( initv, 'v' ) ) {
 | |
|             LAPACKE_zge_trans( matrix_order, n, mm, vr, ldvr, vr_t, ldvr_t );
 | |
|         }
 | |
|         /* Call LAPACK function and adjust info */
 | |
|         LAPACK_zhsein( &job, &eigsrc, &initv, select, &n, h_t, &ldh_t, w, vl_t,
 | |
|                        &ldvl_t, vr_t, &ldvr_t, &mm, m, work, rwork, ifaill,
 | |
|                        ifailr, &info );
 | |
|         if( info < 0 ) {
 | |
|             info = info - 1;
 | |
|         }
 | |
|         /* Transpose output matrices */
 | |
|         if( LAPACKE_lsame( job, 'b' ) || LAPACKE_lsame( job, 'l' ) ) {
 | |
|             LAPACKE_zge_trans( LAPACK_COL_MAJOR, n, mm, vl_t, ldvl_t, vl,
 | |
|                                ldvl );
 | |
|         }
 | |
|         if( LAPACKE_lsame( job, 'b' ) || LAPACKE_lsame( job, 'r' ) ) {
 | |
|             LAPACKE_zge_trans( LAPACK_COL_MAJOR, n, mm, vr_t, ldvr_t, vr,
 | |
|                                ldvr );
 | |
|         }
 | |
|         /* Release memory and exit */
 | |
|         if( LAPACKE_lsame( job, 'b' ) || LAPACKE_lsame( job, 'r' ) ) {
 | |
|             LAPACKE_free( vr_t );
 | |
|         }
 | |
| exit_level_2:
 | |
|         if( LAPACKE_lsame( job, 'b' ) || LAPACKE_lsame( job, 'l' ) ) {
 | |
|             LAPACKE_free( vl_t );
 | |
|         }
 | |
| exit_level_1:
 | |
|         LAPACKE_free( h_t );
 | |
| exit_level_0:
 | |
|         if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
 | |
|             LAPACKE_xerbla( "LAPACKE_zhsein_work", info );
 | |
|         }
 | |
|     } else {
 | |
|         info = -1;
 | |
|         LAPACKE_xerbla( "LAPACKE_zhsein_work", info );
 | |
|     }
 | |
|     return info;
 | |
| }
 |