Controller.java

package org.opentrafficsim.road.network.factory.opendrive;

import java.io.Serializable;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

import javax.naming.NamingException;

import org.djunits.unit.DurationUnit;
import org.djunits.value.vdouble.scalar.Duration;
import org.opentrafficsim.core.gtu.GTUException;
import org.opentrafficsim.core.network.NetworkException;
import org.opentrafficsim.road.network.lane.object.trafficlight.SimpleTrafficLight;
import org.opentrafficsim.road.network.lane.object.trafficlight.TrafficLightColor;

import nl.tudelft.simulation.dsol.SimRuntimeException;
import nl.tudelft.simulation.dsol.simulators.AnimatorInterface;
import nl.tudelft.simulation.dsol.simulators.DEVSSimulatorInterface;

/**
 * Traffic Light Controller
 * <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/docs/license.html">OpenTrafficSim License</a>.
 * <p>
 * @version $Revision: 1155 $, $LastChangedDate: 2015-07-26 01:01:13 +0200 (Sun, 26 Jul 2015) $, by $Author: averbraeck $,
 *          initial version 15 jul. 2015 <br>
 * @author <a href="http://www.tbm.tudelft.nl/averbraeck">Alexander Verbraeck</a>
 * @author <a href="http://www.tudelft.nl/pknoppers">Peter Knoppers</a>
 */
public class Controller implements Serializable
{

    /** */
    private static final long serialVersionUID = 20150715L;

    /** */
    private DEVSSimulatorInterface.TimeDoubleUnit simulator;

    /** */
    private String id;

    /** */
    private Map<Integer, Set<SimpleTrafficLight>> trafficLights = new HashMap<Integer, Set<SimpleTrafficLight>>();

    /**
     * @param name String; the name of the OnOffTrafficLight
     * @param simulator DEVSSimulatorInterface.TimeDoubleUnit; the simulator to avoid NullPointerExceptions
     * @throws GTUException when GTU cannot be created.
     * @throws NamingException if an error occurs when adding the animation handler
     * @throws NetworkException when the GTU cannot be placed on the given lane
     */
    public Controller(final String name, final DEVSSimulatorInterface.TimeDoubleUnit simulator)
            throws GTUException, NetworkException, NamingException
    {
        this.id = name;
        this.simulator = simulator;

        try
        {
            // new DefaultBlockOnOffAnimation(this, getSimulator());
            // animation
            if (simulator instanceof AnimatorInterface)
            {
                // TODO
            }
            simulator.scheduleEventRel(Duration.ZERO, this, this, "unBlock", null);
        }
        catch (SimRuntimeException exception)
        {
            exception.printStackTrace();
        }
    }

