Ghost User · d650803c · 92030734 · 014a1c77 · 59c39fcf · 71d8972c
--- a/src/routing/choosers/UtilityMessageChooser.java 0 → 100644
+++ b/src/routing/choosers/UtilityMessageChooser.java 0 → 100644
+    public Collection<Tuple<Message, Connection>> findOtherMessages(
+            Collection<Message> messages, List<Connection> connections) {
+        Collection<Tuple<Message, Connection>> chosenMessages = new ArrayList<>();
+
+        // Add ordinary messages.
+        for (Connection con : connections) {
+            DTNHost neighbor = con.getOtherNode(this.attachedHost);
+            for (Message m : messages) {
+                if (!m.isFinalRecipient(neighbor) && this.shouldBeSent(m, neighbor)) {
+                    chosenMessages.add(new Tuple<>(m, con));
+                }
+            }
+        }
+
+        // Wrap useful data stored at host in data messages to neighbors and add them to the messages to sent.
+        chosenMessages.addAll(DatabaseApplicationUtil.createDataMessages(
--- a/src/test/DisasterRouterTest.java
+++ b/src/test/DisasterRouterTest.java
+
+        // Prepare message with medium replications density.
+        Message popularMessage = new Message(h1, h6, "M1", 0);
+        h1.createNewMessage(popularMessage);
+        h0.createNewMessage(popularMessage);
+        h1.connect(h0);
+        h1.connect(h5);
+        this.clock.advance(DisasterRouterTestUtils.RD_WINDOW_LENGTH);
+        disconnect(h1);
+        this.updateAllNodes();
+
+        // Check that we only send to the neighbor with high encounter value AND high power.
+        this.mc.reset();
+        h1.connect(h2);
+        this.updateAllNodes();
+        Assert.assertFalse("Should not send to social neighbor with low power value.", this.mc.next());
--- a/src/test/UtilityMessageChooserTest.java 0 → 100644
+++ b/src/test/UtilityMessageChooserTest.java 0 → 100644
+    }
+
+    /**
+     * Checks the provided message-connection tuple list for the existence of a tuple mapping a message with the
+     * provided ID to a connection where the host which is not {@link #attachedHost} is the provided host.
+     *
+     * @param messages List to check.
+     * @param id Message ID to look for.
+     * @param host Host to look for.
+     * @return True if such a message can be found.
+     */
+    private boolean messageToHostsExists(Collection<Tuple<Message, Connection>> messages, String id, DTNHost host) {
+        boolean messageFound = false;
+        for (Tuple<Message, Connection> tuple : messages) {
+            if (tuple.getKey().getId().equals(id) && tuple.getValue().getOtherNode(this.attachedHost).equals(host)) {
+                messageFound = true;
--- a/configurations/evaluation/settingsDisasterRouter.txt
+++ b/configurations/evaluation/settingsDisasterRouter.txt
 ReplicationsDensity.windowLength = 120

 # Parameters for message choosing
-# ! TODO: Uncommented this once message choosing is implemented !
-# UtilityMessageChooser.dpWeight = 0.95
-# UtilityMessageChooser.powerWeight = 0.05
-# UtilityMessageChooser.prophetPlusWeight = 0.65
-# UtilityMessageChooser.rdWeight = 0.25
--- a/src/routing/choosers/UtilityMessageChooser.java 0 → 100644
+++ b/src/routing/choosers/UtilityMessageChooser.java 0 → 100644
+    private double computeUtility(Message m, DisasterRouter otherRouter) {
+        return this.deliveryPredictabilityWeight * otherRouter.getDeliveryPredictability(m)
+                + this.powerWeight * otherRouter.remainingEnergyRatio()
+                + this.replicationsDensityWeight * (1 - this.attachedRouter.getReplicationsDensity(m))
+                + this.encounterValueWeight * this.attachedRouter.computeEncounterValueRatio(otherRouter);
+    }
+
+    /**
+     * Determines whether the provided message should be sent to the provided host right now.
+     * This is only the case if the host is not transferring, does not know the message yet, and the message - host
+     * pair's utility is sufficiently high.
+     * @param m Message to check.
+     * @param otherHost Host to check.
+     * @return True iff the message should be sent.
+     */
+    private boolean shouldBeSent(Message m, DTNHost otherHost) {
--- a/src/test/DisasterRouterTest.java
+++ b/src/test/DisasterRouterTest.java
     */
    public void testNonDirectMessageSorting() {
        // Create messages to sort.
-        DisasterData data = new DisasterData(DisasterData.DataType.MARKER, 0, 0, new Coord(0, 0));
-        Message vipDataMessages = new DataMessage(h1, h3, "D1", data, 0, VERY_HIGH_PRIORITY);
--- a/src/test/DisasterRouterTestUtils.java
+++ b/src/test/DisasterRouterTestUtils.java
    public static final int PRIORITY_THRESHOLD = 4;
    public static final double DP_WEIGHT = 0.8;

+    /* Constants needed for message choosing. */