import qualified Data.List as L
import System.Environment
import Data.Array.IArray
import Data.Function

main = do
    (sourcefile:_) <- getArgs
    sourcedata <- readFile sourcefile
    let tships = lines sourcedata
        header = take 3 tships
        rest = drop 3 $ tships
        parsedlist = map split rest
        arrayedlist = map mkarray parsedlist
        processedlist = map preprocess arrayedlist
        finallist = map process processedlist
        listedfinal = map mklist finallist
        stuff = unlines . map unsplit $ listedfinal
    writeFile "done.txt" stuff

mkarray list = listArray (0, (length list - 1)) list
mklist myarray = elems myarray

split :: String -> [String]
split = filter (not . L.any (== ';')) . L.groupBy (on (==) (== ';')) -- like words, but for ;. Also like Data.ByteString split, but partially applied with ';'


unsplit :: [String] -> String
unsplit = L.intercalate ";"

process :: Array -> Array
process myarray = myarray // [(2,show fixp),(3,show fixy),(4,show fixroll),(7,show fixspeed),(8,show fixaccel)]
    where   pitch = read $ myarray ! 2
        yaw = read $ myarray ! 3
        roll = read $ myarray ! 4
        speed = read $ myarray ! 7
        accel = read $ myarray ! 8
        normalize num mid factor = num + (mid - num) / factor
        fixp = normalize pitch 1 5
        fixy = normalize yaw 1 5
        fixroll = normalize roll 1 5
        fixspeed = normalize speed 500000 15
        fixaccel = normalize (2 * accel) 200000 15

preprocess :: Array -> Array
preprocess myarray
    | tunings > 15 = fixarray myarray
    | otherwise = myarray
    where   tunings = read $ myarray ! 26
        fixarray passedarray = passedarray // [(7, show fixedspeed),(8, show fixedaccel),(26, "15")]
        fixedspeed = originalspeed * ( 1 + (tunings - 15) * 0.1)
        fixedaccel =  originalaccel * ( 1 + originalspeed / (fixedspeed - originalspeed))
        originalspeed = read $ myarray ! 7
        originalaccel = read $ myarray ! 8

[1 of 1] Compiling Main             ( new  2.hs, new  2.o )

new  2.hs:30:11:
    `Array' is not applied to enough type arguments
    Expected kind `??', but `Array' has kind `* -> * -> *'
    In the type signature for `process': process :: Array -> Array

new  2.hs:44:14:
    `Array' is not applied to enough type arguments
    Expected kind `??', but `Array' has kind `* -> * -> *'
    In the type signature for `preprocess':
      preprocess :: Array -> Array