import java.util.List;

// KMLGenerator.java
// by Morgan Wahl

/**
 * Class to generate KML output for Traveling-Salesman Approximation.
 * <ul>
 * <lh>Three sets of paths can be added to:
 * <li>white ones for all the edges in the graph
 * <li>thick, orange ones for the minimum spanning tree
 * <li>thick, green ones for the final path derived from a euclidean tour of
 * the spanning tree
 * </ul>
 */ 
public class KMLGenerator {

	private List<Place> places;

	private List<Connection> origConnections;
	private List<Connection> mstConnections;
	private List<Connection> routeConnections;

	public KMLGenerator(List<Place> places, List<Connection> origConnections, List<Connection> mstConnections,
			List<Connection> routeConnections) {
		this.places = places;
		this.origConnections = origConnections;
		this.mstConnections = mstConnections;
		this.routeConnections = routeConnections;
	}

	/**
	 * returns the KML for the paths contained in this KMLGenerator
	 */
	public String toString() {
		// TODO use actual XML classes
		// yes, this is really ugly, but it works.
		StringBuffer kml = new StringBuffer();
		kml.append("<?xml version=\"1.0\" encoding=\"UTF-8\"?>");
		kml.append("<kml xmlns=\"http://earth.google.com/kml/2.0\">");
		kml.append("<Document>\n");
		kml.append("<name>Approximate TSP</name>");
		kml.append("<open>1</open>");
		// places style
		kml.append("<Style id=\"places\">\n<IconStyle>\n<Icon>\n");
		kml.append("<href>root://icons/palette-4.png</href>\n");
		kml.append("<x>32</x>\n<y>128</y>\n<w>32</w>\n<h>32</h>\n");
		kml.append("</Icon>\n</IconStyle>");
		kml.append("</Style>\n");
		// whole graph style
		kml.append("<Style id=\"graph\">\n");
		kml.append("<LineStyle>\n");
		kml.append("<color>7fffffff</color>\n");
		kml.append("</LineStyle>\n");
		kml.append("</Style>");
		// mst style
		kml.append("<Style id=\"mst\">");
		kml.append("<LineStyle>");
		kml.append("<color>ff00aaff</color>");
		kml.append("<width>4</width>");
		kml.append("</LineStyle>");
		kml.append("</Style>\n");
		// route style
		kml.append("<Style id=\"route\">\n");
		kml.append("<LineStyle>\n");
		kml.append("<color>ff7fff55</color>\n");
		kml.append("<width>2</width>\n");
		kml.append("</LineStyle>\n");
		kml.append("</Style>\n");

		// the place-names
		kml.append("<Folder>\n");
		kml.append("<name>places</name>\n");
		kml.append("<visibility>0</visibility>\n");
		for (Place p : places) {
			kml.append("<Placemark>\n");
			kml.append("<styleUrl>#places</styleUrl>\n");
			kml.append("<name>" + p.getName() + "</name>\n");
			kml.append("<Point>\n");
			kml.append("<coordinates>\n");
			kml.append(p.getLon() + "," + p.getLat() + ",0\n");
			kml.append("</coordinates>\n");
			kml.append("</Point>\n");
			kml.append("</Placemark>");
		}
		kml.append("</Folder>\n");

		// the whole graph
		kml.append("<Folder id=\"edges\">\n");
		kml.append("<name>whole graph</name>\n");
		kml.append("<visibility>0</visibility>\n");
		for (Connection p : origConnections) {
			kml.append(connectionToKML(p, "graph"));
		}
		kml.append("</Folder>\n");

		// the minimum-spaning tree
		kml.append("<Folder>\n");
		kml.append("<name>MST</name>\n");
		kml.append("<visibility>1</visibility>\n");
		for (Connection p : mstConnections) {
			kml.append(connectionToKML(p, "mst"));
		}
		kml.append("</Folder>\n");

		// the solution route
		kml.append("<Placemark>\n");
		kml.append("<name>route</name>\n");
		kml.append("<styleUrl>#route</styleUrl>\n");
		kml.append("<visibility>1</visibility>\n");
		kml.append("<LineString>\n");
		kml.append("<tessellate>1</tessellate>\n");
		kml.append("<altitudeMode>clampToGround</altitudeMode>\n");
		kml.append("<coordinates>\n");
		for (Connection p : routeConnections) {
			kml.append(p.getLon1() + "," + p.getLat1() + " ");
		}
		kml.append("</coordinates>\n");
		kml.append("</LineString>\n");
		kml.append("</Placemark>");

		kml.append("</Document>\n");
		kml.append("</kml>\n");

		return kml.toString();
	}

	private String connectionToKML(Connection p, String styleUrl) {
		StringBuffer kml = new StringBuffer();
		kml.append("<Placemark>\n");
		kml.append("<name>" + p.getLabel() + "</name>\n");
		kml.append("<styleUrl>" + styleUrl + "</styleUrl>\n");
		kml.append("<LineString>\n");
		kml.append("<tessellate>1</tessellate>\n");
		kml.append("<altitudeMode>clampToGround</altitudeMode>\n");
		kml.append("<coordinates>\n");
		kml.append(p.getLon1() + "," + p.getLat1() + " ");
		kml.append(p.getLon2() + "," + p.getLat2() + "\n");
		kml.append("</coordinates>\n");
		kml.append("</LineString>\n");
		kml.append("</Placemark>");
		return kml.toString();
	}
}