File.hs

module TPDB.Input.File where

import TPDB.Data
import TPDB.Convert
import qualified TPDB.ARI

import qualified TPDB.Input.Memory as TIM

import qualified Data.Text.Lazy as T
import qualified Data.Text.Lazy.IO as T
import qualified Data.ByteString as BS
import System.FilePath.Posix ( takeExtension )

-- | read input from file with given name.
-- can have extension .srs, .trs, .xml, .ari
-- unknown extension is considered as .xml, because of 
-- http://starexec.forumotion.com/t60-restore-file-extension-for-renamed-benchmarks

get :: FilePath 
         -> IO ( Either (TRS Identifier Identifier) 
                        ( SRS Identifier ) )
get f = do
    m <- getE f
    case m of
        Right x -> return x 
        Left err -> error err

getE :: FilePath 
         -> IO (Either String
                 ( Either (TRS Identifier Identifier) 
                        ( SRS Identifier ) ) )
getE f = case takeExtension f of
  ".ari" -> do
    s <- BS.readFile f
    return $ fmap Left $ TPDB.ARI.get s
  _ ->  do
    s <- T.readFile f
    TIM.get f s

get_trs f = do
    x <- get f
    return $ case x of
        Right x -> srs2trs x
        Left  x -> x

getE_trs f = do
    e <- getE f
    return $ case e of
        Right x -> Right $ case x of
            Right x -> srs2trs x
            Left  x -> x
        Left e -> Left e

get_srs f = do
    x <- get f
    return $ case x of
        Right x -> x
        Left  x -> case trs2srs x of
            Nothing -> error "not an SRS"
            Just x -> x