added -nofun and -nocat options to vt

src-3.0/PGF/Macros.hs

@@ -34,6 +34,9 @@ lookType :: PGF -> CId -> Type
 lookType pgf f = 
   fst $ lookMap (error $ "lookType " ++ show f) f (funs (abstract pgf))
 
+lookValCat :: PGF -> CId -> CId
+lookValCat pgf = valCat . lookType pgf
+
 lookParser :: PGF -> CId -> Maybe ParserInfo
 lookParser pgf lang = Map.lookup lang (concretes pgf) >>= parser
 

src-3.0/PGF/VisualizeTree.hs

@@ -20,12 +20,13 @@ module PGF.VisualizeTree ( visualizeTrees
 
 import PGF.CId (prCId)
 import PGF.Data
+import PGF.Macros (lookValCat)
 
-visualizeTrees :: Bool -> [Tree] -> String
-visualizeTrees digr = unlines . map (prGraph digr . tree2graph digr)
+visualizeTrees :: PGF -> (Bool,Bool) -> [Tree] -> String
+visualizeTrees pgf funscats = unlines . map (prGraph False . tree2graph pgf funscats)
 
-tree2graph :: Bool -> Tree -> [String]
-tree2graph digr = prf [] where
+tree2graph :: PGF -> (Bool,Bool) -> Tree -> [String]
+tree2graph pgf (funs,cats) = prf [] where
   prf ps t = case t of
     Fun cid trees -> 
       let (nod,lab) = prn ps cid in
@@ -33,10 +34,15 @@ tree2graph digr = prf [] where
         [       pra (j:ps) nod t | (j,t) <- zip [0..] trees] ++
         concat [prf (j:ps) t     | (j,t) <- zip [0..] trees]
   prn ps cid = 
-    let lab = "\"" ++ prCId cid ++ "\""
+    let
+      fun = if funs then prCId cid else ""
+      cat = if cats then prCat cid else ""
+      colon = if funs && cats then " : " else ""
+      lab = "\"" ++ fun ++ colon ++ cat ++ "\""
     in (show(show (ps :: [Int])),lab)
   pra i nod t@(Fun cid _) =  nod ++ arr ++ fst (prn i cid) ++ " [style = \"solid\"];"
-  arr = if digr then " -> " else " -- "
+  arr = " -- " -- if digr then " -> " else " -- "
+  prCat = prCId . lookValCat pgf
 
 prGraph digr ns = concat $ map (++"\n") $ [graph ++ "{\n"] ++ ns ++ ["}"] where
   graph = if digr then "digraph" else "graph"