GUI (work in progress..)
[master-thesis.git] / Parasitemia / Parasitemia / GUI / GUI.fs
index 6f11379..2aadefe 100644 (file)
@@ -10,8 +10,6 @@ open System.Windows.Controls
 open System.Diagnostics
 open Microsoft.Win32 // For the common dialogs.
 
-//open Emgu.CV
-//open Emgu.CV.Structure
 open Emgu.CV.WPF
 
 open Config
@@ -25,7 +23,8 @@ let run (defaultConfig: Config) =
 
     // Utils.log <- (fun m -> log mainWindow m)
 
-    let currentImageMargin = 0.1 // 10 %.
+    let colorRBCHealthy = Brushes.YellowGreen
+    let colorRBCInfected = Brushes.Red
 
     let state = State.State()
 
@@ -50,11 +49,44 @@ let run (defaultConfig: Config) =
         state.PatientID <- txtPatient.Text
 
     let setCurrentImage (srcImg: SourceImage) =
+        state.CurrentImage <- Some srcImg
+
+        // Highlight the preview.
+        stackPreviews.Children
+        |> Seq.cast<Views.ImageSourcePreview>
+        |> Seq.iter (fun preview -> preview.border.BorderThickness <- Thickness(if preview.lblImageNumber.Content = box srcImg.num then 3. else 0.))
+
         canvasCurrentImage.Height <- float srcImg.img.Height
         canvasCurrentImage.Width <- float srcImg.img.Width
         canvasCurrentImage.Background <- ImageBrush(BitmapSourceConvert.ToBitmapSource(srcImg.img))
-        (*for rbc in srcImg.rbcs do
-            let rectangle = Rectangle(Height = float rbc.size.Height, Width = float rbc.size.Width, )*)
+
+        // Remove all image canvas children and add the RBC.
+        canvasCurrentImage.Children.Clear()
+        for rbc in srcImg.rbcs do
+            let rectangle =
+                Rectangle(
+                    Height = rbc.size.Height,
+                    Width = rbc.size.Width,
+                    Stroke = (if rbc.infected then colorRBCInfected else colorRBCHealthy),
+                    StrokeThickness = 1.,
+                    Fill = SolidColorBrush(Color.FromArgb(0uy, 0uy, 0uy, 0uy)),
+                    Tag = rbc,
+                    Opacity = if rbc.infected then 1. else 0.)
+            Canvas.SetLeft(rectangle, rbc.center.X - rbc.size.Width / 2.)
+            Canvas.SetTop(rectangle, rbc.center.Y - rbc.size.Height / 2.)
+            canvasCurrentImage.Children.Add(rectangle) |> ignore
+            rectangle.MouseEnter.AddHandler(
+                fun obj args -> match obj with
+                                | :? Rectangle as r ->
+                                    r.StrokeThickness <- 3.
+                                    if not (r.Tag :?> RBC).infected then r.Opacity <- 1.
+                                | _ -> ())
+            rectangle.MouseLeave.AddHandler(
+                fun obj args -> match obj with
+                                | :? Rectangle as r ->
+                                    r.StrokeThickness <- 1.
+                                    if not (r.Tag :?> RBC).infected then r.Opacity <- 0.
+                                | _ -> ())
 
     let addPreview (srcImg: SourceImage) =
         let imgCtrl = Views.ImageSourcePreview(Margin = Thickness(3.))
@@ -63,11 +95,15 @@ let run (defaultConfig: Config) =
         let height = srcImg.img.Height * width / srcImg.img.Width
         imgCtrl.imagePreview.Source <- BitmapSourceConvert.ToBitmapSource(srcImg.img.Resize(width, height, Emgu.CV.CvEnum.Inter.Cubic))
         stackPreviews.Children.Add(imgCtrl) |> ignore
+        imgCtrl.MouseLeftButtonUp.AddHandler(fun obj args -> setCurrentImage (state.SourceImages |> Seq.find (fun i -> box i.num = (obj :?> Views.ImageSourcePreview).lblImageNumber.Content)))
 
     let updatePreviews () =
         stackPreviews.Children.Clear ()
