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
|
{-# LANGUAGE OverloadedStrings #-}
module GHC.StgToJS.Regs
( StgReg (..)
, Special(..)
, sp
, stack
, r1, r2, r3, r4
, regsFromR1
, regsFromR2
, jsRegsFromR1
, jsRegsFromR2
, StgRet (..)
, jsRegToInt
, intToJSReg
, jsReg
, maxReg
, minReg
)
where
import GHC.Prelude
import GHC.JS.Syntax
import GHC.JS.Make
import GHC.Data.FastString
import Data.Array
import Data.Char
-- | General purpose "registers"
--
-- The JS backend arbitrarily supports 128 registers
data StgReg
= R1 | R2 | R3 | R4 | R5 | R6 | R7 | R8
| R9 | R10 | R11 | R12 | R13 | R14 | R15 | R16
| R17 | R18 | R19 | R20 | R21 | R22 | R23 | R24
| R25 | R26 | R27 | R28 | R29 | R30 | R31 | R32
| R33 | R34 | R35 | R36 | R37 | R38 | R39 | R40
| R41 | R42 | R43 | R44 | R45 | R46 | R47 | R48
| R49 | R50 | R51 | R52 | R53 | R54 | R55 | R56
| R57 | R58 | R59 | R60 | R61 | R62 | R63 | R64
| R65 | R66 | R67 | R68 | R69 | R70 | R71 | R72
| R73 | R74 | R75 | R76 | R77 | R78 | R79 | R80
| R81 | R82 | R83 | R84 | R85 | R86 | R87 | R88
| R89 | R90 | R91 | R92 | R93 | R94 | R95 | R96
| R97 | R98 | R99 | R100 | R101 | R102 | R103 | R104
| R105 | R106 | R107 | R108 | R109 | R110 | R111 | R112
| R113 | R114 | R115 | R116 | R117 | R118 | R119 | R120
| R121 | R122 | R123 | R124 | R125 | R126 | R127 | R128
deriving (Eq, Ord, Show, Enum, Bounded, Ix)
-- | Stack registers
data Special
= Stack
| Sp
deriving (Show, Eq)
-- | Return registers
--
-- Extra results from foreign calls can be stored here (while first result is
-- directly returned)
data StgRet = Ret1 | Ret2 | Ret3 | Ret4 | Ret5 | Ret6 | Ret7 | Ret8 | Ret9 | Ret10
deriving (Eq, Ord, Show, Enum, Bounded, Ix)
instance ToJExpr Special where
toJExpr Stack = var "h$stack"
toJExpr Sp = var "h$sp"
instance ToJExpr StgReg where
toJExpr r = registers ! r
instance ToJExpr StgRet where
toJExpr r = rets ! r
---------------------------------------------------
-- helpers
---------------------------------------------------
sp :: JExpr
sp = toJExpr Sp
stack :: JExpr
stack = toJExpr Stack
r1, r2, r3, r4 :: JExpr
r1 = toJExpr R1
r2 = toJExpr R2
r3 = toJExpr R3
r4 = toJExpr R4
jsRegToInt :: StgReg -> Int
jsRegToInt = (+1) . fromEnum
intToJSReg :: Int -> StgReg
intToJSReg r = toEnum (r - 1)
jsReg :: Int -> JExpr
jsReg r = toJExpr (intToJSReg r)
maxReg :: Int
maxReg = jsRegToInt maxBound
minReg :: Int
minReg = jsRegToInt minBound
-- | List of registers, starting from R1
regsFromR1 :: [StgReg]
regsFromR1 = enumFrom R1
-- | List of registers, starting from R2
regsFromR2 :: [StgReg]
regsFromR2 = tail regsFromR1
-- | List of registers, starting from R1 as JExpr
jsRegsFromR1 :: [JExpr]
jsRegsFromR1 = fmap toJExpr regsFromR1
-- | List of registers, starting from R2 as JExpr
jsRegsFromR2 :: [JExpr]
jsRegsFromR2 = tail jsRegsFromR1
---------------------------------------------------
-- caches
---------------------------------------------------
-- cache JExpr representing StgReg
registers :: Array StgReg JExpr
registers = listArray (minBound, maxBound) (map regN regsFromR1)
where
regN r
| fromEnum r < 32 = var . mkFastString . ("h$"++) . map toLower . show $ r
| otherwise = IdxExpr (var "h$regs")
(toJExpr ((fromEnum r) - 32))
-- cache JExpr representing StgRet
rets :: Array StgRet JExpr
rets = listArray (minBound, maxBound) (map retN (enumFrom Ret1))
where
retN = var . mkFastString . ("h$"++) . map toLower . show
|