6531da82fce59908bda470c5bd61489d50cc8dd1
[master-thesis.git] / Parasitemia / Parasitemia / Program.fs
1 module Parasitemia.Main
2
3 open System
4 open System.IO
5 open System.Windows
6 open System.Windows.Media
7 open System.Windows.Markup
8 open System.Windows.Shapes
9 open System.Windows.Controls
10 open System.Drawing
11 open System.Diagnostics
12
13 open Emgu.CV
14 open Emgu.CV.Structure
15 open Emgu.CV.WPF
16
17 open Config
18
19 let display (window : Views.MainWindow) (img : IImage) =
20 let imgControl = window.Root.FindName("img") :?> Controls.Image
21 imgControl.Source <- BitmapSourceConvert.ToBitmapSource(img)
22
23 let log (window : Views.MainWindow) (mess : string) =
24 let txtLog = window.Root.FindName("txtLog") :?> Controls.TextBlock
25 txtLog.Text <- txtLog.Text + mess + "\n"
26
27
28 type Input =
29 | File of string
30 | Dir of string
31
32 type RunningMode =
33 | CmdLine of Input * string
34 | Window
35
36 type Arguments = RunningMode * bool
37
38 let parseArgs (args: string[]) : Arguments =
39
40 let output = Array.tryFindIndex ((=) "--output") args
41
42 let runningMode =
43 match Array.tryFindIndex ((=) "--folder") args, output with
44 | Some i, Some i_output when i < args.Length - 2 && i_output < args.Length - 2 ->
45 CmdLine ((Dir args.[i+1]), args.[i_output + 1])
46 | _ ->
47 match Array.tryFindIndex ((=) "--file") args, output with
48 | Some i, Some i_output when i < args.Length - 2 && i_output < args.Length - 2 ->
49 CmdLine ((File args.[i+1]), args.[i_output + 1])
50 |_ ->
51 Window
52
53 runningMode, Array.exists ((=) "--debug") args
54
55
56 [<EntryPoint>]
57 let main args =
58 match parseArgs args with
59 | mode, debug ->
60 let config =
61 Config(
62 { scale = 1.
63
64 initialAreaOpen = 2000
65
66 minRbcRadius = -0.32
67 maxRbcRadius = 0.32
68
69 preFilterSigma = 1.7 // 1.5
70
71 factorNbPick = 1.0
72 factorWindowSize = 2.0
73
74 darkStainLevel = 0.25 // Lower -> more sensitive. 0.3
75 maxDarkStainRatio = 0.1 // 10 %
76
77 infectionArea = 0.012 // 1.2 %
78 infectionLevel = 1.12 // Lower -> more sensitive.
79
80 stainArea = 0.08 // 8 %
81 stainLevel = 1.1 // Lower -> more sensitive.
82 maxStainRatio = 0.12 // 12 %
83
84 standardDeviationMaxRatio = 0.5 // 0.55
85 minimumCellArea = 0.5 })
86
87 match mode with
88 | CmdLine (input, output) ->
89 if debug
90 then
91 config.Debug <- DebugOn output
92
93 Directory.CreateDirectory output |> ignore
94
95 use logFile = new StreamWriter(new FileStream(Path.Combine(output, "log.txt"), FileMode.Append, FileAccess.Write))
96 Utils.log <- (fun m -> logFile.WriteLine(m))
97 Utils.log (sprintf "=== New run : %A %A ===" DateTime.Now (if debug then "[DEBUG]" else "[RELEASE]"))
98
99 let files = match input with
100 | File file -> [ file ]
101 | Dir dir -> Directory.EnumerateFiles dir |> List.ofSeq
102
103 use resultFile = new StreamWriter(new FileStream(Path.Combine(output, "results.txt"), FileMode.Append, FileAccess.Write))
104
105 for file in files do
106 try
107 use img = new Image<Bgr, byte>(file)
108 Utils.log (sprintf "== File: %A" file)
109
110 let cells = Utils.logTime "Whole analyze" (fun () -> ImageAnalysis.doAnalysis img (Path.GetFileNameWithoutExtension(FileInfo(file).Name)) config)
111 let total, infected = Utils.countCells cells
112 fprintf resultFile "File: %s %d %d %.2f\n" file total infected (100. * (float infected) / (float total))
113 with
114 | :? IOException as ex -> Utils.log (sprintf "Unable to open the image '%A': %A" file ex)
115 0
116
117 | Window ->
118 let app = new Application()
119 let mainWindow = Views.MainWindow()
120
121 if debug
122 then
123 config.Debug <- DebugOn "."
124
125 Utils.log <- (fun m -> log mainWindow m)
126
127 //display mainWindow img
128 mainWindow.Root.Show()
129 app.Run()