Support both IO and clean version of Generate monad.
[hs-java.git] / JVM / Builder / Instructions.hs
1 -- | This module exports shortcuts for some of JVM instructions (which are defined in JVM.Assembler).
2 -- These functions get Constants, put them into constants pool and generate instruction using index
3 -- of constant in the pool.
4 module JVM.Builder.Instructions where
5
6 import Data.Word
7 import qualified Data.ByteString.Lazy as B
8
9 import JVM.ClassFile
10 import JVM.Assembler
11 import JVM.Builder.Monad
12
13 nop :: Generator g => g ()
14 nop = i0 NOP
15 aconst_null :: Generator g => g ()
16 aconst_null = i0 ACONST_NULL
17 iconst_m1 :: Generator g => g ()
18 iconst_m1 = i0 ICONST_M1
19 iconst_0 :: Generator g => g ()
20 iconst_0 = i0 ICONST_0
21 iconst_1 :: Generator g => g ()
22 iconst_1 = i0 ICONST_1
23 iconst_2 :: Generator g => g ()
24 iconst_2 = i0 ICONST_2
25 iconst_3 :: Generator g => g ()
26 iconst_3 = i0 ICONST_3
27 iconst_4 :: Generator g => g ()
28 iconst_4 = i0 ICONST_4
29 iconst_5 :: Generator g => g ()
30 iconst_5 = i0 ICONST_5
31 lconst_0 :: Generator g => g ()
32 lconst_0 = i0 LCONST_0
33 lconst_1 :: Generator g => g ()
34 lconst_1 = i0 LCONST_1
35 fconst_0 :: Generator g => g ()
36 fconst_0 = i0 FCONST_0
37 fconst_1 :: Generator g => g ()
38 fconst_1 = i0 FCONST_1
39 fconst_2 :: Generator g => g ()
40 fconst_2 = i0 FCONST_2
41 dconst_0 :: Generator g => g ()
42 dconst_0 = i0 DCONST_0
43 dconst_1 :: Generator g => g ()
44 dconst_1 = i0 DCONST_1
45
46 bipush :: Generator g => Word8 -> g ()
47 bipush x = i0 (BIPUSH x)
48 sipush :: Generator g => Word16 -> g ()
49 sipush x = i0 (SIPUSH x)
50
51 ldc1 :: Generator g => Constant Direct -> g ()
52 ldc1 x = i8 LDC1 x
53 ldc2 :: Generator g => Constant Direct -> g ()
54 ldc2 x = i1 LDC2 x
55 ldc2w :: Generator g => Constant Direct -> g ()
56 ldc2w x = i1 LDC2W x
57 iload :: Generator g => Constant Direct -> g ()
58 iload x = i8 ILOAD x
59 lload :: Generator g => Constant Direct -> g ()
60 lload x = i8 LLOAD x
61 fload :: Generator g => Constant Direct -> g ()
62 fload x = i8 FLOAD x
63 dload :: Generator g => Constant Direct -> g ()
64 dload x = i8 DLOAD x
65 aload :: Generator g => Constant Direct -> g ()
66 aload x = i8 ALOAD x
67
68 iload_ :: Generator g => IMM -> g ()
69 iload_ x = i0 (ILOAD_ x)
70 lload_ :: Generator g => IMM -> g ()
71 lload_ x = i0 (LLOAD_ x)
72 fload_ :: Generator g => IMM -> g ()
73 fload_ x = i0 (FLOAD_ x)
74 dload_ :: Generator g => IMM -> g ()
75 dload_ x = i0 (DLOAD_ x)
76 aload_ :: Generator g => IMM -> g ()
77 aload_ x = i0 (ALOAD_ x)
78
79 iaload :: Generator g => g ()
80 iaload = i0 IALOAD
81 laload :: Generator g => g ()
82 laload = i0 LALOAD
83 faload :: Generator g => g ()
84 faload = i0 FALOAD
85 daload :: Generator g => g ()
86 daload = i0 DALOAD
87 aaload :: Generator g => g ()
88 aaload = i0 AALOAD
89 caload :: Generator g => g ()
90 caload = i0 CALOAD
91 saload :: Generator g => g ()
92 saload = i0 SALOAD
93
94 istore :: Generator g => Constant Direct -> g ()
95 istore x = i8 ISTORE x
96 lstore :: Generator g => Constant Direct -> g ()
97 lstore x = i8 LSTORE x
98 fstore :: Generator g => Constant Direct -> g ()
99 fstore x = i8 FSTORE x
100 dstore :: Generator g => Constant Direct -> g ()
101 dstore x = i8 DSTORE x
102 astore :: Generator g => Constant Direct -> g ()
103 astore x = i8 ASTORE x
104
105 istore_ :: Generator g => Word8 -> g ()
106 istore_ x = i0 (ISTORE x)
107 lstore_ :: Generator g => Word8 -> g ()
108 lstore_ x = i0 (LSTORE x)
109 fstore_ :: Generator g => Word8 -> g ()
110 fstore_ x = i0 (FSTORE x)
111 dstore_ :: Generator g => Word8 -> g ()
112 dstore_ x = i0 (DSTORE x)
113 astore_ :: Generator g => Word8 -> g ()
114 astore_ x = i0 (ASTORE x)
115
116 iastore :: Generator g => g ()
117 iastore = i0 IASTORE
118 lastore :: Generator g => g ()
119 lastore = i0 LASTORE
120 fastore :: Generator g => g ()
121 fastore = i0 FASTORE
122 dastore :: Generator g => g ()
123 dastore = i0 DASTORE
124 aastore :: Generator g => g ()
125 aastore = i0 AASTORE
126 bastore :: Generator g => g ()
127 bastore = i0 BASTORE
128 castore :: Generator g => g ()
129 castore = i0 CASTORE
130 sastore :: Generator g => g ()
131 sastore = i0 SASTORE
132
133 pop :: Generator g => g ()
134 pop     = i0 POP    
135 pop2 :: Generator g => g ()
136 pop2    = i0 POP2   
137 dup :: Generator g => g ()
138 dup     = i0 DUP    
139 dup_x1 :: Generator g => g ()
140 dup_x1  = i0 DUP_X1 
141 dup_x2 :: Generator g => g ()
142 dup_x2  = i0 DUP_X2 
143 dup2 :: Generator g => g ()
144 dup2    = i0 DUP2   
145 dup2_x1 :: Generator g => g ()
146 dup2_x1 = i0 DUP2_X1
147 dup2_x2 :: Generator g => g ()
148 dup2_x2 = i0 DUP2_X2
149 swap :: Generator g => g ()
150 swap    = i0 SWAP   
151 iadd :: Generator g => g ()
152 iadd    = i0 IADD   
153 ladd :: Generator g => g ()
154 ladd    = i0 LADD   
155 fadd :: Generator g => g ()
156 fadd    = i0 FADD   
157 dadd :: Generator g => g ()
158 dadd    = i0 DADD   
159 isub :: Generator g => g ()
160 isub    = i0 ISUB   
161 lsub :: Generator g => g ()
162 lsub    = i0 LSUB   
163 fsub :: Generator g => g ()
164 fsub    = i0 FSUB   
165 dsub :: Generator g => g ()
166 dsub    = i0 DSUB   
167 imul :: Generator g => g ()
168 imul    = i0 IMUL   
169 lmul :: Generator g => g ()
170 lmul    = i0 LMUL   
171 fmul :: Generator g => g ()
172 fmul    = i0 FMUL   
173 dmul :: Generator g => g ()
174 dmul    = i0 DMUL   
175 idiv :: Generator g => g ()
176 idiv    = i0 IDIV   
177 ldiv :: Generator g => g ()
178 ldiv    = i0 LDIV   
179 fdiv :: Generator g => g ()
180 fdiv    = i0 FDIV   
181 ddiv :: Generator g => g ()
182 ddiv    = i0 DDIV   
183 irem :: Generator g => g ()
184 irem    = i0 IREM   
185 lrem :: Generator g => g ()
186 lrem    = i0 LREM   
187 frem :: Generator g => g ()
188 frem    = i0 FREM   
189 drem :: Generator g => g ()
190 drem    = i0 DREM   
191 ineg :: Generator g => g ()
192 ineg    = i0 INEG   
193 lneg :: Generator g => g ()
194 lneg    = i0 LNEG   
195 fneg :: Generator g => g ()
196 fneg    = i0 FNEG   
197 dneg :: Generator g => g ()
198 dneg    = i0 DNEG   
199 ishl :: Generator g => g ()
200 ishl    = i0 ISHL   
201 lshl :: Generator g => g ()
202 lshl    = i0 LSHL   
203 ishr :: Generator g => g ()
204 ishr    = i0 ISHR   
205 lshr :: Generator g => g ()
206 lshr    = i0 LSHR   
207 iushr :: Generator g => g ()
208 iushr   = i0 IUSHR  
209 lushr :: Generator g => g ()
210 lushr   = i0 LUSHR  
211 iand :: Generator g => g ()
212 iand    = i0 IAND   
213 land :: Generator g => g ()
214 land    = i0 LAND   
215 ior :: Generator g => g ()
216 ior     = i0 IOR    
217 lor :: Generator g => g ()
218 lor     = i0 LOR    
219 ixor :: Generator g => g ()
220 ixor    = i0 IXOR   
221 lxor :: Generator g => g ()
222 lxor    = i0 LXOR   
223
224 iinc :: Generator g => Word8 -> Word8 -> g ()
225 iinc x y = i0 (IINC x y)
226
227 i2l :: Generator g => g ()
228 i2l  = i0 I2L 
229 i2f :: Generator g => g ()
230 i2f  = i0 I2F 
231 i2d :: Generator g => g ()
232 i2d  = i0 I2D 
233 l2i :: Generator g => g ()
234 l2i  = i0 L2I 
235 l2f :: Generator g => g ()
236 l2f  = i0 L2F 
237 l2d :: Generator g => g ()
238 l2d  = i0 L2D 
239 f2i :: Generator g => g ()
240 f2i  = i0 F2I 
241 f2l :: Generator g => g ()
242 f2l  = i0 F2L 
243 f2d :: Generator g => g ()
244 f2d  = i0 F2D 
245 d2i :: Generator g => g ()
246 d2i  = i0 D2I 
247 d2l :: Generator g => g ()
248 d2l  = i0 D2L 
249 d2f :: Generator g => g ()
250 d2f  = i0 D2F 
251 i2b :: Generator g => g ()
252 i2b  = i0 I2B 
253 i2c :: Generator g => g ()
254 i2c  = i0 I2C 
255 i2s :: Generator g => g ()
256 i2s  = i0 I2S 
257 lcmp :: Generator g => g ()
258 lcmp = i0 LCMP
259
260 -- | Wide instruction
261 wide :: Generator g => (Word8 -> Instruction) -> Constant Direct -> g ()
262 wide fn c = do
263   ix <- addToPool c
264   let ix0 = fromIntegral (ix `div` 0x100) :: Word8
265       ix1 = fromIntegral (ix `mod` 0x100) :: Word8
266   i0 (WIDE ix0 $ fn ix1)
267
268 new :: Generator g => B.ByteString -> g ()
269 new cls =
270   i1 NEW (CClass cls)
271
272 newArray :: Generator g => ArrayType -> g ()
273 newArray t =
274   i0 (NEWARRAY $ atype2byte t)
275
276 allocNewArray :: Generator g => B.ByteString -> g ()
277 allocNewArray cls =
278   i1 ANEWARRAY (CClass cls)
279
280 invokeVirtual :: Generator g => B.ByteString -> NameType Method -> g ()
281 invokeVirtual cls sig =
282   i1 INVOKEVIRTUAL (CMethod cls sig)
283
284 invokeStatic :: Generator g => B.ByteString -> NameType Method -> g ()
285 invokeStatic cls sig =
286   i1 INVOKESTATIC (CMethod cls sig)
287
288 invokeSpecial :: Generator g => B.ByteString -> NameType Method -> g ()
289 invokeSpecial cls sig =
290   i1 INVOKESPECIAL (CMethod cls sig)
291
292 getStaticField :: Generator g => B.ByteString -> NameType Field -> g ()
293 getStaticField cls sig =
294   i1 GETSTATIC (CField cls sig)
295
296 loadString :: Generator g => B.ByteString -> g ()
297 loadString str =
298   i8 LDC1 (CString str)
299
300 allocArray :: Generator g => B.ByteString -> g ()
301 allocArray cls =
302   i1 ANEWARRAY (CClass cls)
303