X-Git-Url: http://git.euphorik.ch/?p=master-thesis.git;a=blobdiff_plain;f=Parasitemia%2FParasitemia%2FProgram.fs;h=038f827c754aa3deb22f0c08dff0b43b69f17bb3;hp=56672bd0646bda268189cde7006f6c29e9b02e92;hb=807437584bdd3d7b7b33be8282472f81f8cce606;hpb=ba64921fb9a0c36cd8cf802cbf1b2c0f79bc25f6

diff --git a/Parasitemia/Parasitemia/Program.fs b/Parasitemia/Parasitemia/Program.fs
index 56672bd..038f827 100644
--- a/Parasitemia/Parasitemia/Program.fs
+++ b/Parasitemia/Parasitemia/Program.fs
@@ -1,52 +1,115 @@
 module Parasitemia.Main
 
+open System
 open System.IO
-open System.Windows
-open System.Windows.Media
-open System.Windows.Markup
-open System.Windows.Shapes
-open System.Windows.Controls
-open System.Drawing
-open System.Diagnostics
+open System.Threading
 
 open Emgu.CV
 open Emgu.CV.Structure
-open Emgu.CV.WPF
 
 open Config
 
-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"
-   
-[<System.STAThreadAttribute>]
-do
-    Utils.dprintfn "OpenCV test"
-
-    let app = new Application()
-    let mainWindow = Views.MainWindow()
-
-    Utils.log <- (fun m -> log mainWindow m)
-
-    let config = {
-        doGSigma1 = 1.5
-        doGSigma2 = 20.0
-        doGLowFreqPercentageReduction = 0.9
-        scale = 1.0
-    }
-   
-    //use img = new Image<Rgb, byte>("../../../../src/Tests_hough/images/1508133543-7-4-120-0001.png")
-    //use img = new Image<Bgr, byte>("../../../../src/Tests_hough/images/rbc_single.png")
-    //use img = new Image<Bgr, byte>("../../../../src/Tests_hough/images/rbc_single_oblong_2.png")
-    use img = new Image<Bgr, byte>("../../../../src/Tests_hough/images/two_rbc_1.png")
-   
-    let result = ImageAnalysis.doAnalysis img config
-
-    display mainWindow result.img
-    mainWindow.Root.Show()
-
-    app.Run() |> ignore
+type Input =
+    | File of string
+    | Dir of string
+
+type RunningMode =
+    | CmdLine of Input * string
+    | Window
+
+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
+
+    runningMode, Array.exists ((=) "--debug") args
+
+
+[<EntryPoint>]
+[<STAThread()>]
+let main args =
+    match parseArgs args with
+    | mode, debug ->
+        let config =
+            Config(
+              {
+                initialAreaOpen = 2000
+                ratioSecondAreaOpen = 1.f / 3.f
+
+                minRbcRadius = -0.3f
+                maxRbcRadius = 0.3f
+
+                preFilterSigma = 1.7 // 1.5
+
+                factorNbPick = 1.0
+
+                darkStainLevel = 0.25 // 0.3
+                maxDarkStainRatio = 0.1 // 10 %
+
+                infectionArea = 0.012f // 1.2 %
+                infectionLevel = 1.12 // Lower -> more sensitive.
+
+                stainArea = 0.08f // 8 %
+                stainLevel = 1.1 // Lower -> more sensitive.
+                maxStainRatio = 0.12 // 12 %
+
+                standardDeviationMaxRatio = 0.5 // 0.55
+                minimumCellAreaFactor = 0.5f })
+
+        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), new Image<Bgr, byte>(file) ]
+
+
+            Utils.logTime "Whole analyze" (fun () ->
+                let results = ImageAnalysis.doMultipleAnalysis images config None
+
+                for id, _, cells in results do
+                    let total, infected = Utils.countCells cells
+                    fprintf resultFile "File: %s %d %d %.2f\n" id total infected (100. * (float infected) / (float total)))
+
+            //Utils.log (sprintf "== File: %A" file)
+            //with
+            //| :? IOException as ex -> Utils.log (sprintf "Unable to open the image '%A': %A" file ex)
+            0
+
+        | Window ->
+            (*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