{- |

Module : $Header$

Copyright : (c) Christian Maeder and Uni Bremen 2002-2003

Licence : similar to LGPL, see HetCATS/LICENCE.txt or LIZENZ.txt

Maintainer : hets@tzi.de

Stability : provisional

Portability : portable

analyse type decls

-}

module HasCASL.TypeDecl where

import HasCASL.As

import HasCASL.Le

import HasCASL.ClassAna

import HasCASL.Unify

import qualified Common.Lib.Map as Map

import Common.Lexer

import Common.Id

import Common.AS_Annotation

import Common.Lib.State

import Data.Maybe

import Data.List

import Common.Result

import Common.GlobalAnnotations

import HasCASL.ClassAna

import HasCASL.TypeAna

import HasCASL.DataAna

import HasCASL.VarDecl

import HasCASL.TypeCheck

-- | add a supertype to a given type id

addSuperType :: Type -> Id -> State Env ()

addSuperType t i =

do tk <- gets typeMap

case Map.lookup i tk of

Nothing -> return () -- previous error

Just (TypeInfo ok ks sups defn) ->

putTypeMap $ Map.insert i

(TypeInfo ok ks (t:sups) defn)

tk

idsToTypePatterns :: [Maybe Id] -> [TypePattern]

idsToTypePatterns mis = map ( \ i -> TypePattern i [] [] )

$ catMaybes mis

anaFormula :: GlobalAnnos -> Annoted Term -> State Env (Maybe (Annoted Term))

anaFormula ga at =

do mt <- resolveTerm ga (Just logicalType) $ item at

return $ case mt of Nothing -> Nothing

Just e -> Just at { item = e }

anaVars :: Vars -> Type -> Result [VarDecl]

anaVars (Var v) t = return [VarDecl v t Other []]

anaVars (VarTuple vs _) t =

case t of

ProductType ts _ ->

if length ts == length vs then

let lrs = zipWith anaVars vs ts

lms = map maybeResult lrs in

if all isJust lms then

return $ concatMap fromJust lms

else Result (concatMap diags lrs) Nothing

else Result [mkDiag Error "wrong arity" t] Nothing

_ -> Result [mkDiag Error "product type expected instead" t] Nothing

-- | analyse a 'TypeItem'

anaTypeItem :: GlobalAnnos -> GenKind -> Instance -> TypeItem

-> State Env TypeItem

anaTypeItem _ _ inst (TypeDecl pats kind ps) =

do ak <- anaKind kind

let Result ds (Just is) = convertTypePatterns pats

addDiags ds

mis <- mapM (addTypePattern NoTypeDefn inst ak) is

return $ TypeDecl (idsToTypePatterns mis) ak ps

anaTypeItem _ _ inst (SubtypeDecl pats t ps) =

do let Result ds (Just is) = convertTypePatterns pats

addDiags ds

mis <- mapM (addTypePattern NoTypeDefn inst star) is

let newPats = idsToTypePatterns mis

mt <- anaStarType t

case mt of

Nothing -> return $ TypeDecl newPats star ps

Just newT -> do mapM_ (addSuperType newT) $ map fst is

return $ SubtypeDecl newPats newT ps

anaTypeItem _ _ inst (IsoDecl pats ps) =

do let Result ds (Just is) = convertTypePatterns pats

js = map fst is

addDiags ds

mis <- mapM (addTypePattern NoTypeDefn inst star) is

mapM_ ( \ i -> mapM_ (addSuperType (TypeName i star 0)) js) js

return $ IsoDecl (idsToTypePatterns mis) ps

anaTypeItem ga _ inst (SubtypeDefn pat v t f ps) =

do let Result ds m = convertTypePattern pat

addDiags ds

mty <- anaStarType t

let Result es mvds = case mty of

Nothing -> Result [] Nothing

Just ty -> anaVars v ty

addDiags es

as <- gets assumps

