package org.opentrafficsim.road.gtu.colorer;
   
   import java.awt.Color;
   import java.io.Serializable;
   import java.util.ArrayList;
   import java.util.Collections;
   import java.util.LinkedHashMap;
   import java.util.List;
   import java.util.Map;
  
  import org.opentrafficsim.core.animation.gtu.colorer.GTUColorer;
  import org.opentrafficsim.core.animation.gtu.colorer.IDGTUColorer;
  import org.opentrafficsim.core.gtu.GTU;
  import org.opentrafficsim.road.gtu.lane.perception.LanePerception;
  import org.opentrafficsim.road.gtu.lane.perception.mental.Fuller;
  import org.opentrafficsim.road.gtu.lane.perception.mental.Fuller.IdentifiableTask;
  import org.opentrafficsim.road.gtu.lane.perception.mental.Fuller.Task;
  import org.opentrafficsim.road.gtu.lane.perception.mental.Mental;
  import org.opentrafficsim.road.gtu.lane.perception.mental.sdm.DefaultDistraction;
  
  /**
   * <p>
   * Copyright (c) 2013-2019 Delft University of Technology, PO Box 5, 2600 AA, Delft, the Netherlands. All rights reserved. <br>
   * BSD-style license. See <a href="http://opentrafficsim.org/node/13">OpenTrafficSim License</a>.
   * <p>
   * @version $Revision$, $LastChangedDate$, by $Author$, initial version 6 nov. 2018 <br>
   * @author <a href="http://www.tbm.tudelft.nl/averbraeck">Alexander Verbraeck</a>
   * @author <a href="http://www.tudelft.nl/pknoppers">Peter Knoppers</a>
   * @author <a href="http://www.transport.citg.tudelft.nl">Wouter Schakel</a>
   */
  
  public class DistractionColorer implements GTUColorer, Serializable
  {
      /** */
      private static final long serialVersionUID = 20181106L;
  
      /** None color. */
      private static final Color NONE = Color.MAGENTA.darker().darker();
  
      /** The legend. */
      private final ArrayList<LegendEntry> legend = new ArrayList<>();
  
      /** Colors. */
      private final Map<String, Color> colors = new LinkedHashMap<>();
  
      /**
       * Constructor.
       * @param distractions DefaultDistraction...; DefaultDistraction... distractions to color
       */
      public DistractionColorer(final DefaultDistraction... distractions)
      {
          this.legend.add(new LegendEntry(NONE, "None", "No distraction"));
          for (int i = 0; i < distractions.length; i++)
          {
              Color c = IDGTUColorer.LEGEND.get(i % 10).getColor();
              this.colors.put(distractions[i].getId(), c);
              this.legend.add(new LegendEntry(c, distractions[i].getDescription(), distractions[i].getDescription()));
          }
      }
  
      /** {@inheritDoc} */
      @Override
      public Color getColor(final GTU gtu)
      {
          if (gtu.getTacticalPlanner().getPerception() instanceof LanePerception)
          {
              Mental mental = ((LanePerception) gtu.getTacticalPlanner().getPerception()).getMental();
              if (mental != null && mental instanceof Fuller)
              {
                  for (Task task : ((Fuller) mental).getTasks())
                  {
                      if (task instanceof IdentifiableTask)
                      {
                          String id = ((IdentifiableTask) task).getId();
                          if (this.colors.containsKey(id))
                          {
                              return this.colors.get(id);
                          }
                      }
                  }
              }
          }
          return NONE;
      }
  
      /** {@inheritDoc} */
      @Override
      public final List<LegendEntry> getLegend()
      {
          return Collections.unmodifiableList(this.legend);
      }
  
      /** {@inheritDoc} */
      @Override
      public final String toString()
      {
          return "Distraction";
      }
  
 }