aboutsummaryrefslogtreecommitdiff
path: root/process-database.hs
diff options
context:
space:
mode:
Diffstat (limited to 'process-database.hs')
-rwxr-xr-xprocess-database.hs8
1 files changed, 6 insertions, 2 deletions
diff --git a/process-database.hs b/process-database.hs
index bfbd4b7..a9eb97c 100755
--- a/process-database.hs
+++ b/process-database.hs
@@ -152,12 +152,16 @@ main :: IO ()
main = do
(csvfname, outfname) <- getArgs >>= \case
[csvfname, outfname] -> return (csvfname, outfname)
- _ -> die "Usage: process-database <database.csv> <output.bin>"
+ _ -> die "Usage: process-database <database.csv> <output.bin>\n\
+ \Additionally reads <database-extra.csv> and <table.txt>."
(hira2kata, kata2id, nIDs) <- readTable <$> readFile "table.txt"
let any2id = Map.map (kata2id Map.!) hira2kata <> kata2id
- rows <- concatMap (normalise any2id) . tail . readCSV <$> readFile csvfname
+ rows <- do
+ rows1 <- tail . readCSV <$> readFile csvfname
+ rows2 <- tail . readCSV <$> readFile "database-extra.csv"
+ return (concatMap (normalise any2id) (rows1 ++ rows2))
let trie = makeTrie 0 rows
-- print trie