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
|
#include "rb_lapack.h"
extern VOID csymv_(char* uplo, integer* n, complex* alpha, complex* a, integer* lda, complex* x, integer* incx, complex* beta, complex* y, integer* incy);
static VALUE
rblapack_csymv(int argc, VALUE *argv, VALUE self){
VALUE rblapack_uplo;
char uplo;
VALUE rblapack_alpha;
complex alpha;
VALUE rblapack_a;
complex *a;
VALUE rblapack_x;
complex *x;
VALUE rblapack_incx;
integer incx;
VALUE rblapack_beta;
complex beta;
VALUE rblapack_y;
complex *y;
VALUE rblapack_incy;
integer incy;
VALUE rblapack_y_out__;
complex *y_out__;
integer lda;
integer n;
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 y = NumRu::Lapack.csymv( uplo, alpha, a, x, incx, beta, y, incy, [:usage => usage, :help => help])\n\n\nFORTRAN MANUAL\n SUBROUTINE CSYMV( UPLO, N, ALPHA, A, LDA, X, INCX, BETA, Y, INCY )\n\n* Purpose\n* =======\n*\n* CSYMV performs the matrix-vector operation\n*\n* y := alpha*A*x + beta*y,\n*\n* where alpha and beta are scalars, x and y are n element vectors and\n* A is an n by n symmetric matrix.\n*\n\n* Arguments\n* ==========\n*\n* UPLO (input) CHARACTER*1\n* On entry, UPLO specifies whether the upper or lower\n* triangular part of the array A is to be referenced as\n* follows:\n*\n* UPLO = 'U' or 'u' Only the upper triangular part of A\n* is to be referenced.\n*\n* UPLO = 'L' or 'l' Only the lower triangular part of A\n* is to be referenced.\n*\n* Unchanged on exit.\n*\n* N (input) INTEGER\n* On entry, N specifies the order of the matrix A.\n* N must be at least zero.\n* Unchanged on exit.\n*\n* ALPHA (input) COMPLEX\n* On entry, ALPHA specifies the scalar alpha.\n* Unchanged on exit.\n*\n* A (input) COMPLEX array, dimension ( LDA, N )\n* Before entry, with UPLO = 'U' or 'u', the leading n by n\n* upper triangular part of the array A must contain the upper\n* triangular part of the symmetric matrix and the strictly\n* lower triangular part of A is not referenced.\n* Before entry, with UPLO = 'L' or 'l', the leading n by n\n* lower triangular part of the array A must contain the lower\n* triangular part of the symmetric matrix and the strictly\n* upper triangular part of A is not referenced.\n* Unchanged on exit.\n*\n* LDA (input) INTEGER\n* On entry, LDA specifies the first dimension of A as declared\n* in the calling (sub) program. LDA must be at least\n* max( 1, N ).\n* Unchanged on exit.\n*\n* X (input) COMPLEX array, dimension at least\n* ( 1 + ( N - 1 )*abs( INCX ) ).\n* Before entry, the incremented array X must contain the N-\n* element vector x.\n* Unchanged on exit.\n*\n* INCX (input) INTEGER\n* On entry, INCX specifies the increment for the elements of\n* X. INCX must not be zero.\n* Unchanged on exit.\n*\n* BETA (input) COMPLEX\n* On entry, BETA specifies the scalar beta. When BETA is\n* supplied as zero then Y need not be set on input.\n* Unchanged on exit.\n*\n* Y (input/output) COMPLEX array, dimension at least\n* ( 1 + ( N - 1 )*abs( INCY ) ).\n* Before entry, the incremented array Y must contain the n\n* element vector y. On exit, Y is overwritten by the updated\n* vector y.\n*\n* INCY (input) INTEGER\n* On entry, INCY specifies the increment for the elements of\n* Y. INCY must not be zero.\n* Unchanged on exit.\n*\n\n* =====================================================================\n*\n\n");
return Qnil;
}
if (rb_hash_aref(rblapack_options, sUsage) == Qtrue) {
printf("%s\n", "USAGE:\n y = NumRu::Lapack.csymv( uplo, alpha, a, x, incx, beta, y, incy, [: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_uplo = argv[0];
rblapack_alpha = argv[1];
rblapack_a = argv[2];
rblapack_x = argv[3];
rblapack_incx = argv[4];
rblapack_beta = argv[5];
rblapack_y = argv[6];
rblapack_incy = argv[7];
if (argc == 8) {
} else if (rblapack_options != Qnil) {
} else {
}
uplo = StringValueCStr(rblapack_uplo)[0];
if (!NA_IsNArray(rblapack_a))
rb_raise(rb_eArgError, "a (3th argument) must be NArray");
if (NA_RANK(rblapack_a) != 2)
rb_raise(rb_eArgError, "rank of a (3th argument) must be %d", 2);
lda = NA_SHAPE0(rblapack_a);
n = NA_SHAPE1(rblapack_a);
if (NA_TYPE(rblapack_a) != NA_SCOMPLEX)
rblapack_a = na_change_type(rblapack_a, NA_SCOMPLEX);
a = NA_PTR_TYPE(rblapack_a, complex*);
incx = NUM2INT(rblapack_incx);
incy = NUM2INT(rblapack_incy);
alpha.r = (real)NUM2DBL(rb_funcall(rblapack_alpha, rb_intern("real"), 0));
alpha.i = (real)NUM2DBL(rb_funcall(rblapack_alpha, rb_intern("imag"), 0));
beta.r = (real)NUM2DBL(rb_funcall(rblapack_beta, rb_intern("real"), 0));
beta.i = (real)NUM2DBL(rb_funcall(rblapack_beta, rb_intern("imag"), 0));
if (!NA_IsNArray(rblapack_x))
rb_raise(rb_eArgError, "x (4th argument) must be NArray");
if (NA_RANK(rblapack_x) != 1)
rb_raise(rb_eArgError, "rank of x (4th argument) must be %d", 1);
if (NA_SHAPE0(rblapack_x) != (1 + ( n - 1 )*abs( incx )))
rb_raise(rb_eRuntimeError, "shape 0 of x must be %d", 1 + ( n - 1 )*abs( incx ));
if (NA_TYPE(rblapack_x) != NA_SCOMPLEX)
rblapack_x = na_change_type(rblapack_x, NA_SCOMPLEX);
x = NA_PTR_TYPE(rblapack_x, complex*);
if (!NA_IsNArray(rblapack_y))
rb_raise(rb_eArgError, "y (7th argument) must be NArray");
if (NA_RANK(rblapack_y) != 1)
rb_raise(rb_eArgError, "rank of y (7th argument) must be %d", 1);
if (NA_SHAPE0(rblapack_y) != (1 + ( n - 1 )*abs( incy )))
rb_raise(rb_eRuntimeError, "shape 0 of y must be %d", 1 + ( n - 1 )*abs( incy ));
if (NA_TYPE(rblapack_y) != NA_SCOMPLEX)
rblapack_y = na_change_type(rblapack_y, NA_SCOMPLEX);
y = NA_PTR_TYPE(rblapack_y, complex*);
{
na_shape_t shape[1];
shape[0] = 1 + ( n - 1 )*abs( incy );
rblapack_y_out__ = na_make_object(NA_SCOMPLEX, 1, shape, cNArray);
}
y_out__ = NA_PTR_TYPE(rblapack_y_out__, complex*);
MEMCPY(y_out__, y, complex, NA_TOTAL(rblapack_y));
rblapack_y = rblapack_y_out__;
y = y_out__;
csymv_(&uplo, &n, &alpha, a, &lda, x, &incx, &beta, y, &incy);
return rblapack_y;
}
void
init_lapack_csymv(VALUE mLapack, VALUE sH, VALUE sU, VALUE zero){
sHelp = sH;
sUsage = sU;
rblapack_ZERO = zero;
rb_define_module_function(mLapack, "csymv", rblapack_csymv, -1);
}
|
