mirror of
https://github.com/macocianradu/RosettaHaskellCompiler.git
synced 2026-03-18 13:00:08 +00:00
40 lines
1.9 KiB
Haskell
40 lines
1.9 KiB
Haskell
{-# LANGUAGE OverloadedStrings #-}
|
|
|
|
module PrettyPrinter.Type where
|
|
|
|
import Prettyprinter
|
|
import PrettyPrinter.General
|
|
import Model.Type
|
|
|
|
printType :: Type -> String
|
|
printType (MakeType name (Just (MakeType super _ _ _)) description attributes) = printType (MakeType name Nothing description (superToAttribute super:attributes))
|
|
printType (MakeType _ (Just (BasicType _)) _ _) = error "Can't extend basic types"
|
|
printType (MakeType name Nothing description attributes) =
|
|
show $ printDescription description (vcat [nest 4 $ vcat ("data" <+> pretty name <+> "=" <+> "Make" <> pretty name <+> "{": printAttributes attributes), "}", ""])
|
|
printType (BasicType name) = show $ pretty name
|
|
|
|
--printSuperType :: Maybe Type -> Doc a
|
|
--printSuperType (Just (MakeType name _ _ _)) = "super" <+> "::" <+> pretty name
|
|
--printSuperType (Just (BasicType _)) = error "Can't extend basic types"
|
|
--printSuperType Nothing = emptyDoc
|
|
|
|
superToAttribute :: String -> TypeAttribute
|
|
superToAttribute name = MakeTypeAttribute "super" (MakeType name Nothing Nothing []) (Bounds (1, 1)) (Just "Pointer to super class")
|
|
|
|
printAttributes :: [TypeAttribute] -> [Doc a]
|
|
printAttributes [] = []
|
|
printAttributes [at] = [printAttribute at]
|
|
printAttributes (at : ats) = (printAttribute at <> ",") : printAttributes ats
|
|
|
|
printAttribute :: TypeAttribute -> Doc a
|
|
printAttribute (MakeTypeAttribute name typ crd description) =
|
|
printDescription description
|
|
(pretty name <+> "::" <+> printCardinality (MakeTypeAttribute name typ crd description))
|
|
|
|
printCardinality :: TypeAttribute -> Doc a
|
|
printCardinality (MakeTypeAttribute _ typ (Bounds (x, y)) _)
|
|
| x == 0 && y == 1 = "Maybe" <+> pretty (typeName typ)
|
|
| x == 1 && y == 1 = pretty (typeName typ)
|
|
| otherwise = "[" <> pretty (typeName typ) <> "]"
|
|
printCardinality (MakeTypeAttribute _ typ NoBounds _) = "[" <> pretty (typeName typ) <> "]"
|
|
printCardinality (MakeTypeAttribute _ typ (OneBound _) _) = "[" <> pretty (typeName typ) <> "]" |