mv <- case mvds of

Nothing -> return Nothing

Just vds -> do checkUniqueVars vds

mvs <- mapM addVarDecl vds

if all isJust mvs then

return $ Just v

else return Nothing

mt <- anaFormula ga f

putAssumps as

case m of

Nothing -> return $ TypeDecl [] star ps

Just i -> case (mt, mv) of

(Just newF, Just _) -> do

let newT = fromJust mty

mi <- addTypePattern (Supertype v newT $ item newF)

inst star i

addSuperType newT $ fst i

case mi of

Nothing -> return $ TypeDecl [] star ps

Just _ -> return $ SubtypeDefn

(TypePattern (fst i) (snd i) [])

v newT newF ps

_ -> do mi <- addTypePattern NoTypeDefn inst star i

return $ TypeDecl (idsToTypePatterns [mi]) star ps

anaTypeItem _ _ inst (AliasType pat mk sc ps) =

do let Result ds m = convertTypePattern pat

addDiags ds

(ik, mt) <- anaPseudoType mk sc

case m of

Just i -> case mt of

Nothing -> return $ TypeDecl [] star ps

Just nsc -> do let newPty = generalize nsc

addTypePattern (AliasTypeDefn newPty)

inst ik i

return $ AliasType

(TypePattern (fst i) (snd i)[])

(Just ik) newPty ps

_ -> return $ TypeDecl [] star ps

anaTypeItem _ gk inst (Datatype d) =

do newD <- anaDatatype gk inst d

return $ Datatype newD

-- | analyse a 'DatatypeDecl'

anaDatatype :: GenKind -> Instance -> DatatypeDecl -> State Env DatatypeDecl

anaDatatype genKind inst (DatatypeDecl pat kind alts derivs ps) =

do k <- anaKind kind

checkKinds pat star k

cs <- mapM anaClassId derivs

let jcs = catMaybes cs

mapM (checkKinds pat star) jcs

let newDerivs = foldr( \ ck l -> case ck of

ClassKind ci _ -> ci:l

_ -> l) [] jcs

Result ds m = convertTypePattern pat

addDiags ds

case m of

Nothing -> return $ DatatypeDecl pat k [] newDerivs ps

Just (i, as) ->

do let dt = TypeName i (typeArgsListToKind as k) 0

newAlts <- anaAlts dt

$ map item alts

mapM_ ( \ (Construct c tc p sels) -> do

let ty = simpleTypeScheme $ getConstrType dt p tc

addOpId c ty

[] (ConstructData i)

mapM_ ( \ (Select s ts pa) ->

addOpId s (simpleTypeScheme $

getSelType dt pa ts)

[] (SelectData [ConstrInfo c ty] i)

) sels) newAlts

mi <- addTypePattern (DatatypeDefn genKind [] newAlts)

inst k (i, as)

return $ DatatypeDecl (TypePattern i as []) k

(case mi of Nothing -> []

Just _ -> alts)

newDerivs ps

-- | analyse a pseudo type (represented as a 'TypeScheme')

anaPseudoType :: Maybe Kind -> TypeScheme -> State Env (Kind, Maybe TypeScheme)

anaPseudoType mk (TypeScheme tArgs (q :=> ty) p) =

do k <- case mk of

Nothing -> return Nothing

Just j -> anaKindM j

tm <- gets typeMap -- save global variables

mapM_ anaTypeVarDecl tArgs

(sk, mt) <- anaType (k, ty)

let newK = typeArgsListToKind tArgs sk

case mk of

Nothing -> return ()

Just j -> checkKinds ty j newK

putTypeMap tm -- forget local variables

case mt of

Nothing -> return (newK, Nothing)

Just newTy -> return (newK, Just $ TypeScheme tArgs (q :=> newTy) p)

-- | add a type pattern

addTypePattern :: TypeDefn -> Instance -> Kind -> (Id, [TypeArg])

