import sim.field.network.Edge; //导入依赖的package包/类
/** Return a set containing the current person's friends in the social graph.
 */
public Set<Person> getFriends() {
  Bag myEdges = sim.socialNetwork.getEdges(this, null);
  Set<Person> friends = new HashSet<Person>();

  for (Object e1 : myEdges) {
    Person from = (Person) ((Edge) e1).from();
    Person to = (Person) ((Edge) e1).to();

    if (from == this) {
      friends.add(to);
    } else {
      friends.add(from);
    }
  }
  return friends;
}
 

import sim.field.network.Edge; //导入依赖的package包/类
/** Return a set containing the common friend with "other". This would be handled by a 
 * PSI operation in practice.
 *
 * @param other The person with whom we should compute mutual friends.
 */
public Set<Object> getSharedFriends(Person other) {
  Bag myEdges = sim.socialNetwork.getEdges(this, null);
  Bag otherEdges = sim.socialNetwork.getEdges(other, null);

  Set<Object> sharedFriends = new HashSet<Object>();
  for (Object e1 : myEdges) {
    for (Object e2 : otherEdges) {
      // There has to be some way to do this more elegantly?
      Object myFrom = ((Edge) e1).from();
      Object myTo = ((Edge) e1).to();
      Object otherFrom = ((Edge) e2).from();
      Object otherTo = ((Edge) e2).to();

      Object myFriend = (myFrom == this) ? myTo : myFrom;
      Object otherFriend = (otherFrom == other) ? otherTo : otherFrom;

      // System.out.println(myFrom + " " + myTo + " " + otherFrom + " " + otherTo);
      if (myFriend == otherFriend) {
        sharedFriends.add(myFriend);
      }
    }
  }
  return sharedFriends;
}
 

import sim.field.network.Edge; //导入依赖的package包/类
@Test
public void initializeNeighbourhood() {
	
	Bag edges = new Bag(3);
	edges.add(new Edge(agent, agent1, 1));
	edges.add(new Edge(agent, agent2, 1));
	edges.add(new Edge(agent, agent3, 1));
	Mockito.when(network.getEdgesIn(agent)).thenReturn(edges);
	
	Mockito.reset(agent);
	agent.initialize(true, network);
	assertNotNull(agent.getNeighbourhood());
	Mockito.verify(agent, Mockito.times(1)).calculateNeighbourhood(network);
	assertEquals(3, agent.getNeighbourhood().size());

	Mockito.reset(agent);
	agent.initialize(true, network);
	assertNotNull(agent.getNeighbourhood());
	Mockito.verify(agent, Mockito.times(0)).calculateNeighbourhood(network);
	assertEquals(3, agent.getNeighbourhood().size());

}
 

import sim.field.network.Edge; //导入依赖的package包/类
@Override
public void draw(Object object, Graphics2D graphics, DrawInfo2D info) {
    EdgeDrawInfo2D ei = (EdgeDrawInfo2D) info;
    Edge e = (Edge) object;

    final int startX = (int) ei.draw.x;
    final int startY = (int) ei.draw.y;
    final int endX = (int) ei.secondPoint.x;
    final int endY = (int) ei.secondPoint.y;
    final int midX = (int) (ei.draw.x + ei.secondPoint.x) / 2;
    final int midY = (int) (ei.draw.y + ei.secondPoint.y) / 2;

    graphics.setColor(Color.yellow);
    graphics.drawLine(startX, startY, endX, endY);
    
    Link link = (Link) e.getInfo();
    graphics.setColor(Color.blue);
    graphics.setFont(labelFont);
    int width = graphics.getFontMetrics().stringWidth(link.getID());
    graphics.drawString( link.getID(), midX - width / 2, midY );

}
 

import sim.field.network.Edge; //导入依赖的package包/类
@SuppressWarnings("unchecked")
public Bag getSellers(final Party buyer) {
	final Bag edges = network.getEdgesIn(buyer);
	final Bag sellers = new Bag();
	for (final Iterator<Edge> iterEdges = edges.iterator(); iterEdges.hasNext();) {
		sellers.add(iterEdges.next().from());
	}
	return sellers;
}
 

import sim.field.network.Edge; //导入依赖的package包/类
@SuppressWarnings("unchecked")
public Bag getBuyers(final Party seller) {
	final Bag edges = network.getEdgesOut(seller);
	final Bag buyers = new Bag();
	for (final Iterator<Edge> iterEdges = edges.iterator(); iterEdges.hasNext();) {
		buyers.add(iterEdges.next().to());
	}
	return buyers;
}
 

import sim.field.network.Edge; //导入依赖的package包/类
public void draw(Object object, Graphics2D graphics, DrawInfo2D info) {
	Edge e = (Edge) object;

	Link link = (Link) e.getInfo();
	Color color = null;
	if (link.getClass().equals(SensorLink.class)) {
		color = Color.green;
	} else if (link.getClass().equals(RoutePathLink.class)) {
		color = Color.black;
	}

	this.drawSimpleLink(link, object, graphics, info, color);

}
 

import sim.field.network.Edge; //导入依赖的package包/类
/**
 * Calculate the neighbourhood of this agent by the given network.
 * 
 * @return agent
 */
protected Bag calculateNeighbourhood(Network network) {

	Bag edges = new Bag(network.getEdgesIn(this));
	Bag neighbours = new Bag();
	for (int i = 0, si = edges.size(); i < si; i++) {
		Edge edge = (Edge) edges.get(i);
		Agent neighbour = (Agent) edge.getOtherNode(this);
		neighbours.add(neighbour);
	}
	return neighbours;
}
 

import sim.field.network.Edge; //导入依赖的package包/类
/**
 * To draw a link
 * 
 * @param link
 */
public void drawLink(Link link) {
    List<Device> linkedDevices = link.getLinkedDevices();
    for (int i = 0; i<linkedDevices.size(); i++) {
        Device from = linkedDevices.get(i);
        for (int j = i+1 ; j<linkedDevices.size(); j++) {
            Device to = linkedDevices.get(j);
            Edge e = new Edge(from, to, link);
            links.addEdge(e);
        }
    }
}
 

import sim.field.network.Edge; //导入依赖的package包/类
/**
 * To draw a link
 * 
 * @param link
 */
public void drawLink(Link link) {
    List<Device> linkedDevices = link.getLinkedDevices();
    for (int i = 0; i < linkedDevices.size(); i++) {
        Device from = linkedDevices.get(i);
        for (int j = i + 1; j < linkedDevices.size(); j++) {
            Device to = linkedDevices.get(j);
            Edge e = new Edge(from, to, link);
            links.addEdge(e);
        }
    }
}
 

import sim.field.network.Edge; //导入依赖的package包/类
public void draw(Object object, Graphics2D graphics, DrawInfo2D info) {
    Edge e = (Edge) object;

    Link link = (Link) e.getInfo();
    if (link.getCapacity() == 2) {
        this.drawSimpleLink(link, object, graphics, info);
    } else if (link.getCapacity() > 2) {
        this.drawComplexLink(link, object, graphics, info);
    }

}
 

import sim.field.network.Edge; //导入依赖的package包/类
public String getLabel(Edge e) {
	return "";
}