namespace RealTimeGraphEx.DX2D
{
    using SharpDX;
    using SharpDX.Direct2D1;
    using SharpDX.Direct3D10;
    using SharpDX.DXGI;
    using System;
    using System.ComponentModel;
    using System.Diagnostics;
    using System.Threading;
    using System.Windows;
    using Device = SharpDX.Direct3D10.Device1;
    using FeatureLevel = SharpDX.Direct3D10.FeatureLevel;

    public abstract class Direct2DControl : System.Windows.Controls.Image
    {
        private Device Device;
        private Texture2D RenderTarget;
        private DX10ImageSource D3DSurface;
        private readonly Stopwatch RenderTimer;
        private RenderTarget m_d2dRenderTarget;
        protected SharpDX.Direct2D1.Factory m_d2dFactory;
        private bool enableRendering;
        //private Stopwatch renderWatch;

        //public int FrameRate { get; set; }

        public Direct2DControl()
        {
            //renderWatch = new Stopwatch();
            //FrameRate = 60;
            this.RenderTimer = new Stopwatch();
            this.Loaded += this.Window_Loaded;
            this.Unloaded += this.Window_Closing;
            this.Stretch = System.Windows.Media.Stretch.Fill;
        }

        public void EnableRendering()
        {
            enableRendering = true;
        }

        public void DisableRendering()
        {
            enableRendering = false;
        }

        private void Window_Loaded(object sender, RoutedEventArgs e)
        {
            if (Direct2DControl.IsInDesignMode)
            {
                return;
            }

            this.StartD3D();
            this.StartRendering();
        }

        private void Window_Closing(object sender, RoutedEventArgs e)
        {
            if (Direct2DControl.IsInDesignMode)
            {
                return;
            }

            this.StopRendering();
            this.EndD3D();
        }

        private void StartD3D()
        {
            this.Device = new Device(DriverType.Hardware, DeviceCreationFlags.BgraSupport, FeatureLevel.Level_10_0);

            this.D3DSurface = new DX10ImageSource();
            this.D3DSurface.IsFrontBufferAvailableChanged += OnIsFrontBufferAvailableChanged;

            this.CreateAndBindTargets();

            this.Source = this.D3DSurface;
        }

        private void EndD3D()
        {
            this.D3DSurface.IsFrontBufferAvailableChanged -= OnIsFrontBufferAvailableChanged;
            this.Source = null;

            Disposer.SafeDispose(ref this.m_d2dRenderTarget);
            Disposer.SafeDispose(ref this.m_d2dFactory);
            Disposer.SafeDispose(ref this.D3DSurface);
            Disposer.SafeDispose(ref this.RenderTarget);
            Disposer.SafeDispose(ref this.Device);
        }

        private void CreateAndBindTargets()
        {
            if (this.D3DSurface == null) return;

            this.D3DSurface.SetRenderTargetDX10(null);

            Disposer.SafeDispose(ref this.m_d2dRenderTarget);
            Disposer.SafeDispose(ref this.m_d2dFactory);
            Disposer.SafeDispose(ref this.RenderTarget);

            var width = Math.Max((int)this.ActualWidth, 100);
            var height = Math.Max((int)this.ActualHeight, 100);

            var colordesc = new Texture2DDescription
            {
                BindFlags = BindFlags.RenderTarget | BindFlags.ShaderResource,
                Format = Format.B8G8R8A8_UNorm,
                Width = width,
                Height = height,
                MipLevels = 1,
                SampleDescription = new SampleDescription(1, 0),
                Usage = ResourceUsage.Default,
                OptionFlags =