module Parasitemia.GUI.Main
+open System
open System.IO
open System.Linq
open System.Windows
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"
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
+ 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
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) =
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)
""
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(srcImg.config.FormattedRadius))
+ 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 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 rec setAsInfected (rbc: RBC) (infected: bool) =
- rbc.SetAsInfected infected
+ state.SetAsInfected rbc infected
canvasCurrentImage.Children
|> Seq.cast<Views.RBCFrame>
|> Seq.iter
updateGlobalParasitemia ()
and RBCFrame (rbc: RBC) : Views.RBCFrame =
- let f = RBCFrameFromExisting rbc (Views.RBCFrame())
- f.menuRBCSetAsHealthy.Click.AddHandler(fun obj args -> setAsInfected (f.Tag :?> RBC) false)
- f.menuRBCSetAsInfected.Click.AddHandler(fun obj args -> setAsInfected (f.Tag :?> RBC) true)
- f.ContextMenuOpening.AddHandler(
+ let frame = RBCFrameFromExisting rbc (Views.RBCFrame())
+ frame.SetValue(Panel.ZIndexProperty, Int32.MaxValue - rbc.num) // To be sure the
+ frame.menuRBCSetAsHealthy.Click.AddHandler(fun obj args -> setAsInfected (frame.Tag :?> RBC) false)
+ frame.menuRBCSetAsInfected.Click.AddHandler(fun obj args -> setAsInfected (frame.Tag :?> RBC) true)
+ frame.ContextMenuOpening.AddHandler(
fun obj args ->
- if (f.Tag :?> RBC).infected
+ if (frame.Tag :?> RBC).infected
then
- f.menuRBCSetAsHealthy.Visibility <- Visibility.Visible
- f.menuRBCSetAsInfected.Visibility <- Visibility.Collapsed
+ frame.menuRBCSetAsHealthy.Visibility <- Visibility.Visible
+ frame.menuRBCSetAsInfected.Visibility <- Visibility.Collapsed
else
- f.menuRBCSetAsHealthy.Visibility <- Visibility.Collapsed
- f.menuRBCSetAsInfected.Visibility <- Visibility.Visible)
+ frame.menuRBCSetAsHealthy.Visibility <- Visibility.Collapsed
+ frame.menuRBCSetAsInfected.Visibility <- Visibility.Visible)
- f.ContextMenuClosing.AddHandler(fun obj args -> if not f.IsMouseOver then highlightRBCFrame f false )
- f.MouseEnter.AddHandler(fun obj args -> highlightRBCFrame f true)
- f.MouseLeave.AddHandler(fun obj args -> if not f.grid.ContextMenu.IsOpen then highlightRBCFrame f false)
- f
+ frame.ContextMenuClosing.AddHandler(fun obj args -> if not frame.IsMouseOver then highlightRBCFrame frame false )
+ frame.MouseEnter.AddHandler(fun obj args -> highlightRBCFrame frame true)
+ frame.MouseLeave.AddHandler(fun obj args -> if not frame.grid.ContextMenu.IsOpen then highlightRBCFrame frame false)
+ frame
and updateRBCFramesPreview () =
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
stackRBC.Children.RemoveRange(currentPreview, stackRBC.Children.Count - currentPreview)
| _ -> ()
+ updateViewportPreview ()
+
let updateRBCFramesCurrent () =
match state.CurrentImage with
| Some srcImg ->
RBCFrameFromExisting rbc (canvasCurrentImage.Children.[currentCanvas] :?> Views.RBCFrame)
else
let f = RBCFrame rbc
+ f.Root.Opacity <- 0.7
canvasCurrentImage.Children.Add(f) |> ignore
f
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 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)))
+
+ // Zoom to a mouse position into the control 'imgCtrl'.
+ let zoomTo (mousePos: Point) =
+ let canvasW = canvasCurrentImage.ActualWidth * currentScale
+ let canvasH = canvasCurrentImage.ActualHeight * currentScale
+ let centerX = (mousePos.X - imgCtrl.BorderThickness.Left) / (imgCtrl.ActualWidth - imgCtrl.BorderThickness.Left) * canvasW
+ let centerY = (mousePos.Y - imgCtrl.BorderThickness.Top) / (imgCtrl.ActualHeight - imgCtrl.BorderThickness.Top) * canvasH
+ scrollViewCurrentImage.ScrollToHorizontalOffset(centerX - scrollViewCurrentImage.ViewportWidth / 2. + borderCurrentImage.BorderThickness.Left)
+ scrollViewCurrentImage.ScrollToVerticalOffset(centerY - scrollViewCurrentImage.ViewportHeight / 2. + borderCurrentImage.BorderThickness.Top)
+
+ imgCtrl.MouseLeftButtonDown.AddHandler(fun obj args ->
+ setCurrentImage (state.SourceImages |> Seq.find (fun srcImg -> (srcImg :> Object) = imgCtrl.Tag))
+ imgCtrl.UpdateLayout()
+ zoomTo (args.GetPosition(imgCtrl))
+ imgCtrl.CaptureMouse() |> ignore)
+
+ imgCtrl.MouseMove.AddHandler(fun obj args ->
+ if imgCtrl.IsMouseCaptured
+ then
+ zoomTo (args.GetPosition(imgCtrl)))
+
+ imgCtrl.MouseLeftButtonUp.AddHandler(fun obj args ->
+ if imgCtrl.IsMouseCaptured
+ then
+ imgCtrl.ReleaseMouseCapture())
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.LostFocus.AddHandler(fun obj args -> state.PatientID <- txtPatient.Text)
+
+ menuExit.Click.AddHandler(fun obj args ->
+ askSaveCurrent ()
+ mainWindow.Root.Close())
- loadFile.Click.AddHandler(fun obj args ->
+ menuSaveFile.Click.AddHandler(fun obj args -> saveCurrentDocument ())
+
+ 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
then
- let srcImg = state.AddSourceImage(dialog.FileName)
+ let srcImg = state.AddSourceImage dialog.FileName defaultConfig
addPreview srcImg
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
+ 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) =
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.
let deltaY = scrollStartPosition.Y - position.Y
scrollViewCurrentImage.ScrollToHorizontalOffset(deltaX + scrollStartOffsetX)
scrollViewCurrentImage.ScrollToVerticalOffset(deltaY + scrollStartOffsetY)
+
args.Handled <- true)
borderCurrentImage.PreviewMouseLeftButtonUp.AddHandler(fun obj args ->
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