X-Git-Url: http://git.euphorik.ch/?a=blobdiff_plain;f=Parasitemia%2FParasitemia%2FProgram.fs;h=2aa3ee4f77c929454e2ef67ba1f0655d78d8545f;hb=4147bfad3c85dcf4fe7d291abf7d356379d88b4f;hp=fd9a2f94236b663d317cafa6a234cc9e0a3d1c01;hpb=dec96d50e56e1932bbfa09e6bedf90d6b707ccbd;p=master-thesis.git

diff --git a/Parasitemia/Parasitemia/Program.fs b/Parasitemia/Parasitemia/Program.fs
index fd9a2f9..2aa3ee4 100644
--- a/Parasitemia/Parasitemia/Program.fs
+++ b/Parasitemia/Parasitemia/Program.fs
@@ -1,71 +1,91 @@
 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 = {
-        scale = 1.0
-
-        doGSigma1 = 1.5
-        doGSigma2 = 20.0
-        doGLowFreqPercentageReduction = 0.9
-
-        darkStainLevel = 0.839     
-        
-        stainSigma = 10.0
-        stainLevel = 0.9
-        stainSpreadRequired = 3.0
-    
-        infectionSigma = 2.0
-        infectionLevel = 0.762
-        infectionPixelsRequired = 1 }
-   
-    ///// ELLIPSES /////
-    //use img = new Image<Bgr, 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_4.png")
-    //use img = new Image<Bgr, byte>("../../../../src/Tests_hough/images/strange_rbc_1.png")
-    //use img = new Image<Bgr, byte>("../../../../src/Tests_hough/images/rbc_single_blurred.png")   
-    //use img = new Image<Bgr, byte>("../../../../src/Tests_hough/images/lot.png")
-
-
-    ///// PARASITES /////
-    use img = new Image<Bgr, byte>("../../../../src/Parasites/images/1.png")
-    
-//    KdTree.test3 ()
-//    Utils.dprintfn "area: %A" area
-
-    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(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), 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