add graphviz output

Author: ocramz

.travis.yml

@@ -15,7 +15,6 @@ addons:
 env:
 - ARGS="--resolver lts"
 - ARGS="--resolver nightly"
-## - ARGS="--resolver lts-11.5"
 - ARGS="--resolver lts-12.14"
 - ARGS="--resolver nightly-2019-08-04"
 

app/Main.hs

@@ -0,0 +1,27 @@
+{-# language OverloadedStrings #-}
+module Main where
+
+import Numeric.Statistics.Inference.Bayes.Exact.VariableElimination (student)
+
+import Algebra.Graph.Export (Doc, literal, render, export)
+import Algebra.Graph (Graph)
+
+import qualified Data.Text as T
+import qualified Data.Text.IO as T (writeFile)
+
+
+-- vDoc x   = literal (show x) <> "\n"
+-- eDoc x y = literal (show x) <> " -> " <> literal (show y) <> "\n"
+
+vDoc x   = literal [x] <> ";\n"
+eDoc x y = literal [x] <> " -> " <> literal [y] <> ";\n"
+
+withDigraph :: String -> Doc String -> Doc String
+withDigraph name body =
+  literal (unwords ["digraph", name, "{\n"]) <>
+  body <>
+  literal "}\n"
+
+main :: IO ()
+main = T.writeFile "graph.example.student.dot" $
+           T.pack $ render $ withDigraph "student" $ export vDoc eDoc student

bayesian-inference.cabal

@@ -46,6 +46,16 @@ library
                      , hspec
                      , QuickCheck
 
+executable graph-export
+  default-language:    Haskell2010
+  ghc-options:         -threaded -rtsopts -with-rtsopts=-N
+  hs-source-dirs:      app
+  main-is:             Main.hs
+  build-depends:       base
+                     , bayesian-inference
+                     , algebraic-graphs
+                     , text
+
 test-suite spec
   default-language:    Haskell2010
   ghc-options:         -Wall
@@ -70,14 +80,14 @@ test-suite doctest
                      , doctest
                      , exceptions
 
-test-suite property
-  default-language:    Haskell2010
-  ghc-options:         -Wall
-  type:                exitcode-stdio-1.0
-  hs-source-dirs:      test
-  main-is:             Property.hs
-  build-depends:       base
-                     , bayesian-inference
+-- test-suite property
+--   default-language:    Haskell2010
+--   ghc-options:         -Wall
+--   type:                exitcode-stdio-1.0
+--   hs-source-dirs:      test
+--   main-is:             Property.hs
+--   build-depends:       base
+--                      , bayesian-inference
 
 source-repository head
   type:     git

makefile

@@ -0,0 +1,4 @@
+graph-render:
+	stack build
+	stack exec -- graph-export
+	dot -Tps graph.example.student.dot -o graph.example.student.ps

src/Numeric/Statistics/Inference/Bayes/Exact/VariableElimination.hs

@@ -77,7 +77,7 @@ fromList :: (Foldable t, Monad m) => t a -> Lex m (IM.IntMap a)
 fromList xs = foldlM (flip insert) IM.empty xs
 
 -- | sequential sum-product elimination of graph factors
---
+-- 
 -- >>> elims student "cdihg"
 -- fromList [(5,{ 'j' 'l' 's' }),(12,{ 'j' 'l' 's' })]
 elims :: (Monad m, TG.ToGraph g, Foldable t, Ord (TG.ToVertex g)) =>

test/DocTest.hs

@@ -2,8 +2,8 @@ module Main where
 
 import Test.DocTest (doctest)
 
-
+main :: IO ()
 main = doctest [
-  "test/Bayes/Exact/VariableElimination.hs"
+  "src/Numeric/Statistics/Inference/Bayes/Exact/VariableElimination.hs"
 
         ]