Adds MQTT Value to KMeans Classifier

examples/classifier/BasicPacketClassifier.java

@@ -9,6 +9,7 @@ import java.util.Map.Entry;
 import de.tu_darmstadt.tk.SmartHomeNetworkSim.core.Link;
 import de.tu_darmstadt.tk.SmartHomeNetworkSim.core.Packet;
 import de.tu_darmstadt.tk.SmartHomeNetworkSim.core.PacketSniffer;
+import de.tu_darmstadt.tk.SmartHomeNetworkSim.core.protocols.packets.MQTTpublishPacket;
 import weka.core.Attribute;
 import weka.core.DenseInstance;
 import weka.core.Instance;
@@ -180,6 +181,11 @@ public abstract class BasicPacketClassifier implements PacketSniffer {
 		
 		// Packets per second
 		instance.setValue(6, getEstimatedPacketsPerSecond(link, packet));
+		// MQTT Value
+		if(packet instanceof MQTTpublishPacket)
+			instance.setValue(7, ((MQTTpublishPacket)packet).getValue());
+		else
+			instance.setValue(7, 0);
 		
 		return instance;
 	}
@@ -255,6 +261,7 @@ public abstract class BasicPacketClassifier implements PacketSniffer {
 		Attribute pps = new Attribute("Packets-per-second", false);
 		//pps.setWeight(20);
 		atts.add(pps);
+		atts.add(new Attribute("PacketValue", false));
 		//atts.add(new Attribute("Anomaly", false));
 
 		/*