module Feldspar.Compiler.Backend.C.Plugin.VariableRoleAssigner where
import Data.List
import Feldspar.Transformation
data VariableRoleAssigner = VariableRoleAssigner
data Parameters = Parameters
{ inParametersVRA :: [String]
, outParametersVRA :: [String]
}
instance Transformation VariableRoleAssigner where
type From VariableRoleAssigner = ()
type To VariableRoleAssigner = ()
type Down VariableRoleAssigner = Parameters
type Up VariableRoleAssigner = ()
type State VariableRoleAssigner = ()
instance Transformable VariableRoleAssigner Variable where
transform t s d v = Result v' () () where
v' = v { varRole = if (varName v `elem` outParametersVRA d) ||
(isStruct v && (varName v `elem` inParametersVRA d))
then Pointer else Value
, varLabel = ()
}
instance Transformable VariableRoleAssigner Definition where
transform t s d p@(Procedure n i o pr inf1 inf2) = defaultTransform t s d' p where
d' = Parameters
{ inParametersVRA = map varName i
, outParametersVRA = map varName o
}
transform t s d p = defaultTransform t s d p
instance Plugin VariableRoleAssigner where
type ExternalInfo VariableRoleAssigner = ()
executePlugin self@VariableRoleAssigner externalInfo procedure =
result $ transform self () (Parameters [] []) procedure
isStruct :: Variable () -> Bool
isStruct v = case varType v of
(StructType types) -> True
otherwise -> False