-> State Env (Maybe Id) -- , [TypeArg]))

addTypePattern defn inst kind (i, as) =

do newAs <- mapM anaTypeVarDecl as

let fullKind = typeArgsListToKind (catMaybes newAs) kind

addTypeId defn inst fullKind i

-- | extent a kind to expect further type arguments

typeArgsListToKind :: [TypeArg] -> Kind -> Kind

typeArgsListToKind tArgs k =

if null tArgs then k

else typeArgsListToKind (init tArgs)

(FunKind (( \ (TypeArg _ xk _ _) -> xk) $ last tArgs) k [])

-- | convert type patterns

convertTypePatterns :: [TypePattern] -> Result [(Id, [TypeArg])]

convertTypePatterns [] = Result [] $ Just []

convertTypePatterns (s:r) =

let Result d m = convertTypePattern s

Result ds (Just l) = convertTypePatterns r

in case m of

Nothing -> Result (d++ds) $ Just l

Just i -> Result (d++ds) $ Just (i:l)

illegalTypePattern, illegalTypePatternArg :: TypePattern -> Result a

illegalTypePattern tp = fatal_error ("illegal type pattern: " ++

showPretty tp "") $ getMyPos tp

illegalTypePatternArg tp = fatal_error ("illegal type pattern argument: " ++

showPretty tp "") $ getMyPos tp

-- | convert a 'TypePattern'

convertTypePattern :: TypePattern -> Result (Id, [TypeArg])

convertTypePattern (TypePattern t as _) = return (t, as)

convertTypePattern tp@(TypePatternToken t) =

if isPlace t then illegalTypePattern tp

else return (simpleIdToId t, [])

convertTypePattern tp@(MixfixTypePattern

[ra, ri@(TypePatternToken inTok), rb]) =

if head (tokStr inTok) `elem` signChars

then let inId = Id [Token place $ getMyPos ra, inTok,

Token place $ getMyPos rb] [] [] in

case (ra, rb) of

(TypePatternToken (Token "__" _),

TypePatternToken (Token "__" _)) -> return (inId, [])

_ -> do a <- convertToTypeArg ra

b <- convertToTypeArg rb

return (inId, [a, b])

else case ra of

TypePatternToken t1 -> do

a <- convertToTypeArg ri

b <- convertToTypeArg rb

return (simpleIdToId t1, [a, b])

_ -> illegalTypePattern tp

convertTypePattern tp@(MixfixTypePattern (TypePatternToken t1 : rp)) =

if isPlace t1 then

case rp of

[TypePatternToken inId, TypePatternToken t2] ->

if isPlace t2 && head (tokStr inId) `elem` signChars

then return (Id [t1,inId,t2] [] [], [])

else illegalTypePattern tp

_ -> illegalTypePattern tp

else do as <- mapM convertToTypeArg rp

return (simpleIdToId t1, as)

convertTypePattern (BracketTypePattern bk [ap] ps) =

case bk of

Parens -> convertTypePattern ap

_ -> let (o, c) = getBrackets bk

tid = Id [Token o $ head ps, Token place $ getMyPos ap,

Token c $ last ps] [] [] in

case ap of

TypePatternToken t -> if isPlace t then

return (tid, [])

else return (tid, [TypeArg (simpleIdToId t) MissingKind Other []])

_ -> do a <- convertToTypeArg ap

return (tid, [a])

convertTypePattern tp = illegalTypePattern tp

convertToTypeArg :: TypePattern -> Result TypeArg

convertToTypeArg tp@(TypePatternToken t) =

if isPlace t then illegalTypePatternArg tp

else return $ TypeArg (simpleIdToId t) MissingKind Other []

convertToTypeArg (TypePatternArg a _) = return a

convertToTypeArg (BracketTypePattern Parens [tp] _) = convertToTypeArg tp

convertToTypeArg tp = illegalTypePatternArg tp