package org.opentrafficsim.web.animation.d2;
   
   import java.awt.Color;
   import java.awt.geom.AffineTransform;
   import java.rmi.RemoteException;
   import java.util.ArrayList;
   import java.util.LinkedHashMap;
   import java.util.LinkedHashSet;
   import java.util.List;
  import java.util.Map;
  import java.util.Set;
  import java.util.SortedSet;
  import java.util.TreeSet;
  
  import org.djutils.draw.bounds.Bounds;
  import org.djutils.draw.bounds.Bounds2d;
  import org.djutils.draw.point.Point;
  import org.djutils.event.Event;
  import org.djutils.event.EventListener;
  import org.opentrafficsim.animation.gtu.colorer.GtuColorerManager;
  import org.opentrafficsim.base.logger.Logger;
  import org.opentrafficsim.core.dsol.OtsSimulatorInterface;
  import org.opentrafficsim.web.animation.HtmlGraphics2d;
  
  import nl.tudelft.simulation.dsol.animation.Locatable;
  import nl.tudelft.simulation.dsol.animation.d2.Renderable2dComparator;
  import nl.tudelft.simulation.dsol.animation.d2.Renderable2dInterface;
  import nl.tudelft.simulation.dsol.animation.gis.GisMapInterface;
  import nl.tudelft.simulation.dsol.animation.gis.GisRenderable2d;
  import nl.tudelft.simulation.dsol.experiment.Replication;
  import nl.tudelft.simulation.naming.context.ContextInterface;
  import nl.tudelft.simulation.naming.context.util.ContextUtil;
  
  /**
   * The AnimationPanel to display animated (Locatable) objects. Added the possibility to witch layers on and off. By default all
   * layers will be drawn, so no changes to existing software need to be made.
   * <p>
   * Copyright (c) 2003-2024 Delft University of Technology, Jaffalaan 5, 2628 BX Delft, the Netherlands. All rights reserved.
   * BSD-style license. See <a href="https://opentrafficsim.org/docs/v2/license.html">OpenTrafficSim License</a>.
   * </p>
   * @author <a href="https://github.com/averbraeck">Alexander Verbraeck</a>
   */
  public class HtmlAnimationPanel extends HtmlGridPanel implements EventListener
  {
      /** the elements of this panel. */
      private SortedSet<Renderable2dInterface<? extends Locatable>> elements =
              new TreeSet<Renderable2dInterface<? extends Locatable>>(new Renderable2dComparator());
  
      /** filter for types to be shown or not. */
      private Map<Class<? extends Locatable>, Boolean> visibilityMap = new LinkedHashMap<>();
  
      /** cache of the classes that are hidden. */
      private Set<Class<? extends Locatable>> hiddenClasses = new LinkedHashSet<>();
  
      /** cache of the classes that are shown. */
      private Set<Class<? extends Locatable>> shownClasses = new LinkedHashSet<>();
  
      /** the simulator. */
      private OtsSimulatorInterface simulator;
  
      /** the eventContext. */
      private ContextInterface context = null;
  
      /** a line that helps the user to see where s/he is dragging. */
      private int[] dragLine = new int[4];
  
      /** enable drag line. */
      private boolean dragLineEnabled = false;
  
      /** List of drawable objects. */
      private List<Renderable2dInterface<? extends Locatable>> elementList = new ArrayList<>();
  
      /** dirty flag for the list. */
      private boolean dirtyElements = false;
  
      /** Map of toggle names to toggle animation classes. */
      private Map<String, Class<? extends Locatable>> toggleLocatableMap = new LinkedHashMap<>();
  
      /** Set of animation classes to toggle buttons. */
      private Map<Class<? extends Locatable>, ToggleButtonInfo> toggleButtonMap = new LinkedHashMap<>();
  
      /** Set of GIS layer names to toggle GIS layers . */
      private Map<String, GisMapInterface> toggleGISMap = new LinkedHashMap<>();
  
      /** Set of GIS layer names to toggle buttons. */
      private Map<String, ToggleButtonInfo> toggleGISButtonMap = new LinkedHashMap<>();
  
      /** List of buttons in the right order. */
      private List<ToggleButtonInfo> toggleButtons = new ArrayList<>();
  
      /** GTU colorer manager. */
      private final GtuColorerManager gtuColorerManager = new GtuColorerManager(Color.WHITE);
  
      /** the margin factor 'around' the extent. */
      public static final double EXTENT_MARGIN_FACTOR = 0.05;
  
      /**
       * constructs a new AnimationPanel.
       * @param homeExtent the extent of the panel
      * @param simulator the simulator of which we want to know the events for animation
      * @throws RemoteException on network error for one of the listeners
      */
     public HtmlAnimationPanel(final Bounds2d homeExtent, final OtsSimulatorInterface simulator) throws RemoteException
     {
         super(homeExtent);
         super.showGrid = true;
         this.simulator = simulator;
         simulator.addListener(this, Replication.START_REPLICATION_EVENT);
     }
 
     /**
      * Returns the GTU colorer manager.
      * @return GTU colorer manager
      */
     public GtuColorerManager getGtuColorerManager()
     {
         return this.gtuColorerManager;
     }
 
     @Override
     public void paintComponent(final HtmlGraphics2d g2)
     {
         // draw the grid.
         super.paintComponent(g2);
 
         // update drawable elements when necessary
         if (this.dirtyElements)
         {
             synchronized (this.elementList)
             {
                 this.elementList.clear();
                 this.elementList.addAll(this.elements);
                 this.dirtyElements = false;
             }
         }
 
         // draw the animation elements.
         for (Renderable2dInterface<? extends Locatable> element : this.elementList)
         {
             // destroy has been called?
             if (element.getSource() == null)
             {
                 objectRemoved(element);
             }
             else if (isShowElement(element))
             {
                 AffineTransform at = (AffineTransform) g2.getTransform().clone();
                 element.paintComponent(g2, this.getExtent(), this.getSize(), this.renderableScale, this);
                 g2.setTransform(at);
             }
         }
 
         // draw drag line if enabled.
         if (this.dragLineEnabled)
         {
             g2.setColor(Color.BLACK);
             g2.drawLine(this.dragLine[0], this.dragLine[1], this.dragLine[2], this.dragLine[3]);
             this.dragLineEnabled = false;
         }
     }
 
     /**
      * Test whether the element needs to be shown on the screen or not.
      * @param element the renderable element to test
      * @return whether the element needs to be shown or not
      */
     public boolean isShowElement(final Renderable2dInterface<? extends Locatable> element)
     {
         return element.getSource() == null ? false : isShowClass(element.getSource().getClass());
     }
 
     /**
      * Test whether a certain class needs to be shown on the screen or not. The class needs to implement Locatable, otherwise it
      * cannot be shown at all.
      * @param locatableClass the class to test
      * @return whether the class needs to be shown or not
      */
     public boolean isShowClass(final Class<? extends Locatable> locatableClass)
     {
         if (this.hiddenClasses.contains(locatableClass))
         {
             return false;
         }
         else
         {
             boolean show = true;
             if (!this.shownClasses.contains(locatableClass))
             {
                 for (Class<? extends Locatable> lc : this.visibilityMap.keySet())
                 {
                     if (lc.isAssignableFrom(locatableClass))
                     {
                         if (!this.visibilityMap.get(lc))
                         {
                             show = false;
                         }
                     }
                 }
                 // add to the right cache
                 if (show)
                 {
                     this.shownClasses.add(locatableClass);
                 }
                 else
                 {
                     this.hiddenClasses.add(locatableClass);
                 }
             }
             return show;
         }
     }
 
     @SuppressWarnings("unchecked")
     @Override
     public void notify(final Event event)
     {
         if (event.getType().equals(ContextInterface.OBJECT_ADDED_EVENT))
         {
             objectAdded((Renderable2dInterface<? extends Locatable>) ((Object[]) event.getContent())[2]);
         }
 
         else if (event.getType().equals(ContextInterface.OBJECT_REMOVED_EVENT))
         {
             objectRemoved((Renderable2dInterface<? extends Locatable>) ((Object[]) event.getContent())[2]);
         }
 
         else if // (this.simulator.getSourceId().equals(event.getSourceId()) &&
         (event.getType().equals(Replication.START_REPLICATION_EVENT))
         {
             synchronized (this.elementList)
             {
                 this.elements.clear();
                 try
                 {
                     if (this.context != null)
                     {
                         this.context.removeListener(this, ContextInterface.OBJECT_ADDED_EVENT);
                         this.context.removeListener(this, ContextInterface.OBJECT_REMOVED_EVENT);
                     }
 
                     this.context =
                             ContextUtil.lookupOrCreateSubContext(this.simulator.getReplication().getContext(), "animation/2D");
                     this.context.addListener(this, ContextInterface.OBJECT_ADDED_EVENT);
                     this.context.addListener(this, ContextInterface.OBJECT_REMOVED_EVENT);
                     for (Object element : this.context.values())
                     {
                         objectAdded((Renderable2dInterface<? extends Locatable>) element);
                     }
                     this.repaint();
                 }
                 catch (Exception exception)
                 {
                     Logger.ots().warn(exception, "notify");
                 }
             }
         }
     }
 
     /**
      * Add a locatable object to the animation.
      * @param element the element to add to the animation
      */
     public void objectAdded(final Renderable2dInterface<? extends Locatable> element)
     {
         synchronized (this.elementList)
         {
             this.elements.add(element);
             this.dirtyElements = true;
         }
     }
 
     /**
      * Remove a locatable object from the animation.
      * @param element the element to add to the animation
      */
     public void objectRemoved(final Renderable2dInterface<? extends Locatable> element)
     {
         synchronized (this.elementList)
         {
             this.elements.remove(element);
             this.dirtyElements = true;
         }
     }
 
     /**
      * Calculate the full extent based on the current positions of the objects.
      * @return the full extent of the animation.
      */
     public synchronized Bounds2d fullExtent()
     {
         double minX = Double.MAX_VALUE;
         double maxX = -Double.MAX_VALUE;
         double minY = Double.MAX_VALUE;
         double maxY = -Double.MAX_VALUE;
         try
         {
             for (Renderable2dInterface<? extends Locatable> renderable : this.elementList)
             {
                 if (renderable.getSource() == null)
                 {
                     continue;
                 }
                 Point<?> l = renderable.getSource().getLocation();
                 if (l != null)
                 {
                     Bounds<?, ?> b = renderable.getSource().getRelativeBounds();
                     minX = Math.min(minX, l.getX() + b.getMinX());
                     minY = Math.min(minY, l.getY() + b.getMinY());
                     maxX = Math.max(maxX, l.getX() + b.getMaxX());
                     maxY = Math.max(maxY, l.getY() + b.getMaxY());
                 }
             }
         }
         catch (Exception e)
         {
             // ignore
         }
 
         minX -= EXTENT_MARGIN_FACTOR * Math.abs(maxX - minX);
         minY -= EXTENT_MARGIN_FACTOR * Math.abs(maxY - minY);
         maxX += EXTENT_MARGIN_FACTOR * Math.abs(maxX - minX);
         maxY += EXTENT_MARGIN_FACTOR * Math.abs(maxY - minY);
 
         return new Bounds2d(minX, maxX, minY, maxY);
     }
 
     /**
      * resets the panel to its an extent that covers all displayed objects.
      */
     public synchronized void zoomAll()
     {
         setExtent(computeVisibleExtent(fullExtent()).get());
         this.repaint();
     }
 
     /**
      * Set a class to be shown in the animation to true.
      * @param locatableClass the class for which the animation has to be shown.
      */
     public void showClass(final Class<? extends Locatable> locatableClass)
     {
         this.visibilityMap.put(locatableClass, true);
         this.shownClasses.clear();
         this.hiddenClasses.clear();
         this.repaint();
     }
 
     /**
      * Set a class to be hidden in the animation to true.
      * @param locatableClass the class for which the animation has to be hidden.
      */
     public void hideClass(final Class<? extends Locatable> locatableClass)
     {
         this.visibilityMap.put(locatableClass, false);
         this.shownClasses.clear();
         this.hiddenClasses.clear();
         this.repaint();
     }
 
     /**
      * Toggle a class to be displayed in the animation to its reverse value.
      * @param locatableClass the class for which a visible animation has to be turned off or vice versa.
      */
     public void toggleClass(final Class<? extends Locatable> locatableClass)
     {
         if (!this.visibilityMap.containsKey(locatableClass))
         {
             showClass(locatableClass);
         }
         this.visibilityMap.put(locatableClass, !this.visibilityMap.get(locatableClass));
         this.shownClasses.clear();
         this.hiddenClasses.clear();
         this.repaint();
     }
 
     /**
      * Return animation elements.
      * @return the set of animation elements.
      */
     public final SortedSet<Renderable2dInterface<? extends Locatable>> getElements()
     {
         return this.elements;
     }
 
     /**
      * Returns drag line.
      * @return returns the dragLine.
      */
     public final int[] getDragLine()
     {
         return this.dragLine;
     }
 
     /**
      * Returns drag line enabled.
      * @return returns the dragLineEnabled.
      */
     public final boolean isDragLineEnabled()
     {
         return this.dragLineEnabled;
     }
 
     /**
      * Sets drag line enabled.
      * @param dragLineEnabled the dragLineEnabled to set.
      */
     public final void setDragLineEnabled(final boolean dragLineEnabled)
     {
         this.dragLineEnabled = dragLineEnabled;
     }
 
     /**********************************************************************************************************/
     /******************************************* TOGGLES ******************************************************/
     /**********************************************************************************************************/
 
     /**
      * Add a button for toggling an animatable class on or off.
      * @param name the name of the button
      * @param locatableClass the class for which the button holds (e.g., GTU.class)
      * @param toolTipText the tool tip text to show when hovering over the button
      * @param initiallyVisible whether the class is initially shown or not
      */
     public final void addToggleAnimationButtonText(final String name, final Class<? extends Locatable> locatableClass,
             final String toolTipText, final boolean initiallyVisible)
     {
         ToggleButtonInfo.LocatableClass buttonInfo =
                 new ToggleButtonInfo.LocatableClass(name, locatableClass, toolTipText, initiallyVisible);
         if (initiallyVisible)
         {
             showClass(locatableClass);
         }
         else
         {
             hideClass(locatableClass);
         }
         this.toggleButtons.add(buttonInfo);
         this.toggleLocatableMap.put(name, locatableClass);
         this.toggleButtonMap.put(locatableClass, buttonInfo);
     }
 
     /**
      * Show a Locatable class based on the name.
      * @param name the name of the class to show
      */
     public final void showClass(final String name)
     {
         showClass(this.toggleLocatableMap.get(name));
     }
 
     /**
      * Hide a Locatable class based on the name.
      * @param name the name of the class to hide
      */
     public final void hideClass(final String name)
     {
         hideClass(this.toggleLocatableMap.get(name));
     }
 
     /**
      * Add a text to explain animatable classes.
      * @param text the text to show
      */
     public final void addToggleText(final String text)
     {
         this.toggleButtons.add(new ToggleButtonInfo.Text(text, true));
     }
 
     /**
      * Add buttons for toggling all GIS layers on or off.
      * @param header the name of the group of layers
      * @param gisMap the GIS map for which the toggles have to be added
      * @param toolTipText the tool tip text to show when hovering over the button
      */
     public final void addAllToggleGISButtonText(final String header, final GisRenderable2d gisMap, final String toolTipText)
     {
         addToggleText(" ");
         addToggleText(header);
         for (String layerName : gisMap.getMap().getLayerMap().keySet())
         {
             addToggleGISButtonText(layerName, layerName, gisMap, toolTipText);
         }
     }
 
     /**
      * Add a button to toggle a GIS Layer on or off.
      * @param layerName the name of the layer
      * @param displayName the name to display next to the tick box
      * @param gisMap the map
      * @param toolTipText the tool tip text
      */
     public final void addToggleGISButtonText(final String layerName, final String displayName, final GisRenderable2d gisMap,
             final String toolTipText)
     {
         ToggleButtonInfo.Gis buttonInfo = new ToggleButtonInfo.Gis(displayName, layerName, toolTipText, true);
         this.toggleButtons.add(buttonInfo);
         this.toggleGISMap.put(layerName, gisMap.getMap());
         this.toggleGISButtonMap.put(layerName, buttonInfo);
     }
 
     /**
      * Set a GIS layer to be shown in the animation to true.
      * @param layerName the name of the GIS-layer that has to be shown.
      */
     public final void showGISLayer(final String layerName)
     {
         GisMapInterface gisMap = this.toggleGISMap.get(layerName);
         if (gisMap != null)
         {
             gisMap.showLayer(layerName);
             this.toggleGISButtonMap.get(layerName).setVisible(true);
         }
     }
 
     /**
      * Set a GIS layer to be hidden in the animation to true.
      * @param layerName the name of the GIS-layer that has to be hidden.
      */
     public final void hideGISLayer(final String layerName)
     {
         GisMapInterface gisMap = this.toggleGISMap.get(layerName);
         if (gisMap != null)
         {
             gisMap.hideLayer(layerName);
             this.toggleGISButtonMap.get(layerName).setVisible(false);
         }
     }
 
     /**
      * Toggle a GIS layer to be displayed in the animation to its reverse value.
      * @param layerName the name of the GIS-layer that has to be turned off or vice versa.
      */
     public final void toggleGISLayer(final String layerName)
     {
         GisMapInterface gisMap = this.toggleGISMap.get(layerName);
         if (gisMap != null)
         {
             if (gisMap.getVisibleLayers().contains(gisMap.getLayerMap().get(layerName)))
             {
                 gisMap.hideLayer(layerName);
                 this.toggleGISButtonMap.get(layerName).setVisible(false);
             }
             else
             {
                 gisMap.showLayer(layerName);
                 this.toggleGISButtonMap.get(layerName).setVisible(true);
             }
         }
     }
 
     /**
      * Returns toggle buttons.
      * @return toggleButtons
      */
     public final List<ToggleButtonInfo> getToggleButtons()
     {
         return this.toggleButtons;
     }
 
 }