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
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
|
(* ocamlgsl - OCaml interface to GSL *)
(* Copyright () 2002 - Olivier Andrieu *)
(* distributed under the terms of the GPL version 2 *)
open Bigarray
type complex_double_vector_bigarr =
(Complex.t, complex64_elt, c_layout) Bigarray.Array1.t
type vector = complex_double_vector_bigarr
let create ?init len =
let barr = Array1.create complex64 c_layout len in
begin match init with
| None -> ()
| Some x -> Array1.fill barr x
end ;
barr
let length = Array1.dim
let of_array arr =
Array1.of_array complex64 c_layout arr
let to_array v =
Array.init (Array1.dim v) (Array1.get v)
let of_complex_array arr =
let n = (Array.length arr) / 2 in
let barr = create n in
for i=0 to pred n do
barr.{i} <- Gsl_complex.get arr i
done ;
barr
let to_complex_array barr =
let n = Array1.dim barr in
let arr = Array.create (2*n) 0. in
for i=0 to pred n do
Gsl_complex.set arr i barr.{i}
done ;
arr
let get (v : vector) i = Array1.get v i
let set (v : vector) i x = Array1.set v i x
let set_all = Array1.fill
let set_zero v =
set_all v Complex.zero
let set_basis v i =
set_zero v ;
set v i Complex.one
let subvector v ~off ~len =
Array1.sub v off len
let memcpy ~src:v ~dst:w =
if length v <> length w
then invalid_arg "Gsl_vector.memcpy" ;
Array1.blit v w
let copy v =
let w = create (length v) in
memcpy v w ;
w
let swap_element v i j =
let d = get v i in
let d' = get v j in
set v j d ;
set v i d'
let reverse v =
let len = length v in
for i=0 to pred (len/2) do
swap_element v i (pred len - i)
done
module Single =
struct
type complex_float_vector_bigarr =
(Complex.t, complex32_elt, c_layout) Bigarray.Array1.t
type vector = complex_float_vector_bigarr
let create ?init len =
let barr = Array1.create complex32 c_layout len in
begin match init with
| None -> ()
| Some x -> Array1.fill barr x
end ;
barr
let length = length
let of_array arr =
Array1.of_array complex32 c_layout arr
let to_array = to_array
let of_complex_array arr =
let n = (Array.length arr) / 2 in
let barr = create n in
for i=0 to pred n do
barr.{i} <- Gsl_complex.get arr i
done ;
barr
let to_complex_array barr =
let n = Array1.dim barr in
let arr = Array.create (2*n) 0. in
for i=0 to pred n do
Gsl_complex.set arr i barr.{i}
done ;
arr
let get (v : vector) i = Array1.get v i
let set (v : vector) i x = Array1.set v i x
let set_all = set_all
let set_zero = set_zero
let set_basis v i =
set_zero v ;
set v i Complex.one
let subvector = subvector
let memcpy = memcpy
let copy v =
let w = create (length v) in
memcpy v w ;
w
let swap_element v i j =
let d = get v i in
let d' = get v j in
set v j d ;
set v i d'
let reverse v =
let len = length v in
for i=0 to pred (len/2) do
swap_element v i (pred len - i)
done
end
|
