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
+[<System.Runtime.InteropServices.DllImport "kernel32.dll">]
+extern bool AttachConsole (int dwProcessId)
[<EntryPoint>]
-[<STAThread()>]
+[<STAThread ()>]
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<Bgr, byte>(file) ]
-
- Log.LogWithTime("Whole analyze", Severity.USER, (fun () ->
- match ParasitemiaCore.Analysis.doMultipleAnalysis images None with
- | Some results ->
- for id, cells 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 cells
- fprintf resultFile "File: %s %d %d %.2f (diameter: %A)\n" id total infected (100. * (float infected) / (float total)) config.RBCRadius
- | None ->
- fprintf resultFile "Analysis aborted"
- Some ())) |> ignore
-
- Log.RmListener(listener)
- 0
-
- | Window fileToOpen ->
- if debug then config.Debug <- DebugOn "."
- GUI.run config fileToOpen
-
- Log.User("Parasitemia UI closed")
- result
-
- with
- | 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<Bgr, byte> (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
projects / master-thesis.git / blobdiff