git.iko.soy/project-m36
on fbfa2ea help sign in
project-m36/src/lib/ProjectM36/DataConstructorDef.hs
20 lines 831 B history raw
1module ProjectM36.DataConstructorDef where
2import ProjectM36.Base
3import qualified ProjectM36.TypeConstructor as TC
4import qualified Data.Set as S
6emptyDataConstructor :: DataConstructorName -> DataConstructorDef
7emptyDataConstructor name' = DataConstructorDef name' []
9name :: DataConstructorDef -> DataConstructorName
10name (DataConstructorDef name' _) = name'
12fields :: DataConstructorDef -> [DataConstructorDefArg]
13fields (DataConstructorDef _ args) = args
15typeVars :: DataConstructorDef -> S.Set TypeVarName
16typeVars (DataConstructorDef _ tConsArgs) = S.unions $ map typeVarsInDefArg tConsArgs
18typeVarsInDefArg :: DataConstructorDefArg -> S.Set TypeVarName
19typeVarsInDefArg (DataConstructorDefTypeConstructorArg tCons) = TC.typeVars tCons
20typeVarsInDefArg (DataConstructorDefTypeVarNameArg pVarName) = S.singleton pVarName