blob: a11f7fcdc789ac7332f85afa28d69f5aac659e70 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
|
import System.Environment
import System.IO
import System.IO.Error
import Control.Exception
import Data.List
import Parser
import Equation
main :: IO ()
main = catchIOError tryMain handler
where handler e
| isUserError e = putStrLn $ trimUserError (show e)
| otherwise = putStrLn "Error"
where trimUserError s = init $ tail $ dropWhile (/='(') s
tryMain :: IO ()
tryMain = do
args <- getArgs
checkArgs args
equ <- checkParsing (head args)
let reduced = reduce equ
l = filterNull $ left reduced
putStrLn $ "Reduced From: " ++ show reduced
putStrLn $ "Polynomial degree: " ++ (show $ degree l)
case l of [] -> putStrLn "Infinite solutions"
[_] -> putStrLn "No solution"
_ -> putSolutions l
checkArgs :: [String] -> IO ()
checkArgs args
| length args == 0 = fail "Usage ./computor equation"
| length args > 1 = fail "Too many arguments"
| otherwise = return ()
checkParsing :: String -> IO Equation
checkParsing input = case parse Parser.equationP input
of Nothing -> fail "Couldnt parse equation"
Just (equ, "") -> return equ
Just (_, s) -> fail "Couldnt parse equation yo"
putSolutions :: Polynomial -> IO ()
putSolutions p
| degree p > 2 = fail "The polynomial degree is strictly greater then 2, can't solve."
| otherwise = putStr $ intercalate "\n" (map show (solve p))
|
