module Parasitemia.Main

open System
open System.IO
open System.Threading

open Emgu.CV
open Emgu.CV.Structure

open Config

type Input =
    | File of string
    | Dir of string

type RunningMode =
    | CmdLine of Input * string // A file or a directory to process and the output directory.
    | Window of string option // An optional path to a file to open can be given in window mode.

type Arguments = RunningMode * bool

let parseArgs (args: string[]) : Arguments =

    let output = Array.tryFindIndex ((=) "--output") args

    let runningMode =
        match Array.tryFindIndex ((=) "--folder") args, output with
        | Some i, Some i_output when i < args.Length - 2 && i_output < args.Length - 2 ->
            CmdLine ((Dir args.[i+1]), args.[i_output + 1])
        | _ ->
            match Array.tryFindIndex ((=) "--file") args, output with
            | Some i, Some i_output when i < args.Length - 2 && i_output < args.Length - 2 ->
                CmdLine ((File args.[i+1]), args.[i_output + 1])
            |_ ->
                Window (if args.Length > 0 && not (args.[0].StartsWith("--")) then Some args.[0] else None)

    runningMode, Array.exists ((=) "--debug") args

[<EntryPoint>]
[<STAThread()>]
let main args =

    let e = Ellipse.ellipse2 -11.4 -7.8 -0.169811 -23.75 0.8 -3.885714 -19. 1.5

    match parseArgs args with
    | mode, debug ->
        let config = Config(defaultParameters)

        match mode with
        | CmdLine (input, output) ->
            if debug
            then
                config.Debug <- DebugOn output

            Directory.CreateDirectory output |> ignore

            use logFile = new StreamWriter(new FileStream(Path.Combine(output, "log.txt"), FileMode.Append, FileAccess.Write))
            Utils.log <- (fun m -> logFile.WriteLine(m))
            Utils.log (sprintf "=== New run : %A %A ===" DateTime.Now  (if debug then "[DEBUG]" else "[RELEASE]"))

            let files = match input with
                        | File file -> [ file ]
                        | Dir dir -> Directory.EnumerateFiles dir |> List.ofSeq

            use resultFile = new StreamWriter(new FileStream(Path.Combine(output, "results.txt"), FileMode.Append, FileAccess.Write))

            //try
            let images = [ for file in files -> Path.GetFileNameWithoutExtension(FileInfo(file).Name), config.Copy(), new Image<Bgr, byte>(file) ]


            Utils.logTime "Whole analyze" (fun () ->
                let results = ImageAnalysis.doMultipleAnalysis images None

                for id, cells in results do
                    let config = images |> List.pick (fun (id', config', _) -> if id' = id then Some config' else None)
                    let total, infected = Utils.countCells cells
                    fprintf resultFile "File: %s %d %d %.2f (diameter: %A)\n" id total infected (100. * (float infected) / (float total)) config.RBCRadius)

            //Utils.log (sprintf "== File: %A" file)
            //with
            //| :? IOException as ex -> Utils.log (sprintf "Unable to open the image '%A': %A" file ex)
            0

        | Window fileToOpen ->
            (*let display (window : Views.MainWindow) (img : IImage) =
               let imgControl = window.Root.FindName("img") :?> Controls.Image
               imgControl.Source <- BitmapSourceConvert.ToBitmapSource(img)

            let log (window : Views.MainWindow) (mess : string) =
               let txtLog = window.Root.FindName("txtLog") :?> Controls.TextBlock
               txtLog.Text <- txtLog.Text + mess + "\n"*)

            if debug then config.Debug <- DebugOn "."
            GUI.Main.run config fileToOpen