X-Git-Url: http://git.euphorik.ch/?p=master-thesis.git;a=blobdiff_plain;f=Parasitemia%2FParasitemia%2FGUI%2FGUI.fs;h=f88bc3cc66fd7a3a87cc18634fb489994d43a537;hp=8aef25154d9429de7ff2c573fcfd6e94d9b245ba;hb=807437584bdd3d7b7b33be8282472f81f8cce606;hpb=8013d6a586604e443332e6e9a715c86df444a875 diff --git a/Parasitemia/Parasitemia/GUI/GUI.fs b/Parasitemia/Parasitemia/GUI/GUI.fs index 8aef251..f88bc3c 100644 --- a/Parasitemia/Parasitemia/GUI/GUI.fs +++ b/Parasitemia/Parasitemia/GUI/GUI.fs @@ -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) (rbc: RBC) : Emgu.CV.Image = 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 "" 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 |> 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 - |> 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 + |> 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 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 |> 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