package org.opentrafficsim.demo.lanechange;

import java.awt.BasicStroke;
import java.awt.BorderLayout;
import java.awt.Color;
import java.rmi.RemoteException;
import java.util.ArrayList;

import javax.naming.NamingException;
import javax.swing.JPanel;
import javax.swing.SwingUtilities;
import javax.swing.event.EventListenerList;

import nl.tudelft.simulation.dsol.SimRuntimeException;
import nl.tudelft.simulation.dsol.gui.swing.TablePanel;
import nl.tudelft.simulation.dsol.simulators.SimulatorInterface;

import org.jfree.chart.ChartFactory;
import org.jfree.chart.ChartPanel;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.StandardChartTheme;
import org.jfree.chart.axis.NumberAxis;
import org.jfree.chart.plot.PlotOrientation;
import org.jfree.chart.plot.XYPlot;
import org.jfree.chart.renderer.xy.XYLineAndShapeRenderer;
import org.jfree.data.DomainOrder;
import org.jfree.data.general.DatasetChangeEvent;
import org.jfree.data.general.DatasetChangeListener;
import org.jfree.data.general.DatasetGroup;
import org.jfree.data.xy.XYDataset;
import org.opentrafficsim.core.dsol.OTSModelInterface;
import org.opentrafficsim.core.dsol.OTSSimTimeDouble;
import org.opentrafficsim.core.geometry.OTSGeometryException;
import org.opentrafficsim.core.geometry.OTSPoint3D;
import org.opentrafficsim.core.gtu.GTUType;
import org.opentrafficsim.core.network.NetworkException;
import org.opentrafficsim.core.network.OTSNode;
import org.opentrafficsim.core.network.factory.LaneFactory;
import org.opentrafficsim.core.network.lane.Lane;
import org.opentrafficsim.core.network.lane.LaneType;
import org.opentrafficsim.core.network.route.CompleteRoute;
import org.opentrafficsim.core.network.route.CompleteLaneBasedRouteNavigator;
import org.opentrafficsim.core.unit.LengthUnit;
import org.opentrafficsim.core.unit.SpeedUnit;
import org.opentrafficsim.core.unit.TimeUnit;
import org.opentrafficsim.core.value.vdouble.scalar.DoubleScalar;
import org.opentrafficsim.core.value.vdouble.scalar.DoubleScalar.Abs;
import org.opentrafficsim.core.value.vdouble.scalar.DoubleScalar.Rel;
import org.opentrafficsim.gui.SimulatorFrame;
import org.opentrafficsim.simulationengine.SimpleSimulator;

/**
 * <p>
 * Copyright (c) 2013-2015 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/docs/license.html">OpenTrafficSim License</a>.
 * <p>
 * $LastChangedDate$, @version $Revision$, by $Author$,
 * initial version 15 apr. 2015 <br>
 * @author <a href="http://www.tudelft.nl/pknoppers">Peter Knoppers</a>
 */
public class SuitabilityGraph implements OTSModelInterface
{
    /** */
    private static final long serialVersionUID = 20150415L;

    /** The JPanel that contains all the graphs. */
    private JPanel graphPanel;

    /** Number of lanes on the main roadway (do not set higher than size of colorTable). */
    private static final int LANECOUNT = 4;

    /** Speed limit values in km/h. */
    private static final double[] SPEEDLIMITS = {30, 50, 80, 120};

    /** Arrangements of lanes to aim for. Negative numbers indicate lanes on right side of the roadway. */
    private static final int[] TARGETLANES = {1, 2, -2, -1};

    /** Time horizon for lane changes. */
    private DoubleScalar.Rel<TimeUnit> timeHorizon = new DoubleScalar.Rel<TimeUnit>(100, TimeUnit.SECOND);

    /** Time range for graphs (also adjusts distance range). */
    private DoubleScalar.Rel<TimeUnit> timeRange = new DoubleScalar.Rel<TimeUnit>(110, TimeUnit.SECOND);

    /** Colors that correspond to the lanes; taken from electrical resistor color codes. */
    private static final Color[] COLORTABLE = {new Color(160, 82, 45) /* brown */, Color.RED, Color.ORANGE, Color.YELLOW,
        Color.GREEN, Color.BLUE, new Color(199, 21, 133) /* violet */, Color.GRAY, Color.WHITE};

    /** The graphs. */
    private JFreeChart[][] charts;

