using Tango.Editors;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Windows;
using System.Windows.Media;
using System.Windows.Media.Imaging;

/// <exclude/>
/// <summary>
/// A collection of <see cref="FrameworkElement"/> extension methods. 
/// </summary>
public static class FrameworkElementExtensions
{
    /// <summary>
    /// Renders the element to a bitmap source.
    /// </summary>
    /// <param name="element">The element.</param>
    /// <param name="forceRedraw">if set to <c>true</c> forces the redraw of the element and prevents a freezing effect.</param>
    /// <returns></returns>
    public static BitmapSource TakeSnapshot(this FrameworkElement element, bool forceRedraw = true)
    {
        var bitmap = UIHelper.TakeSnapshot(element, new Size(element.ActualWidth, element.ActualHeight));

        var parent = element.Parent;

        if (parent != null && forceRedraw)
        {
            UIHelper.RemoveChild(parent, element);
            element.UpdateLayout();
            UIHelper.AddChild(parent, element);
            element.UpdateLayout();
        }

        return bitmap;
    }

    /// <summary>
    /// Renders the element to a bitmap source of the specified size.
    /// </summary>
    /// <param name="element">The element.</param>
    /// <param name="width">The bitmap width.</param>
    /// <param name="height">The bitmap height.</param>
    /// <param name="forceRedraw">if set to <c>true</c> forces the redraw of the element and prevents a freezing effect.</param>
    /// <returns></returns>
    public static BitmapSource TakeSnapshot(this FrameworkElement element, int width, int height, bool forceRedraw = true)
    {
        var bitmap = UIHelper.TakeSnapshot(element, new Size(element.ActualWidth, element.ActualHeight));
        TransformedBitmap resizedBitmap = new TransformedBitmap(bitmap, new ScaleTransform(width / bitmap.Width, height / bitmap.Height, 0, 0));

        var parent = element.Parent;

        if (parent != null && forceRedraw)
        {
            UIHelper.RemoveChild(parent, element);
            element.UpdateLayout();
            UIHelper.AddChild(parent, element);
            element.UpdateLayout();
        }

        return resizedBitmap;
    }
}