1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132
|
#include "rb_lapack.h"
extern VOID zlagtm_(char* trans, integer* n, integer* nrhs, doublereal* alpha, doublecomplex* dl, doublecomplex* d, doublecomplex* du, doublecomplex* x, integer* ldx, doublereal* beta, doublecomplex* b, integer* ldb);
static VALUE
rblapack_zlagtm(int argc, VALUE *argv, VALUE self){
VALUE rblapack_trans;
char trans;
VALUE rblapack_alpha;
doublereal alpha;
VALUE rblapack_dl;
doublecomplex *dl;
VALUE rblapack_d;
doublecomplex *d;
VALUE rblapack_du;
doublecomplex *du;
VALUE rblapack_x;
doublecomplex *x;
VALUE rblapack_beta;
doublereal beta;
VALUE rblapack_b;
doublecomplex *b;
VALUE rblapack_b_out__;
doublecomplex *b_out__;
integer n;
integer ldx;
integer nrhs;
integer ldb;
VALUE rblapack_options;
if (argc > 0 && TYPE(argv[argc-1]) == T_HASH) {
argc--;
rblapack_options = argv[argc];
if (rb_hash_aref(rblapack_options, sHelp) == Qtrue) {
printf("%s\n", "USAGE:\n b = NumRu::Lapack.zlagtm( trans, alpha, dl, d, du, x, beta, b, [:usage => usage, :help => help])\n\n\nFORTRAN MANUAL\n SUBROUTINE ZLAGTM( TRANS, N, NRHS, ALPHA, DL, D, DU, X, LDX, BETA, B, LDB )\n\n* Purpose\n* =======\n*\n* ZLAGTM performs a matrix-vector product of the form\n*\n* B := alpha * A * X + beta * B\n*\n* where A is a tridiagonal matrix of order N, B and X are N by NRHS\n* matrices, and alpha and beta are real scalars, each of which may be\n* 0., 1., or -1.\n*\n\n* Arguments\n* =========\n*\n* TRANS (input) CHARACTER*1\n* Specifies the operation applied to A.\n* = 'N': No transpose, B := alpha * A * X + beta * B\n* = 'T': Transpose, B := alpha * A**T * X + beta * B\n* = 'C': Conjugate transpose, B := alpha * A**H * X + beta * B\n*\n* N (input) INTEGER\n* The order of the matrix A. N >= 0.\n*\n* NRHS (input) INTEGER\n* The number of right hand sides, i.e., the number of columns\n* of the matrices X and B.\n*\n* ALPHA (input) DOUBLE PRECISION\n* The scalar alpha. ALPHA must be 0., 1., or -1.; otherwise,\n* it is assumed to be 0.\n*\n* DL (input) COMPLEX*16 array, dimension (N-1)\n* The (n-1) sub-diagonal elements of T.\n*\n* D (input) COMPLEX*16 array, dimension (N)\n* The diagonal elements of T.\n*\n* DU (input) COMPLEX*16 array, dimension (N-1)\n* The (n-1) super-diagonal elements of T.\n*\n* X (input) COMPLEX*16 array, dimension (LDX,NRHS)\n* The N by NRHS matrix X.\n* LDX (input) INTEGER\n* The leading dimension of the array X. LDX >= max(N,1).\n*\n* BETA (input) DOUBLE PRECISION\n* The scalar beta. BETA must be 0., 1., or -1.; otherwise,\n* it is assumed to be 1.\n*\n* B (input/output) COMPLEX*16 array, dimension (LDB,NRHS)\n* On entry, the N by NRHS matrix B.\n* On exit, B is overwritten by the matrix expression\n* B := alpha * A * X + beta * B.\n*\n* LDB (input) INTEGER\n* The leading dimension of the array B. LDB >= max(N,1).\n*\n\n* =====================================================================\n*\n\n");
return Qnil;
}
if (rb_hash_aref(rblapack_options, sUsage) == Qtrue) {
printf("%s\n", "USAGE:\n b = NumRu::Lapack.zlagtm( trans, alpha, dl, d, du, x, beta, b, [:usage => usage, :help => help])\n");
return Qnil;
}
} else
rblapack_options = Qnil;
if (argc != 8 && argc != 8)
rb_raise(rb_eArgError,"wrong number of arguments (%d for 8)", argc);
rblapack_trans = argv[0];
rblapack_alpha = argv[1];
rblapack_dl = argv[2];
rblapack_d = argv[3];
rblapack_du = argv[4];
rblapack_x = argv[5];
rblapack_beta = argv[6];
rblapack_b = argv[7];
if (argc == 8) {
} else if (rblapack_options != Qnil) {
} else {
}
trans = StringValueCStr(rblapack_trans)[0];
if (!NA_IsNArray(rblapack_d))
rb_raise(rb_eArgError, "d (4th argument) must be NArray");
if (NA_RANK(rblapack_d) != 1)
rb_raise(rb_eArgError, "rank of d (4th argument) must be %d", 1);
n = NA_SHAPE0(rblapack_d);
if (NA_TYPE(rblapack_d) != NA_DCOMPLEX)
rblapack_d = na_change_type(rblapack_d, NA_DCOMPLEX);
d = NA_PTR_TYPE(rblapack_d, doublecomplex*);
if (!NA_IsNArray(rblapack_x))
rb_raise(rb_eArgError, "x (6th argument) must be NArray");
if (NA_RANK(rblapack_x) != 2)
rb_raise(rb_eArgError, "rank of x (6th argument) must be %d", 2);
ldx = NA_SHAPE0(rblapack_x);
nrhs = NA_SHAPE1(rblapack_x);
if (NA_TYPE(rblapack_x) != NA_DCOMPLEX)
rblapack_x = na_change_type(rblapack_x, NA_DCOMPLEX);
x = NA_PTR_TYPE(rblapack_x, doublecomplex*);
if (!NA_IsNArray(rblapack_b))
rb_raise(rb_eArgError, "b (8th argument) must be NArray");
if (NA_RANK(rblapack_b) != 2)
rb_raise(rb_eArgError, "rank of b (8th argument) must be %d", 2);
ldb = NA_SHAPE0(rblapack_b);
if (NA_SHAPE1(rblapack_b) != nrhs)
rb_raise(rb_eRuntimeError, "shape 1 of b must be the same as shape 1 of x");
if (NA_TYPE(rblapack_b) != NA_DCOMPLEX)
rblapack_b = na_change_type(rblapack_b, NA_DCOMPLEX);
b = NA_PTR_TYPE(rblapack_b, doublecomplex*);
alpha = NUM2DBL(rblapack_alpha);
if (!NA_IsNArray(rblapack_du))
rb_raise(rb_eArgError, "du (5th argument) must be NArray");
if (NA_RANK(rblapack_du) != 1)
rb_raise(rb_eArgError, "rank of du (5th argument) must be %d", 1);
if (NA_SHAPE0(rblapack_du) != (n-1))
rb_raise(rb_eRuntimeError, "shape 0 of du must be %d", n-1);
if (NA_TYPE(rblapack_du) != NA_DCOMPLEX)
rblapack_du = na_change_type(rblapack_du, NA_DCOMPLEX);
du = NA_PTR_TYPE(rblapack_du, doublecomplex*);
if (!NA_IsNArray(rblapack_dl))
rb_raise(rb_eArgError, "dl (3th argument) must be NArray");
if (NA_RANK(rblapack_dl) != 1)
rb_raise(rb_eArgError, "rank of dl (3th argument) must be %d", 1);
if (NA_SHAPE0(rblapack_dl) != (n-1))
rb_raise(rb_eRuntimeError, "shape 0 of dl must be %d", n-1);
if (NA_TYPE(rblapack_dl) != NA_DCOMPLEX)
rblapack_dl = na_change_type(rblapack_dl, NA_DCOMPLEX);
dl = NA_PTR_TYPE(rblapack_dl, doublecomplex*);
beta = NUM2DBL(rblapack_beta);
{
na_shape_t shape[2];
shape[0] = ldb;
shape[1] = nrhs;
rblapack_b_out__ = na_make_object(NA_DCOMPLEX, 2, shape, cNArray);
}
b_out__ = NA_PTR_TYPE(rblapack_b_out__, doublecomplex*);
MEMCPY(b_out__, b, doublecomplex, NA_TOTAL(rblapack_b));
rblapack_b = rblapack_b_out__;
b = b_out__;
zlagtm_(&trans, &n, &nrhs, &alpha, dl, d, du, x, &ldx, &beta, b, &ldb);
return rblapack_b;
}
void
init_lapack_zlagtm(VALUE mLapack, VALUE sH, VALUE sU, VALUE zero){
sHelp = sH;
sUsage = sU;
rblapack_ZERO = zero;
rb_define_module_function(mLapack, "zlagtm", rblapack_zlagtm, -1);
}
|