    /**
     * Start the program.
     * @param args String[]; command line arguments (not used)
     * @throws RemoteException on communications failure
     * @throws SimRuntimeException should never happen
     */
    public static void main(final String[] args) throws RemoteException, SimRuntimeException
    {
        SwingUtilities.invokeLater(new Runnable()
        {
            @Override
            public void run()
            {
                SuitabilityGraph suitabilityGraph = new SuitabilityGraph();
                new SimulatorFrame("Suitability graph", suitabilityGraph.getPanel());
                try
                {
                    suitabilityGraph.drawPlots();
                }
                catch (RemoteException | NamingException | NetworkException | SimRuntimeException | OTSGeometryException exception)
                {
                    exception.printStackTrace();
                }
            }
        });
    }

    /**
     * Draw the plots.
     * @throws NetworkException on network inconsistency
     * @throws NamingException on ???
     * @throws RemoteException on communications failure
     * @throws SimRuntimeException on ???
     * @throws OTSGeometryException
     */
    protected final void drawPlots() throws RemoteException, NamingException, NetworkException, SimRuntimeException,
        OTSGeometryException
    {
        SimpleSimulator simulator =
            new SimpleSimulator(new DoubleScalar.Abs<TimeUnit>(0, TimeUnit.SI), new DoubleScalar.Rel<TimeUnit>(0,
                TimeUnit.SI), new DoubleScalar.Rel<TimeUnit>(99999, TimeUnit.SI), this);
        final int rows = SPEEDLIMITS.length;
        final int columns = TARGETLANES.length;
        for (int row = 0; row < rows; row++)
        {
            int targetLaneConfiguration = TARGETLANES[row];
            for (int column = 0; column < columns; column++)
            {
                DoubleScalar.Abs<SpeedUnit> speedLimit =
                    new DoubleScalar.Abs<SpeedUnit>(SPEEDLIMITS[column], SpeedUnit.KM_PER_HOUR);
                double mainLength = speedLimit.getSI() * this.timeRange.getSI();
                OTSNode<String> from = new OTSNode<String>("From", new OTSPoint3D(-mainLength, 0, 0));
                OTSNode<String> branchPoint = new OTSNode<String>("From", new OTSPoint3D(0, 0, 0));
                LaneType<String> laneType = new LaneType<String>("CarLane");
                GTUType<String> gtuType = GTUType.makeGTUType("Car");
                laneType.addCompatibility(gtuType);
                Lane[] lanes =
                    LaneFactory.makeMultiLane("Test road", from, branchPoint, null, LANECOUNT, laneType, speedLimit,
                        simulator);
                OTSNode<String> destination =
                    new OTSNode<String>("Destination", new OTSPoint3D(1000, targetLaneConfiguration > 0 ? 100 : -100, 0));
                LaneFactory.makeMultiLane("DestinationLink", branchPoint, destination, null, Math
                    .abs(targetLaneConfiguration), targetLaneConfiguration > 0 ? 0 : LANECOUNT + targetLaneConfiguration, 0,
                    laneType, speedLimit, simulator);
                OTSNode<String> nonDestination =
                    new OTSNode<String>("Non-Destination", new OTSPoint3D(1000, targetLaneConfiguration > 0 ? -100 : 100, 0));
                LaneFactory.makeMultiLane("Non-DestinationLink", branchPoint, nonDestination, null, LANECOUNT
                    - Math.abs(targetLaneConfiguration), targetLaneConfiguration > 0 ? LANECOUNT - targetLaneConfiguration
                    : 0, 0, laneType, speedLimit, simulator);
                CompleteRoute route = new CompleteRoute("route");
                route.addNode(from);
                route.addNode(branchPoint);
                route.addNode(destination);
                CompleteLaneBasedRouteNavigator navigator = new CompleteLaneBasedRouteNavigator(route);
                SuitabilityData dataset = (SuitabilityData) ((XYPlot) (this.charts[row][column].getPlot())).getDataset();
                for (int laneIndex = 0; laneIndex < LANECOUNT; laneIndex++)
                {
                    int key = dataset.addSeries("Lane " + (laneIndex + 1));
                    Lane<String, String> lane = lanes[laneIndex];
                    for (int position = 0; position <= mainLength; position += 10)
                    {
                        DoubleScalar.Rel<LengthUnit> longitudinalPosition =
                            new DoubleScalar.Rel<LengthUnit>(position, LengthUnit.METER);
                        DoubleScalar.Rel<LengthUnit> suitability =
                            navigator.suitability(lane, longitudinalPosition, gtuType, this.timeHorizon);
                        if (suitability.getSI() <= mainLength)
                        {
                            dataset.addXYPair(key, mainLength - position, suitability.getSI());
                        }
                    }
                    dataset.reGraph();
                }
            }
        }
    }

