module Parasitemia.GUI.Main

open System.IO
open System.Linq
open System.Windows
open System.Windows.Media
open System.Windows.Markup
open System.Windows.Shapes
open System.Windows.Controls
open System.Diagnostics
open Microsoft.Win32 // For the common dialogs.

//open Emgu.CV
//open Emgu.CV.Structure
open Emgu.CV.WPF

open Config
open Types

let run (defaultConfig: Config) =
    let app = new Application()
    let mainWindow = Views.MainWindow()
    let ctrl (name: string): 'a =
        mainWindow.Root.FindName(name) :?> 'a

    // Utils.log <- (fun m -> log mainWindow m)

    let currentImageMargin = 0.1 // 10 %.

    let state = State.State()

    let exit: MenuItem = ctrl "menuExit"
    let saveFile: MenuItem = ctrl "menuSave"
    let loadFile: MenuItem = ctrl "menuOpen"
    let newFile: MenuItem = ctrl "menuNew"

    let addSourceImage: MenuItem = ctrl "menuAddSourceImage"
    let txtPatient: TextBox = ctrl "txtPatient"
    let stackPreviews: StackPanel = ctrl "stackPreviews"
    let butStartAnalysis: Button = ctrl "butStartAnalysis"

    let scrollViewCurrentImage: ScrollViewer = ctrl "scrollViewCurrentImage"
    let borderCurrentImage: Border = ctrl "borderCurrentImage"
    let canvasCurrentImage: Canvas = ctrl "canvasCurrentImage"

    // Initializations.

    // Operations.
    let synchronizeState () =
        state.PatientID <- txtPatient.Text

    let setCurrentImage (srcImg: SourceImage) =
        canvasCurrentImage.Height <- float srcImg.img.Height
        canvasCurrentImage.Width <- float srcImg.img.Width
        canvasCurrentImage.Background <- ImageBrush(BitmapSourceConvert.ToBitmapSource(srcImg.img))
        (*for rbc in srcImg.rbcs do
            let rectangle = Rectangle(Height = float rbc.size.Height, Width = float rbc.size.Width, )*)

    let addPreview (srcImg: SourceImage) =
        let imgCtrl = Views.ImageSourcePreview(Margin = Thickness(3.))
        imgCtrl.lblImageNumber.Content <- srcImg.num
        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

    let updatePreviews () =
        stackPreviews.Children.Clear ()
        for srcImg in state.SourceImages do
            addPreview srcImg

    let updateGUI () =
        txtPatient.Text <- state.PatientID
        updatePreviews ()

    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 = Pia.extension, Filter = Pia.filter);
            let res = dialog.ShowDialog()
            if res.HasValue && res.Value
            then
                state.FilePath <- dialog.FileName
                state.Save()
        else
            state.Save())

    loadFile.Click.AddHandler(fun obj args ->
        // TODO: if current state not saved and not empty, ask to save it.
        let dialog = OpenFileDialog(Filter = Pia.filter)
        let res = dialog.ShowDialog()
        if res.HasValue && res.Value
        then
            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.
        state.Reset()
        updateGUI())

    addSourceImage.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)
            addPreview srcImg
            if state.SourceImages.Count() = 1
            then
                setCurrentImage srcImg)

    butStartAnalysis.Click.AddHandler(fun obj args -> ())

    // Zoom on the current image.
    let adjustCurrentImageMargins () =
        borderCurrentImage.BorderThickness <-
            Thickness(scrollViewCurrentImage.ViewportWidth / 2., scrollViewCurrentImage.ViewportHeight / 2., scrollViewCurrentImage.ViewportWidth / 2., scrollViewCurrentImage.ViewportHeight / 2.)

    scrollViewCurrentImage.Loaded.AddHandler(fun obj args -> adjustCurrentImageMargins ())
    scrollViewCurrentImage.SizeChanged.AddHandler(fun obj args -> adjustCurrentImageMargins ())

    let mutable currentScale = 1.
    let mutable maxScale = 5.
    let mutable minScale = 0.1
    let currentImageScaleTransform = ScaleTransform()
    borderCurrentImage.LayoutTransform <- currentImageScaleTransform
    borderCurrentImage.PreviewMouseWheel.AddHandler(fun obj args ->
        let scaleFactor = if args.Delta > 0 then 2.0 else 0.5
        if scaleFactor > 1. && currentScale < maxScale || scaleFactor < 1. && currentScale > minScale
        then
            let previousScale = currentScale
            currentScale <-
                let newScale = currentScale * scaleFactor
                if newScale > maxScale then maxScale elif newScale < minScale then minScale else newScale
            let realScaleFactor = currentScale / previousScale

            let centerX = scrollViewCurrentImage.HorizontalOffset + scrollViewCurrentImage.ViewportWidth / 2. - canvasCurrentImage.Margin.Left
            let centerY = scrollViewCurrentImage.VerticalOffset + scrollViewCurrentImage.ViewportHeight / 2. - canvasCurrentImage.Margin.Top

            canvasCurrentImage.Margin <- Thickness(canvasCurrentImage.Margin.Top * realScaleFactor)
            currentImageScaleTransform.ScaleX <- currentScale
            currentImageScaleTransform.ScaleY <- currentScale

            scrollViewCurrentImage.ScrollToHorizontalOffset(centerX * realScaleFactor - scrollViewCurrentImage.ViewportWidth / 2. + canvasCurrentImage.Margin.Left)
            scrollViewCurrentImage.ScrollToVerticalOffset(centerY * realScaleFactor - scrollViewCurrentImage.ViewportHeight / 2. + canvasCurrentImage.Margin.Top)
        args.Handled <- true)

    // Pan on the current image.
    let mutable scrollStartPosition = Point(0., 0.)
    let mutable scrollStartOffsetX = 0.
    let mutable scrollStartOffsetY = 0.
    borderCurrentImage.PreviewMouseLeftButtonDown.AddHandler(fun obj args ->
        scrollStartPosition <- args.GetPosition(scrollViewCurrentImage)
        scrollStartOffsetX <- scrollViewCurrentImage.HorizontalOffset
        scrollStartOffsetY <- scrollViewCurrentImage.VerticalOffset
        borderCurrentImage.Cursor <- Input.Cursors.ScrollAll
        borderCurrentImage.CaptureMouse() |> ignore
        args.Handled <- true)

    borderCurrentImage.PreviewMouseMove.AddHandler(fun obj args ->
        if borderCurrentImage.IsMouseCaptured
        then
            let position = args.GetPosition(scrollViewCurrentImage)
            let deltaX = scrollStartPosition.X - position.X
            let deltaY = scrollStartPosition.Y - position.Y
            scrollViewCurrentImage.ScrollToHorizontalOffset(deltaX + scrollStartOffsetX)
            scrollViewCurrentImage.ScrollToVerticalOffset(deltaY + scrollStartOffsetY)
            args.Handled <- true)

    borderCurrentImage.PreviewMouseLeftButtonUp.AddHandler(fun obj args ->
        if borderCurrentImage.IsMouseCaptured
        then
            borderCurrentImage.Cursor <- Input.Cursors.Arrow
            borderCurrentImage.ReleaseMouseCapture()
            args.Handled <- true)


    (*let txtPatient: Controls.TextBox = ctrl "txtPatient"
    txtPatient.TextChanged.AddHandler(fun obj args ->
        state.PatientID <- txtPatient.Text)*)

     (*saveFileDialog1.Filter = "txt files (*.txt)|*.txt|All files (*.*)|*.*"  ;
     saveFileDialog1.FilterIndex = 2 ;
     saveFileDialog1.RestoreDirectory = true ;*)

    // display mainWindow img
    mainWindow.Root.Show()
    app.Run()