simplemessage.java

DTNSim2 is a simulator for Delay-Tolerant Networks (DTNs) written in Java. It is based on Sushant Ja

/**
 * implementations/SimpleMessage.java
 */
package implementations;

import java.util.ArrayList;

import simulator.Message;
import simulator.Node;

/**
 * 
 */
public class SimpleMessage extends Message
{
	private int copiesToMake = 1;

	// Not actually sent over contacts, but used locally in nodes to mark nodes
	// which already received the message.
	private ArrayList<Node> nodesWithCopy = new ArrayList<Node>();

	public SimpleMessage(int id, double creationTime, Node source, Node dest, int dLen)
	{
		super(id, creationTime, source, dest, dLen);
	}

	public SimpleMessage(SimpleMessage org)
	{
		super(org);

		// New copy will "receive" half of copies number,
		// and this node will leave the rest for itself

		copiesToMake = org.copiesToMake / 2;

		org.copiesToMake = org.copiesToMake - copiesToMake;
	}

	public void setCopiesToMake(int c)
	{
		if (c > 0)
			copiesToMake = c;
	}

	public int getCopiesToMake()
	{
		return copiesToMake;
	}

	public void removeNodeWithCopy(Node n)
	{
		nodesWithCopy.remove(n);
	}

	public void addNodeWithCopy(Node n)
	{
		if (!nodesWithCopy.contains(n))
			nodesWithCopy.add(n);
	}

	public boolean hasCopy(Node n)
	{
		return nodesWithCopy.contains(n);
	}
}