    /**
     * Instantiate the class.
     */
    public SuitabilityGraph()
    {
        this.graphPanel = new JPanel(new BorderLayout());
        final int rows = SPEEDLIMITS.length;
        final int columns = TARGETLANES.length;
        TablePanel chartsPanel = new TablePanel(rows, rows);
        this.graphPanel.add(chartsPanel, BorderLayout.CENTER);
        this.charts = new JFreeChart[rows][columns];
        for (int row = 0; row < rows; row++)
        {
            int targetLaneConfiguration = TARGETLANES[row];
            String targetLaneDescription =
                String.format("%s lane %s exit", Math.abs(targetLaneConfiguration) == 1 ? "single" : "double",
                    targetLaneConfiguration > 0 ? "left" : "right");
            for (int column = 0; column < columns; column++)
            {
                DoubleScalar.Abs<SpeedUnit> speedLimit =
                    new DoubleScalar.Abs<SpeedUnit>(SPEEDLIMITS[column], SpeedUnit.KM_PER_HOUR);
                JFreeChart chart =
                    createChart(String.format("Speed limit %.0f%s, %s", speedLimit.getInUnit(), speedLimit.getUnit(),
                        targetLaneDescription), speedLimit);
                chartsPanel.setCell(new ChartPanel(chart), column, row);
                this.charts[row][column] = chart;
            }
        }
    }

    /**
     * @param caption String; the caption for the chart
     * @param speedLimit DoubleScalar.Abs&lt;SpeedUnit&gt;; the speed limit
     * @return JFreeChart; the newly created graph
     */
    private JFreeChart createChart(final String caption, final DoubleScalar.Abs<SpeedUnit> speedLimit)
    {
        ChartFactory.setChartTheme(new StandardChartTheme("JFree/Shadow", false));
        XYDataset chartData = new SuitabilityData();
        JFreeChart chartPanel =
            ChartFactory.createXYLineChart(caption, "", "", chartData, PlotOrientation.VERTICAL, true, false, false);
        chartPanel.setBorderVisible(true);
        chartPanel.setBorderPaint(new Color(192, 192, 192));
        NumberAxis timeAxis = new NumberAxis("\u2192 " + "Remaining time to junction [s]");
        double distanceRange = this.timeRange.getSI() * speedLimit.getSI();
        NumberAxis distanceAxis = new NumberAxis("\u2192 " + "Remaining distance to junction [m]");
        distanceAxis.setRange(0, distanceRange);
        distanceAxis.setInverted(true);
        distanceAxis.setStandardTickUnits(NumberAxis.createIntegerTickUnits());
        timeAxis.setAutoRangeIncludesZero(true);
        timeAxis.setRange(0, this.timeRange.getSI());
        timeAxis.setInverted(true);
        timeAxis.setStandardTickUnits(NumberAxis.createIntegerTickUnits());
        // time axis gets messed up on auto range (probably due to all data being relative to distance axis)
        // ((XYPlot) chartPanel.getPlot()).setDomainAxis(1, timeAxis);
        NumberAxis yAxis = new NumberAxis("\u2192 " + "Distance to vacate lane [m]");
        yAxis.setAutoRangeIncludesZero(true);
        yAxis.setRange(-0.1, distanceRange);
        chartPanel.getXYPlot().setDomainAxis(distanceAxis);
        chartPanel.getXYPlot().setRangeAxis(yAxis);
        final XYLineAndShapeRenderer renderer = (XYLineAndShapeRenderer) chartPanel.getXYPlot().getRenderer();
        renderer.setBaseLinesVisible(true);
        renderer.setBaseShapesVisible(false);
        // Set paint color and stroke for each series
        for (int index = 0; index < LANECOUNT; index++)
        {
            renderer.setSeriesPaint(index, COLORTABLE[index]);
            renderer.setSeriesStroke(index, new BasicStroke(4.0f));
        }

        return chartPanel;
    }

    /**
     * Return the JPanel that contains all the graphs.
     * @return JPanel
     */
    public final JPanel getPanel()
    {
        return this.graphPanel;
    }

