{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE TypeFamilies #-}
module Futhark.IR.Prop.Patterns
(
paramIdent,
paramType,
paramDeclType,
patElemIdent,
patElemType,
setPatElemDec,
patIdents,
patNames,
patTypes,
patSize,
basicPat,
)
where
import Futhark.IR.Prop.Types (DeclTyped (..), Typed (..))
import Futhark.IR.Syntax
paramType :: Typed dec => Param dec -> Type
paramType :: Param dec -> Type
paramType = Param dec -> Type
forall t. Typed t => t -> Type
typeOf
paramDeclType :: DeclTyped dec => Param dec -> DeclType
paramDeclType :: Param dec -> DeclType
paramDeclType = Param dec -> DeclType
forall t. DeclTyped t => t -> DeclType
declTypeOf
paramIdent :: Typed dec => Param dec -> Ident
paramIdent :: Param dec -> Ident
paramIdent Param dec
param = VName -> Type -> Ident
Ident (Param dec -> VName
forall dec. Param dec -> VName
paramName Param dec
param) (Param dec -> Type
forall t. Typed t => t -> Type
typeOf Param dec
param)
patElemIdent :: Typed dec => PatElemT dec -> Ident
patElemIdent :: PatElemT dec -> Ident
patElemIdent PatElemT dec
pelem = VName -> Type -> Ident
Ident (PatElemT dec -> VName
forall dec. PatElemT dec -> VName
patElemName PatElemT dec
pelem) (PatElemT dec -> Type
forall t. Typed t => t -> Type
typeOf PatElemT dec
pelem)
patElemType :: Typed dec => PatElemT dec -> Type
patElemType :: PatElemT dec -> Type
patElemType = PatElemT dec -> Type
forall t. Typed t => t -> Type
typeOf
setPatElemDec :: PatElemT oldattr -> newattr -> PatElemT newattr
setPatElemDec :: PatElemT oldattr -> newattr -> PatElemT newattr
setPatElemDec PatElemT oldattr
pe newattr
x = (oldattr -> newattr) -> PatElemT oldattr -> PatElemT newattr
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap (newattr -> oldattr -> newattr
forall a b. a -> b -> a
const newattr
x) PatElemT oldattr
pe
patIdents :: Typed dec => PatT dec -> [Ident]
patIdents :: PatT dec -> [Ident]
patIdents = (PatElemT dec -> Ident) -> [PatElemT dec] -> [Ident]
forall a b. (a -> b) -> [a] -> [b]
map PatElemT dec -> Ident
forall dec. Typed dec => PatElemT dec -> Ident
patElemIdent ([PatElemT dec] -> [Ident])
-> (PatT dec -> [PatElemT dec]) -> PatT dec -> [Ident]
forall b c a. (b -> c) -> (a -> b) -> a -> c
. PatT dec -> [PatElemT dec]
forall dec. PatT dec -> [PatElemT dec]
patElems
patNames :: PatT dec -> [VName]
patNames :: PatT dec -> [VName]
patNames = (PatElemT dec -> VName) -> [PatElemT dec] -> [VName]
forall a b. (a -> b) -> [a] -> [b]
map PatElemT dec -> VName
forall dec. PatElemT dec -> VName
patElemName ([PatElemT dec] -> [VName])
-> (PatT dec -> [PatElemT dec]) -> PatT dec -> [VName]
forall b c a. (b -> c) -> (a -> b) -> a -> c
. PatT dec -> [PatElemT dec]
forall dec. PatT dec -> [PatElemT dec]
patElems
patTypes :: Typed dec => PatT dec -> [Type]
patTypes :: PatT dec -> [Type]
patTypes = (Ident -> Type) -> [Ident] -> [Type]
forall a b. (a -> b) -> [a] -> [b]
map Ident -> Type
identType ([Ident] -> [Type]) -> (PatT dec -> [Ident]) -> PatT dec -> [Type]
forall b c a. (b -> c) -> (a -> b) -> a -> c
. PatT dec -> [Ident]
forall dec. Typed dec => PatT dec -> [Ident]
patIdents
patSize :: PatT dec -> Int
patSize :: PatT dec -> Int
patSize (Pat [PatElemT dec]
xs) = [PatElemT dec] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [PatElemT dec]
xs
basicPat :: [Ident] -> PatT Type
basicPat :: [Ident] -> PatT Type
basicPat [Ident]
values =
[PatElemT Type] -> PatT Type
forall dec. [PatElemT dec] -> PatT dec
Pat ([PatElemT Type] -> PatT Type) -> [PatElemT Type] -> PatT Type
forall a b. (a -> b) -> a -> b
$ (Ident -> PatElemT Type) -> [Ident] -> [PatElemT Type]
forall a b. (a -> b) -> [a] -> [b]
map Ident -> PatElemT Type
patElem [Ident]
values
where
patElem :: Ident -> PatElemT Type
patElem (Ident VName
name Type
t) = VName -> Type -> PatElemT Type
forall dec. VName -> dec -> PatElemT dec
PatElem VName
name Type
t