* Add the analysis window.
[master-thesis.git] / Parasitemia / Parasitemia / GUI / GUI.fs
index 8aef251..f88bc3c 100644 (file)
@@ -1,5 +1,6 @@
 module Parasitemia.GUI.Main
 
+open System
 open System.IO
 open System.Linq
 open System.Windows
@@ -21,25 +22,24 @@ let run (defaultConfig: Config) =
     let ctrl (name: string): 'a =
         mainWindow.Root.FindName(name) :?> 'a
 
-    // Utils.log <- (fun m -> log mainWindow m)
-
     let colorRBCHealthy = Brushes.YellowGreen
     let colorRBCInfected = Brushes.Red
 
     let state = State.State()
     let mutable currentScale = 1.
+    let mutable displayHealthy = false
 
-    let exit: MenuItem = ctrl "menuExit"
-    let saveFile: MenuItem = ctrl "menuSave"
-    let loadFile: MenuItem = ctrl "menuOpen"
-    let newFile: MenuItem = ctrl "menuNew"
+    let menuExit: MenuItem = ctrl "menuExit"
+    let menuSaveFile: MenuItem = ctrl "menuSave"
+    let menuLoadFile: MenuItem = ctrl "menuOpen"
+    let menuNewFile: MenuItem = ctrl "menuNew"
+    let menuAddSourceImage: MenuItem = ctrl "menuAddSourceImage"
+    let menuStartAnalysis: MenuItem = ctrl "menuStartAnalysis"
+    let menuHightlightRBC: MenuItem = ctrl "menuHightlightRBC"
 
-    let addSourceImage: MenuItem = ctrl "menuAddSourceImage"
     let txtPatient: TextBox = ctrl "txtPatient"
     let txtGlobalParasitemia: TextBox = ctrl "txtGlobalParasitemia"
 
-    let butStartAnalysis: Button = ctrl "butStartAnalysis"
-
     let stackPreviews: StackPanel = ctrl "stackPreviews"
 
     let scrollViewCurrentImage: ScrollViewer = ctrl "scrollViewCurrentImage"
@@ -51,16 +51,23 @@ let run (defaultConfig: Config) =
     let stackRBC: StackPanel = ctrl "stackRBC"
 
     // Initializations.
+    menuHightlightRBC.IsChecked <- displayHealthy
 
     // Utils.
     let extractRBCPreview (img: Emgu.CV.Image<Emgu.CV.Structure.Bgr, byte>) (rbc: RBC) : Emgu.CV.Image<Emgu.CV.Structure.Bgr, byte> =
         let rbcWidth = rbc.size.Width
         let rbcHeight = rbc.size.Height
-        img.GetSubRect(System.Drawing.Rectangle(System.Drawing.Point(rbc.center.X - rbcWidth / 2. |> Utils.roundInt, rbc.center.Y - rbcHeight / 2. |> Utils.roundInt),
-                                                System.Drawing.Size(Utils.roundInt rbcWidth, Utils.roundInt rbcHeight)))
+        let x = rbc.center.X - rbcWidth / 2. |> Utils.roundInt
+        let y = rbc.center.Y - rbcHeight / 2. |> Utils.roundInt
+        let w = Utils.roundInt rbcWidth
+        let h = Utils.roundInt rbcHeight
+        //Utils.dprintfn "w: %A, h: %A, cx: %A, cy: %A, img.w: %A, img.h: %A" w h x y img.Width img.Height
+        img.GetSubRect(System.Drawing.Rectangle(System.Drawing.Point((if x < 0 then 0 else x), (if y < 0 then 0 else y)),
+                                                System.Drawing.Size((if x + w >= img.Width then img.Width - x else w),
+                                                                    (if y + h >= img.Height then img.Height - y else h))))
 
     let setRBCFrameStyle (rbc: RBC) (frame: Views.RBCFrame) =
-        frame.Opacity <- if rbc.setManually || rbc.infected then 1. else 0.
+        frame.Opacity <- if displayHealthy || rbc.setManually || rbc.infected then 1. else 0.
         let color = if rbc.infected then colorRBCInfected else colorRBCHealthy
         frame.manuallyAdded.Visibility <- if rbc.setManually then Visibility.Visible else Visibility.Hidden
         frame.manuallyAdded.Fill <- color
@@ -73,7 +80,7 @@ let run (defaultConfig: Config) =
         frame.Tag <- rbc
         setRBCFrameStyle rbc frame
         frame.border.StrokeThickness <- 1.
-        frame.lblRBCNumber.Content <- rbc.num
+        frame.txtRBCNumber.Text <- rbc.num.ToString()
         frame
 
     let highlightRBCFrame (frame: Views.RBCFrame) (highlight: bool) =
