Started with visualization.

src/de/tud/kom/p2psim/impl/topology/views/fiveg/ModelBasedSegmentDatabase.java

@@ -25,9 +25,14 @@ import java.util.HashSet;
 
 import de.tud.kom.p2psim.api.linklayer.mac.MacAddress;
 import de.tud.kom.p2psim.api.scenario.ConfigurationException;
+import de.tud.kom.p2psim.impl.topology.views.VisualizationTopologyView.VisualizationInjector;
 import de.tud.kom.p2psim.impl.topology.views.fiveg.models.AbstractModel;
 import de.tud.kom.p2psim.impl.topology.views.fiveg.utils.Direction;
 import de.tud.kom.p2psim.impl.topology.views.fiveg.utils.ParameterType;
+import de.tud.kom.p2psim.impl.topology.views.visualization.world.FiveGVisualization;
+import de.tud.kom.p2psim.impl.topology.views.visualization.world.ModelFiveGVisualization;
+import de.tudarmstadt.maki.simonstrator.api.Event;
+import de.tudarmstadt.maki.simonstrator.api.EventHandler;
 import de.tudarmstadt.maki.simonstrator.api.Time;
 
 public class ModelBasedSegmentDatabase
@@ -47,6 +52,22 @@ public class ModelBasedSegmentDatabase
 	protected Entry createEntryFor(int segmentID, boolean isCloudlet) {
 		return new ModelBasedEntry(segmentID);
 	}
