TernaryTrees-0.0.1: Main.hs

module Main where
    
import Data.Binary
-- import Data.Set.TernarySet
import Data.Set.StringSet
import System.IO
import System.Environment

main = do
    (file:_) <- getArgs -- get file name
    contents <- readFile file -- read the file contents
    -- input <- getContents
    let wds = words contents -- separate the words
        tree = fromList $ wds -- put them in the tree
        newname = (file ++ ".bin")
    -- print . treeSize $ tree
    putStr "All input words are in dictionary: "
    print . all (`isElem` tree) $ wds -- make sure all words are actually in the tree
    putStr "Same number of words as input: "
    print (numEntries tree == length wds) -- make sure the same number of words are in the tree
    putStr ("Writing " ++ newname ++ "... ")
    encodeFile newname tree -- write the tree to a file as "filename.bin"
    putStr "done.\nReading data back in... "
    ntree <- decodeFile newname -- read in the file and decode it
    putStr "done.\nRead in data matches original: "
    print (tree == ntree) -- check the read in tree is the same as the one we wrote
    -- print tree
    putStrLn "\n-- Enter a word to see if it is in the dictionary (^C to exit):"
    interact' (("-- " ++) . show . (`isElem` tree)) -- enter a word to see if it's in the tree

interact' :: (String -> String) -> IO ()
interact' f = do
    line <- getLine
    putStrLn (f line)
    interact' f