X-Git-Url: http://git.euphorik.ch/?a=blobdiff_plain;f=Parasitemia%2FParasitemiaUI%2FProgram.fs;h=5b5edd09c28236eb1bc1ff0f4e6d463f812b5885;hb=HEAD;hp=74f0d359930dac0227d8d6a8a5203db544254b09;hpb=97c24aa168f06f507fdff79429038d78a2c33326;p=master-thesis.git diff --git a/Parasitemia/ParasitemiaUI/Program.fs b/Parasitemia/ParasitemiaUI/Program.fs index 74f0d35..5b5edd0 100644 --- a/Parasitemia/ParasitemiaUI/Program.fs +++ b/Parasitemia/ParasitemiaUI/Program.fs @@ -12,86 +12,95 @@ open Logger open ParasitemiaCore.Utils open ParasitemiaCore.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 +[] +extern bool AttachConsole (int dwProcessId) [] -[] +[] let main args = - try - Log.User("Starting of Parasitemia UI ...") - - let result = - 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)) - let listener = { new IListener with member this.NewEntry severity mess = logFile.WriteLine(mess) } - Log.AddListener(listener) - - Log.User (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)) - - - let images = [ for file in files -> Path.GetFileNameWithoutExtension(FileInfo(file).Name), config.Copy(), new Image(file) ] - - Log.LogWithTime("Whole analyze", Severity.USER, (fun () -> - let results = ParasitemiaCore.Analysis.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 = countCells cells - fprintf resultFile "File: %s %d %d %.2f (diameter: %A)\n" id total infected (100. * (float infected) / (float total)) config.RBCRadius)) - - Log.RmListener(listener) - 0 - - | Window fileToOpen -> - if debug then config.Debug <- DebugOn "." - GUI.run config fileToOpen - - Log.User("Parasitemia UI closed") - result - - with - | _ as ex -> - Log.Fatal("Error: {0}", ex) - 1 \ No newline at end of file + // To redirect stdout to the attached console. + AttachConsole -1 |> ignore // -1 to attach to the parent process. + + Log.LogDirectory <- Constants.USER_DIRECTORY_LOG + +#if DEBUG + Log.ClearLogFiles () +#endif + + Log.AvoidRepeatingIdenticalMessages <- true + + if Array.exists (fun e -> e = "--help" || e = "-h") args then + Args.showArgsHelp () + 0 + else + try + Log.Info ">>>>> Starting of Parasitemia UI ..." + // Log.Info "UI Version: %s" Utils.UIVersion TODO + + let result = + match Args.parse args with + | mode, debug -> + let config = Config defaultParameters + +#if DEBUG + Log.DebugLoggingEnabled <- true +#else + Log.DebugLoggingEnabled <- debug +#endif + + Log.Info "Command line arguments: %s" (args |> String.concat " ") + Log.Info "Configuration: %O" config + // Log.Info "Settings from %s:\n%s" Settings.Instance.Path Settings.Instance.AsJson TODO + + match mode with + | Args.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)) + let listener = { new IListener with member this.NewEntry severity header mess = logFile.WriteLine (header + " : " + mess) } + Log.AddListener listener + + Log.Info "=== New run : %O %s ===" DateTime.Now (if debug then "[DEBUG]" else "[RELEASE]") + + let files = + match input with + | Args.File file -> [ file ] + | Args.Dir dir -> Directory.EnumerateFiles dir |> List.ofSeq + + use resultFile = new StreamWriter (new FileStream (Path.Combine (output, "results.txt"), FileMode.Append, FileAccess.Write)) + + let images = [ for file in files -> Path.GetFileNameWithoutExtension (FileInfo(file).Name), config.Copy (), new Image (file) ] + + Log.LogWithTime Types.Severity.INFO ( + fun () -> + match ParasitemiaCore.Analysis.doMultipleAnalysis images None with + | Some results -> + for id, result in results do + let config, img = images |> List.pick (fun (id', config', img') -> if id' = id then Some (config', img') else None) + img.Dispose () + let total, infected = countCells result.Cells + fprintf resultFile "File: %s %d %d %.2f (diameter: %O)\n" id total infected (100. * (float infected) / (float total)) config.RBCRadius + | None -> + fprintf resultFile "Analysis aborted" + Some () + ) "Whole analyze" |> ignore + + Log.RemoveListener listener + 0 + + | Args.Window fileToOpen -> + if debug then config.Debug <- DebugOn "." + GUI.run config fileToOpen + + Log.Info "<<<<< Parasitemia UI closed" + Log.Close () + result + + with + | ex -> + Log.Fatal "Error: %A" ex + Log.Close () + 1 \ No newline at end of file