TestGen.hs

   1 {-# LANGUAGE OverloadedStrings #-}
   2
   3 import qualified Data.ByteString.Lazy as B
   4
   5 import JVM.Types
   6 import JVM.ClassFile
   7 import JVM.Converter
   8 import JVM.Assembler
   9 import JVM.Generator
  10 import JVM.Dump
  11
  12 initNT :: NameType Method
  13 initNT = NameType "<init>" $ MethodSignature [] ReturnsVoid
  14
  15 helloNT :: NameType Method
  16 helloNT = NameType "hello" $ MethodSignature [IntType] ReturnsVoid
  17
  18 printlnNT :: NameType Method
  19 printlnNT = NameType "println" $ MethodSignature [ObjectType "java/lang/String"] ReturnsVoid
  20
  21 outNT :: NameType Field
  22 outNT = NameType "out" $ ObjectType "java/io/PrintStream"
  23
  24 valueOfNT :: NameType Method
  25 valueOfNT = NameType "valueOf" $ MethodSignature [IntType] (Returns $ ObjectType "java/lang/Integer")
  26
  27 printfNT :: NameType Method
  28 printfNT = NameType "printf" $ MethodSignature [ObjectType "java/lang/String",
  29                                                 Array Nothing $ ObjectType "java/lang/Object"] (Returns $ ObjectType "java/io/PrintStream")
  30
  31 test :: Generate ()
  32 test = do
  33   newMethod [ACC_PUBLIC] "<init>" [] ReturnsVoid $ do
  34       i0 $ ALOAD_ I0
  35       i1 INVOKESPECIAL (CMethod "java/lang/Object" initNT)
  36       i0 RETURN
  37
  38   newMethod [ACC_PUBLIC, ACC_STATIC] "main" [Array Nothing $ ObjectType "java/lang/String"] ReturnsVoid $ do
  39       i0 ICONST_5
  40       i1 INVOKESTATIC (CMethod "Test" helloNT)
  41       i0 RETURN
  42
  43   newMethod [ACC_PUBLIC, ACC_STATIC] "hello" [IntType] ReturnsVoid $ do
  44       i1 GETSTATIC (CField "java/lang/System" outNT)
  45       i8 LDC1 (CString "Здравствуй, мир!")
  46       i1 INVOKEVIRTUAL (CMethod "java/io/PrintStream" printlnNT)
  47       i1 GETSTATIC (CField "java/lang/System" outNT)
  48       i8 LDC1 (CString "Argument: %d\n")
  49       i0 ICONST_1
  50       i1 ANEWARRAY (CClass "java/lang/Object")
  51       i0 DUP
  52       i0 ICONST_0
  53       i0 (ILOAD_ I0)
  54       i1 INVOKESTATIC (CMethod "java/lang/Integer" valueOfNT)
  55       i0 AASTORE
  56       i1 INVOKEVIRTUAL (CMethod "java/io/PrintStream" printfNT)
  57       i0 POP
  58       i0 RETURN
  59
  60 testClass = generate "Test" test
  61
  62 main = do
  63   B.writeFile "Test.class" (encodeClass testClass)
  64