nkjp-0.3.1: tools/nkjp2text.hs

{-# LANGUAGE OverloadedStrings #-}

import           Control.Monad (forM_)
import           System.Environment (getArgs)
import           Data.Foldable (foldMap)
import qualified Data.Text.Lazy as L
import qualified Data.Text.Lazy.IO as L

import qualified Text.NKJP.Named as Ne
import qualified Text.NKJP.Morphosyntax as Mx

main :: IO ()
main = do
    [teiPath] <- getArgs
    fs <- Ne.readTrees teiPath
    forM_ fs $ \ts ->
        L.putStrLn . L.strip . showLeaves . getLeaves $ ts
  where
    getLeaves  = concatMap (foldMap getRight)
    showLeaves = L.concat . map showLeaf
    showLeaf x = if Mx.nps x
        then Mx.orth x
        else " " `L.append` Mx.orth x
    getRight (Right x) = [x]
    getRight _         = []