@@ -81,10 +88,10 @@ let run (defaultConfig: Config) =
         if highlight
         then
             frame.border.StrokeThickness <- 3.
-            if not rbc.infected && not rbc.setManually then frame.Opacity <- 1.
+            if not rbc.infected && not rbc.setManually && not displayHealthy then frame.Opacity <- 1.
         else
             frame.border.StrokeThickness <- 1.
-            if not rbc.infected && not rbc.setManually then frame.Opacity <- 0.
+            if not rbc.infected && not rbc.setManually && not displayHealthy then frame.Opacity <- 0.
 
     let zoomToRBC (rbc: RBC) =
         scrollViewCurrentImage.ScrollToHorizontalOffset(rbc.center.X * currentScale - scrollViewCurrentImage.ViewportWidth / 2. + borderCurrentImage.BorderThickness.Left)
@@ -96,20 +103,31 @@ let run (defaultConfig: Config) =
             ""
         else
             let percent = 100. * (float nbInfected) / (float nbTotal)
-            sprintf "%.2f %% (%d / %d)" percent nbInfected nbTotal
+            sprintf "%.1f %% (%d / %d)" percent nbInfected nbTotal
 
     let updateCurrentImageInformation () =
         match state.CurrentImage with
         | Some srcImg ->
             let parasitemiaStr = parasitemiaText (state.ImageParasitemia srcImg)
-            txtImageInformation.Text <- sprintf "Parasitemia: %s" parasitemiaStr
+            txtImageInformation.Inlines.Clear()
+            txtImageInformation.Inlines.Add(Documents.Run("Parasitemia: ", FontWeight = FontWeights.Bold))
+            txtImageInformation.Inlines.Add(parasitemiaStr)
+            txtImageInformation.Inlines.Add(Documents.LineBreak())
+
+            txtImageInformation.Inlines.Add(Documents.Run("Average erytrocyte diameter: ", FontWeight = FontWeights.Bold))
+            txtImageInformation.Inlines.Add(Documents.Run(string (Utils.roundInt <| 2. * srcImg.rbcRadius)))
+            txtImageInformation.Inlines.Add(Documents.Run(" px"))
+            txtImageInformation.Inlines.Add(Documents.LineBreak())
+
+            txtImageInformation.Inlines.Add(Documents.Run("Last analysis: ", FontWeight = FontWeights.Bold))
+            txtImageInformation.Inlines.Add(Documents.Run(if srcImg.dateLastAnalysis.Ticks = 0L then "<Never>" else srcImg.dateLastAnalysis.ToLocalTime().ToString()))
         | _ -> ()
 
     let updateGlobalParasitemia () =
         txtGlobalParasitemia.Text <- parasitemiaText state.GlobalParasitemia
 
     let rec setAsInfected (rbc: RBC) (infected: bool) =
-        rbc.SetAsInfected infected
+        state.SetAsInfected rbc infected
         canvasCurrentImage.Children
         |> Seq.cast<Views.RBCFrame>
         |> Seq.iter
@@ -144,14 +162,14 @@ let run (defaultConfig: Config) =
         match state.CurrentImage with
         | Some srcImg ->
             let mutable currentPreview = 0
-            for rbc in srcImg.rbcs |> List.filter (fun rbc -> rbc.infected) do
+            for rbc in srcImg.rbcs |> List.filter (fun rbc -> displayHealthy || rbc.infected) do
                 let previewInfected =
                     if currentPreview < stackRBC.Children.Count
                     then
                         RBCFrameFromExisting rbc (stackRBC.Children.[currentPreview] :?> Views.RBCFrame)
                     else
                         let f = RBCFrame rbc
-                        f.MouseLeftButtonUp.AddHandler(fun obj args -> zoomToRBC ((obj :?> Views.RBCFrame).Tag :?> RBC))
+                        f.MouseLeftButtonUp.AddHandler(fun obj args -> zoomToRBC (f.Tag :?> RBC))
                         stackRBC.Children.Add(f) |> ignore
                         f
 
@@ -187,81 +205,134 @@ let run (defaultConfig: Config) =
                 canvasCurrentImage.Children.[i].Visibility <- Visibility.Hidden
         | _ -> ()
 
-    // Operations.
-    let synchronizeState () =
-        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.))
+    let saveCurrentDocument () =
+        if state.FilePath = ""
+        then
+            let dialog = SaveFileDialog(AddExtension = true, DefaultExt = PiaZ.extension, Filter = PiaZ.filter);
+            let res = dialog.ShowDialog()
+            if res.HasValue && res.Value
+            then
+                state.FilePath <- dialog.FileName
+                state.Save()
+        else
+            state.Save()
 
