rinde.sim.pdptw.common.RouteRenderer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of rinsim-problem Show documentation
The newest version!
/**
 * 
 */
package rinde.sim.pdptw.common;

import static com.google.common.collect.Sets.newHashSet;

import java.util.Set;

import javax.annotation.Nullable;

import org.eclipse.swt.graphics.GC;

import rinde.sim.core.graph.Point;
import rinde.sim.core.model.ModelProvider;
import rinde.sim.core.model.ModelReceiver;
import rinde.sim.core.model.pdp.PDPModel;
import rinde.sim.core.model.road.RoadModel;
import rinde.sim.ui.renderers.CanvasRenderer;
import rinde.sim.ui.renderers.ViewPort;
import rinde.sim.ui.renderers.ViewRect;

import com.google.common.base.Optional;
import com.google.common.collect.ImmutableList;

/**
 * A renderer that draws the route for any {@link RouteFollowingVehicle}s that
 * exist in the {@link RoadModel}.
 * 
 * @author Rinde van Lon 
 */
public class RouteRenderer implements CanvasRenderer, ModelReceiver {

  Optional rm;
  Optional pm;

  /**
   * Create a new route renderer.
   */
  public RouteRenderer() {
    rm = Optional.absent();
    pm = Optional.absent();
  }

  @Override
  public void renderStatic(GC gc, ViewPort vp) {}

  @Override
  public void renderDynamic(GC gc, ViewPort vp, long time) {
    final Set vehicles = rm.get().getObjectsOfType(
        RouteFollowingVehicle.class);
    for (final RouteFollowingVehicle v : vehicles) {
      final Set seen = newHashSet();
      final Point from = rm.get().getPosition(v);
      int prevX = vp.toCoordX(from.x);
      int prevY = vp.toCoordY(from.y);

      for (final DefaultParcel parcel : ImmutableList.copyOf(v.getRoute())) {
        Point to;
        if (pm.get().getParcelState(parcel).isPickedUp()
            || seen.contains(parcel)) {
          to = parcel.dto.destinationLocation;
        } else {
          to = parcel.dto.pickupLocation;
        }
        seen.add(parcel);
        final int x = vp.toCoordX(to.x);
        final int y = vp.toCoordY(to.y);
        gc.drawLine(prevX, prevY, x, y);

        prevX = x;
        prevY = y;
      }
    }
  }

  @Override
  @Nullable
  public ViewRect getViewRect() {
    return null;
  }

  @Override
  public void registerModelProvider(ModelProvider mp) {
    rm = Optional.fromNullable(mp.getModel(RoadModel.class));
    pm = Optional.fromNullable(mp.getModel(PDPModel.class));
  }
}