    /**
     * 
     */
    protected void unBlock()
    {
        try
        {
            this.simulator.scheduleEventRel(new Duration(96.0, DurationUnit.SECOND), this, this, "unBlock", null);
        }
        catch (SimRuntimeException exception)
        {
            exception.printStackTrace();
        }

        if (this.trafficLights.containsKey(3))
        {
            for (SimpleTrafficLight light : this.trafficLights.get(3))
            {
                try
                {
                    // System.out.println("traffic light 3 at time " + this.simulator.getSimulatorTime() + " is " +
                    // ((TrafficLight) light).getTrafficLightColor().toString());
                    this.simulator.scheduleEventRel(Duration.ZERO, this, light, "setTrafficLightColor",
                            new Object[] { TrafficLightColor.GREEN });
                    this.simulator.scheduleEventRel(new Duration(15.0, DurationUnit.SECOND), this, light,
                            "setTrafficLightColor", new Object[] { TrafficLightColor.YELLOW });
                    this.simulator.scheduleEventRel(new Duration(24.0, DurationUnit.SECOND), this, light,
                            "setTrafficLightColor", new Object[] { TrafficLightColor.RED });
                }
                catch (SimRuntimeException exception)
                {
                    exception.printStackTrace();
                }
            }
        }
        if (this.trafficLights.containsKey(4))
        {
            for (SimpleTrafficLight light : this.trafficLights.get(4))
            {
                try
                {
                    this.simulator.scheduleEventRel(Duration.ZERO, this, light, "setTrafficLightColor",
                            new Object[] { TrafficLightColor.GREEN });
                    this.simulator.scheduleEventRel(new Duration(15.0, DurationUnit.SECOND), this, light,
                            "setTrafficLightColor", new Object[] { TrafficLightColor.YELLOW });
                    this.simulator.scheduleEventRel(new Duration(24.0, DurationUnit.SECOND), this, light,
                            "setTrafficLightColor", new Object[] { TrafficLightColor.RED });
                }
                catch (SimRuntimeException exception)
                {
                    exception.printStackTrace();
                }
            }
        }
        if (this.trafficLights.containsKey(6))
        {
            for (SimpleTrafficLight light : this.trafficLights.get(6))
            {
                try
                {
                    // System.out.println("traffic light 6 at time " + this.simulator.getSimulatorTime());

                    this.simulator.scheduleEventRel(new Duration(24.0, DurationUnit.SECOND), this, light,
                            "setTrafficLightColor", new Object[] { TrafficLightColor.GREEN });
                    this.simulator.scheduleEventRel(new Duration(39.0, DurationUnit.SECOND), this, light,
                            "setTrafficLightColor", new Object[] { TrafficLightColor.YELLOW });
                    this.simulator.scheduleEventRel(new Duration(48.0, DurationUnit.SECOND), this, light,
                            "setTrafficLightColor", new Object[] { TrafficLightColor.RED });
                }
                catch (SimRuntimeException exception)
                {
                    exception.printStackTrace();
                }
            }
        }
        if (this.trafficLights.containsKey(7))
        {
            for (SimpleTrafficLight light : this.trafficLights.get(7))
            {
                try
                {
                    this.simulator.scheduleEventRel(new Duration(24.0, DurationUnit.SECOND), this, light,
                            "setTrafficLightColor", new Object[] { TrafficLightColor.GREEN });
                    this.simulator.scheduleEventRel(new Duration(39.0, DurationUnit.SECOND), this, light,
                            "setTrafficLightColor", new Object[] { TrafficLightColor.YELLOW });
                    this.simulator.scheduleEventRel(new Duration(48.0, DurationUnit.SECOND), this, light,
                            "setTrafficLightColor", new Object[] { TrafficLightColor.RED });
                }
                catch (SimRuntimeException exception)
                {
                    exception.printStackTrace();
                }
            }
        }
        if (this.trafficLights.containsKey(9))
        {
            for (SimpleTrafficLight light : this.trafficLights.get(9))
            {
                try
                {
                    // System.out.println("traffic light 9 at time " + this.simulator.getSimulatorTime());

                    this.simulator.scheduleEventRel(new Duration(48.0, DurationUnit.SECOND), this, light,
                            "setTrafficLightColor", new Object[] { TrafficLightColor.GREEN });
                    this.simulator.scheduleEventRel(new Duration(63.0, DurationUnit.SECOND), this, light,
                            "setTrafficLightColor", new Object[] { TrafficLightColor.YELLOW });
                    this.simulator.scheduleEventRel(new Duration(72.0, DurationUnit.SECOND), this, light,
                            "setTrafficLightColor", new Object[] { TrafficLightColor.RED });
                }
                catch (SimRuntimeException exception)
                {
                    exception.printStackTrace();
                }
            }
        }
        if (this.trafficLights.containsKey(10))
        {
            for (SimpleTrafficLight light : this.trafficLights.get(10))
            {
                try
                {
                    this.simulator.scheduleEventRel(new Duration(48.0, DurationUnit.SECOND), this, light,
                            "setTrafficLightColor", new Object[] { TrafficLightColor.GREEN });
                    this.simulator.scheduleEventRel(new Duration(63.0, DurationUnit.SECOND), this, light,
                            "setTrafficLightColor", new Object[] { TrafficLightColor.YELLOW });
                    this.simulator.scheduleEventRel(new Duration(72.0, DurationUnit.SECOND), this, light,
                            "setTrafficLightColor", new Object[] { TrafficLightColor.RED });
                }
                catch (SimRuntimeException exception)
                {
                    exception.printStackTrace();
                }
            }
        }
        if (this.trafficLights.containsKey(12))
        {
            for (SimpleTrafficLight light : this.trafficLights.get(12))
            {
                try
                {
                    // System.out.println("traffic light 12 at time " + this.simulator.getSimulatorTime());

                    this.simulator.scheduleEventRel(new Duration(72.0, DurationUnit.SECOND), this, light,
                            "setTrafficLightColor", new Object[] { TrafficLightColor.GREEN });
                    this.simulator.scheduleEventRel(new Duration(87.0, DurationUnit.SECOND), this, light,
                            "setTrafficLightColor", new Object[] { TrafficLightColor.YELLOW });
                    this.simulator.scheduleEventRel(new Duration(96.0, DurationUnit.SECOND), this, light,
                            "setTrafficLightColor", new Object[] { TrafficLightColor.RED });
                }
                catch (SimRuntimeException exception)
                {
                    exception.printStackTrace();
                }
            }
        }
        if (this.trafficLights.containsKey(13))
        {
            for (SimpleTrafficLight light : this.trafficLights.get(13))
            {
                try
                {
                    this.simulator.scheduleEventRel(new Duration(72.0, DurationUnit.SECOND), this, light,
                            "setTrafficLightColor", new Object[] { TrafficLightColor.GREEN });
                    this.simulator.scheduleEventRel(new Duration(87.0, DurationUnit.SECOND), this, light,
                            "setTrafficLightColor", new Object[] { TrafficLightColor.YELLOW });
                    this.simulator.scheduleEventRel(new Duration(96.0, DurationUnit.SECOND), this, light,
                            "setTrafficLightColor", new Object[] { TrafficLightColor.RED });
                }
                catch (SimRuntimeException exception)
                {
                    exception.printStackTrace();
                }
            }
        }
    }

    /**
     * @param sequence2 int; sequence2
     * @param trafficLight SimpleTrafficLight; trafficLight
     */
    public void addTrafficLight(int sequence2, SimpleTrafficLight trafficLight)
    {
        if (!this.trafficLights.containsKey(sequence2))
        {
            Set<SimpleTrafficLight> trafficLightsSet = new HashSet<SimpleTrafficLight>();
            this.trafficLights.put(sequence2, trafficLightsSet);
        }
        this.trafficLights.get(sequence2).add(trafficLight);
    }

    /** {@inheritDoc} */
    @Override
    public final String toString()
    {
        return "Controller [id=" + this.id + ", trafficLights=" + this.trafficLights + "]";
    }

}