sascha@2502: package de.intevation.flys.client.server; sascha@2502: sascha@2504: import de.intevation.artifacts.common.ArtifactNamespaceContext; sascha@2504: sascha@2504: import de.intevation.artifacts.common.utils.ClientProtocolUtils; sascha@2504: import de.intevation.artifacts.common.utils.JSON; sascha@2504: import de.intevation.artifacts.common.utils.StringUtils; sascha@2504: import de.intevation.artifacts.common.utils.XMLUtils; sascha@2504: sascha@2504: import de.intevation.artifacts.httpclient.exceptions.ConnectionException; sascha@2504: sascha@2504: import de.intevation.artifacts.httpclient.http.HttpClient; sascha@2504: import de.intevation.artifacts.httpclient.http.HttpClientImpl; sascha@2504: sascha@2506: import de.intevation.artifacts.httpclient.http.response.DocumentResponseHandler; sascha@2506: sascha@2504: import de.intevation.flys.client.shared.model.MapConfig; sascha@2510: import de.intevation.flys.client.shared.MapUtils; sascha@2504: sascha@2506: import java.io.BufferedInputStream; sascha@2506: import java.io.BufferedOutputStream; sascha@2506: import java.io.File; sascha@2506: import java.io.FileInputStream; sascha@2506: import java.io.FileOutputStream; sascha@2506: import java.io.IOException; sascha@2506: import java.io.InputStream; sascha@2506: import java.io.OutputStream; sascha@2506: sascha@2506: import java.util.ArrayList; sascha@2506: import java.util.Collections; sascha@2506: import java.util.LinkedHashMap; sascha@2506: import java.util.List; sascha@2506: import java.util.Map; sascha@2506: sascha@2506: import javax.servlet.ServletException; sascha@2506: sascha@2506: import javax.servlet.http.HttpServlet; sascha@2506: import javax.servlet.http.HttpServletRequest; sascha@2506: import javax.servlet.http.HttpServletResponse; sascha@2506: sascha@2502: import org.apache.log4j.Logger; sascha@2502: sascha@2502: import org.mapfish.print.MapPrinter; sascha@2502: sascha@2502: import org.mapfish.print.output.OutputFactory; sascha@2502: import org.mapfish.print.output.OutputFormat; sascha@2502: sascha@2502: import org.mapfish.print.utils.PJsonObject; sascha@2502: sascha@2503: import org.w3c.dom.Document; sascha@2504: import org.w3c.dom.Element; sascha@2504: import org.w3c.dom.NodeList; sascha@2503: sascha@2502: public class MapPrintServiceImpl sascha@2502: extends HttpServlet sascha@2502: { sascha@2502: private static final Logger log = sascha@2502: Logger.getLogger(MapPrintServiceImpl.class); sascha@2502: sascha@2506: protected static class Layer implements Comparable { sascha@2504: sascha@2506: protected int pos; sascha@2506: protected String url; sascha@2506: protected String layers; sascha@2506: protected String description; sascha@2504: sascha@2506: public Layer() { sascha@2506: } sascha@2504: sascha@2506: public boolean setup(Element element) { sascha@2504: sascha@2506: Element parent = (Element)element.getParentNode(); sascha@2506: String parentName = parent.getAttribute("name"); sascha@2506: if (!(parentName.equals("map") sascha@2506: || parentName.equals("floodmap"))) { sascha@2506: return false; sascha@2504: } sascha@2506: sascha@2506: String ns = ArtifactNamespaceContext.NAMESPACE_URI; sascha@2506: sascha@2506: String visible = element.getAttributeNS(ns, "visible"); sascha@2506: String active = element.getAttributeNS(ns, "active"); sascha@2506: sascha@2506: if (visible.equals("0") || active.equals("0")) { sascha@2506: return false; sascha@2506: } sascha@2506: sascha@2506: url = element.getAttributeNS(ns, "url"); sascha@2506: layers = element.getAttributeNS(ns, "layers"); sascha@2506: description = element.getAttributeNS(ns, "description"); sascha@2506: sascha@2506: try { sascha@2506: pos = Integer.parseInt(element.getAttributeNS(ns, "pos")); sascha@2506: } sascha@2506: catch (NumberFormatException nfe) { sascha@2506: return false; sascha@2506: } sascha@2506: sascha@2506: return true; sascha@2506: } sascha@2506: sascha@2506: public Map toMap() { sascha@2504: Map layer = new LinkedHashMap(); sascha@2504: sascha@2504: layer.put("type", "WMS"); sascha@2504: List subLayers = new ArrayList(1); sascha@2506: subLayers.add(layers); sascha@2504: layer.put("layers", subLayers); sascha@2506: layer.put("baseURL", url); sascha@2506: layer.put("format", "image/png"); // TODO: Make configurable. sascha@2504: sascha@2506: return layer; sascha@2506: } sascha@2506: sascha@2506: @Override sascha@2506: public int compareTo(Layer other) { sascha@2506: int d = pos - other.pos; sascha@2506: if (d < 0) return -1; sascha@2506: return d > 0 ? +1 : 0; sascha@2506: } sascha@2506: } // class Layer sascha@2506: sascha@2506: protected static String generateSpec( sascha@2506: Document descDocument, sascha@2508: MapConfig mapConfig, sascha@2508: Double minX, Double minY, sascha@2508: Double maxX, Double maxY sascha@2506: ) { sascha@2506: Map spec = new LinkedHashMap(); sascha@2506: spec.put("layout", "A4 portrait"); sascha@2506: spec.put("title", "FLYS Druck"); sascha@2506: spec.put("srs", "EPSG:" + mapConfig.getSrid()); sascha@2506: spec.put("dpi", Integer.valueOf(254)); sascha@2508: spec.put("units", "m"); sascha@2510: spec.put("geodaetic", "true"); sascha@2506: spec.put("outputFormat", "pdf"); sascha@2506: sascha@2506: String ns = ArtifactNamespaceContext.NAMESPACE_URI; sascha@2506: sascha@2510: List ls = new ArrayList(); sascha@2510: { Layer l = new Layer(); sascha@2510: sascha@2510: NodeList facets = descDocument.getElementsByTagNameNS(ns, "facet"); sascha@2510: sascha@2510: for (int i = 0, N = facets.getLength(); i < N; ++i) { sascha@2510: Element element = (Element)facets.item(i); sascha@2510: if (l.setup(element)) { sascha@2510: ls.add(l); sascha@2510: l = new Layer(); sascha@2510: } sascha@2506: } sascha@2506: } sascha@2506: sascha@2506: // Establish Z order. sascha@2506: Collections.sort(ls); sascha@2506: sascha@2506: List layers = new ArrayList(ls.size()); sascha@2506: sascha@2510: for (int i = ls.size()-1; i >= 0; --i) { sascha@2510: layers.add(ls.get(i).toMap()); sascha@2504: } sascha@2504: sascha@2504: spec.put("layers", layers); sascha@2504: spec.put("name", "Name"); sascha@2504: sascha@2504: List pages = new ArrayList(1); sascha@2504: sascha@2504: sascha@2504: Map page = new LinkedHashMap(); sascha@2504: sascha@2504: List bounds = new ArrayList(4); sascha@2510: bounds.add(minX); sascha@2510: bounds.add(minY); sascha@2510: bounds.add(maxX); sascha@2510: bounds.add(maxY); sascha@2510: page.put("bbox", bounds); sascha@2504: sascha@2510: /* sascha@2510: bounds.add(Double.valueOf((minX+maxX)*0.5)); sascha@2510: bounds.add(Double.valueOf((minY+maxY)*0.5)); sascha@2510: sascha@2510: page.put("center", bounds); sascha@2510: page.put("scale", Integer.valueOf(50000)); sascha@2510: */ sascha@2504: sascha@2504: page.put("mapTitle", "FLYS Karte"); sascha@2504: page.put("comment", "Eine mit FLYS gedruckte Karte."); sascha@2504: page.put("rotation", Integer.valueOf(0)); sascha@2504: sascha@2504: pages.add(page); sascha@2510: spec.put("pages", pages); sascha@2510: sascha@2510: List legends = new ArrayList(layers.size()); sascha@2510: sascha@2510: for (Layer layer: ls) { sascha@2510: Map legend = new LinkedHashMap(); sascha@2510: List classes = new ArrayList(1); sascha@2510: Map clazz = new LinkedHashMap(); sascha@2510: String lgu = MapUtils.getLegendGraphicUrl(layer.url, layer.layers); sascha@2510: clazz.put("icon", lgu); sascha@2510: clazz.put("name", layer.description); sascha@2510: classes.add(clazz); sascha@2510: legend.put("classes", classes); sascha@2510: legend.put("name", layer.description); sascha@2510: legends.add(legend); sascha@2510: } sascha@2510: sascha@2510: spec.put("legends", legends); sascha@2504: sascha@2504: return JSON.toJSONString(spec); sascha@2504: } sascha@2502: sascha@2502: sascha@2503: @Override sascha@2503: public void doGet(HttpServletRequest req, HttpServletResponse resp) sascha@2503: throws ServletException, IOException sascha@2503: { sascha@2503: log.info("MapPrintServiceImpl.doGet"); sascha@2503: sascha@2503: String uuid = req.getParameter("uuid"); sascha@2503: sascha@2503: if (uuid == null || !StringUtils.checkUUID(uuid)) { sascha@2503: throw new ServletException("Missing or misspelled UUID"); sascha@2503: } sascha@2503: sascha@2503: String minXS = req.getParameter("minx"); sascha@2503: String maxXS = req.getParameter("maxx"); sascha@2503: String minYS = req.getParameter("miny"); sascha@2503: String maxYS = req.getParameter("maxy"); sascha@2503: sascha@2508: Double minX = null; sascha@2508: Double maxX = null; sascha@2508: Double minY = null; sascha@2508: Double maxY = null; sascha@2503: sascha@2508: if (minXS != null && maxXS != null sascha@2515: && minYS != null && maxYS != null) { sascha@2515: log.debug("all parameters found -> parsing"); sascha@2508: try { sascha@2508: minX = Double.parseDouble(minXS); sascha@2508: minY = Double.parseDouble(minYS); sascha@2508: maxX = Double.parseDouble(maxXS); sascha@2508: maxY = Double.parseDouble(maxYS); sascha@2508: } sascha@2508: catch (NumberFormatException nfe) { sascha@2508: throw new ServletException("Misspelled minX, minY, maxX or maxY"); sascha@2508: } sascha@2503: } sascha@2503: sascha@2503: String mapType = req.getParameter("maptype"); sascha@2503: sascha@2508: if (mapType == null || !mapType.equals("floodmap")) { sascha@2508: mapType = "map"; sascha@2503: } sascha@2503: sascha@2503: String url = getURL(); sascha@2503: sascha@2506: Document requestOut = sascha@2506: ClientProtocolUtils.newOutCollectionDocument( sascha@2506: uuid, mapType, mapType); sascha@2506: Document requestDesc = sascha@2506: ClientProtocolUtils.newDescribeCollectionDocument(uuid); sascha@2503: sascha@2506: Document outDocument; sascha@2506: Document descDocument; sascha@2506: sascha@2503: try { sascha@2503: HttpClient client = new HttpClientImpl(url); sascha@2506: sascha@2508: descDocument = (Document)client.doCollectionAction( sascha@2508: requestDesc, uuid, new DocumentResponseHandler()); sascha@2508: sascha@2506: InputStream is = client.collectionOut( sascha@2506: requestOut, uuid, mapType); sascha@2506: sascha@2503: try { sascha@2506: outDocument = XMLUtils.parseDocument(is); sascha@2503: } sascha@2503: finally { sascha@2503: is.close(); sascha@2506: is = null; sascha@2503: } sascha@2506: sascha@2503: } sascha@2503: catch (ConnectionException ce) { sascha@2503: log.error(ce); sascha@2503: throw new ServletException(ce); sascha@2503: } sascha@2503: sascha@2508: MapConfig mapConfig = MapHelper.parseConfig(outDocument); sascha@2508: sascha@2508: if (minX == null) { sascha@2515: log.debug("parameters missing -> fallback to max extent"); sascha@2508: String [] parts = mapConfig.getMaxExtent().split("\\s+"); sascha@2508: if (parts.length < 4) { sascha@2508: throw new ServletException( sascha@2508: "Max extent has less than 4 values"); sascha@2508: } sascha@2508: try { sascha@2508: minX = Double.valueOf(parts[0]); sascha@2508: minY = Double.valueOf(parts[1]); sascha@2508: maxX = Double.valueOf(parts[2]); sascha@2508: maxY = Double.valueOf(parts[3]); sascha@2508: } sascha@2508: catch (NumberFormatException nfe) { sascha@2508: throw new ServletException(nfe); sascha@2508: } sascha@2508: } sascha@2510: if (log.isDebugEnabled()) { sascha@2510: log.debug("minX: " + minX); sascha@2510: log.debug("maxX: " + maxX); sascha@2510: log.debug("minY: " + minY); sascha@2510: log.debug("maxY: " + maxY); sascha@2510: } sascha@2508: sascha@2504: String spec = generateSpec( sascha@2506: descDocument, sascha@2508: mapConfig, sascha@2504: minX, minY, sascha@2504: maxX, maxY); sascha@2503: sascha@2504: if (log.isDebugEnabled()) { sascha@2510: log.debug("Generated spec:"); sascha@2504: log.debug(spec); sascha@2504: //System.err.println(spec); sascha@2504: } sascha@2503: sascha@2504: producePDF(spec, resp); sascha@2503: } sascha@2503: sascha@2503: protected String getURL() throws ServletException { sascha@2503: String url = getServletContext().getInitParameter("server-url"); sascha@2503: if (url == null) { sascha@2503: throw new ServletException("Missing server-url"); sascha@2503: } sascha@2503: return url; sascha@2503: } sascha@2503: sascha@2503: protected MapPrinter getMapPrinter() throws ServletException, IOException { sascha@2502: String configPath = getInitParameter("config"); sascha@2502: if (configPath == null) { sascha@2502: throw new ServletException("Missing configuration in web.xml"); sascha@2502: } sascha@2502: sascha@2502: File configFile = new File(configPath); sascha@2502: if (!configFile.isAbsolute()) { sascha@2502: configFile = new File(getServletContext().getRealPath(configPath)); sascha@2502: } sascha@2502: sascha@2502: if (!configFile.isFile() || !configFile.canRead()) { sascha@2502: throw new ServletException("Cannot read '" + configFile + "'"); sascha@2502: } sascha@2502: sascha@2502: return new MapPrinter(configFile); sascha@2502: } sascha@2502: sascha@2503: sascha@2503: protected void producePDF(String json, HttpServletResponse resp) sascha@2503: throws ServletException, IOException sascha@2502: { sascha@2504: PJsonObject jsonSpec = MapPrinter.parseSpec(json); sascha@2502: sascha@2502: MapPrinter printer = getMapPrinter(); sascha@2502: sascha@2502: OutputFormat outputFormat = OutputFactory.create( sascha@2502: printer.getConfig(), jsonSpec); sascha@2502: sascha@2502: resp.setHeader("Content-Disposition", "attachment;filename=print.pdf"); sascha@2502: resp.setHeader("Content-Type", "application/pdf"); sascha@2502: sascha@2502: // XXX: Streaming the generated PDF directly sascha@2502: // to the request out does not work. :-/ sascha@2502: File tmpFile = File.createTempFile("map-printing", null); sascha@2502: sascha@2502: try { sascha@2502: OutputStream out = sascha@2502: new BufferedOutputStream( sascha@2502: new FileOutputStream(tmpFile)); sascha@2502: try { sascha@2502: outputFormat.print(printer, jsonSpec, out, ""); sascha@2502: out.flush(); sascha@2502: } sascha@2502: catch (Exception e) { sascha@2502: log.error(e); sascha@2502: throw new ServletException(e); sascha@2502: } sascha@2502: finally { sascha@2502: printer.stop(); sascha@2502: out.close(); sascha@2502: } sascha@2502: InputStream in = sascha@2502: new BufferedInputStream( sascha@2502: new FileInputStream(tmpFile)); sascha@2502: out = resp.getOutputStream(); sascha@2502: try { sascha@2502: byte [] buf = new byte[4096]; sascha@2502: int r; sascha@2502: while ((r = in.read(buf)) >= 0) { sascha@2502: out.write(buf, 0, r); sascha@2502: } sascha@2502: out.flush(); sascha@2502: } sascha@2502: finally { sascha@2502: in.close(); sascha@2502: out.close(); sascha@2502: } sascha@2502: } sascha@2502: finally { sascha@2502: if (tmpFile.exists()) { sascha@2502: tmpFile.delete(); sascha@2502: } sascha@2502: } sascha@2502: } sascha@2502: } sascha@2502: // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :