Parasitemia/ParasitemiaUI/GUI.fs

   1 module ParasitemiaUI.GUI
   2
   3 open System
   4 open System.IO
   5 open System.Linq
   6 open System.Windows
   7 open System.Windows.Media
   8 open System.Windows.Markup
   9 open System.Windows.Shapes
  10 open System.Windows.Controls
  11 open System.Diagnostics
  12 open Microsoft.Win32 // For the common dialogs.
  13
  14 open Emgu.CV.WPF
  15
  16 open Logger
  17
  18 open ParasitemiaCore.Config
  19 open ParasitemiaCore.Utils
  20 open Types
  21
  22 let run (defaultConfig : Config) (fileToOpen : string option) =
  23     let app = new Application ()
  24     let win = Views.MainWindow ()
  25
  26     let state = State.State defaultConfig
  27     let mutable currentScale = 1.
  28     let mutable displayHealthy = false
  29     let warningBelowNumberOfRBC = 1000
  30
  31     // Initializations.
  32     let canvasCurrentImageColor = win.canvasCurrentImage.Background
  33     win.menuHightlightRBC.IsChecked <- displayHealthy
  34
  35     // Utils.
  36     let extractRBCPreview (img : Emgu.CV.Image<Emgu.CV.Structure.Bgr, byte>) (rbc : RBC) : Emgu.CV.Image<Emgu.CV.Structure.Bgr, byte> =
  37         let rbcWidth = rbc.size.Width
  38         let rbcHeight = rbc.size.Height
  39         let x = rbc.center.X - rbcWidth / 2. |> roundInt
  40         let y = rbc.center.Y - rbcHeight / 2. |> roundInt
  41         let w = roundInt rbcWidth
  42         let h = roundInt rbcHeight
  43         img.GetSubRect (
  44             System.Drawing.Rectangle (
  45                 System.Drawing.Point ((if x < 0 then 0 else x), (if y < 0 then 0 else y)),
  46                 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))
  47             )
  48         )
  49
  50     let setRBCFrameStyle (srcImg : SourceImage) (rbc : RBC) (frame : Views.RBCFrame) =
  51         frame.Opacity <- if displayHealthy || rbc.setManually || rbc.infected then 1. else 0.
  52         let color = if rbc.infected then srcImg.InfectedRBCColor else srcImg.HealthyRBCColor
  53         frame.manuallyAdded.Visibility <- if rbc.setManually then Visibility.Visible else Visibility.Hidden
  54         frame.manuallyAdded.Fill <- color
  55         frame.border.Stroke <- color
  56
  57     let RBCFrameFromExisting (srcImg : SourceImage) (rbc : RBC) (frame : Views.RBCFrame) : Views.RBCFrame =
  58         frame.Visibility <- Visibility.Visible
  59         frame.Height <- rbc.size.Height
  60         frame.Width <- rbc.size.Width
  61         frame.Tag <- rbc
  62         setRBCFrameStyle srcImg rbc frame
  63         frame.border.StrokeThickness <- 1.
  64         frame.txtRBCNumber.Text <- string rbc.num
  65         frame
  66
  67     let updateDocumentStatus () =
  68         win.txtDocumentStatus.Text <- if state.FilePath = "" then "<New document>" else state.FilePath
  69
  70     let statusMessageTimer = Threading.DispatcherTimer ()
  71     statusMessageTimer.Tick.AddHandler (fun obj args -> statusMessageTimer.Stop (); win.txtMessageStatus.Text <- "")
  72     statusMessageTimer.Interval <- TimeSpan (0, 0, 2)
  73     let displayStatusMessage (message : string) =
  74         win.txtMessageStatus.Text <- message
  75         statusMessageTimer.Stop ()
  76         statusMessageTimer.Start ()
  77
  78     let highlightRBCFrame (frame : Views.RBCFrame) (highlight : bool) =
  79         let rbc = frame.Tag :?> RBC
  80         if highlight then
  81             frame.border.StrokeThickness <- 3.
  82             if not rbc.infected && not rbc.setManually && not displayHealthy then frame.Opacity <- 1.
  83         else
  84             frame.border.StrokeThickness <- 1.
  85             if not rbc.infected && not rbc.setManually && not displayHealthy then frame.Opacity <- 0.
  86
  87     let zoomToRBC (rbc : RBC) =
  88         win.scrollViewCurrentImage.ScrollToHorizontalOffset (rbc.center.X * currentScale - win.scrollViewCurrentImage.ViewportWidth / 2. + win.borderCurrentImage.BorderThickness.Left)
  89         win.scrollViewCurrentImage.ScrollToVerticalOffset (rbc.center.Y * currentScale - win.scrollViewCurrentImage.ViewportHeight / 2. + win.borderCurrentImage.BorderThickness.Top)
  90
  91
  92     let txtImageName_TextChanged =
  93         TextChangedEventHandler (fun obj args -> state.CurrentImage |> Option.iter (fun srcImg -> state.SetName srcImg win.txtImageName.Text))
  94
  95     let updateCurrentImageInformation () =
  96         win.txtImageName.TextChanged.RemoveHandler (txtImageName_TextChanged)
  97         win.txtImageInformation1.Inlines.Clear ()
  98         win.txtImageInformation2.Inlines.Clear ()
  99         win.txtImageName.Text <- ""
 100
 101         match state.CurrentImage with
 102         | Some srcImg ->
 103             win.gridImageInformation.Visibility <- Visibility.Visible
 104             win.txtImageName.Text <- srcImg.name
 105             win.txtImageName.TextChanged.AddHandler txtImageName_TextChanged
 106
 107             // The left part.
 108             let parasitemiaStr = Utils.percentText (state.ImageParasitemia srcImg)
 109             win.txtImageInformation1.Inlines.Add (Documents.Run ("Parasitemia: ", FontWeight = FontWeights.Bold))
 110             win.txtImageInformation1.Inlines.Add parasitemiaStr
 111             win.txtImageInformation1.Inlines.Add (Documents.LineBreak ())
 112
 113             win.txtImageInformation1.Inlines.Add (Documents.Run ("Last analysis: ", FontWeight = FontWeights.Bold))
 114             win.txtImageInformation1.Inlines.Add (Documents.Run (if srcImg.dateLastAnalysis.Ticks = 0L then "<Never>" else string (srcImg.dateLastAnalysis.ToLocalTime())))
 115
 116             // The right part part.
 117             win.txtImageInformation2.Inlines.Add (Documents.Run ("Added infected erythrocyte: ", FontWeight = FontWeights.Bold))
 118             win.txtImageInformation2.Inlines.Add (Documents.Run ((state.ImageNbManuallyChangedRBCStr srcImg true) + " " + (state.ImageManuallyChangedRBCStr srcImg true)))
 119             win.txtImageInformation2.Inlines.Add (Documents.LineBreak ())
 120             win.txtImageInformation2.Inlines.Add (Documents.Run ("Removed infected erythrocyte: ", FontWeight = FontWeights.Bold))
 121             win.txtImageInformation2.Inlines.Add (Documents.Run ((state.ImageNbManuallyChangedRBCStr srcImg false) + " " + (state.ImageManuallyChangedRBCStr srcImg false)))
 122
 123         | _ ->
 124             win.gridImageInformation.Visibility <- Visibility.Hidden
 125
 126     let updateGlobalParasitemia () =
 127         win.txtGlobalParasitemia.Inlines.Clear ()
 128         let total, infected = state.GlobalParasitemia
 129         win.txtGlobalParasitemia.Inlines.Add (Documents.Run (Utils.percentText (total, infected), FontWeight = FontWeights.Bold))
 130         if total > 0 && total < warningBelowNumberOfRBC then
 131             win.txtGlobalParasitemia.Inlines.Add (
 132                 Documents.Run (
 133                     sprintf " Warning: the number of erythrocytes should be above %d" warningBelowNumberOfRBC,
 134                     FontWeight = FontWeights.Bold,
 135                     Foreground = Brushes.Red
 136                 )
 137             )
 138
 139     let updateViewportPreview () =
 140         for preview in win.stackPreviews.Children |> Seq.cast<Views.ImageSourcePreview> do
 141             let srcImg = preview.Tag :?> SourceImage
 142             if Some srcImg = state.CurrentImage then
 143                 preview.viewport.Visibility <- Visibility.Visible
 144
 145                 let canvasWidth = win.canvasCurrentImage.ActualWidth * currentScale
 146                 let canvasHeight = win.canvasCurrentImage.ActualHeight * currentScale
 147                 let previewWidth = (preview.ActualWidth - preview.BorderThickness.Left - preview.BorderThickness.Right)
 148                 let previewHeight = (preview.ActualHeight - preview.BorderThickness.Top - preview.BorderThickness.Bottom)
 149
 150                 let marginLeft = previewWidth * (win.scrollViewCurrentImage.HorizontalOffset - win.borderCurrentImage.BorderThickness.Left) / canvasWidth - 2.
 151                 let marginRight = previewWidth * (canvasWidth - (win.scrollViewCurrentImage.HorizontalOffset - win.borderCurrentImage.BorderThickness.Right) - win.scrollViewCurrentImage.ViewportWidth) / canvasWidth - 2.
 152                 let marginTop = previewHeight * (win.scrollViewCurrentImage.VerticalOffset - win.borderCurrentImage.BorderThickness.Top) / canvasHeight - 2.
 153                 let marginBottom = previewHeight * (canvasHeight - (win.scrollViewCurrentImage.VerticalOffset - win.borderCurrentImage.BorderThickness.Bottom) - win.scrollViewCurrentImage.ViewportHeight) / canvasHeight - 2.
 154
 155                 preview.viewport.Margin <-
 156                     Thickness (
 157                         marginLeft,
 158                         marginTop,
 159                         marginRight,
 160                         marginBottom
 161                     )
 162             else
 163                 preview.viewport.Visibility <- Visibility.Hidden
 164
 165     let rec setAsInfected (srcImg : SourceImage) (rbc : RBC) (infected : bool) =
 166         state.SetAsInfected rbc infected
 167         win.canvasCurrentImage.Children
 168         |> Seq.cast<Views.RBCFrame>
 169         |> Seq.iter (
 170             fun frame ->
 171                 if (frame.Tag :?> RBC) = rbc then
 172                     setRBCFrameStyle srcImg rbc frame
 173         )
 174         updateRBCFramesPreview ()
 175         updateCurrentImageInformation ()
 176         updateGlobalParasitemia ()
 177
 178     and RBCFrame (srcImg : SourceImage) (rbc : RBC) : Views.RBCFrame =
 179         let frame = RBCFrameFromExisting srcImg rbc (Views.RBCFrame ())
 180         frame.SetValue (Panel.ZIndexProperty, Int32.MaxValue - rbc.num) // To be sure the
 181         frame.menuRBCSetAsHealthy.Click.AddHandler (fun obj args -> setAsInfected srcImg (frame.Tag :?> RBC) false)
 182         frame.menuRBCSetAsInfected.Click.AddHandler (fun obj args -> setAsInfected srcImg (frame.Tag :?> RBC) true)
 183         frame.ContextMenuOpening.AddHandler (
 184             fun obj args ->
 185                 if (frame.Tag :?> RBC).infected then
 186                     frame.menuRBCSetAsHealthy.Visibility <- Visibility.Visible
 187                     frame.menuRBCSetAsInfected.Visibility <- Visibility.Collapsed
 188                 else
 189                     frame.menuRBCSetAsHealthy.Visibility <- Visibility.Collapsed
 190                     frame.menuRBCSetAsInfected.Visibility <- Visibility.Visible
 191         )
 192         frame.ContextMenuClosing.AddHandler (fun obj args -> if not frame.IsMouseOver then highlightRBCFrame frame false )
 193         frame.MouseEnter.AddHandler (fun obj args -> highlightRBCFrame frame true)
 194         frame.MouseLeave.AddHandler (fun obj args -> if not frame.grid.ContextMenu.IsOpen then highlightRBCFrame frame false)
 195         frame
 196
 197     and updateRBCFramesPreview () =
 198         match state.CurrentImage with
 199         | Some srcImg ->
 200             let mutable currentPreview = 0
 201             for rbc in srcImg.rbcs |> List.filter (fun rbc -> displayHealthy || rbc.infected) do
 202                 let previewInfected =
 203                     if currentPreview < win.stackRBC.Children.Count then
 204                         RBCFrameFromExisting srcImg rbc (win.stackRBC.Children.[currentPreview] :?> Views.RBCFrame)
 205                     else
 206                         let f = RBCFrame srcImg rbc
 207                         f.MouseLeftButtonUp.AddHandler (fun obj args -> zoomToRBC (f.Tag :?> RBC))
 208                         win.stackRBC.Children.Add f |> ignore
 209                         f
 210
 211                 currentPreview <- currentPreview + 1
 212
 213                 previewInfected.Height <- win.stackRBC.ActualHeight
 214                 previewInfected.Width <- win.stackRBC.ActualHeight * rbc.size.Width / rbc.size.Height
 215                 previewInfected.border.Fill <- ImageBrush (BitmapSourceConvert.ToBitmapSource (extractRBCPreview srcImg.img rbc))
 216
 217             win.stackRBC.Children.RemoveRange (currentPreview, win.stackRBC.Children.Count - currentPreview)
 218         | _ -> ()
 219
 220         updateViewportPreview ()
 221
 222     let updateRBCFramesCurrent () =
 223         match state.CurrentImage with
 224         | Some srcImg ->
 225             let mutable currentCanvas = 0
 226             for rbc in srcImg.rbcs do
 227                 let frame =
 228                     if currentCanvas < win.canvasCurrentImage.Children.Count then
 229                         RBCFrameFromExisting srcImg rbc (win.canvasCurrentImage.Children.[currentCanvas] :?> Views.RBCFrame)
 230                     else
 231                         let f = RBCFrame srcImg rbc
 232                         win.canvasCurrentImage.Children.Add f |> ignore
 233                         f
 234
 235                 currentCanvas <- currentCanvas + 1
 236
 237                 Canvas.SetLeft (frame, rbc.center.X - rbc.size.Width / 2.)
 238                 Canvas.SetTop (frame, rbc.center.Y - rbc.size.Height / 2.)
 239
 240             for i in currentCanvas .. win.canvasCurrentImage.Children.Count - 1 do
 241                 win.canvasCurrentImage.Children.[i].Visibility <- Visibility.Hidden
 242         | _ -> ()
 243
 244     let askDocumentPathToSave () : string option =
 245         let dialog = SaveFileDialog (AddExtension = true, DefaultExt = PiaZ.extension, Filter = PiaZ.filter)
 246
 247         if state.FilePath <> "" then
 248             dialog.FileName <- FileInfo(state.FilePath).Name
 249         elif state.PatientID <> "" then
 250             dialog.FileName <- state.PatientID + PiaZ.extension
 251
 252         let res = dialog.ShowDialog ()
 253         if res.HasValue && res.Value then
 254             Some dialog.FileName
 255         else
 256             None
 257
 258     let saveCurrentDocument () =
 259         try
 260             if state.FilePath = "" then
 261                 match askDocumentPathToSave () with
 262                 | Some filepath ->
 263                     state.FilePath <- filepath
 264                     state.Save ()
 265                 | _ -> ()
 266             else
 267                 state.Save ()
 268             updateDocumentStatus ()
 269             displayStatusMessage "Document saved"
 270         with
 271         | :? IOException as ex ->
 272             Log.Error "%O" ex
 273             MessageBox.Show (sprintf "The document cannot be save in \"%s\"" state.FilePath, "Error saving the document", MessageBoxButton.OK, MessageBoxImage.Error) |> ignore
 274
 275     let saveCurrentDocumentAsNewFile () =
 276         match askDocumentPathToSave () with
 277         | Some filepath ->
 278             state.FilePath <- filepath
 279             saveCurrentDocument ()
 280         | _ -> ()
 281
 282     // Ask the use to save the current document if neccessary.
 283     let askSaveCurrent () =
 284         if state.AlteredSinceLastSave then
 285             match MessageBox.Show ("Would you like to save the current document?", "Saving the current document", MessageBoxButton.YesNo, MessageBoxImage.Question) with
 286             | MessageBoxResult.Yes -> saveCurrentDocument ()
 287             | _ -> ()
 288
 289     let updateCurrentImage () =
 290         match state.CurrentImage with
 291         | Some srcImg ->
 292             win.imgLogos.Visibility <- Visibility.Collapsed
 293
 294             // Highlight the preview.
 295             win.stackPreviews.Children
 296             |> Seq.cast<Views.ImageSourcePreview>
 297             |> Seq.iter (fun preview -> preview.border.BorderThickness <- Thickness (if preview.Tag = (srcImg :> Object) then 3. else 0.))
 298
 299             win.canvasCurrentImage.Height <- float srcImg.img.Height
 300             win.canvasCurrentImage.Width <- float srcImg.img.Width
 301             win.canvasCurrentImage.Background <- ImageBrush (BitmapSourceConvert.ToBitmapSource (srcImg.img))
 302
 303             updateRBCFramesCurrent ()
 304             updateRBCFramesPreview ()
 305
 306         | None ->
 307             win.imgLogos.Visibility <- Visibility.Visible
 308
 309             win.stackRBC.Children.Clear ()
 310             win.canvasCurrentImage.Children.Clear ()
 311             win.canvasCurrentImage.Background <- canvasCurrentImageColor
 312
 313         updateCurrentImageInformation ()
 314
 315     let setCurrentImage (srcImg : SourceImage) =
 316         if state.CurrentImage.IsNone || state.CurrentImage.Value <> srcImg then
 317             state.CurrentImage <- Some srcImg
 318             updateCurrentImage ()
 319
 320     let addPreview (srcImg : SourceImage) =
 321         let imgCtrl = Views.ImageSourcePreview (Margin = Thickness 3.)
 322
 323         imgCtrl.menuRemoveImage.Click.AddHandler (
 324             fun obj args ->
 325                 win.stackPreviews.Children.Remove (imgCtrl)
 326                 let srcImg = imgCtrl.Tag :?> SourceImage
 327                 let currentRemoved = Some srcImg = state.CurrentImage
 328                 state.RemoveSourceImage srcImg
 329                 if currentRemoved then
 330                     updateCurrentImage ()
 331
 332                 updateGlobalParasitemia ()
 333
 334                 // Update image numbers.
 335                 win.stackPreviews.Children |> Seq.cast<Views.ImageSourcePreview> |> Seq.iter (fun imgPreview -> imgPreview.txtImageNumber.Text <- (imgPreview.Tag :?> SourceImage).num.ToString ())
 336         )
 337
 338         imgCtrl.Tag <- srcImg
 339         imgCtrl.txtImageNumber.Text <- string srcImg.num
 340         let width = 200
 341         let height = srcImg.img.Height * width / srcImg.img.Width
 342         imgCtrl.imagePreview.Source <- BitmapSourceConvert.ToBitmapSource (srcImg.img.Resize (width, height, Emgu.CV.CvEnum.Inter.Cubic))
 343         win.stackPreviews.Children.Add imgCtrl |> ignore
 344
 345         // Zoom to a mouse position into the control 'imgCtrl'.
 346         let zoomTo (mousePos : Point) =
 347             let canvasW = win.canvasCurrentImage.ActualWidth * currentScale
 348             let canvasH = win.canvasCurrentImage.ActualHeight * currentScale
 349             let centerX = (mousePos.X - imgCtrl.BorderThickness.Left) / (imgCtrl.ActualWidth - imgCtrl.BorderThickness.Left) * canvasW
 350             let centerY = (mousePos.Y - imgCtrl.BorderThickness.Top) / (imgCtrl.ActualHeight - imgCtrl.BorderThickness.Top) * canvasH
 351             win.scrollViewCurrentImage.ScrollToHorizontalOffset (centerX - win.scrollViewCurrentImage.ViewportWidth / 2. + win.borderCurrentImage.BorderThickness.Left)
 352             win.scrollViewCurrentImage.ScrollToVerticalOffset (centerY - win.scrollViewCurrentImage.ViewportHeight / 2. + win.borderCurrentImage.BorderThickness.Top)
 353
 354         imgCtrl.MouseLeftButtonDown.AddHandler (
 355             fun obj args ->
 356                 setCurrentImage (state.SourceImages |> Seq.find (fun srcImg -> (srcImg :> Object) = imgCtrl.Tag))
 357                 imgCtrl.UpdateLayout ()
 358                 zoomTo (args.GetPosition (imgCtrl))
 359                 imgCtrl.CaptureMouse () |> ignore
 360         )
 361
 362         imgCtrl.MouseMove.AddHandler (
 363             fun obj args ->
 364                 if imgCtrl.IsMouseCaptured then
 365                     zoomTo (args.GetPosition imgCtrl)
 366         )
 367
 368         imgCtrl.MouseLeftButtonUp.AddHandler (
 369             fun obj args ->
 370                 if imgCtrl.IsMouseCaptured then
 371                     imgCtrl.ReleaseMouseCapture ()
 372         )
 373
 374     let updatePreviews () =
 375         win.stackPreviews.Children.Clear ()
 376         for srcImg in state.SourceImages do
 377             addPreview srcImg
 378         updateCurrentImage ()
 379
 380     let updateGUI () =
 381         win.txtPatient.Text <- state.PatientID
 382         updatePreviews ()
 383         updateGlobalParasitemia ()
 384         updateDocumentStatus ()
 385
 386     let loadFile (filepath : string) =
 387         askSaveCurrent ()
 388         let previousFilePath = state.FilePath
 389         try
 390             state.FilePath <- filepath
 391             state.Load ()
 392             updateGUI ()
 393         with
 394         | :? IOException as ex ->
 395             Log.Error "%O" ex
 396             state.FilePath <- previousFilePath
 397             MessageBox.Show (sprintf "The document cannot be loaded from \"%s\"" filepath, "Error loading the document", MessageBoxButton.OK, MessageBoxImage.Error) |> ignore
 398
 399     let askLoadFile () =
 400         let dialog = OpenFileDialog (Filter = PiaZ.filter)
 401         let res = dialog.ShowDialog ()
 402         if res.HasValue && res.Value then
 403             loadFile dialog.FileName
 404
 405     let newFile () =
 406         askSaveCurrent ()
 407         state.Reset ()
 408         updateGUI ()
 409
 410     let exportResults () =
 411         let extension = ".txt"
 412         let dialog = SaveFileDialog (AddExtension = true, DefaultExt = extension)
 413
 414         if state.FilePath <> "" then
 415             dialog.FileName <- Path.GetFileNameWithoutExtension state.FilePath + extension
 416         elif state.PatientID <> "" then
 417             dialog.FileName <- state.PatientID + extension
 418
 419         let res = dialog.ShowDialog ()
 420         if res.HasValue && res.Value then
 421             try
 422                 Export.exportResults state dialog.FileName
 423             with
 424             | :? IOException as ex ->
 425                 Log.Error "%O" ex
 426                 MessageBox.Show (sprintf "The results cannot be exported in \"%s\"" state.FilePath, "Error exporting the files", MessageBoxButton.OK, MessageBoxImage.Error) |> ignore
 427
 428     let importImage () =
 429         let dialog = OpenFileDialog (Filter = "Image Files|*.png;*.jpg;*.tif;*.tiff", Multiselect = true)
 430         let res = dialog.ShowDialog ()
 431         if res.HasValue && res.Value then
 432             let noSourceImage = state.SourceImages.Count () = 0
 433
 434             for filename in dialog.FileNames do
 435                 try
 436                     let srcImg = state.AddSourceImage filename defaultConfig
 437                     addPreview srcImg
 438                 with
 439                 | _ as ex ->
 440                     Log.Error "%O" ex
 441                     MessageBox.Show (sprintf "Unable to read the image from \"%s\"" filename, "Error adding an image", MessageBoxButton.OK, MessageBoxImage.Error) |> ignore
 442
 443             updateGlobalParasitemia ()
 444
 445             if noSourceImage then
 446                 updateCurrentImage ()
 447
 448     let showAnalysisWindow () =
 449         if Analysis.showWindow win state then
 450             updateGlobalParasitemia ()
 451             updateCurrentImage ()
 452
 453     let setHighlightRBC (highlight : bool) =
 454         displayHealthy <- highlight
 455         updateRBCFramesPreview ()
 456         updateRBCFramesCurrent ()
 457
 458     win.txtPatient.TextChanged.AddHandler (fun obj args -> state.PatientID <- win.txtPatient.Text)
 459
 460     win.menuExit.Click.AddHandler (fun obj args -> win.Close ())
 461     win.menuSave.Click.AddHandler (fun obj args -> saveCurrentDocument ())
 462     win.menuSaveAs.Click.AddHandler (fun obj args -> saveCurrentDocumentAsNewFile ())
 463     win.menuOpen.Click.AddHandler (fun obj args -> askLoadFile ())
 464     win.menuNew.Click.AddHandler (fun obj args -> newFile ())
 465     win.menuExportResults.Click.AddHandler (fun obj args -> exportResults ())
 466
 467     win.menuAddSourceImage.Click.AddHandler (fun obj args -> importImage ())
 468
 469     win.menuAnalysis.SubmenuOpened.AddHandler (fun obj args -> win.menuStartAnalysis.IsEnabled <- state.SourceImages.Count () > 0)
 470
 471     win.menuStartAnalysis.Click.AddHandler (fun obj args -> showAnalysisWindow ())
 472
 473     win.menuHightlightRBC.Click.AddHandler (fun obj args -> setHighlightRBC win.menuHightlightRBC.IsChecked)
 474
 475     win.menuAbout.Click.AddHandler (fun obj args -> About.showWindow win)
 476
 477     win.Closing.AddHandler (fun obj args -> askSaveCurrent ())
 478
 479     // Zoom on the current image.
 480     let adjustCurrentImageBorders (deltaX : float) (deltaY : float) =
 481         win.borderCurrentImage.BorderThickness <-
 482             Thickness (
 483                 (win.scrollViewCurrentImage.ViewportWidth + deltaX) / 2.,
 484                 (win.scrollViewCurrentImage.ViewportHeight + deltaY) / 2.,
 485                 (win.scrollViewCurrentImage.ViewportWidth + deltaX) / 2.,
 486                 (win.scrollViewCurrentImage.ViewportHeight + deltaY) / 2.
 487             )
 488
 489     win.canvasCurrentImage.SizeChanged.AddHandler (
 490         fun obj args ->
 491             let deltaX = args.NewSize.Width - args.PreviousSize.Width
 492             let deltaY = args.NewSize.Height - args.PreviousSize.Height
 493             if deltaX > 0.5 || deltaY > 0.5 then
 494                 adjustCurrentImageBorders 0.0 0.0
 495                 // Center the view at the center of the image initialy.
 496                 win.scrollViewCurrentImage.UpdateLayout ()
 497                 win.scrollViewCurrentImage.ScrollToHorizontalOffset (win.borderCurrentImage.ActualWidth / 2. - win.scrollViewCurrentImage.ViewportWidth / 2.)
 498                 win.scrollViewCurrentImage.ScrollToVerticalOffset (win.borderCurrentImage.ActualHeight / 2. - win.scrollViewCurrentImage.ViewportHeight / 2.)
 499     )
 500
 501     win.scrollViewCurrentImage.SizeChanged.AddHandler (
 502         fun obj args ->
 503             let deltaX = args.NewSize.Width - args.PreviousSize.Width
 504             let deltaY = args.NewSize.Height - args.PreviousSize.Height
 505             adjustCurrentImageBorders deltaX deltaY
 506             win.scrollViewCurrentImage.ScrollToHorizontalOffset (win.scrollViewCurrentImage.HorizontalOffset + deltaX / 8.)
 507             win.scrollViewCurrentImage.ScrollToVerticalOffset (win.scrollViewCurrentImage.VerticalOffset + deltaY / 8.)
 508     )
 509
 510     let mutable maxScale = 4.
 511     let mutable minScale = 0.25
 512     let currentImageScaleTransform = ScaleTransform ()
 513     win.canvasCurrentImage.LayoutTransform <- currentImageScaleTransform
 514     win.borderCurrentImage.PreviewMouseWheel.AddHandler (
 515         fun obj args ->
 516             let scaleFactor = if args.Delta > 0 then 2.0 else 0.5
 517             if scaleFactor > 1. && currentScale < maxScale || scaleFactor < 1. && currentScale > minScale then
 518                 let previousScale = currentScale
 519                 currentScale <-
 520                     let newScale = currentScale * scaleFactor
 521                     if newScale > maxScale then maxScale elif newScale < minScale then minScale else newScale
 522                 let realScaleFactor = currentScale / previousScale
 523
 524                 let centerX = win.scrollViewCurrentImage.HorizontalOffset + win.scrollViewCurrentImage.ViewportWidth / 2. - win.borderCurrentImage.BorderThickness.Left
 525                 let centerY = win.scrollViewCurrentImage.VerticalOffset + win.scrollViewCurrentImage.ViewportHeight / 2. - win.borderCurrentImage.BorderThickness.Top
 526
 527                 currentImageScaleTransform.ScaleX <- currentScale
 528                 currentImageScaleTransform.ScaleY <- currentScale
 529
 530                 win.scrollViewCurrentImage.ScrollToHorizontalOffset (centerX * realScaleFactor - win.scrollViewCurrentImage.ViewportWidth / 2. + win.borderCurrentImage.BorderThickness.Left)
 531                 win.scrollViewCurrentImage.ScrollToVerticalOffset (centerY * realScaleFactor - win.scrollViewCurrentImage.ViewportHeight / 2. + win.borderCurrentImage.BorderThickness.Top)
 532
 533             args.Handled <- true
 534     )
 535
 536     // Pan on the current image.
 537     let mutable scrollStartPosition = Point (0., 0.)
 538     let mutable scrollStartOffsetX = 0.
 539     let mutable scrollStartOffsetY = 0.
 540     win.borderCurrentImage.PreviewMouseLeftButtonDown.AddHandler (
 541         fun obj args ->
 542             scrollStartPosition <- args.GetPosition win.scrollViewCurrentImage
 543             scrollStartOffsetX <- win.scrollViewCurrentImage.HorizontalOffset
 544             scrollStartOffsetY <- win.scrollViewCurrentImage.VerticalOffset
 545             win.borderCurrentImage.Cursor <- Input.Cursors.ScrollAll
 546             win.borderCurrentImage.CaptureMouse () |> ignore
 547             args.Handled <- true
 548     )
 549
 550     win.borderCurrentImage.PreviewMouseMove.AddHandler (
 551         fun obj args ->
 552             if win.borderCurrentImage.IsMouseCaptured then
 553                 let position = args.GetPosition win.scrollViewCurrentImage
 554                 let deltaX = scrollStartPosition.X - position.X
 555                 let deltaY = scrollStartPosition.Y - position.Y
 556                 win.scrollViewCurrentImage.ScrollToHorizontalOffset (deltaX + scrollStartOffsetX)
 557                 win.scrollViewCurrentImage.ScrollToVerticalOffset (deltaY + scrollStartOffsetY)
 558
 559                 args.Handled <- true
 560     )
 561
 562     win.borderCurrentImage.PreviewMouseLeftButtonUp.AddHandler (
 563         fun obj args ->
 564             if win.borderCurrentImage.IsMouseCaptured then
 565                 win.borderCurrentImage.Cursor <- Input.Cursors.Arrow
 566                 win.borderCurrentImage.ReleaseMouseCapture ()
 567                 args.Handled <- true
 568     )
 569
 570     // Shortcuts.
 571     // Save.
 572     win.InputBindings.Add (
 573         Input.KeyBinding (
 574             ViewModule.FunCommand ((fun obj -> saveCurrentDocument ()), (fun obj -> true)),
 575             Input.KeyGesture (Input.Key.S, Input.ModifierKeys.Control)
 576         )
 577     ) |> ignore
 578
 579     // Save as.
 580     win.InputBindings.Add (
 581         Input.KeyBinding (
 582             ViewModule.FunCommand ((fun obj -> saveCurrentDocumentAsNewFile ()), (fun obj -> true)),
 583             Input.KeyGesture (Input.Key.S, Input.ModifierKeys.Control ||| Input.ModifierKeys.Shift)
 584         )
 585     ) |> ignore
 586
 587     // Open.
 588     win.InputBindings.Add (
 589         Input.KeyBinding (
 590             ViewModule.FunCommand ((fun obj -> askLoadFile ()), (fun obj -> true)),
 591             Input.KeyGesture (Input.Key.O, Input.ModifierKeys.Control)
 592         )
 593     ) |> ignore
 594
 595     // New file.
 596     win.InputBindings.Add (
 597         Input.KeyBinding (
 598             ViewModule.FunCommand ((fun obj -> newFile ()), (fun obj -> true)),
 599             Input.KeyGesture (Input.Key.N, Input.ModifierKeys.Control)
 600         )
 601     ) |> ignore
 602
 603     // Export results.
 604     win.InputBindings.Add (
 605         Input.KeyBinding (
 606             ViewModule.FunCommand ((fun obj -> exportResults ()), (fun obj -> true)),
 607             Input.KeyGesture (Input.Key.E, Input.ModifierKeys.Control)
 608         )
 609     ) |> ignore
 610
 611     // Import an image.
 612     win.InputBindings.Add (
 613         Input.KeyBinding (
 614             ViewModule.FunCommand ((fun obj -> importImage ()), (fun obj -> true)),
 615             Input.KeyGesture (Input.Key.A, Input.ModifierKeys.Control)
 616         )
 617     ) |> ignore
 618
 619     // Show analysis dialog.
 620     win.InputBindings.Add (
 621         Input.KeyBinding (
 622             ViewModule.FunCommand ((fun obj -> showAnalysisWindow ()), (fun obj -> state.SourceImages.Count () > 0)),
 623             Input.KeyGesture (Input.Key.Y, Input.ModifierKeys.Control)
 624         )
 625     ) |> ignore
 626
 627     // Toggle RBC highlight.
 628     win.InputBindings.Add (
 629         Input.KeyBinding (
 630             ViewModule.FunCommand (
 631                 (
 632                     fun obj ->
 633                         win.menuHightlightRBC.IsChecked <- not win.menuHightlightRBC.IsChecked
 634                         setHighlightRBC win.menuHightlightRBC.IsChecked
 635                 ),
 636                 (fun obj -> true)
 637             ),
 638             Input.KeyGesture (Input.Key.H, Input.ModifierKeys.Control)
 639         )
 640     ) |> ignore
 641
 642     // Viewport preview.
 643     win.scrollViewCurrentImage.ScrollChanged.AddHandler (fun obj args -> updateViewportPreview ())
 644
 645     updateDocumentStatus ()
 646     win.gridImageInformation.Visibility <- Visibility.Hidden
 647
 648     win.Show ()
 649
 650     match fileToOpen with
 651     | Some filepath -> loadFile filepath
 652     | None -> ()
 653
 654     app.Run ()