+	
+	@Override
+	public void setEnableVis(boolean enableVis) {
+		if (enableVis == true) {
+			Event.scheduleImmediately(new EventHandler() {
+
+				@Override
+				public void eventOccurred(Object content, int type) {
+					VisualizationInjector.injectComponent("5G 2 " + _visId++, -1,
+							new ModelFiveGVisualization(
+									ModelBasedSegmentDatabase.this, true),
+							false, true);
+				}
+			}, null, 0);
+		}
+	}
 
 	/**
 	 * Add a model to Database

src/de/tud/kom/p2psim/impl/topology/views/visualization/world/ModelFiveGVisualization.java

+/*
+ * Copyright (c) 2005-2010 KOM – Multimedia Communications Lab
+ *
+ * This file is part of PeerfactSim.KOM.
+ * 
+ * PeerfactSim.KOM is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * any later version.
+ * 
+ * PeerfactSim.KOM is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ * 
+ * You should have received a copy of the GNU General Public License
+ * along with PeerfactSim.KOM.  If not, see <http://www.gnu.org/licenses/>.
+ *
+ */
+
+package de.tud.kom.p2psim.impl.topology.views.visualization.world;
+
+import de.tud.kom.p2psim.impl.topology.views.FiveGTopologyView;
+import de.tud.kom.p2psim.impl.topology.views.VisualizationTopologyView.VisualizationInjector;
+import de.tud.kom.p2psim.impl.topology.views.fiveg.AbstractGridBasedTopologyDatabase;
+import de.tud.kom.p2psim.impl.topology.views.fiveg.FiveGTopologyDatabase.Entry;
+import de.tudarmstadt.maki.simonstrator.api.Rate;
+import de.tudarmstadt.maki.simonstrator.api.Time;
+
+import javax.swing.*;
+import java.awt.*;
+import java.awt.geom.Rectangle2D;
+import java.awt.image.BufferedImage;
+
+/**
+ * Visualization for the {@link FiveGTopologyView}
+ * 
+ * Added 06.10.2016 Nils Richerzhagen, Clemens Krug
+ * Functionality to 'destroy' cells using SHIFT + Left-click. Enable/Disable using the enableCellDestruction
+ * 
+ * @author Bjoern Richerzhagen
+ * @version 1.0, Nov 5, 2015
+ */
+public class ModelFiveGVisualization extends JComponent implements VisualizationInjector.MouseClickListener {
+
+	private final boolean enableCellDestruction;
+	
+	protected BufferedImage image;
+
+	private final AbstractGridBasedTopologyDatabase database;
+
+	public ModelFiveGVisualization(AbstractGridBasedTopologyDatabase database, boolean enableCellDestruction) {
+		setBounds(0, 0, VisualizationInjector.getWorldX(),
+				VisualizationInjector.getWorldY());
+		setOpaque(true);
+		setVisible(true);
+
+		this.database = database;
+		image = new BufferedImage(VisualizationInjector.getWorldX(),
+				VisualizationInjector.getWorldY(), BufferedImage.TYPE_INT_ARGB);
+
+		this.enableCellDestruction = enableCellDestruction;
+		
+		VisualizationInjector.addMouseListener(this);
+	}
+
+	@Override
+	protected void paintComponent(Graphics g) {
+		super.paintComponent(g);
+
+		redraw();
+
+		Graphics2D g2 = (Graphics2D) g;
+		g2.drawImage(image, 0, 0, null);
+	}
+
+	protected void redraw() {
+
+		Graphics2D g2 = (Graphics2D) image.getGraphics();
+
+		Composite c = g2.getComposite();
+		g2.setComposite(AlphaComposite.getInstance(AlphaComposite.CLEAR, 0.0f));
+		Rectangle2D.Double rect = new Rectangle2D.Double(0, 0,
+				VisualizationInjector.getWorldX(),
+				VisualizationInjector.getWorldY());
+		g2.fill(rect);
+		g2.setComposite(c);
+
+		try {
+			drawEntries(g2);
+		} catch (RuntimeException e) {
+			e.printStackTrace();
+			System.exit(0);
+		}
+	}
+
+	private void drawEntries(Graphics2D g2) {
+		// Iterate over grid coordinates
+		int stepSize = database.getGridSize();
+
+		double maxLatency = 0;
+		double minLatency = Double.MAX_VALUE;
+		boolean isUpload = false;
+
+		for (int x = 0; x < VisualizationInjector.getWorldX(); x += stepSize) {
+			for (int y = 0; y < VisualizationInjector
+					.getWorldY(); y += stepSize) {
+				// TODO add checkbox for cloudlets?
+				Entry entry = database.getEntryFor(database.getSegmentID(x, y),
+						false);
+				if (entry == null) {
+					continue;
+				}
+			}
+		}
+
+		for (int x = 0; x < VisualizationInjector.getWorldX(); x += stepSize) {
+			for (int y = 0; y < VisualizationInjector
+					.getWorldY(); y += stepSize) {
+
+				// TODO add checkbox for cloudlets?
+				Entry entry = database.getEntryFor(database.getSegmentID(x, y),
+						false);
+				if (entry == null) {
+					continue;
+				}
+
+				System.out.println(entry.getLatency(isUpload));
+				System.out.println(entry.getDropProbability(isUpload));
+				System.out.println(entry.getBandwidth(isUpload));
+
+				// TODO add checkbox for upload/download toggle?
+
+				// Latency
+				double latencyFactor = (entry.getLatency(isUpload) - minLatency)
+						/ (maxLatency - minLatency);
+				g2.setColor(
+							new Color(255, 0, 0, 10 + (int) (40 * latencyFactor)));
+				g2.fillRect(x, y, stepSize, stepSize);
+
+				// Drop-Prob
+				g2.setColor(new Color(255, 0, 0,
+						10 + (int) (100 * entry.getDropProbability(isUpload))));
+				float strokeWidth = (float) entry.getDropProbability(isUpload);
+				g2.setStroke(new BasicStroke((10 * strokeWidth)));
+				g2.drawRect(x, y, stepSize, stepSize);
+				g2.setColor(new Color(255, 255, 255, 255));
+				g2.drawString("L: "
+						+ entry.getLatency(isUpload) / Time.MILLISECOND + " ms",
+						x + 10, y + 15);
+				g2.drawString(
+						"D: " + (int) (entry.getDropProbability(isUpload) * 100)
+								+ " %",
+						x + 10, y + 25);
+				g2.drawString("BW: "
+						+ (int) (entry.getBandwidth(isUpload) / Rate.kbit_s)
+						+ " kBit/s", x + 10, y + 35);
+				if(!entry.isAvailable()) {
+					g2.drawString("!DEAD!", x + 30, y + 70);
+				}
+			}
+		}
+
+	}
+
+	@Override
+	public void mouseClicked(int x, int y, int modifier)
+	{
+		// 17 == Shift
+		if(enableCellDestruction && modifier == 17)
+		{
+			int segID = database.getSegmentID(x, y);
+			Entry entry = database.getEntryFor(segID, false);
+			entry.setAvailability(!entry.isAvailable());
+		}
+	}
+}