// --------------------------------------------------------------------------------------------------------------------

// <copyright file="GridLinesVisual3D.cs" company="Helix 3D Toolkit">

//   http://helixtoolkit.codeplex.com, license: MIT

// </copyright>

// --------------------------------------------------------------------------------------------------------------------



namespace HelixToolkit.Wpf

{

    using System;

    using System.Windows;

    using System.Windows.Media;

    using System.Windows.Media.Media3D;



    /// <summary>

    /// A visual element that shows a set of grid lines.

    /// </summary>

    public class GridLinesVisual3D : MeshElement3D

    {

        /// <summary>

        /// Identifies the <see cref="Center"/> dependency property.

        /// </summary>

        public static readonly DependencyProperty CenterProperty = DependencyProperty.Register(

            "Center", typeof(Point3D), typeof(GridLinesVisual3D), new UIPropertyMetadata(new Point3D(), GeometryChanged));



        /// <summary>

        /// Identifies the <see cref="MinorDistance"/> dependency property.

        /// </summary>

        public static readonly DependencyProperty MinorDistanceProperty = DependencyProperty.Register(

            "MinorDistance", typeof(double), typeof(GridLinesVisual3D), new PropertyMetadata(2.5, GeometryChanged));



        /// <summary>

        /// Identifies the <see cref="LengthDirection"/> dependency property.

        /// </summary>

        public static readonly DependencyProperty LengthDirectionProperty =

            DependencyProperty.Register(

                "LengthDirection",

                typeof(Vector3D),

                typeof(GridLinesVisual3D),

                new UIPropertyMetadata(new Vector3D(1, 0, 0), GeometryChanged));



        /// <summary>

        /// Identifies the <see cref="Length"/> dependency property.

        /// </summary>

        public static readonly DependencyProperty LengthProperty = DependencyProperty.Register(

            "Length", typeof(double), typeof(GridLinesVisual3D), new PropertyMetadata(200.0, GeometryChanged));



        /// <summary>

        /// Identifies the <see cref="MajorDistance"/> dependency property.

        /// </summary>

        public static readonly DependencyProperty MajorDistanceProperty = DependencyProperty.Register(

            "MajorDistance", typeof(double), typeof(GridLinesVisual3D), new PropertyMetadata(10.0, GeometryChanged));



        /// <summary>

        /// Identifies the <see cref="Normal"/> dependency property.

        /// </summary>

        public static readonly DependencyProperty NormalProperty = DependencyProperty.Register(

            "Normal",

            typeof(Vector3D),

            typeof(GridLinesVisual3D),

            new UIPropertyMetadata(new Vector3D(0, 0, 1), GeometryChanged));



        /// <summary>

        /// Identifies the <see cref="Thickness"/> dependency property.

        /// </summary>

        public static readonly DependencyProperty ThicknessProperty = DependencyProperty.Register(

            "Thickness", typeof(double), typeof(GridLinesVisual3D), new PropertyMetadata(0.08, GeometryChanged));



        /// <summary>

        /// Identifies the <see cref="Width"/> dependency property.

        /// </summary>

        public static readonly DependencyProperty WidthProperty = DependencyProperty.Register(

            "Width", typeof(double), typeof(GridLinesVisual3D), new PropertyMetadata(200.0, GeometryChanged));



        /// <summary>

        /// The length direction.

        /// </summary>

        private Vector3D lengthDirection;



        /// <summary>

        /// The width direction.

        /// </summary>

        private Vector3D widthDirection;



        /// <summary>

        /// Initializes a new instance of the <see cref = "GridLinesVisual3D" /> class.

        /// </summary>

        public GridLinesVisual3D()

        {

            this.Fill = Brushes.Gray;

        }



        /// <summary>

        /// Gets or sets the center of the grid.

        /// </summary>

        /// <value>The center.</value>

        public Point3D Center

        {

            get

            {

                return (Point3D)this.GetValue(CenterProperty);

            }



            set

            {

                this.SetValue(CenterProperty, value);

            }

        }



        /// <summary>

        /// Gets or sets the length of the grid area.

        /// </summary>

        /// <value>The length.</value>

        public double Length

        {

            get

            {

                return (double)this.GetValue(LengthProperty);

            }



            set

            {

                this.SetValue(LengthProperty, value);

            }

        }



        /// <summary>

        /// Gets or sets the length direction of the grid.

        /// </summary>

        /// <value>The length direction.</value>

        public Vector3D LengthDirection

        {

            get

            {

                return (Vector3D)this.GetValue(LengthDirectionProperty);

            }



            set

            {

                this.SetValue(LengthDirectionProperty, value);

            }

        }



        /// <summary>

        /// Gets or sets the distance between major grid lines.

        /// </summary>

        /// <value>The distance.</value>

        public double MajorDistance

        {

            get

            {

                return (double)this.GetValue(MajorDistanceProperty);

            }



            set

            {

                this.SetValue(MajorDistanceProperty, value);

            }

        }



        /// <summary>

        /// Gets or sets the distance between minor grid lines.

        /// </summary>

        /// <value>The distance.</value>

        public double MinorDistance

        {

            get

            {

                return (double)this.GetValue(MinorDistanceProperty);

            }



            set

            {

                this.SetValue(MinorDistanceProperty, value);

            }

        }



        /// <summary>

        /// Gets or sets the normal vector of the grid plane.

        /// </summary>

        /// <value>The normal.</value>

        public Vector3D Normal

        {

            get

            {

                return (Vector3D)this.GetValue(NormalProperty);

            }



            set

            {

                this.SetValue(NormalProperty, value);

            }

        }



        /// <summary>

