Better datatype formatting.

tomahawkins · Apr 10, 2012 · 484f798 · 484f798
1 parent 75eecc2
commit 484f798
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/Language/CIRC.hs b/Language/CIRC.hs
@@ -130,9 +130,10 @@ codeModule initModuleName initImports rootTypeName moduleName unsortedTypes tran
 
 codeTypeDef :: TypeDef -> String
 codeTypeDef (TypeDef name params ctors) = "data " ++ name ++ " " ++ intercalate " " params ++ "\n  = " ++ 
-  intercalate "\n  | " [ name ++ " " ++ intercalate " " (map codeType args) | CtorDef name args <- ctors' ] ++ "\n"
+  intercalate "\n  | " [ name ++ replicate (m - length name) ' ' ++ " " ++ intercalate " " (map codeType args) | CtorDef name args <- ctors' ] ++ "\n"
   where
   ctors' = sortBy (compare `on` \ (CtorDef n _) -> n) ctors
+  m = maximum [ length n | CtorDef n _ <- ctors ]
 
 codeType :: Type -> String
 codeType a = case a of