{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE DeriveDataTypeable #-}


-- | The plugin to make it all work.

module Rattus.Plugin (plugin, Rattus(..)) where
import Rattus.Plugin.StableSolver
import Rattus.Plugin.ScopeCheck
import Rattus.Plugin.Strictify
import Rattus.Plugin.Utils

import Prelude hiding ((<>))
import GhcPlugins

import qualified Data.Set as Set
import Control.Monad
import Data.Maybe
import Data.Data hiding (tyConName)

import System.Exit


-- | Use this type to mark a Haskell function definition as a Rattus
-- function:
--
-- > {-# ANN myFunction Rattus #-}
-- 
-- Or mark a whole module as consisting of Rattus functions only:
--
-- > {-# ANN module Rattus #-}
--
-- If you use the latter option, you can mark exceptions
-- (i.e. functions that should be treated as ordinary Haskell function
-- definitions) as follows:
--
-- > {-# ANN myFunction NotRattus #-}
--
-- By default all Rattus functions are checked for use of lazy data
-- types, since these may cause memory leaks. If any lazy data types
-- are used, a warning is issued. These warnings can be disabled by
-- annotating the module or the function with 'AllowLazyData'
--
-- > {-# ANN myFunction AllowLazyData #-}
-- >
-- > {-# ANN module AllowLazyData #-}

data Rattus = Rattus | NotRattus | AllowLazyData deriving (Typeable, Typeable Rattus
DataType
Constr
Typeable Rattus =>
(forall (c :: * -> *).
 (forall d b. Data d => c (d -> b) -> d -> c b)
 -> (forall g. g -> c g) -> Rattus -> c Rattus)
-> (forall (c :: * -> *).
    (forall b r. Data b => c (b -> r) -> c r)
    -> (forall r. r -> c r) -> Constr -> c Rattus)
-> (Rattus -> Constr)
-> (Rattus -> DataType)
-> (forall (t :: * -> *) (c :: * -> *).
    Typeable t =>
    (forall d. Data d => c (t d)) -> Maybe (c Rattus))
-> (forall (t :: * -> * -> *) (c :: * -> *).
    Typeable t =>
    (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c Rattus))
-> ((forall b. Data b => b -> b) -> Rattus -> Rattus)
-> (forall r r'.
    (r -> r' -> r)
    -> r -> (forall d. Data d => d -> r') -> Rattus -> r)
-> (forall r r'.
    (r' -> r -> r)
    -> r -> (forall d. Data d => d -> r') -> Rattus -> r)
-> (forall u. (forall d. Data d => d -> u) -> Rattus -> [u])
-> (forall u. Int -> (forall d. Data d => d -> u) -> Rattus -> u)
-> (forall (m :: * -> *).
    Monad m =>
    (forall d. Data d => d -> m d) -> Rattus -> m Rattus)
-> (forall (m :: * -> *).
    MonadPlus m =>
    (forall d. Data d => d -> m d) -> Rattus -> m Rattus)
-> (forall (m :: * -> *).
    MonadPlus m =>
    (forall d. Data d => d -> m d) -> Rattus -> m Rattus)
-> Data Rattus
Rattus -> DataType
Rattus -> Constr
(forall b. Data b => b -> b) -> Rattus -> Rattus
(forall d b. Data d => c (d -> b) -> d -> c b)
-> (forall g. g -> c g) -> Rattus -> c Rattus
(forall b r. Data b => c (b -> r) -> c r)
-> (forall r. r -> c r) -> Constr -> c Rattus
forall a.
Typeable a =>
(forall (c :: * -> *).
 (forall d b. Data d => c (d -> b) -> d -> c b)
 -> (forall g. g -> c g) -> a -> c a)
-> (forall (c :: * -> *).
    (forall b r. Data b => c (b -> r) -> c r)
    -> (forall r. r -> c r) -> Constr -> c a)
-> (a -> Constr)
-> (a -> DataType)
-> (forall (t :: * -> *) (c :: * -> *).
    Typeable t =>
    (forall d. Data d => c (t d)) -> Maybe (c a))
-> (forall (t :: * -> * -> *) (c :: * -> *).
    Typeable t =>
    (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c a))
-> ((forall b. Data b => b -> b) -> a -> a)
-> (forall r r'.
    (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> a -> r)
-> (forall r r'.
    (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> a -> r)
-> (forall u. (forall d. Data d => d -> u) -> a -> [u])
-> (forall u. Int -> (forall d. Data d => d -> u) -> a -> u)
-> (forall (m :: * -> *).
    Monad m =>
    (forall d. Data d => d -> m d) -> a -> m a)
-> (forall (m :: * -> *).
    MonadPlus m =>
    (forall d. Data d => d -> m d) -> a -> m a)
-> (forall (m :: * -> *).
    MonadPlus m =>
    (forall d. Data d => d -> m d) -> a -> m a)
-> Data a
forall u. Int -> (forall d. Data d => d -> u) -> Rattus -> u
forall u. (forall d. Data d => d -> u) -> Rattus -> [u]
forall r r'.
(r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> Rattus -> r
forall r r'.
(r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> Rattus -> r
forall (m :: * -> *).
Monad m =>
(forall d. Data d => d -> m d) -> Rattus -> m Rattus
forall (m :: * -> *).
MonadPlus m =>
(forall d. Data d => d -> m d) -> Rattus -> m Rattus
forall (c :: * -> *).
(forall b r. Data b => c (b -> r) -> c r)
-> (forall r. r -> c r) -> Constr -> c Rattus
forall (c :: * -> *).
(forall d b. Data d => c (d -> b) -> d -> c b)
-> (forall g. g -> c g) -> Rattus -> c Rattus
forall (t :: * -> *) (c :: * -> *).
Typeable t =>
(forall d. Data d => c (t d)) -> Maybe (c Rattus)
forall (t :: * -> * -> *) (c :: * -> *).
Typeable t =>
(forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c Rattus)
$cAllowLazyData :: Constr
$cNotRattus :: Constr
$cRattus :: Constr
$tRattus :: DataType
gmapMo :: (forall d. Data d => d -> m d) -> Rattus -> m Rattus
$cgmapMo :: forall (m :: * -> *).
MonadPlus m =>
(forall d. Data d => d -> m d) -> Rattus -> m Rattus
gmapMp :: (forall d. Data d => d -> m d) -> Rattus -> m Rattus
$cgmapMp :: forall (m :: * -> *).
MonadPlus m =>
(forall d. Data d => d -> m d) -> Rattus -> m Rattus
gmapM :: (forall d. Data d => d -> m d) -> Rattus -> m Rattus
$cgmapM :: forall (m :: * -> *).
Monad m =>
(forall d. Data d => d -> m d) -> Rattus -> m Rattus
gmapQi :: Int -> (forall d. Data d => d -> u) -> Rattus -> u
$cgmapQi :: forall u. Int -> (forall d. Data d => d -> u) -> Rattus -> u
gmapQ :: (forall d. Data d => d -> u) -> Rattus -> [u]
$cgmapQ :: forall u. (forall d. Data d => d -> u) -> Rattus -> [u]
gmapQr :: (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> Rattus -> r
$cgmapQr :: forall r r'.
(r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> Rattus -> r
gmapQl :: (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> Rattus -> r
$cgmapQl :: forall r r'.
(r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> Rattus -> r
gmapT :: (forall b. Data b => b -> b) -> Rattus -> Rattus
$cgmapT :: (forall b. Data b => b -> b) -> Rattus -> Rattus
dataCast2 :: (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c Rattus)
$cdataCast2 :: forall (t :: * -> * -> *) (c :: * -> *).
Typeable t =>
(forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c Rattus)
dataCast1 :: (forall d. Data d => c (t d)) -> Maybe (c Rattus)
$cdataCast1 :: forall (t :: * -> *) (c :: * -> *).
Typeable t =>
(forall d. Data d => c (t d)) -> Maybe (c Rattus)
dataTypeOf :: Rattus -> DataType
$cdataTypeOf :: Rattus -> DataType
toConstr :: Rattus -> Constr
$ctoConstr :: Rattus -> Constr
gunfold :: (forall b r. Data b => c (b -> r) -> c r)
-> (forall r. r -> c r) -> Constr -> c Rattus
$cgunfold :: forall (c :: * -> *).
(forall b r. Data b => c (b -> r) -> c r)
-> (forall r. r -> c r) -> Constr -> c Rattus
gfoldl :: (forall d b. Data d => c (d -> b) -> d -> c b)
-> (forall g. g -> c g) -> Rattus -> c Rattus
$cgfoldl :: forall (c :: * -> *).
(forall d b. Data d => c (d -> b) -> d -> c b)
-> (forall g. g -> c g) -> Rattus -> c Rattus
$cp1Data :: Typeable Rattus
Data, Int -> Rattus -> ShowS
[Rattus] -> ShowS
Rattus -> String
(Int -> Rattus -> ShowS)
-> (Rattus -> String) -> ([Rattus] -> ShowS) -> Show Rattus
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
showList :: [Rattus] -> ShowS
$cshowList :: [Rattus] -> ShowS
show :: Rattus -> String
$cshow :: Rattus -> String
showsPrec :: Int -> Rattus -> ShowS
$cshowsPrec :: Int -> Rattus -> ShowS
Show, Rattus -> Rattus -> Bool
(Rattus -> Rattus -> Bool)
-> (Rattus -> Rattus -> Bool) -> Eq Rattus
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
/= :: Rattus -> Rattus -> Bool
$c/= :: Rattus -> Rattus -> Bool
== :: Rattus -> Rattus -> Bool
$c== :: Rattus -> Rattus -> Bool
Eq)

-- | Use this to enable Rattus' plugin, either by supplying the option
-- @-fplugin=Rattus.Plugin@ directly to GHC. or by including the
-- following pragma in each source file:
-- 
-- > {-# OPTIONS -fplugin=Rattus.Plugin #-}
plugin :: Plugin
plugin :: Plugin
plugin = Plugin
defaultPlugin {
  installCoreToDos :: CorePlugin
installCoreToDos = CorePlugin
install,
  tcPlugin :: TcPlugin
tcPlugin = TcPlugin
tcStable,
  pluginRecompile :: [String] -> IO PluginRecompile
pluginRecompile = [String] -> IO PluginRecompile
purePlugin
  }



install :: [CommandLineOption] -> [CoreToDo] -> CoreM [CoreToDo]
install :: CorePlugin
install _ todo :: [CoreToDo]
todo = [CoreToDo] -> CoreM [CoreToDo]
forall (m :: * -> *) a. Monad m => a -> m a
return (CoreToDo
scPass CoreToDo -> [CoreToDo] -> [CoreToDo]
forall a. a -> [a] -> [a]
: CoreToDo
strPass CoreToDo -> [CoreToDo] -> [CoreToDo]
forall a. a -> [a] -> [a]
: [CoreToDo]
todo)
    where scPass :: CoreToDo
scPass = String -> CorePluginPass -> CoreToDo
CoreDoPluginPass "Rattus scopecheck" CorePluginPass
scopecheckProgram

          strPass :: CoreToDo
strPass = String -> CorePluginPass -> CoreToDo
CoreDoPluginPass "Rattus strictify" CorePluginPass
strictifyProgram

strictifyProgram :: ModGuts -> CoreM ModGuts
strictifyProgram :: CorePluginPass
strictifyProgram guts :: ModGuts
guts = do
  [CoreBind]
newBinds <- (CoreBind -> CoreM CoreBind) -> [CoreBind] -> CoreM [CoreBind]
forall (t :: * -> *) (m :: * -> *) a b.
(Traversable t, Monad m) =>
(a -> m b) -> t a -> m (t b)
mapM (ModGuts -> CoreBind -> CoreM CoreBind
strictify ModGuts
guts) (ModGuts -> [CoreBind]
mg_binds ModGuts
guts)
  CorePluginPass
forall (m :: * -> *) a. Monad m => a -> m a
return ModGuts
guts { mg_binds :: [CoreBind]
mg_binds = [CoreBind]
newBinds }

strictify :: ModGuts -> CoreBind -> CoreM (CoreBind)
strictify :: ModGuts -> CoreBind -> CoreM CoreBind
strictify guts :: ModGuts
guts b :: CoreBind
b@(Rec bs :: [(CoreBndr, Expr CoreBndr)]
bs) = do
  Bool
tr <- ([Bool] -> Bool) -> CoreM [Bool] -> CoreM Bool
forall (m :: * -> *) a1 r. Monad m => (a1 -> r) -> m a1 -> m r
liftM [Bool] -> Bool
forall (t :: * -> *). Foldable t => t Bool -> Bool
or (((CoreBndr, Expr CoreBndr) -> CoreM Bool)
-> [(CoreBndr, Expr CoreBndr)] -> CoreM [Bool]
forall (t :: * -> *) (m :: * -> *) a b.
(Traversable t, Monad m) =>
(a -> m b) -> t a -> m (t b)
mapM (ModGuts -> CoreBndr -> CoreM Bool
shouldTransform ModGuts
guts (CoreBndr -> CoreM Bool)
-> ((CoreBndr, Expr CoreBndr) -> CoreBndr)
-> (CoreBndr, Expr CoreBndr)
-> CoreM Bool
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (CoreBndr, Expr CoreBndr) -> CoreBndr
forall a b. (a, b) -> a
fst) [(CoreBndr, Expr CoreBndr)]
bs)
  if Bool
tr then do
    let vs :: [CoreBndr]
vs = ((CoreBndr, Expr CoreBndr) -> CoreBndr)
-> [(CoreBndr, Expr CoreBndr)] -> [CoreBndr]
forall a b. (a -> b) -> [a] -> [b]
map (CoreBndr, Expr CoreBndr) -> CoreBndr
forall a b. (a, b) -> a
fst [(CoreBndr, Expr CoreBndr)]
bs
    [Expr CoreBndr]
es' <- ((CoreBndr, Expr CoreBndr) -> CoreM (Expr CoreBndr))
-> [(CoreBndr, Expr CoreBndr)] -> CoreM [Expr CoreBndr]
forall (t :: * -> *) (m :: * -> *) a b.
(Traversable t, Monad m) =>
(a -> m b) -> t a -> m (t b)
mapM (\ (v :: CoreBndr
v,e :: Expr CoreBndr
e) -> do
                    Bool
lazy <- ModGuts -> CoreBndr -> CoreM Bool
allowLazyData ModGuts
guts CoreBndr
v
                    SCxt -> Expr CoreBndr -> CoreM (Expr CoreBndr)
strictifyExpr (SrcSpan -> Bool -> SCxt
SCxt (Name -> SrcSpan
nameSrcSpan (Name -> SrcSpan) -> Name -> SrcSpan
forall a b. (a -> b) -> a -> b
$ CoreBndr -> Name
forall a. NamedThing a => a -> Name
getName CoreBndr
v) (Bool -> Bool
not Bool
lazy))Expr CoreBndr
e) [(CoreBndr, Expr CoreBndr)]
bs
    CoreBind -> CoreM CoreBind
forall (m :: * -> *) a. Monad m => a -> m a
return ([(CoreBndr, Expr CoreBndr)] -> CoreBind
forall b. [(b, Expr b)] -> Bind b
Rec ([CoreBndr] -> [Expr CoreBndr] -> [(CoreBndr, Expr CoreBndr)]
forall a b. [a] -> [b] -> [(a, b)]
zip [CoreBndr]
vs [Expr CoreBndr]
es'))
  else CoreBind -> CoreM CoreBind
forall (m :: * -> *) a. Monad m => a -> m a
return CoreBind
b
strictify guts :: ModGuts
guts b :: CoreBind
b@(NonRec v :: CoreBndr
v e :: Expr CoreBndr
e) = do
    Bool
tr <- ModGuts -> CoreBndr -> CoreM Bool
shouldTransform ModGuts
guts CoreBndr
v
    if Bool
tr then do
      Bool
lazy <- ModGuts -> CoreBndr -> CoreM Bool
allowLazyData ModGuts
guts CoreBndr
v
      Expr CoreBndr
e' <- SCxt -> Expr CoreBndr -> CoreM (Expr CoreBndr)
strictifyExpr (SrcSpan -> Bool -> SCxt
SCxt (Name -> SrcSpan
nameSrcSpan (Name -> SrcSpan) -> Name -> SrcSpan
forall a b. (a -> b) -> a -> b
$ CoreBndr -> Name
forall a. NamedThing a => a -> Name
getName CoreBndr
v) (Bool -> Bool
not Bool
lazy)) Expr CoreBndr
e
      CoreBind -> CoreM CoreBind
forall (m :: * -> *) a. Monad m => a -> m a
return (CoreBndr -> Expr CoreBndr -> CoreBind
forall b. b -> Expr b -> Bind b
NonRec CoreBndr
v Expr CoreBndr
e')
    else CoreBind -> CoreM CoreBind
forall (m :: * -> *) a. Monad m => a -> m a
return CoreBind
b


scopecheckProgram :: ModGuts -> CoreM ModGuts
scopecheckProgram :: CorePluginPass
scopecheckProgram guts :: ModGuts
guts = do
  [Bool]
res <- (CoreBind -> CoreM Bool) -> [CoreBind] -> CoreM [Bool]
forall (t :: * -> *) (m :: * -> *) a b.
(Traversable t, Monad m) =>
(a -> m b) -> t a -> m (t b)
mapM (ModGuts -> CoreBind -> CoreM Bool
scopecheck ModGuts
guts) (ModGuts -> [CoreBind]
mg_binds ModGuts
guts)
  if [Bool] -> Bool
forall (t :: * -> *). Foldable t => t Bool -> Bool
and [Bool]
res then CorePluginPass
forall (m :: * -> *) a. Monad m => a -> m a
return ModGuts
guts else IO ModGuts -> CoreM ModGuts
forall (m :: * -> *) a. MonadIO m => IO a -> m a
liftIO IO ModGuts
forall a. IO a
exitFailure


scopecheck :: ModGuts -> CoreBind -> CoreM Bool
scopecheck :: ModGuts -> CoreBind -> CoreM Bool
scopecheck guts :: ModGuts
guts (Rec bs :: [(CoreBndr, Expr CoreBndr)]
bs) = do
  Bool
tr <- ([Bool] -> Bool) -> CoreM [Bool] -> CoreM Bool
forall (m :: * -> *) a1 r. Monad m => (a1 -> r) -> m a1 -> m r
liftM [Bool] -> Bool
forall (t :: * -> *). Foldable t => t Bool -> Bool
or (((CoreBndr, Expr CoreBndr) -> CoreM Bool)
-> [(CoreBndr, Expr CoreBndr)] -> CoreM [Bool]
forall (t :: * -> *) (m :: * -> *) a b.
(Traversable t, Monad m) =>
(a -> m b) -> t a -> m (t b)
mapM (ModGuts -> CoreBndr -> CoreM Bool
shouldTransform ModGuts
guts (CoreBndr -> CoreM Bool)
-> ((CoreBndr, Expr CoreBndr) -> CoreBndr)
-> (CoreBndr, Expr CoreBndr)
-> CoreM Bool
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (CoreBndr, Expr CoreBndr) -> CoreBndr
forall a b. (a, b) -> a
fst) [(CoreBndr, Expr CoreBndr)]
bs)
  if Bool
tr then do
    let vs :: [CoreBndr]
vs = ((CoreBndr, Expr CoreBndr) -> CoreBndr)
-> [(CoreBndr, Expr CoreBndr)] -> [CoreBndr]
forall a b. (a -> b) -> [a] -> [b]
map (CoreBndr, Expr CoreBndr) -> CoreBndr
forall a b. (a, b) -> a
fst [(CoreBndr, Expr CoreBndr)]
bs
    let vs' :: Set CoreBndr
vs' = [CoreBndr] -> Set CoreBndr
forall a. Ord a => [a] -> Set a
Set.fromList [CoreBndr]
vs
    [Bool]
valid <- ((CoreBndr, Expr CoreBndr) -> CoreM Bool)
-> [(CoreBndr, Expr CoreBndr)] -> CoreM [Bool]
forall (t :: * -> *) (m :: * -> *) a b.
(Traversable t, Monad m) =>
(a -> m b) -> t a -> m (t b)
mapM (\ (v :: CoreBndr
v,e :: Expr CoreBndr
e) -> Ctx -> Expr CoreBndr -> CoreM Bool
checkExpr (Maybe (Set CoreBndr) -> CoreBndr -> Ctx
emptyCtx (Set CoreBndr -> Maybe (Set CoreBndr)
forall a. a -> Maybe a
Just Set CoreBndr
vs') CoreBndr
v) Expr CoreBndr
e) [(CoreBndr, Expr CoreBndr)]
bs
    Bool -> CoreM Bool
forall (m :: * -> *) a. Monad m => a -> m a
return ([Bool] -> Bool
forall (t :: * -> *). Foldable t => t Bool -> Bool
and [Bool]
valid)
  else Bool -> CoreM Bool
forall (m :: * -> *) a. Monad m => a -> m a
return Bool
True
scopecheck guts :: ModGuts
guts (NonRec v :: CoreBndr
v e :: Expr CoreBndr
e) = do
    Bool
tr <- ModGuts -> CoreBndr -> CoreM Bool
shouldTransform ModGuts
guts CoreBndr
v
    if Bool
tr then do
      Bool
valid <- Ctx -> Expr CoreBndr -> CoreM Bool
checkExpr (Maybe (Set CoreBndr) -> CoreBndr -> Ctx
emptyCtx Maybe (Set CoreBndr)
forall a. Maybe a
Nothing CoreBndr
v) Expr CoreBndr
e
      Bool -> CoreM Bool
forall (m :: * -> *) a. Monad m => a -> m a
return Bool
valid
    else Bool -> CoreM Bool
forall (m :: * -> *) a. Monad m => a -> m a
return Bool
True

getModuleAnnotations :: Data a => ModGuts -> [a]
getModuleAnnotations :: ModGuts -> [a]
getModuleAnnotations guts :: ModGuts
guts = [a]
anns'
  where anns :: [Annotation]
anns = (Annotation -> Bool) -> [Annotation] -> [Annotation]
forall a. (a -> Bool) -> [a] -> [a]
filter (\a :: Annotation
a-> case Annotation -> CoreAnnTarget
ann_target Annotation
a of
                         ModuleTarget m :: Module
m -> Module
m Module -> Module -> Bool
forall a. Eq a => a -> a -> Bool
== (ModGuts -> Module
mg_module ModGuts
guts)
                         _ -> Bool
False) (ModGuts -> [Annotation]
mg_anns ModGuts
guts)
        anns' :: [a]
anns' = (Annotation -> Maybe a) -> [Annotation] -> [a]
forall a b. (a -> Maybe b) -> [a] -> [b]
mapMaybe (([Word8] -> a) -> Serialized -> Maybe a
forall a. Typeable a => ([Word8] -> a) -> Serialized -> Maybe a
fromSerialized [Word8] -> a
forall a. Data a => [Word8] -> a
deserializeWithData (Serialized -> Maybe a)
-> (Annotation -> Serialized) -> Annotation -> Maybe a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Annotation -> Serialized
ann_value) [Annotation]
anns
  



allowLazyData :: ModGuts -> CoreBndr -> CoreM Bool
allowLazyData :: ModGuts -> CoreBndr -> CoreM Bool
allowLazyData guts :: ModGuts
guts bndr :: CoreBndr
bndr = do
  [Rattus]
l <- ModGuts -> CoreBndr -> CoreM [Rattus]
forall a. Data a => ModGuts -> CoreBndr -> CoreM [a]
annotationsOn ModGuts
guts CoreBndr
bndr :: CoreM [Rattus]
  Bool -> CoreM Bool
forall (m :: * -> *) a. Monad m => a -> m a
return (Rattus
AllowLazyData Rattus -> [Rattus] -> Bool
forall (t :: * -> *) a. (Foldable t, Eq a) => a -> t a -> Bool
`elem` [Rattus]
l)


shouldTransform :: ModGuts -> CoreBndr -> CoreM Bool
shouldTransform :: ModGuts -> CoreBndr -> CoreM Bool
shouldTransform guts :: ModGuts
guts bndr :: CoreBndr
bndr = do
  [Rattus]
l <- ModGuts -> CoreBndr -> CoreM [Rattus]
forall a. Data a => ModGuts -> CoreBndr -> CoreM [a]
annotationsOn ModGuts
guts CoreBndr
bndr :: CoreM [Rattus]
  Bool -> CoreM Bool
forall (m :: * -> *) a. Monad m => a -> m a
return (Rattus
Rattus Rattus -> [Rattus] -> Bool
forall (t :: * -> *) a. (Foldable t, Eq a) => a -> t a -> Bool
`elem` [Rattus]
l Bool -> Bool -> Bool
&& Bool -> Bool
not (Rattus
NotRattus Rattus -> [Rattus] -> Bool
forall (t :: * -> *) a. (Foldable t, Eq a) => a -> t a -> Bool
`elem` [Rattus]
l) Bool -> Bool -> Bool
&& CoreBndr -> Bool
userFunction CoreBndr
bndr)

annotationsOn :: (Data a) => ModGuts -> CoreBndr -> CoreM [a]
annotationsOn :: ModGuts -> CoreBndr -> CoreM [a]
annotationsOn guts :: ModGuts
guts bndr :: CoreBndr
bndr = do
  UniqFM [a]
anns <- ([Word8] -> a) -> ModGuts -> CoreM (UniqFM [a])
forall a.
Typeable a =>
([Word8] -> a) -> ModGuts -> CoreM (UniqFM [a])
getAnnotations [Word8] -> a
forall a. Data a => [Word8] -> a
deserializeWithData ModGuts
guts
  [a] -> CoreM [a]
forall (m :: * -> *) a. Monad m => a -> m a
return ([a] -> CoreM [a]) -> [a] -> CoreM [a]
forall a b. (a -> b) -> a -> b
$
    UniqFM [a] -> [a] -> Unique -> [a]
forall key elt. Uniquable key => UniqFM elt -> elt -> key -> elt
lookupWithDefaultUFM UniqFM [a]
anns [] (CoreBndr -> Unique
varUnique CoreBndr
bndr) [a] -> [a] -> [a]
forall a. [a] -> [a] -> [a]
++
    ModGuts -> [a]
forall a. Data a => ModGuts -> [a]
getModuleAnnotations ModGuts
guts