    /** {@inheritDoc} */
    @Override
    public final void constructModel(final SimulatorInterface<Abs<TimeUnit>, Rel<TimeUnit>, OTSSimTimeDouble> simulator)
        throws SimRuntimeException, RemoteException
    {
        // Do nothing
    }

    /** {@inheritDoc} */
    @Override
    public final SimulatorInterface<Abs<TimeUnit>, Rel<TimeUnit>, OTSSimTimeDouble> getSimulator() throws RemoteException
    {
        return null;
    }

}

/** */
class SuitabilityData implements XYDataset
{

    /** The X values. */
    private ArrayList<ArrayList<Double>> xValues = new ArrayList<ArrayList<Double>>();

    /** The Y values. */
    private ArrayList<ArrayList<Double>> yValues = new ArrayList<ArrayList<Double>>();

    /** The names of the series. */
    private ArrayList<String> seriesKeys = new ArrayList<String>();

    /** List of parties interested in changes of this ContourPlot. */
    private transient EventListenerList listenerList = new EventListenerList();

    /** Not used internally. */
    private DatasetGroup datasetGroup = null;

    /**
     * Redraw this graph (after the underlying data has been changed).
     */
    public final void reGraph()
    {
        notifyListeners(new DatasetChangeEvent(this, null)); // This guess work actually works!
    }

    /**
     * Notify interested parties of an event affecting this TrajectoryPlot.
     * @param event DatasetChangedEvent
     */
    private void notifyListeners(final DatasetChangeEvent event)
    {
        for (DatasetChangeListener dcl : this.listenerList.getListeners(DatasetChangeListener.class))
        {
            dcl.datasetChanged(event);
        }
    }

    /**
     * Add storage for another series of XY values.
     * @param seriesName String; the name of the new series
     * @return int; the index to use to address the new series
     */
    public final int addSeries(final String seriesName)
    {
        this.xValues.add(new ArrayList<Double>());
        this.yValues.add(new ArrayList<Double>());
        this.seriesKeys.add(seriesName);
        return this.xValues.size() - 1;
    }

    /**
     * Add an XY pair to the data.
     * @param seriesKey int; key to the data series
     * @param x double; x value of the pair
     * @param y double; y value of the pair
     */
    public final void addXYPair(final int seriesKey, final double x, final double y)
    {
        this.xValues.get(seriesKey).add(x);
        this.yValues.get(seriesKey).add(y);
    }

    /** {@inheritDoc} */
    @Override
    public final int getSeriesCount()
    {
        return this.seriesKeys.size();
    }

    /** {@inheritDoc} */
    @Override
    public final Comparable<?> getSeriesKey(final int series)
    {
        return this.seriesKeys.get(series);
    }

    /** {@inheritDoc} */
    @Override
    public final int indexOf(@SuppressWarnings("rawtypes") final Comparable seriesKey)
    {
        return this.seriesKeys.indexOf(seriesKey);
    }

    /** {@inheritDoc} */
    @Override
    public final void addChangeListener(final DatasetChangeListener listener)
    {
        this.listenerList.add(DatasetChangeListener.class, listener);
    }

    /** {@inheritDoc} */
    @Override
    public final void removeChangeListener(final DatasetChangeListener listener)
    {
        this.listenerList.remove(DatasetChangeListener.class, listener);
    }

    /** {@inheritDoc} */
    @Override
    public final DatasetGroup getGroup()
    {
        return this.datasetGroup;
    }

    /** {@inheritDoc} */
    @Override
    public final void setGroup(final DatasetGroup group)
    {
        this.datasetGroup = group;
    }

    /** {@inheritDoc} */
    @Override
    public DomainOrder getDomainOrder()
    {
        return DomainOrder.ASCENDING;
    }

    /** {@inheritDoc} */
    @Override
    public final int getItemCount(final int series)
    {
        return this.xValues.get(series).size();
    }

    /** {@inheritDoc} */
    @Override
    public final Number getX(final int series, final int item)
    {
        return this.xValues.get(series).get(item);
    }

    /** {@inheritDoc} */
    @Override
    public final double getXValue(final int series, final int item)
    {
        return this.xValues.get(series).get(item);
    }

    /** {@inheritDoc} */
    @Override
    public final Number getY(final int series, final int item)
    {
        return this.yValues.get(series).get(item);
    }

    /** {@inheritDoc} */
    @Override
    public final double getYValue(final int series, final int item)
    {
        return this.yValues.get(series).get(item);
    }

}