CGI script
[darcs-mirror-polyfix.git] / ParseType.hs
index f251a8a..b3e1715 100644 (file)
@@ -1,6 +1,9 @@
 {-# LANGUAGE FlexibleContexts, PatternSignatures, DeriveDataTypeable #-}
 module ParseType (
          parseType
+       , parseType'
+       , instType
+       , unquantify
        , TypVar(..)
        , Typ(..)
        ) where
@@ -8,7 +11,6 @@ module ParseType (
 import Language.Haskell.Parser (parseModule, ParseResult(..))
 import Language.Haskell.Syntax
 
-import Control.Monad
 import Control.Monad.Error
 import Control.Monad.Reader
 import Data.List
@@ -19,7 +21,12 @@ import Data.Generics.Schemes
 import Data.Char
 import Data.Maybe
 
-newtype TypVar = TypVar Int deriving (Show, Eq, Typeable, Data)
+data TypVar = TypVar Int        -- alpha, beta etc.
+            | TypInst Int Bool  -- t1,t2 etc
+       deriving (Show, Eq, Typeable, Data)
+
+instType :: Bool -> Typ -> Typ
+instType rightSide typ = everywhere (mkT (\(TypVar i) -> TypInst i rightSide)) typ
 
 data Typ    = TVar    TypVar
             | Arrow   Typ     Typ
@@ -32,6 +39,12 @@ data Typ    = TVar    TypVar
             | TEither  Typ     Typ
             deriving (Show, Eq, Typeable, Data)
 
+unquantify :: Typ -> Typ
+unquantify (All     _ t) = unquantify t
+unquantify (AllStar _ t) = unquantify t
+unquantify t             = t
+
+parseType :: String -> Typ
 parseType = either error id . parseType'
 
 -- | A simple type parser.
@@ -62,7 +75,7 @@ createVarMap :: HsType -> M.Map HsName TypVar
 createVarMap hstype = M.fromList $ zip
                        (nub (listify isVar hstype))
                        (map TypVar [1..])
-  where isVar (HsIdent (x:xs)) | isLower x  = True
+  where isVar (HsIdent (x:_)) | isLower x  = True
         isVar _                            =  False
 
 
@@ -85,6 +98,6 @@ simplifiyType t
                                = throwError ("Unsupported type " ++ show t)
 
 quantify :: [TypVar] -> Typ -> Typ
-quantify special t = foldr all t (nub (listify (\(_::TypVar) -> True) t))
-  where all v | v `elem` special = All v
-              | otherwise        = AllStar v
+quantify special t = foldr allQuant t (nub (listify (\(_::TypVar) -> True) t))
+  where allQuant v | v `elem` special = All v
+                   | otherwise        = AllStar v