-        canvasCurrentImage.Height <- float srcImg.img.Height
-        canvasCurrentImage.Width <- float srcImg.img.Width
-        canvasCurrentImage.Background <- ImageBrush(BitmapSourceConvert.ToBitmapSource(srcImg.img))
+    // Ask the use to save the current document if neccessary.
+    let askSaveCurrent () =
+        if state.AlteredSinceLastSave
+        then
+            match MessageBox.Show("Would you like to save the current document?", "Saving the current document", MessageBoxButton.YesNo, MessageBoxImage.Question) with
+            | MessageBoxResult.Yes -> saveCurrentDocument ()
+            | _ -> ()
 
-        updateRBCFramesCurrent ()
-        updateRBCFramesPreview ()
-        updateCurrentImageInformation ()
+    let updateCurrentImage () =
+        match state.CurrentImage with
+        | Some srcImg ->
+            // Highlight the preview.
+            stackPreviews.Children
+            |> Seq.cast<Views.ImageSourcePreview>
+            |> Seq.iter (fun preview -> preview.border.BorderThickness <- Thickness(if preview.Tag = (srcImg :> Object) then 3. else 0.))
+
+            canvasCurrentImage.Height <- float srcImg.img.Height
+            canvasCurrentImage.Width <- float srcImg.img.Width
+            canvasCurrentImage.Background <- ImageBrush(BitmapSourceConvert.ToBitmapSource(srcImg.img))
+
+            updateRBCFramesCurrent ()
+            updateRBCFramesPreview ()
+            updateCurrentImageInformation ()
+        | None ->
+            stackRBC.Children.Clear()
+            canvasCurrentImage.Children.Clear()
+            canvasCurrentImage.Background <- Brushes.Black
 
+    let setCurrentImage (srcImg: SourceImage) =
+        if state.CurrentImage.IsNone || state.CurrentImage.Value <> srcImg
+        then
+            state.CurrentImage <- Some srcImg
+            updateCurrentImage ()
+
+    let updateViewportPreview () =
+        for preview in stackPreviews.Children |> Seq.cast<Views.ImageSourcePreview> do
+            let srcImg = preview.Tag :?> SourceImage
+            if Some srcImg = state.CurrentImage then
+                preview.viewport.Visibility <- Visibility.Visible
+
+                let canvasWidth = canvasCurrentImage.ActualWidth * currentScale
+                let canvasHeight = canvasCurrentImage.ActualHeight * currentScale
+                let previewWidth = (preview.ActualWidth - preview.BorderThickness.Left - preview.BorderThickness.Right)
+                let previewHeight = (preview.ActualHeight - preview.BorderThickness.Top - preview.BorderThickness.Bottom)
+
+                let marginLeft = previewWidth * (scrollViewCurrentImage.HorizontalOffset - borderCurrentImage.BorderThickness.Left) / canvasWidth - 2.
+                let marginRight = previewWidth * (canvasWidth - (scrollViewCurrentImage.HorizontalOffset - borderCurrentImage.BorderThickness.Right) - scrollViewCurrentImage.ViewportWidth) / canvasWidth - 2.
+                let marginTop = previewHeight * (scrollViewCurrentImage.VerticalOffset - borderCurrentImage.BorderThickness.Top) / canvasHeight - 2.
+                let marginBottom = previewHeight * (canvasHeight - (scrollViewCurrentImage.VerticalOffset - borderCurrentImage.BorderThickness.Bottom) - scrollViewCurrentImage.ViewportHeight) / canvasHeight - 2.
+
+                preview.viewport.Margin <-
+                    Thickness(
+                        marginLeft,
+                        marginTop,
+                        marginRight,
+                        marginBottom)
+            else
+                preview.viewport.Visibility <- Visibility.Hidden
 
     let addPreview (srcImg: SourceImage) =
         let imgCtrl = Views.ImageSourcePreview(Margin = Thickness(3.))
-        imgCtrl.lblImageNumber.Content <- srcImg.num
+
+        imgCtrl.menuRemoveImage.Click.AddHandler(fun obj args ->
+            stackPreviews.Children.Remove(imgCtrl)
+            let srcImg = imgCtrl.Tag :?> SourceImage
+            let currentRemoved = Some srcImg = state.CurrentImage
+            state.RemoveSourceImage srcImg
+            if currentRemoved
+            then
+                updateCurrentImage()
+            stackPreviews.Children |> Seq.cast<Views.ImageSourcePreview> |> Seq.iter (fun imgPreview -> imgPreview.txtImageNumber.Text <- (imgPreview.Tag :?> SourceImage).num.ToString()))
+
+        imgCtrl.Tag <- srcImg
+        imgCtrl.txtImageNumber.Text <- srcImg.num.ToString()
         let width = 200
         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)))
