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
|
#include "rb_lapack.h"
extern VOID zlarnv_(integer* idist, integer* iseed, integer* n, doublecomplex* x);
static VALUE
rblapack_zlarnv(int argc, VALUE *argv, VALUE self){
VALUE rblapack_idist;
integer idist;
VALUE rblapack_iseed;
integer *iseed;
VALUE rblapack_n;
integer n;
VALUE rblapack_x;
doublecomplex *x;
VALUE rblapack_iseed_out__;
integer *iseed_out__;
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 x, iseed = NumRu::Lapack.zlarnv( idist, iseed, n, [:usage => usage, :help => help])\n\n\nFORTRAN MANUAL\n SUBROUTINE ZLARNV( IDIST, ISEED, N, X )\n\n* Purpose\n* =======\n*\n* ZLARNV returns a vector of n random complex numbers from a uniform or\n* normal distribution.\n*\n\n* Arguments\n* =========\n*\n* IDIST (input) INTEGER\n* Specifies the distribution of the random numbers:\n* = 1: real and imaginary parts each uniform (0,1)\n* = 2: real and imaginary parts each uniform (-1,1)\n* = 3: real and imaginary parts each normal (0,1)\n* = 4: uniformly distributed on the disc abs(z) < 1\n* = 5: uniformly distributed on the circle abs(z) = 1\n*\n* ISEED (input/output) INTEGER array, dimension (4)\n* On entry, the seed of the random number generator; the array\n* elements must be between 0 and 4095, and ISEED(4) must be\n* odd.\n* On exit, the seed is updated.\n*\n* N (input) INTEGER\n* The number of random numbers to be generated.\n*\n* X (output) COMPLEX*16 array, dimension (N)\n* The generated random numbers.\n*\n\n* Further Details\n* ===============\n*\n* This routine calls the auxiliary routine DLARUV to generate random\n* real numbers from a uniform (0,1) distribution, in batches of up to\n* 128 using vectorisable code. The Box-Muller method is used to\n* transform numbers from a uniform to a normal distribution.\n*\n* =====================================================================\n*\n\n");
return Qnil;
}
if (rb_hash_aref(rblapack_options, sUsage) == Qtrue) {
printf("%s\n", "USAGE:\n x, iseed = NumRu::Lapack.zlarnv( idist, iseed, n, [:usage => usage, :help => help])\n");
return Qnil;
}
} else
rblapack_options = Qnil;
if (argc != 3 && argc != 3)
rb_raise(rb_eArgError,"wrong number of arguments (%d for 3)", argc);
rblapack_idist = argv[0];
rblapack_iseed = argv[1];
rblapack_n = argv[2];
if (argc == 3) {
} else if (rblapack_options != Qnil) {
} else {
}
idist = NUM2INT(rblapack_idist);
n = NUM2INT(rblapack_n);
if (!NA_IsNArray(rblapack_iseed))
rb_raise(rb_eArgError, "iseed (2th argument) must be NArray");
if (NA_RANK(rblapack_iseed) != 1)
rb_raise(rb_eArgError, "rank of iseed (2th argument) must be %d", 1);
if (NA_SHAPE0(rblapack_iseed) != (4))
rb_raise(rb_eRuntimeError, "shape 0 of iseed must be %d", 4);
if (NA_TYPE(rblapack_iseed) != NA_LINT)
rblapack_iseed = na_change_type(rblapack_iseed, NA_LINT);
iseed = NA_PTR_TYPE(rblapack_iseed, integer*);
{
na_shape_t shape[1];
shape[0] = MAX(1,n);
rblapack_x = na_make_object(NA_DCOMPLEX, 1, shape, cNArray);
}
x = NA_PTR_TYPE(rblapack_x, doublecomplex*);
{
na_shape_t shape[1];
shape[0] = 4;
rblapack_iseed_out__ = na_make_object(NA_LINT, 1, shape, cNArray);
}
iseed_out__ = NA_PTR_TYPE(rblapack_iseed_out__, integer*);
MEMCPY(iseed_out__, iseed, integer, NA_TOTAL(rblapack_iseed));
rblapack_iseed = rblapack_iseed_out__;
iseed = iseed_out__;
zlarnv_(&idist, iseed, &n, x);
return rb_ary_new3(2, rblapack_x, rblapack_iseed);
}
void
init_lapack_zlarnv(VALUE mLapack, VALUE sH, VALUE sU, VALUE zero){
sHelp = sH;
sUsage = sU;
rblapack_ZERO = zero;
rb_define_module_function(mLapack, "zlarnv", rblapack_zlarnv, -1);
}
|
