From 0b80b3ef0aa2daaef66f642fdff984c4eb7709f0 Mon Sep 17 00:00:00 2001
From: krasimir <krasimir@chalmers.se>
Date: Wed, 24 Nov 2010 12:51:54 +0000
Subject: command ai in the shell now shows the probability of the tree

---
 src/compiler/GF/Command/Commands.hs | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

(limited to 'src/compiler/GF')

diff --git a/src/compiler/GF/Command/Commands.hs b/src/compiler/GF/Command/Commands.hs
index faaa9e3ab..b4ab5ff4a 100644
--- a/src/compiler/GF/Command/Commands.hs
+++ b/src/compiler/GF/Command/Commands.hs
@@ -902,21 +902,24 @@ allCommands env@(pgf, mos) = Map.fromList [
      exec = \opts arg -> do
        case arg of
          [EFun id] -> case Map.lookup id (funs (abstract pgf)) of
-                        Just fd -> return $ fromString $
-                                      render (ppFun id fd)
+                        Just fd -> do putStrLn $ render (ppFun id fd)
+                                      putStrLn ("Probability: "++show (probTree pgf (EFun id)))
+                                      return void
                         Nothing -> case Map.lookup id (cats (abstract pgf)) of
-                                     Just hyps -> do return $ fromString $
+                                     Just hyps -> do putStrLn $
                                                         render (ppCat id hyps $$
                                                                 if null (functionsToCat pgf id)
                                                                   then empty
                                                                   else space $$
                                                                        vcat [ppFun fid (ty,0,Just [],0) | (fid,ty) <- functionsToCat pgf id])
+                                                     return void
                                      Nothing   -> do putStrLn ("unknown category of function identifier "++show id)
                                                      return void
          [e]         -> case inferExpr pgf e of
                           Left tcErr   -> error $ render (ppTcError tcErr)
-                          Right (e,ty) -> do putStrLn ("Expression: "++showExpr [] e)
-                                             putStrLn ("Type:       "++showType [] ty)
+                          Right (e,ty) -> do putStrLn ("Expression:  "++showExpr [] e)
+                                             putStrLn ("Type:        "++showType [] ty)
+                                             putStrLn ("Probability: "++show (probTree pgf e))
                                              return void
          _           -> do putStrLn "a single identifier or expression is expected from the command"
                            return void,
-- 
cgit v1.2.3