package org.opentrafficsim.core.geometry;
   
   import java.awt.geom.Rectangle2D;
   import java.io.Serializable;
   import java.util.Collection;
   import java.util.Iterator;
   import java.util.LinkedHashSet;
   import java.util.Set;
   
  import org.djutils.exceptions.Throw;
  import org.djutils.logger.CategoryLogger;
  
  /**
   * Set of OtsShape objects and provides methods for fast selection of those objects that intersect an OtsShape. <br>
   * An Ots2dSet internally stores the OtsShapes in a quad tree. At time of construction the minimum cell size is defined. Node
   * expansion is never performed on nodes that are smaller than this limit. <br>
   * Each node (even the non-leaf nodes) store a set of OtsShape. Non-leaf nodes locally store those shapes that completely cover
   * the rectangular area of the node. Such shapes are <b>not</b> also stored in leaf nodes below that node. OtsShapes that
   * partially cover a non-leaf node are stored in each of the leaf nodes below that node that those OtsShapes (partially) cover.
   * Leaf nodes that cannot be expanded (because they are too small) also store all OtsShapes that partially cover the area of the
   * node. <br>
   * If removal of an OtsShape objects results in a leaf becoming empty, that leaf is removed from its parent (which may then
   * itself become empty and removed in turn).
   * <p>
   * Copyright (c) 2013-2023 Delft University of Technology, PO Box 5, 2600 AA, Delft, the Netherlands. All rights reserved. <br>
   * BSD-style license. See <a href="https://opentrafficsim.org/docs/license.html">OpenTrafficSim License</a>.
   * </p>
   * @author <a href="https://github.com/averbraeck">Alexander Verbraeck</a>
   * @author <a href="https://tudelft.nl/staff/p.knoppers-1">Peter Knoppers</a>
   * @author <a href="https://dittlab.tudelft.nl">Wouter Schakel</a>
   */
  public class Ots2dSet implements Set<OtsShape>, Serializable
  {
      /** */
      private static final long serialVersionUID = 20170400L;
  
      /** Set of all shapes used for iterators, etc. */
      private final Set<OtsShape> allShapes = new LinkedHashSet<OtsShape>();
  
      /** How fine will this quad tree divide. This one is copied to each sub-node which is somewhat inefficient. */
      private final double minimumCellSize;
  
      /** Spatial storage for the OtsShapes. */
      private QuadTreeNode quadTree;
  
      /**
       * Construct an empty Ots2dSet for a rectangular region. Objects that do not intersect this region will never be stored in
       * this Ots2dSet. (Trying to add such an OtsShape is <b>not</b> an error; the <code>add</code> method will return false,
       * indicating that the set has not been modified.)
       * @param boundingBox Rectangle2D; the region
       * @param minimumCellSize double; resolution of the underlying quad tree
       * @throws OtsGeometryException when the bounding box covers no surface
       */
      public Ots2dSet(final Rectangle2D boundingBox, final double minimumCellSize) throws OtsGeometryException
      {
          Throw.when(null == boundingBox, NullPointerException.class, "The boundingBox may not be null");
          Throw.when(boundingBox.getWidth() <= 0 || boundingBox.getHeight() <= 0, OtsGeometryException.class,
                  "The boundingBox must have nonzero surface (got %s", boundingBox);
          Throw.when(minimumCellSize <= 0, OtsGeometryException.class, "The minimumCellSize must be > 0 (got %f)",
                  minimumCellSize);
          this.quadTree = new QuadTreeNode(boundingBox);
          this.minimumCellSize = minimumCellSize;
      }
  
      /** {@inheritDoc} */
      @Override
      public final int size()
      {
          return this.allShapes.size();
      }
  
      /** {@inheritDoc} */
      @Override
      public final boolean isEmpty()
      {
          return this.allShapes.isEmpty();
      }
  
      /** {@inheritDoc} */
      @Override
      public final boolean contains(final Object o)
      {
          return this.allShapes.contains(o);
      }
  
      /** {@inheritDoc} */
      @Override
      public final Iterator<OtsShape> iterator()
      {
          return new QuadTreeIterator();
      }
  
      /** {@inheritDoc} */
      @Override
      public final Object[] toArray()
      {
          return this.allShapes.toArray();
      }
  
     /** {@inheritDoc} */
     @Override
     public final <T> T[] toArray(final T[] a)
     {
         return this.allShapes.toArray(a);
     }
 
     /** {@inheritDoc} */
     @Override
     public final boolean add(final OtsShape e)
     {
         if (!this.quadTree.intersects(e))
         {
             return false;
         }
         if (this.allShapes.contains(e))
         {
             return false;
         }
         if (!this.quadTree.add(e))
         {
             CategoryLogger.always().error("add: ERROR object could not be added to the quad tree");
         }
         return this.allShapes.add(e);
     }
 
     /** {@inheritDoc} */
     @Override
     public final boolean remove(final Object o)
     {
         if (!this.allShapes.remove(o))
         {
             return false;
         }
         if (!this.quadTree.remove((OtsShape) o))
         {
             CategoryLogger.always().error("remove: ERROR object could not be removed from the quad tree");
         }
         return true;
     }
 
     /** {@inheritDoc} */
     @Override
     public final boolean containsAll(final Collection<?> c)
     {
         for (Object o : c)
         {
             if (!contains(o))
             {
                 return false;
             }
         }
         return true;
     }
 
     /** {@inheritDoc} */
     @Override
     public final boolean addAll(final Collection<? extends OtsShape> c)
     {
         boolean result = false;
         for (OtsShape s : c)
         {
             if (add(s))
             {
                 result = true;
             }
         }
         return result;
     }
 
     /** {@inheritDoc} */
     @Override
     public final boolean retainAll(final Collection<?> c)
     {
         boolean result = false;
         for (Iterator<OtsShape> it = iterator(); it.hasNext();)
         {
             OtsShape shape = it.next();
             if (!c.contains(shape))
             {
                 it.remove();
                 result = true;
             }
         }
         return result;
     }
 
     /** {@inheritDoc} */
     @Override
     public final boolean removeAll(final Collection<?> c)
     {
         boolean result = false;
         for (Iterator<OtsShape> it = iterator(); it.hasNext();)
         {
             OtsShape shape = it.next();
             if (c.contains(shape))
             {
                 it.remove();
                 result = true;
             }
         }
         return result;
     }
 
     /** {@inheritDoc} */
     @Override
     public final void clear()
     {
         this.quadTree.clear();
         this.allShapes.clear();
     }
 
     /**
      * Return the set of all shapes in this Ots2dSet that intersect the given rectangle.
      * @param rectangle Rectangle2D; the rectangle
      * @return Set&lt;OtsShape&gt;; the shapes that intersect the rectangle
      */
     public final Set<OtsShape> intersectingShapes(final Rectangle2D rectangle)
     {
         return this.quadTree.intersectingShapes(rectangle);
     }
 
     /**
      * Recursively print this Ots2dSet.
      * @param recursionDepth int; maximum depth to recurse
      * @return String
      */
     final String toString(final int recursionDepth)
     {
         return "Ots2dSet [contains " + size() + (1 == this.allShapes.size() ? "shape" : "shapes") + ", minimumCellSize="
                 + this.minimumCellSize + ", quadTree=" + this.quadTree.toString(recursionDepth) + "]";
     }
 
     /** {@inheritDoc} */
     @Override
     public final String toString()
     {
         return toString(0);
     }
 
     /**
      * Return all OtsShapes in this Ots2dSet that intersect a given OtsShape.
      * @param shape OtsShape; the given OtsShape
      * @return Set&lt;OtsShape&gt;; all OtsShapes in this Ots2dSet that intersect <code>shape</code>
      */
     public final Set<OtsShape> intersectingShapes(final OtsShape shape)
     {
         Bounds envelope = shape.getEnvelope();
         Set<OtsShape> result = intersectingShapes(
                 new Rectangle2D.Double(envelope.getMinX(), envelope.getMinY(), envelope.getDeltaX(), envelope.getDeltaY()));
         for (Iterator<OtsShape> it = result.iterator(); it.hasNext();)
         {
             if (!it.next().intersects(shape))
             {
                 it.remove();
             }
         }
         return result;
     }
 
     /**
      * Return an ASCII art rendering of this Ots2dSet.
      * @param recursionDepth int; maximum recursion depth
      * @return String; a somewhat human readable rendering of this Ots2dSet
      */
     public final String toStringGraphic(final int recursionDepth)
     {
         return this.quadTree.toStringGraphic(recursionDepth);
     }
 
     /**
      * Iterator for quad tree. Shall iterate over the local set of shapes and the (up to four) non-null leave nodes.
      */
     class QuadTreeIterator implements Iterator<OtsShape>, Serializable
     {
         /** */
         private static final long serialVersionUID = 20170400L;
 
         /** Underlying iterator that traverses the allShapes Set. */
         @SuppressWarnings("synthetic-access")
         private final Iterator<OtsShape> theIterator = Ots2dSet.this.allShapes.iterator();
 
         /** Remember the last returned result so we can remove it when requested. */
         private OtsShape lastResult = null;
 
         /** {@inheritDoc} */
         @Override
         public final boolean hasNext()
         {
             return this.theIterator.hasNext();
         }
 
         /** {@inheritDoc} */
         @Override
         public final OtsShape next()
         {
             this.lastResult = this.theIterator.next();
             return this.lastResult;
         }
 
         /** {@inheritDoc} */
         @SuppressWarnings("synthetic-access")
         @Override
         public final void remove()
         {
             this.theIterator.remove();
             if (!Ots2dSet.this.quadTree.remove(this.lastResult))
             {
                 CategoryLogger.always().error("iterator.remove: ERROR: could not remove {} from the quad tree",
                         this.lastResult);
             }
         }
 
         /** {@inheritDoc} */
         @Override
         public String toString()
         {
             return "QuadTreeIterator [theIterator=" + this.theIterator + ", lastResult=" + this.lastResult + "]";
         }
 
     }
 
     /**
      * Spatial-aware storage for a set of OtsShape objects.
      */
     class QuadTreeNode implements Serializable
     {
         /** */
         private static final long serialVersionUID = 20170400L;
 
         /** The OtsShapes stored at this node. */
         private Set<OtsShape> shapes = new LinkedHashSet<OtsShape>();
 
         /** The bounding box of this QuadTreeNode. */
         private final Rectangle2D boundingBox;
 
         /** The bounding box of this QuadTreeNode as an OtsShape. */
         private final OtsShape boundingShape;
 
         /**
          * The four leaves of this node in the quad tree. An empty sub tree may be represented by null. If this field is
          * initialized to null; this node may not expand by adding sub-nodes.
          */
         private final QuadTreeNode[] leaves;
 
         /**
          * Construct a new QuadTreeNode.
          * @param boundingBox Rectangle2D; the bounding box of the area of the new QuadTreeNode
          */
         @SuppressWarnings("synthetic-access")
         QuadTreeNode(final Rectangle2D boundingBox)
         {
             this.boundingBox = boundingBox;
             this.boundingShape = rectangleShape(boundingBox);
             this.leaves = boundingBox.getWidth() > Ots2dSet.this.minimumCellSize
                     || boundingBox.getHeight() > Ots2dSet.this.minimumCellSize ? new QuadTreeNode[4] : null;
         }
 
         /**
          * Return a Set containing all OtsShapes in this QuadTreeNode that intersect a rectangular area.
          * @param rectangle Rectangle2D; the area
          * @return Set&lt;OtsShape&gt;; the set
          */
         public Set<OtsShape> intersectingShapes(final Rectangle2D rectangle)
         {
             Set<OtsShape> result = new LinkedHashSet<OtsShape>();
             if (!this.boundingBox.intersects(rectangle))
             {
                 return result;
             }
             if (null == this.leaves)
             {
                 return result;
             }
             for (QuadTreeNode leaf : this.leaves)
             {
                 if (null != leaf && leaf.intersects(rectangle))
                 {
                     result.addAll(leaf.intersectingShapes(rectangle));
                 }
             }
             for (OtsShape shape : this.shapes)
             {
                 OtsShape rectangleShape = rectangleShape(rectangle);
                 if (rectangleShape.intersects(shape))
                 {
                     result.add(shape);
                 }
             }
             return result;
         }
 
         /**
          * Test if this QuadTreeNode intersects a rectangular area.
          * @param rectangle Rectangle2D; the rectangular area
          * @return boolean; true if the rectangular area intersects this QuadTreeNode; false otherwise
          */
         private boolean intersects(final Rectangle2D rectangle)
         {
             return this.boundingBox.intersects(rectangle);
         }
 
         /**
          * Remove all OtsShapes from this QuadTreeNode and cut off all leaves.
          */
         public void clear()
         {
             this.shapes.clear();
             for (int index = 0; index < this.leaves.length; index++)
             {
                 this.leaves[index] = null;
             }
         }
 
         /**
          * Remove an OtsShape from this QuadTreeNode.
          * @param shape OtsShape; the shape that must be removed.
          * @return boolean; true if this node (or a sub-node) was altered; false otherwise
          */
         public boolean remove(final OtsShape shape)
         {
             if (!this.boundingShape.intersects(shape))
             {
                 return false;
             }
             for (OtsShape s : this.shapes)
             {
                 if (shape.equals(s))
                 {
                     this.shapes.remove(shape);
                     return true;
                 }
             }
             boolean result = false;
             for (int index = 0; index < this.leaves.length; index++)
             {
                 QuadTreeNode qtn = this.leaves[index];
                 if (null != qtn)
                 {
                     if (qtn.remove(shape))
                     {
                         result = true;
                         if (qtn.isEmpty())
                         {
                             this.leaves[index] = null; // Cut off empty leaf node
                         }
                     }
                 }
             }
             return result;
         }
 
         /**
          * Check if this QuadTreeNode is empty.
          * @return boolean; true if this QuadTreeNode is empty
          */
         private boolean isEmpty()
         {
             if (!this.shapes.isEmpty())
             {
                 return false;
             }
             if (null == this.leaves)
             {
                 return true;
             }
             for (QuadTreeNode qtn : this.leaves)
             {
                 if (null != qtn)
                 {
                     return false;
                 }
             }
             return true;
         }
 
         /**
          * Test if the area of this QuadTree intersects an OtsShape.
          * @param shape OtsShape; the shape
          * @return boolean; true if the area of this QuadTree intersects the shape; false otherwise
          */
         public boolean intersects(final OtsShape shape)
         {
             return this.boundingShape.intersects(shape);
         }
 
         /**
          * Construct a OtsShape from a Rectangle2D.
          * @param rectangle Rectangle2D; the rectangle
          * @return OtsShape; a new OtsShape
          */
         private OtsShape rectangleShape(final Rectangle2D rectangle)
         {
             double left = rectangle.getMinX();
             double bottom = rectangle.getMinY();
             double right = rectangle.getMaxX();
             double top = rectangle.getMaxY();
             try
             {
                 return new OtsShape(new OtsPoint3d(left, bottom), new OtsPoint3d(right, bottom), new OtsPoint3d(right, top),
                         new OtsPoint3d(left, top));
             }
             catch (OtsGeometryException exception)
             {
                 CategoryLogger.always().error(exception);
                 return null;
             }
         }
 
         /**
          * Add an OtsShape to this QuadTreeNode.
          * @param shape OtsShape; the shape
          * @return boolean; true if this QuadTreeNode changed as a result of this operation
          */
         public final boolean add(final OtsShape shape)
         {
             if (!this.boundingShape.intersects(shape))
             {
                 return false;
             }
             if ((null == this.leaves) || shape.contains(this.boundingBox))
             {
                 // shape belongs in the set of shapes of this node.
                 return this.shapes.add(shape);
             }
             // This node may have leaves and shape does not entirely contain this node. Add shape to all applicable leaves.
             boolean result = false;
             for (int index = 0; index < this.leaves.length; index++)
             {
                 if (null == this.leaves[index])
                 {
                     double subWidth = this.boundingBox.getWidth() / 2;
                     double subHeight = this.boundingBox.getHeight() / 2;
                     if (0 == subWidth)
                     {
                         // loss of precision; degenerate into a binary tree
                         subWidth = this.boundingBox.getWidth();
                     }
                     if (0 == subHeight)
                     {
                         // loss of precision; degenerate into a binary tree
                         subHeight = this.boundingBox.getHeight();
                     }
                     double left = this.boundingBox.getMinX();
                     if (0 != index / 2)
                     {
                         left += subWidth;
                     }
                     double bottom = this.boundingBox.getMinY();
                     if (0 != index % 2)
                     {
                         bottom += subHeight;
                     }
                     Rectangle2D subBox = new Rectangle2D.Double(left, bottom, subWidth, subHeight);
                     if (rectangleShape(subBox).intersects(shape))
                     {
                         // Expand this node by adding a sub node.
                         this.leaves[index] = new QuadTreeNode(subBox);
                         if (this.leaves[index].add(shape))
                         {
                             result = true;
                         }
                         else
                         {
                             throw new Error("Cannot happen: new QuadTreeNode refused to add shape that intersects it");
                         }
                     }
                 }
                 else
                 {
                     // Leaf node already exists. Let the leaf determine if shape should be stored (somewhere) in it.
                     if (this.leaves[index].add(shape))
                     {
                         result = true;
                     }
                 }
             }
             return result;
         }
 
         /**
          * Helper function for toString.
          * @param recursionDepth int; maximum number of levels to print recursively
          * @param index int; index in leaves
          * @return String
          */
         private String printLeaf(final int recursionDepth, final int index)
         {
             QuadTreeNode leaf = this.leaves[index];
             if (null == leaf)
             {
                 return "null";
             }
             if (recursionDepth > 0)
             {
                 return leaf.toString(recursionDepth - 1);
             }
             int leafSize = leaf.shapes.size();
             return leafSize + " shape" + (1 == leafSize ? "" : "s");
         }
 
         /**
          * Recursively print this QuadTreeNode.
          * @param recursionDepth int; maximum depth to recurse
          * @return String
          */
         final String toString(final int recursionDepth)
         {
             return "QuadTreeNode [" + this.shapes.size() + ", bounds=[LB: " + this.boundingBox.getMinX() + ","
                     + this.boundingBox.getMinY() + ", RT: " + this.boundingBox.getMaxX() + "," + this.boundingBox.getMaxY()
                     + "], " + subNodes(recursionDepth) + ", local " + this.shapes.size()
                     + (1 == this.shapes.size() ? " shape" : " shapes") + "]";
         }
 
         /**
          * Print the leaves of this QuadTreeNode.
          * @param recursionDepth int; maximum depth to recurse
          * @return String
          */
         private String subNodes(final int recursionDepth)
         {
             if (null == this.leaves)
             {
                 return "cannot have leaves";
             }
             return "leaves=[LB: " + printLeaf(recursionDepth, 0) + ", RB: " + printLeaf(recursionDepth, 1) + ", LT: "
                     + printLeaf(recursionDepth, 2) + ", RT: " + printLeaf(recursionDepth, 3) + "]";
         }
 
         /** {@inheritDoc} */
         @Override
         public final String toString()
         {
             return toString(0);
         }
 
         /**
          * Return concatenation of a number of copies of a string.
          * @param count int; number of copies to concatenate
          * @param string String; the string to repeat
          * @return String
          */
         private String repeat(final int count, final String string)
         {
             StringBuilder result = new StringBuilder();
             for (int i = 0; i < count; i++)
             {
                 result.append(string);
             }
             return result.toString();
         }
 
         /** Graphic to draw a vertical line. */
         private static final String VLINE = "|";
 
         /** Graphic to draw a horizontal line. */
         private static final String HLINE = "-";
 
         /** Graphic to draw a space. */
         private static final String SPACE = " ";
 
         /** Number of digits to print. */
         private static final int NUMBERSIZE = 6;
 
         /**
          * Similar to toStringGraphic, but with QuadTreeNode argument which can be null. <br>
          * This code is <b>not</b> optimized for performance; the repeated use of String.split is probably expensive.
          * @param qtn QuadTreeNode; the QuadTreeNode to render. Can be null.
          * @param recursionDepth int; levels to recurse
          * @return String
          */
         private String subStringGraphic(final QuadTreeNode qtn, final int recursionDepth)
         {
             StringBuffer result = new StringBuffer();
             if (0 == recursionDepth)
             {
                 if (null == qtn)
                 {
                     result.append(repeat(NUMBERSIZE, SPACE));
                 }
                 else
                 {
                     String numberBuf = String.format("%d", size());
                     int spare = NUMBERSIZE - numberBuf.length();
                     int filled = 0;
                     while (filled < spare / 2)
                     {
                         result.append(SPACE);
                         filled++;
                     }
                     result.append(numberBuf);
                     while (filled < spare)
                     {
                         result.append(SPACE);
                         filled++;
                     }
                     result.append("\n");
                     return result.toString();
                 }
             }
             else
             {
                 String[] left = subStringGraphic(null == qtn || null == qtn.leaves ? null : qtn.leaves[1], recursionDepth - 1)
                         .split("\\n");
                 String[] right = subStringGraphic(null == qtn || null == qtn.leaves ? null : qtn.leaves[3], recursionDepth - 1)
                         .split("\\n");
                 String horizontalLine = null;
                 for (int i = 0; i < left.length; i++)
                 {
                     if (0 == i)
                     {
                         StringBuilder line = new StringBuilder();
                         int width = left[0].length() + 1 + right[0].length();
                         if (null == qtn)
                         {
                             line.append(repeat(width, SPACE));
                         }
                         else
                         {
                             String numberBuf = String.format("%d", qtn.shapes.size());
                             int spare = width - numberBuf.length();
                             line.append(repeat(spare / 2, HLINE));
                             line.append(numberBuf);
                             line.append(repeat(spare - spare / 2, HLINE));
                         }
                         horizontalLine = line.toString();
                     }
                     result.append(left[i]);
                     result.append(null == qtn ? SPACE : VLINE);
                     result.append(right[i]);
                     result.append("\n");
                 }
                 result.append(horizontalLine);
                 result.append("\n");
                 left = subStringGraphic(null == qtn || null == qtn.leaves ? null : qtn.leaves[0], recursionDepth - 1)
                         .split("\\n");
                 right = subStringGraphic(null == qtn || null == qtn.leaves ? null : qtn.leaves[2], recursionDepth - 1)
                         .split("\\n");
                 for (int i = 0; i < left.length; i++)
                 {
                     result.append(left[i]);
                     result.append(null == qtn ? SPACE : VLINE);
                     result.append(right[i]);
                     result.append("\n");
                 }
                 result.append("\n");
             }
             return result.toString();
         }
 
         /**
          * Return a String depicting this QuadTreeNode.
          * @param recursionDepth int; levels to recurse
          * @return String
          */
         public final String toStringGraphic(final int recursionDepth)
         {
             return subStringGraphic(this, recursionDepth);
         }
 
     }
 
 }