blob: 4832887911fb26a31ec2bb7dd9816a1a5c5a801d (
plain)
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
|
;;; nsmc --- n-sphere Monte Carlo method
;;; Copyright © 2021 Arun I <arunisaac@systemreboot.net>
;;; Copyright © 2021 Murugesan Venkatapathi <murugesh@iisc.ac.in>
;;;
;;; This file is part of nsmc.
;;;
;;; nsmc is free software: you can redistribute it and/or modify it
;;; under the terms of the GNU General Public License as published by
;;; the Free Software Foundation, either version 3 of the License, or
;;; (at your option) any later version.
;;;
;;; nsmc is distributed in the hope that it will be useful, but
;;; WITHOUT ANY WARRANTY; without even the implied warranty of
;;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
;;; General Public License for more details.
;;;
;;; You should have received a copy of the GNU General Public License
;;; along with nsmc. If not, see <https://www.gnu.org/licenses/>.
(sc-define-syntax (when condition body ...)
(cond
(condition
body ...)))
(sc-define-syntax (unless condition body ...)
(when (not condition)
body ...))
(sc-define-syntax (incr var)
(set+ var 1))
(sc-define-syntax (for-i index limit body ...)
(for ((define index int 0)
(< index limit)
(set+ index 1))
body ...))
(sc-define-syntax (for-i-step index start max step body ...)
(for ((define index int start)
(<= index max)
(set+ index step))
body ...))
(sc-define-syntax (with-alloc var type allocate free body ...)
(let* ((var type allocate))
body ...
(free var)))
(sc-define-syntax (with-vector var size body ...)
(with-alloc var gsl-vector*
(gsl-vector-alloc size)
gsl-vector-free
body ...))
(sc-define-syntax (with-matrix var m n body ...)
(with-alloc var gsl-matrix*
(gsl-matrix-alloc m n)
gsl-matrix-free
body ...))
(sc-define-syntax (with-square-matrix var n body ...)
(with-alloc var gsl-matrix*
(gsl-matrix-alloc n n)
gsl-matrix-free
body ...))
(sc-define-syntax (with-rng var body ...)
(with-alloc var gsl-rng*
(gsl-rng-alloc gsl-rng-default) gsl-rng-free
body ...))
(sc-define-syntax (with-rstats var body ...)
(with-alloc var gsl-rstat-workspace*
(gsl-rstat-alloc) gsl-rstat-free
body ...))
(sc-define-syntax (with-rstats* var n body ...)
(begin
(declare var (array gsl-rstat-workspace* 3))
(for-i i n
(array-set var i (gsl-rstat-alloc)))
body ...
(for-i i n
(gsl-rstat-free (array-get var i)))))
(sc-define-syntax (with-file fp filename mode body ...)
(with-alloc fp FILE* (begin (fopen filename mode)
(unless fp
(perror filename)
(exit EXIT_FAILURE)))
fclose
body ...))
(sc-define-syntax* (with-data-file fp filename mode body ...)
(let* ((data-file (string-append "data/" filename)))
`(with-file fp ,data-file ,mode
,@(if (string=? mode "w")
`((fprintf stderr "Writing to %s...\n" ,data-file))
'())
,@body)))
|
