{-# LANGUAGE OverloadedStrings #-}
module Mate.Utilities where
-import Data.Char
import Data.Word
-import Data.Binary
import qualified Data.Map as M
import qualified Data.ByteString.Lazy as B
--- import qualified Data.ByteString.Lazy.Char8 as B8
-import Codec.Binary.UTF8.String hiding (encode,decode)
import JVM.ClassFile
+import Mate.Types
+
-- TODO: actually this function already exists in hs-java-0.3!
lookupMethod :: B.ByteString -> Class Resolved -> Maybe (Method Resolved)
| methodName f == name = Just f
| otherwise = look fs
-toString :: B.ByteString -> String
-toString bstr = decodeString $ map (chr . fromIntegral) $ B.unpack bstr
-
-buildMethodID :: Class Resolved -> Word16 -> B.ByteString
-buildMethodID cls idx = (rc `B.append` dot) `B.append` (ntName nt) `B.append` nt'
+buildMethodID :: Class Resolved -> Word16 -> MethodInfo
+buildMethodID cls idx = MethodInfo (ntName nt) rc (ntSignature nt) idx
where
(CMethod rc nt) = (constsPool cls) M.! idx
- nt' = encode $ ntSignature nt
- dot :: B.ByteString
- -- TODO(bernhard): WTF? why -XOverloadedStrings doesn't apply here?
- dot = B.pack $ map (fromIntegral . ord) "."
+
+methodGetArgsCount :: Class Resolved -> Word16 -> Word32
+methodGetArgsCount cls idx = fromIntegral $ length args
+ where
+ (CMethod _ nt) = (constsPool cls) M.! idx
+ (MethodSignature args _) = ntSignature nt
+
+-- TODO(bernhard): Extend it to more than just int, and provide typeinformation
+methodHaveReturnValue :: Class Resolved -> Word16 -> Bool
+methodHaveReturnValue cls idx = case ret of
+ ReturnsVoid -> False;
+ (Returns IntType) -> True;
+ _ -> error "methodHaveReturnValue: todo"
+ where
+ (CMethod _ nt) = (constsPool cls) M.! idx
+ (MethodSignature _ ret) = ntSignature nt