Adds getRoleOfDevice Method to Protocols, Fixes SimpleProtocol.add()

src/main/java/de/tu_darmstadt/tk/SmartHomeNetworkSim/core/Protocol.java

@@ -51,6 +51,21 @@ public interface Protocol {
 	 */
 	public Collection<Port> getDevicesWithRole(int role);
 
+	/**
+	 * Returns the role of the given device, returns -1, if the Device is not part of the protocol
+	 * 
+	 * @param device device which roles should be calculated
+	 * @return role of the device, -1 if it has no role
+	 */
+	public default int getRoleOfDevice(Port device){
+		if(device == null)
+			return -1;
+		for(int i = 0; i<getNumberOfRoles(); i++)
+			if(getDevicesWithRole(i).contains(device))
+				return i;
+		return -1;
+	}
+	
 	/**
 	 * Adds a new SmartDevice to the role, returns {@code true} if it was
 	 * assigned successfully, {@code false} if it wasn't. (Either invalid role

src/main/java/de/tu_darmstadt/tk/SmartHomeNetworkSim/core/simpleImplementation/SimpleProtocol.java

@@ -82,12 +82,12 @@ public class SimpleProtocol implements Protocol {
 	public boolean addDeviceOfRole(Port device, int role) {
 		if(role == 0 && source == null){
 			source = device;
-		}else if(role == 1 && destination == source){
+		}else if(role == 1 && destination == null){
 			destination = device;
 		}else{
 			return false;
 		}
-		return false;
+		return true;
 	}
 	
 	@Override
@@ -103,4 +103,11 @@ public class SimpleProtocol implements Protocol {
 		return "SimpleProtocol";
 	}
 
+	@Override
+	public int getRoleOfDevice(Port device){
+		if(device == null) return -1;
+		else if(device == source) return 0;
+		else if(device == destination) return 1;
+		else return -1;
+	}
 }