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
|
(*
** For writing ATS code
** that translates into Python3
*)
(* ****** ****** *)
#define ATS_DYNLOADFLAG 0
(* ****** ****** *)
//
// HX-2014-08:
// prefix for external names
//
#define
ATS_EXTERN_PREFIX "ats2pypre_"
#define
ATS_STATIC_PREFIX "_ats2pypre_matrixref_"
//
(* ****** ****** *)
//
#define
LIBATSCC_targetloc
"$PATSHOME\
/contrib/libatscc/ATS2-0.3.2"
//
(* ****** ****** *)
//
#staload
UN = "prelude/SATS/unsafe.sats"
//
(* ****** ****** *)
//
#staload "./../basics_py.sats"
//
#staload "./../SATS/integer.sats"
//
(* ****** ****** *)
//
#staload "./../SATS/PYlist.sats"
#staload "./../SATS/intrange.sats"
//
(* ****** ****** *)
//
#staload "./../SATS/matrixref.sats"
//
(* ****** ****** *)
//
#include "{$LIBATSCC}/DATS/matrixref.dats"
//
(* ****** ****** *)
//
(*
assume
matrixref_vt0ype_type(a, m, n) = PYlist(a)
*)
//
(* ****** ****** *)
%{^
######
def ats2pypre_matrixref_make_elt(m, n, x0):
M = []
i0 = 0
mn = m * n
while (i0 < mn): i0 = i0 + 1; M.append(x0)
return M
######
%} // end of [%{^]
(* ****** ****** *)
implement
matrixref_get_at
{a}(A, i, n, j) = let
val A = $UN.cast{PYlist(a)}(A) in PYlist_get_at(A, i*n+j)
end // end of [matrixref_get_at]
(* ****** ****** *)
implement
matrixref_set_at
{a}(A, i, n, j, x) = let
val A = $UN.cast{PYlist(a)}(A) in PYlist_set_at(A, i*n+j, x)
end // end of [matrixref_set_at]
(* ****** ****** *)
%{^
######
def ats2pypre_mtrxszref_make_matrixref(M, m, n):
return { 'matrix' : M, 'nrow' : m, 'ncol' : n }
######
def ats2pypre_mtrxszref_get_nrow(MSZ): return MSZ['nrow']
def ats2pypre_mtrxszref_get_ncol(MSZ): return MSZ['ncol']
######
def ats2pypre_mtrxszref_get_at(MSZ, i, j):
nrow = MSZ['nrow']
ncol = MSZ['ncol']
if (i < 0): raise IndexError('mtrxszref_get_at')
if (j < 0): raise IndexError('mtrxszref_get_at')
if (i >= nrow): raise IndexError('mtrxszref_get_at')
if (j >= ncol): raise IndexError('mtrxszref_get_at')
return MSZ['matrix'][i*ncol+j]
######
def ats2pypre_mtrxszref_set_at(MSZ, i, j, x0):
nrow = MSZ['nrow']
ncol = MSZ['ncol']
if (i < 0): raise IndexError('mtrxszref_set_at')
if (j < 0): raise IndexError('mtrxszref_set_at')
if (i >= nrow): raise IndexError('mtrxszref_set_at')
if (j >= ncol): raise IndexError('mtrxszref_set_at')
MSZ['matrix'][i*ncol+j] = x0; return#_void
######
%} // end of [%{^]
(* ****** ****** *)
(* end of [matrixref.dats] *)
|