-        for srcImg in state.SourceImages do
-            addPreview srcImg
+        if state.SourceImages.Count() > 0
+        then
+            for srcImg in state.SourceImages do
+                addPreview srcImg
+            setCurrentImage (state.SourceImages.First())
 
     let updateGUI () =
         txtPatient.Text <- state.PatientID
@@ -78,7 +114,7 @@ let run (defaultConfig: Config) =
         synchronizeState ()
         if state.FilePath = ""
         then
-            let dialog = SaveFileDialog(AddExtension = true, DefaultExt = Pia.extension, Filter = Pia.filter);
+            let dialog = SaveFileDialog(AddExtension = true, DefaultExt = PiaZ.extension, Filter = PiaZ.filter);
             let res = dialog.ShowDialog()
             if res.HasValue && res.Value
             then
@@ -89,7 +125,7 @@ let run (defaultConfig: Config) =
 
     loadFile.Click.AddHandler(fun obj args ->
         // TODO: if current state not saved and not empty, ask to save it.
-        let dialog = OpenFileDialog(Filter = Pia.filter)
+        let dialog = OpenFileDialog(Filter = PiaZ.filter)
         let res = dialog.ShowDialog()
         if res.HasValue && res.Value
         then
@@ -113,21 +149,44 @@ let run (defaultConfig: Config) =
             then
                 setCurrentImage srcImg)
 
-    butStartAnalysis.Click.AddHandler(fun obj args -> ())
+    butStartAnalysis.Click.AddHandler(fun obj args ->
+        let results = ImageAnalysis.doMultipleAnalysis (state.SourceImages |> Seq.map (fun srcImg -> string srcImg.num, srcImg.img) |> Seq.toList) defaultConfig
+        for id, cells in results do
+            state.SetResult (int id) cells
+        )
 
     // Zoom on the current image.
-    let adjustCurrentImageMargins () =
+    let adjustCurrentImageBorders (deltaX: float) (deltaY: float) =
         borderCurrentImage.BorderThickness <-
-            Thickness(scrollViewCurrentImage.ViewportWidth / 2., scrollViewCurrentImage.ViewportHeight / 2., scrollViewCurrentImage.ViewportWidth / 2., scrollViewCurrentImage.ViewportHeight / 2.)
-
-    scrollViewCurrentImage.Loaded.AddHandler(fun obj args -> adjustCurrentImageMargins ())
-    scrollViewCurrentImage.SizeChanged.AddHandler(fun obj args -> adjustCurrentImageMargins ())
+            Thickness(
+                (scrollViewCurrentImage.ViewportWidth + deltaX) / 2.,
+                (scrollViewCurrentImage.ViewportHeight + deltaY) / 2.,
+                (scrollViewCurrentImage.ViewportWidth + deltaX) / 2.,
+                (scrollViewCurrentImage.ViewportHeight + deltaY) / 2.)
+
+    canvasCurrentImage.SizeChanged.AddHandler(fun obj args ->
+        let deltaX = args.NewSize.Width - args.PreviousSize.Width
+        let deltaY = args.NewSize.Height - args.PreviousSize.Height
+        if deltaX > 0.5 || deltaY > 0.5
+        then
+            adjustCurrentImageBorders 0.0 0.0
+            // Center the view at the center of the image initialy.
+            scrollViewCurrentImage.UpdateLayout()
+            scrollViewCurrentImage.ScrollToHorizontalOffset(borderCurrentImage.ActualWidth / 2. - scrollViewCurrentImage.ViewportWidth / 2.)
+            scrollViewCurrentImage.ScrollToVerticalOffset(borderCurrentImage.ActualHeight / 2. - scrollViewCurrentImage.ViewportHeight / 2.))
+
+    scrollViewCurrentImage.SizeChanged.AddHandler(fun obj args ->
+        let deltaX = args.NewSize.Width - args.PreviousSize.Width
+        let deltaY = args.NewSize.Height - args.PreviousSize.Height
+        adjustCurrentImageBorders deltaX deltaY
+        scrollViewCurrentImage.ScrollToHorizontalOffset(scrollViewCurrentImage.HorizontalOffset + deltaX / 8.)
+        scrollViewCurrentImage.ScrollToVerticalOffset(scrollViewCurrentImage.VerticalOffset + deltaY / 8.))
 
     let mutable currentScale = 1.
