X-Git-Url: http://git.euphorik.ch/?p=master-thesis.git;a=blobdiff_plain;f=Parasitemia%2FParasitemiaUI%2FProgram.fs;h=0f6df128050d7b31be0ea5d03e0e77a7e5de2936;hp=24e33cdeca4fa7d0667d953fa413f747f23dc7ef;hb=e8d14548fc5b127970fccb19bad146512f7ac6d9;hpb=6c70577f0bcc17881c753f7a527d2ff4672bb6aa

diff --git a/Parasitemia/ParasitemiaUI/Program.fs b/Parasitemia/ParasitemiaUI/Program.fs
index 24e33cd..0f6df12 100644
--- a/Parasitemia/ParasitemiaUI/Program.fs
+++ b/Parasitemia/ParasitemiaUI/Program.fs
@@ -22,7 +22,7 @@ type RunningMode =
 
 type Arguments = RunningMode * bool
 
-let parseArgs (args: string[]) : Arguments =
+let parseArgs (args : string[]) : Arguments =
 
     let output = Array.tryFindIndex ((=) "--output") args
 
@@ -39,62 +39,85 @@ let parseArgs (args: string[]) : Arguments =
 
     runningMode, Array.exists ((=) "--debug") args
 
-[<EntryPoint>]
-[<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
+let showArgsHelp () =
+    printfn "Usage of Parasitemia :"
+    printfn "Non-interactive mode:"
+    printfn "  %s (--folder <folder>|--file <file>) --output <folder> [--debug]" System.AppDomain.CurrentDomain.FriendlyName
+    printfn "    --folder <folder> : an input folder containing images to analyze"
+    printfn "    --file <file> : an image file to be analyzed"
+    printfn "    --output <folder> : a folder to put the results"
+    printfn "    --debug : output more information like intermediate images if set"
 
-                    use resultFile = new StreamWriter(new FileStream(Path.Combine(output, "results.txt"), FileMode.Append, FileAccess.Write))
+    printfn "Interactive mode:"
+    printfn "  %s [<document-file>] [--debug]" System.AppDomain.CurrentDomain.FriendlyName
+    printfn "    <document-file> : a PIAZ file to automatically open at startup"
+    printfn "    --debug : output information like intermediate images if set in the current directory"
 
+[<System.Runtime.InteropServices.DllImport("kernel32.dll")>]
+extern bool AttachConsole(int dwProcessId)
 
-                    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 = 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
-                        | 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
-    | _ as ex ->
-        Log.Fatal("Error: {0}", ex)
-        1
\ No newline at end of file
+[<EntryPoint>]
+[<STAThread()>]
+let main args =
+    // To redirect stdout to the attached console.
+    AttachConsole(-1) |> ignore // -1 to attach to the parent process.
+
+    if Array.exists (fun e -> e = "--help" || e = "-h") args then
+        showArgsHelp ()
+        0
+    else
+        try
+            Log.Info "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 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
+                                    | 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 Severity.INFO (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: %O)\n" id total infected (100. * (float infected) / (float total)) config.RBCRadius
+                            | None ->
+                                fprintf resultFile "Analysis aborted"
+                            Some ()) "Whole analyze" |> ignore
+
+                        Log.RmListener(listener)
+                        0
+
+                    | Window fileToOpen ->
+                        if debug then config.Debug <- DebugOn "."
+                        GUI.run config fileToOpen
+
+            Log.Info "Parasitemia UI closed"
+            result
+
+        with
+        | ex ->
+            Log.Fatal "Error: %A" ex
+            1
\ No newline at end of file