package model;

import graph.Edge;

import graph.Node;

import java.util.ArrayList;

import java.util.LinkedList;

import java.util.List;

public class TravelMap {

private List<Node> nodes = new LinkedList<>();

private List<Edge> edges = new LinkedList<>();

/**

* Default constructor for model.TravelMap

*/

public TravelMap() {

}

/**

* method for adding an edge to the travel map

*

* @param node1 starting node of the edge

* @param node2 ending node of the edge

* @param cost the cost of the edge

*/

public void addEdge(Node node1, Node node2, int cost) {

Edge temporaryEdge = new Edge(node1, node2, cost);

edges.add(temporaryEdge);

}

/**

* method for adding either a oneWay or a twoWay edge

*

* @param node1 Starting graph.Edge

* @param node2 Ending graph.Edge

* @param cost the cost of the edge

* @param isTwoWay true if it is a twoWay edge false if it is a oneWay edge

*/

public void addEdge(Node node1, Node node2, int cost, boolean isTwoWay) {

Edge temporaryEdge = new Edge(node1, node2, cost, isTwoWay);

edges.add(temporaryEdge);

}

public void addNode(Node node) {

nodes.add(node);

}

/**

* Getter for nodes

*

* @return a string containing all of nodes

*/

public String getNodesToString() {

StringBuilder allNodes = new StringBuilder();

for (Node node : nodes) {

allNodes.append(" ").append(node.toString()).append("\n");

}

return allNodes.toString();

}

public List<Node> getNodes() {

return this.nodes;

}

public void setNodes(ArrayList<Node> nodes) {

this.nodes = nodes;

}

/**

* Getter for edges

*

* @return a list for edges

*/

public List<Edge> getEdges() {

return edges;

}

public void setEdges(ArrayList<Edge> edges) {

this.edges = edges;

}

@Override

public String toString() {

return "TravelMap{" +

"nodes=" + nodes.toString() +

", edges=" + edges.toString() +

'}';

}

}