-    let mutable maxScale = 5.
-    let mutable minScale = 0.1
+    let mutable maxScale = 4.
+    let mutable minScale = 0.25
     let currentImageScaleTransform = ScaleTransform()
-    borderCurrentImage.LayoutTransform <- currentImageScaleTransform
+    canvasCurrentImage.LayoutTransform <- currentImageScaleTransform
     borderCurrentImage.PreviewMouseWheel.AddHandler(fun obj args ->
         let scaleFactor = if args.Delta > 0 then 2.0 else 0.5
         if scaleFactor > 1. && currentScale < maxScale || scaleFactor < 1. && currentScale > minScale
@@ -138,15 +197,15 @@ let run (defaultConfig: Config) =
                 if newScale > maxScale then maxScale elif newScale < minScale then minScale else newScale
             let realScaleFactor = currentScale / previousScale
 
-            let centerX = scrollViewCurrentImage.HorizontalOffset + scrollViewCurrentImage.ViewportWidth / 2. - canvasCurrentImage.Margin.Left
-            let centerY = scrollViewCurrentImage.VerticalOffset + scrollViewCurrentImage.ViewportHeight / 2. - canvasCurrentImage.Margin.Top
+            let centerX = scrollViewCurrentImage.HorizontalOffset + scrollViewCurrentImage.ViewportWidth / 2. - borderCurrentImage.BorderThickness.Left
+            let centerY = scrollViewCurrentImage.VerticalOffset + scrollViewCurrentImage.ViewportHeight / 2. - borderCurrentImage.BorderThickness.Top
 
-            canvasCurrentImage.Margin <- Thickness(canvasCurrentImage.Margin.Top * realScaleFactor)
+            //canvasCurrentImage.Margin <- Thickness(canvasCurrentImage.Margin.Top * realScaleFactor)
             currentImageScaleTransform.ScaleX <- currentScale
             currentImageScaleTransform.ScaleY <- currentScale
 
-            scrollViewCurrentImage.ScrollToHorizontalOffset(centerX * realScaleFactor - scrollViewCurrentImage.ViewportWidth / 2. + canvasCurrentImage.Margin.Left)
-            scrollViewCurrentImage.ScrollToVerticalOffset(centerY * realScaleFactor - scrollViewCurrentImage.ViewportHeight / 2. + canvasCurrentImage.Margin.Top)
+            scrollViewCurrentImage.ScrollToHorizontalOffset(centerX * realScaleFactor - scrollViewCurrentImage.ViewportWidth / 2. + borderCurrentImage.BorderThickness.Left)
+            scrollViewCurrentImage.ScrollToVerticalOffset(centerY * realScaleFactor - scrollViewCurrentImage.ViewportHeight / 2. + borderCurrentImage.BorderThickness.Top)
         args.Handled <- true)
 
     // Pan on the current image.
@@ -178,15 +237,6 @@ let run (defaultConfig: Config) =
             borderCurrentImage.ReleaseMouseCapture()
             args.Handled <- true)
 
-
-    (*let txtPatient: Controls.TextBox = ctrl "txtPatient"
-    txtPatient.TextChanged.AddHandler(fun obj args ->
-        state.PatientID <- txtPatient.Text)*)
-
-     (*saveFileDialog1.Filter = "txt files (*.txt)|*.txt|All files (*.*)|*.*"  ;
-     saveFileDialog1.FilterIndex = 2 ;
-     saveFileDialog1.RestoreDirectory = true ;*)
-
     // display mainWindow img
     mainWindow.Root.Show()
     app.Run()
\ No newline at end of file