        /// Gets or sets the thickness of the grid lines.

        /// </summary>

        /// <value>The thickness.</value>

        public double Thickness

        {

            get

            {

                return (double)this.GetValue(ThicknessProperty);

            }



            set

            {

                this.SetValue(ThicknessProperty, value);

            }

        }



        /// <summary>

        /// Gets or sets the width of the grid area (perpendicular to the length direction).

        /// </summary>

        /// <value>The width.</value>

        public double Width

        {

            get

            {

                return (double)this.GetValue(WidthProperty);

            }



            set

            {

                this.SetValue(WidthProperty, value);

            }

        }



        /// <summary>

        /// Do the tesselation and return the <see cref="MeshGeometry3D"/>.

        /// </summary>

        /// <returns>

        /// </returns>

        protected override MeshGeometry3D Tessellate()

        {

            this.lengthDirection = this.LengthDirection;

            this.lengthDirection.Normalize();

            this.widthDirection = Vector3D.CrossProduct(this.Normal, this.lengthDirection);

            this.widthDirection.Normalize();



            var mesh = new MeshBuilder(true, false);

            double minX = -this.Width / 2;

            double minY = -this.Length / 2;

            double maxX = this.Width / 2;

            double maxY = this.Length / 2;



            double x = minX;

            double eps = this.MinorDistance / 10;

            while (x <= maxX + eps)

            {

                double t = this.Thickness;

                if (IsMultipleOf(x, this.MajorDistance))

                {

                    t *= 2;

                }



                this.AddLineX(mesh, x, minY, maxY, t);

                x += this.MinorDistance;

            }



            double y = minY;

            while (y <= maxY + eps)

            {

                double t = this.Thickness;

                if (IsMultipleOf(y, this.MajorDistance))

                {

                    t *= 2;

                }



                this.AddLineY(mesh, y, minX, maxX, t);

                y += this.MinorDistance;

            }



            var m = mesh.ToMesh();

            m.Freeze();

            return m;

        }



        /// <summary>

        /// Determines whether y is a multiple of d.

        /// </summary>

        /// <param name="y">

        /// The y.

        /// </param>

        /// <param name="d">

        /// The d.

        /// </param>

        /// <returns>

        /// The is multiple of.

        /// </returns>

        private static bool IsMultipleOf(double y, double d)

        {

            double y2 = d * (int)(y / d);

            return Math.Abs(y - y2) < 1e-3;

        }



        /// <summary>

        /// The add line x.

        /// </summary>

        /// <param name="mesh">

        /// The mesh.

        /// </param>

        /// <param name="x">

        /// The x.

        /// </param>

        /// <param name="minY">

        /// The min y.

        /// </param>

        /// <param name="maxY">

        /// The max y.

        /// </param>

        /// <param name="thickness">

        /// The thickness.

        /// </param>

        private void AddLineX(MeshBuilder mesh, double x, double minY, double maxY, double thickness)

        {

            int i0 = mesh.Positions.Count;

            mesh.Positions.Add(this.GetPoint(x - thickness / 2, minY));

            mesh.Positions.Add(this.GetPoint(x - thickness / 2, maxY));

            mesh.Positions.Add(this.GetPoint(x + thickness / 2, maxY));

            mesh.Positions.Add(this.GetPoint(x + thickness / 2, minY));

            mesh.Normals.Add(this.Normal);

            mesh.Normals.Add(this.Normal);

            mesh.Normals.Add(this.Normal);

            mesh.Normals.Add(this.Normal);

            mesh.TriangleIndices.Add(i0);

            mesh.TriangleIndices.Add(i0 + 1);

            mesh.TriangleIndices.Add(i0 + 2);

            mesh.TriangleIndices.Add(i0 + 2);

            mesh.TriangleIndices.Add(i0 + 3);

            mesh.TriangleIndices.Add(i0);

        }



        /// <summary>

        /// The add line y.

        /// </summary>

        /// <param name="mesh">

        /// The mesh.

        /// </param>

        /// <param name="y">

        /// The y.

        /// </param>

        /// <param name="minX">

        /// The min x.

        /// </param>

        /// <param name="maxX">

        /// The max x.

        /// </param>

        /// <param name="thickness">

        /// The thickness.

        /// </param>

        private void AddLineY(MeshBuilder mesh, double y, double minX, double maxX, double thickness)

        {

            int i0 = mesh.Positions.Count;

            mesh.Positions.Add(this.GetPoint(minX, y + thickness / 2));

            mesh.Positions.Add(this.GetPoint(maxX, y + thickness / 2));

            mesh.Positions.Add(this.GetPoint(maxX, y - thickness / 2));

            mesh.Positions.Add(this.GetPoint(minX, y - thickness / 2));

            mesh.Normals.Add(this.Normal);

            mesh.Normals.Add(this.Normal);

            mesh.Normals.Add(this.Normal);

            mesh.Normals.Add(this.Normal);

            mesh.TriangleIndices.Add(i0);

            mesh.TriangleIndices.Add(i0 + 1);

            mesh.TriangleIndices.Add(i0 + 2);

            mesh.TriangleIndices.Add(i0 + 2);

            mesh.TriangleIndices.Add(i0 + 3);

            mesh.TriangleIndices.Add(i0);

        }



        /// <summary>

        /// Gets a point on the plane.

        /// </summary>

        /// <param name="x">

        /// The x coordinate.

        /// </param>

        /// <param name="y">

        /// The y coordinate.

        /// </param>

        /// <returns>

        /// </returns>

        private Point3D GetPoint(double x, double y)

        {

            return this.Center + this.widthDirection * x + this.lengthDirection * y;

        }



    }

}