/**
* Copyright 2011, Big Switch Networks, Inc.
* Originally created by David Erickson, Stanford University
*
* Licensed under the Apache License, Version 2.0 (the "License"); you may
* not use this file except in compliance with the License. You may obtain
* a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
* License for the specific language governing permissions and limitations
* under the License.
**/
package net.floodlightcontroller.routing;
import java.util.HashMap;
import org.projectfloodlight.openflow.types.DatapathId;
import net.floodlightcontroller.linkdiscovery.Link;
public class BroadcastTree {
protected HashMap<DatapathId, Link> links;
protected HashMap<DatapathId, Integer> costs;
public BroadcastTree() {
links = new HashMap<DatapathId, Link>();
costs = new HashMap<DatapathId, Integer>();
}
public BroadcastTree(HashMap<DatapathId, Link> links, HashMap<DatapathId, Integer> costs) {
this.links = links;
this.costs = costs;
}
public Link getTreeLink(DatapathId node) {
return links.get(node);
}
public int getCost(DatapathId node) {
if (costs.get(node) == null) return -1;
return (costs.get(node));
}
public HashMap<DatapathId, Link> getLinks() {
return links;
}
public void addTreeLink(DatapathId myNode, Link link) {
links.put(myNode, link);
}
public String toString() {
StringBuffer sb = new StringBuffer();
for(DatapathId n: links.keySet()) {
sb.append("[" + n.toString() + ": cost=" + costs.get(n) + ", " + links.get(n) + "]");
}
return sb.toString();
}
public HashMap<DatapathId, Integer> getCosts() {
return costs;
}
}