git.iko.soy/Idris-dev
on master help sign in
javascript: inline more functions
2ac38b8raichoo
src/IRTS/CodegenJavaScript.hs+14−26
compare with master
unifiedsplit
@@ -633,33 +633,11 @@ inlineFunctions js =
633
633
634
634
635 inlineAble :: Int -> String -> [String] -> JS -> Maybe JS
635 inlineAble :: Int -> String -> [String] -> JS -> Maybe JS
636− inlineAble 1 fun args body
636+ inlineAble 1 fun args (JSReturn body)
637 | nonRecur fun body =
637 | nonRecur fun body =
638− inlineAble' body
638+ if all (<= 1) (map ($ body) (map countIDOccurences args))
639− where
639+ then Just body
640− inlineAble' :: JS -> Maybe JS
641− inlineAble' (
642− JSReturn js@(JSNew "__IDRRT__Con" [JSNum _, JSArray vals])
643− )
644− | and $ map (\x -> isJSIdent x || isJSConstant x) vals = Just js
645−
646− inlineAble' (
647− JSReturn js@(JSNew "__IDRRT__Cont" [JSFunction [] (
648− JSReturn (JSApp (JSIdent _) args)
649− )])
650− )
651− | and $ map (\x -> isJSIdent x || isJSConstant x) args = Just js
652−
653− inlineAble' (
654− JSReturn js@(JSIndex (JSProj (JSApp (JSIdent _) args) "vars") _)
655− )
656− | and $ map (\x -> isJSIdent x || isJSConstant x) args = Just js
657−
658− inlineAble' _ = Nothing
640+ else Nothing
659−
660− isJSIdent js
661− | JSIdent _ <- js = True
662− | otherwise = False
663
641
664 inlineAble _ _ _ _ = Nothing
642 inlineAble _ _ _ _ = Nothing
665
643
@@ -695,6 +673,16 @@ inlineFunctions js =
695 countAll name js = sum $ map (countInvokations name) js
673 countAll name js = sum $ map (countInvokations name) js
696
674
697
675
676+ countIDOccurences :: String -> JS -> Int
677+ countIDOccurences name = foldJS match (+) 0
678+ where
679+ match :: JS -> Int
680+ match js
681+ | JSIdent ident <- js
682+ , name == ident = 1
683+ | otherwise = 0
684+
685+
698 countInvokations :: String -> JS -> Int
686 countInvokations :: String -> JS -> Int
699 countInvokations name = foldJS match (+) 0
687 countInvokations name = foldJS match (+) 0
700 where
688 where