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