StochasticDistractionModel.java
- package org.opentrafficsim.road.gtu.lane.perception.mental.sdm;
- import java.rmi.RemoteException;
- import java.util.LinkedHashMap;
- import java.util.LinkedHashSet;
- import java.util.LinkedList;
- import java.util.List;
- import java.util.Map;
- import java.util.Queue;
- import java.util.Set;
- import org.djutils.event.Event;
- import org.djutils.event.EventListener;
- import org.djutils.exceptions.Throw;
- import org.djutils.exceptions.Try;
- import org.opentrafficsim.core.dsol.OtsSimulatorInterface;
- import org.opentrafficsim.core.gtu.Gtu;
- import org.opentrafficsim.core.network.Network;
- import org.opentrafficsim.road.gtu.lane.LaneBasedGtu;
- import org.opentrafficsim.road.gtu.lane.perception.mental.Fuller;
- import org.opentrafficsim.road.gtu.lane.perception.mental.Mental;
- import org.opentrafficsim.road.gtu.lane.perception.mental.Task;
- import org.opentrafficsim.road.network.RoadNetwork;
- import nl.tudelft.simulation.dsol.SimRuntimeException;
- /**
- * Stochastic Distraction Model by Manuel Lindorfer.
- * <p>
- * Copyright (c) 2013-2024 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://github.com/peter-knoppers">Peter Knoppers</a>
- * @author <a href="https://github.com/wjschakel">Wouter Schakel</a>
- */
- public class StochasticDistractionModel implements EventListener
- {
- /** Whether to allow multi-tasking. */
- private final boolean allowMultiTasking;
- /** List of distractions. */
- private final List<Distraction> distractions;
- /** Simulator. */
- private final OtsSimulatorInterface simulator;
- /** Network. */
- private final RoadNetwork network;
- /** Set of distracted GTUs. */
- private final Set<String> distractedGTUs = new LinkedHashSet<>();
- /** Queue of distractions per GTU. */
- private final Map<String, Queue<Distraction>> distractionQueues = new LinkedHashMap<>();
- /**
- * Constructor. This model will react to GTU's being created in simulation and apply distractions.
- * @param allowMultiTasking whether to allow multi-tasking
- * @param distractions list of distractions
- * @param simulator simulator
- * @param network network
- */
- public StochasticDistractionModel(final boolean allowMultiTasking, final List<Distraction> distractions,
- final OtsSimulatorInterface simulator, final RoadNetwork network)
- {
- Throw.whenNull(distractions, "List of tasks may not be null.");
- Throw.whenNull(simulator, "Simulator may not be null.");
- Throw.whenNull(network, "Network may not be null.");
- this.allowMultiTasking = allowMultiTasking;
- this.distractions = distractions;
- this.simulator = simulator;
- this.network = network;
- network.addListener(this, Network.GTU_ADD_EVENT);
- network.addListener(this, Network.GTU_REMOVE_EVENT);
- }
- /**
- * Start a distraction.
- * @param gtu gtu to start the distraction on
- * @param distraction distraction
- * @param scheduleNext whether to schedule the next distraction (not if starting from queue)
- * @throws SimRuntimeException on time error
- */
- public void startDistraction(final LaneBasedGtu gtu, final Distraction distraction, final boolean scheduleNext)
- throws SimRuntimeException
- {
- if (gtu.isDestroyed())
- {
- return;
- }
- String gtuId = gtu.getId();
- if (this.allowMultiTasking || !this.distractedGTUs.contains(gtuId))
- {
- // start the distraction now
- if (!this.allowMultiTasking)
- {
- this.distractedGTUs.add(gtuId);
- }
- Task task = distraction.getTask(gtu);
- ((Fuller) gtu.getTacticalPlanner().getPerception().getMental()).addTask(task);
- // stop the distraction
- this.simulator.scheduleEventRel(distraction.nextDuration(), this, "stopDistraction", new Object[] {gtu, task});
- }
- else
- {
- // need to queue distraction
- if (!this.distractionQueues.containsKey(gtuId))
- {
- this.distractionQueues.put(gtuId, new LinkedList<>());
- }
- this.distractionQueues.get(gtuId).add(distraction);
- }
- if (scheduleNext)
- {
- // schedule next distraction
- this.simulator.scheduleEventRel(distraction.nextInterArrival(), this, "startDistraction",
- new Object[] {gtu, distraction, true});
- }
- }
- /**
- * Stops a distraction task.
- * @param gtu gtu to stop the task for
- * @param task task to stop
- * @throws SimRuntimeException on time error
- */
- public void stopDistraction(final LaneBasedGtu gtu, final Task task) throws SimRuntimeException
- {
- if (gtu.isDestroyed())
- {
- return;
- }
- String gtuId = gtu.getId();
- ((Fuller) gtu.getTacticalPlanner().getPerception().getMental()).removeTask(task);
- // start next distraction if any in queue
- if (!this.allowMultiTasking)
- {
- this.distractedGTUs.remove(gtuId);
- if (this.distractionQueues.containsKey(gtuId))
- {
- Queue<Distraction> queue = this.distractionQueues.get(gtuId);
- Distraction distraction = queue.poll();
- startDistraction(gtu, distraction, false);
- if (queue.isEmpty())
- {
- this.distractionQueues.remove(gtuId);
- }
- }
- }
- }
- @Override
- public void notify(final Event event) throws RemoteException
- {
- if (event.getType().equals(Network.GTU_ADD_EVENT))
- {
- // The GTU is not initialized yet, so we can't obtain the tactical planner
- String gtuId = (String) event.getContent();
- Gtu gtu = this.network.getGTU(gtuId);
- gtu.addListener(this, Gtu.MOVE_EVENT);
- }
- else if (event.getType().equals(Gtu.MOVE_EVENT))
- {
- String gtuId = (String) ((Object[]) event.getContent())[0];
- LaneBasedGtu gtu = (LaneBasedGtu) this.network.getGTU(gtuId);
- Mental mental = gtu.getTacticalPlanner().getPerception().getMental();
- if (mental != null && mental instanceof Fuller)
- {
- for (Distraction distraction : this.distractions)
- {
- if (distraction.nextExposure())
- {
- Try.execute(
- () -> this.simulator.scheduleEventRel(distraction.nextInterArrival(), this, "startDistraction",
- new Object[] {gtu, distraction, true}),
- "Exception while scheduling distraction start.");
- }
- }
- }
- gtu.removeListener(this, Gtu.MOVE_EVENT);
- }
- else if (event.getType().equals(Network.GTU_REMOVE_EVENT))
- {
- String gtuId = (String) event.getContent();
- if (!this.allowMultiTasking)
- {
- this.distractedGTUs.remove(gtuId);
- }
- this.distractionQueues.remove(gtuId);
- }
- }
- }