+        imgCtrl.MouseLeftButtonUp.AddHandler(fun obj args -> setCurrentImage (state.SourceImages |> Seq.find (fun srcImg -> (srcImg :> Object) = imgCtrl.Tag)))
 
     let updatePreviews () =
         stackPreviews.Children.Clear ()
-        if state.SourceImages.Count() > 0
-        then
-            for srcImg in state.SourceImages do
-                addPreview srcImg
-            match state.CurrentImage with
-            | Some srcImg -> setCurrentImage srcImg
-            | _ -> ()
+        for srcImg in state.SourceImages do
+            addPreview srcImg
+        updateCurrentImage ()
 
     let updateGUI () =
         txtPatient.Text <- state.PatientID
         updatePreviews ()
         updateGlobalParasitemia ()
 
-    exit.Click.AddHandler(fun obj args -> mainWindow.Root.Close())
-    saveFile.Click.AddHandler(fun obj args ->
-        synchronizeState ()
-        if state.FilePath = ""
-        then
-            let dialog = SaveFileDialog(AddExtension = true, DefaultExt = PiaZ.extension, Filter = PiaZ.filter);
-            let res = dialog.ShowDialog()
-            if res.HasValue && res.Value
-            then
-                state.FilePath <- dialog.FileName
-                state.Save()
-        else
-            state.Save())
+    txtPatient.TextChanged.AddHandler(fun obj args -> state.PatientID <- txtPatient.Text)
+
+    menuExit.Click.AddHandler(fun obj args ->
+        askSaveCurrent ()
+        mainWindow.Root.Close())
+
+    menuSaveFile.Click.AddHandler(fun obj args -> saveCurrentDocument ())
 
-    loadFile.Click.AddHandler(fun obj args ->
+    menuLoadFile.Click.AddHandler(fun obj args ->
         // TODO: if current state not saved and not empty, ask to save it.
         let dialog = OpenFileDialog(Filter = PiaZ.filter)
         let res = dialog.ShowDialog()
         if res.HasValue && res.Value
         then
+            askSaveCurrent ()
             state.FilePath <- dialog.FileName
             state.Load()
             updateGUI ())
 
-    newFile.Click.AddHandler(fun obj args ->
-        // TODO: if current state not saved and not empty, ask to save it.
+    menuNewFile.Click.AddHandler(fun obj args ->
+        askSaveCurrent ()
         state.Reset()
         updateGUI())
 
-    addSourceImage.Click.AddHandler(fun obj args ->
+    menuAddSourceImage.Click.AddHandler(fun obj args ->
         let dialog = OpenFileDialog(Filter = "Image Files|*.png;*.jpg;*.tif;*.tiff")
         let res = dialog.ShowDialog()
         if res.HasValue && res.Value
@@ -271,18 +342,18 @@ let run (defaultConfig: Config) =
             updateGlobalParasitemia ()
             if state.SourceImages.Count() = 1
             then
-                setCurrentImage srcImg)
+                updateCurrentImage ())
 
-    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
-        updateGlobalParasitemia ()
+    menuStartAnalysis.Click.AddHandler(fun obj args ->
+        if Analysis.showWindow mainWindow.Root state defaultConfig
+        then
+            updateGlobalParasitemia ()
+            updateCurrentImage ())
 
-        // Refresh current image.
-        match state.CurrentImage with
-        | Some srcImg -> setCurrentImage srcImg
-        | _ -> ())
+    menuHightlightRBC.Click.AddHandler(fun obj args ->
+        displayHealthy <- menuHightlightRBC.IsChecked
+        updateRBCFramesPreview ()
+        updateRBCFramesCurrent ())
 
     // Zoom on the current image.
     let adjustCurrentImageBorders (deltaX: float) (deltaY: float) =
@@ -328,12 +399,12 @@ let run (defaultConfig: Config) =
             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)
             currentImageScaleTransform.ScaleX <- currentScale
             currentImageScaleTransform.ScaleY <- currentScale
 
             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.
@@ -356,6 +427,7 @@ let run (defaultConfig: Config) =
             let deltaY = scrollStartPosition.Y - position.Y
             scrollViewCurrentImage.ScrollToHorizontalOffset(deltaX + scrollStartOffsetX)
             scrollViewCurrentImage.ScrollToVerticalOffset(deltaY + scrollStartOffsetY)
+
             args.Handled <- true)
 
     borderCurrentImage.PreviewMouseLeftButtonUp.AddHandler(fun obj args ->
@@ -365,6 +437,8 @@ let run (defaultConfig: Config) =
             borderCurrentImage.ReleaseMouseCapture()
             args.Handled <- true)
 
-    // display mainWindow img
+    // Viewport preview.
+    scrollViewCurrentImage.ScrollChanged.AddHandler(fun obj args -> updateViewportPreview ())
+
     mainWindow.Root.Show()
     app.Run()
\ No newline at end of file