001 //$HeadURL: svn+ssh://jwilden@svn.wald.intevation.org/deegree/base/branches/2.5_testing/src/org/deegree/portal/standard/wms/control/LegendListener.java $ 002 /*---------------------------------------------------------------------------- 003 This file is part of deegree, http://deegree.org/ 004 Copyright (C) 2001-2009 by: 005 Department of Geography, University of Bonn 006 and 007 lat/lon GmbH 008 009 This library is free software; you can redistribute it and/or modify it under 010 the terms of the GNU Lesser General Public License as published by the Free 011 Software Foundation; either version 2.1 of the License, or (at your option) 012 any later version. 013 This library is distributed in the hope that it will be useful, but WITHOUT 014 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS 015 FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more 016 details. 017 You should have received a copy of the GNU Lesser General Public License 018 along with this library; if not, write to the Free Software Foundation, Inc., 019 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA 020 021 Contact information: 022 023 lat/lon GmbH 024 Aennchenstr. 19, 53177 Bonn 025 Germany 026 http://lat-lon.de/ 027 028 Department of Geography, University of Bonn 029 Prof. Dr. Klaus Greve 030 Postfach 1147, 53001 Bonn 031 Germany 032 http://www.geographie.uni-bonn.de/deegree/ 033 034 e-mail: info@deegree.org 035 ----------------------------------------------------------------------------*/ 036 package org.deegree.portal.standard.wms.control; 037 038 import java.awt.Color; 039 import java.awt.Graphics; 040 import java.awt.Rectangle; 041 import java.awt.geom.Rectangle2D; 042 import java.awt.image.BufferedImage; 043 import java.io.FileOutputStream; 044 import java.io.IOException; 045 import java.net.MalformedURLException; 046 import java.net.URL; 047 import java.util.ArrayList; 048 import java.util.HashMap; 049 import java.util.Map; 050 import java.util.StringTokenizer; 051 052 import javax.servlet.http.HttpServletRequest; 053 import javax.servlet.http.HttpSession; 054 055 import org.deegree.enterprise.control.FormEvent; 056 import org.deegree.enterprise.control.RPCMember; 057 import org.deegree.enterprise.control.RPCMethodCall; 058 import org.deegree.enterprise.control.RPCParameter; 059 import org.deegree.enterprise.control.RPCStruct; 060 import org.deegree.enterprise.control.RPCWebEvent; 061 import org.deegree.framework.log.ILogger; 062 import org.deegree.framework.log.LoggerFactory; 063 import org.deegree.framework.util.IDGenerator; 064 import org.deegree.framework.util.ImageUtils; 065 import org.deegree.ogcwebservices.InconsistentRequestException; 066 import org.deegree.ogcwebservices.wms.operation.GetLegendGraphic; 067 import org.deegree.portal.context.GeneralExtension; 068 import org.deegree.portal.context.IOSettings; 069 import org.deegree.portal.context.ViewContext; 070 071 /** 072 * will be called if the client forces a dynamic legend. 073 * 074 * 075 * @author <a href="mailto:lupp@lat-lon.de">Katharina Lupp</a> 076 * @version $Revision: 21678 $ $Date: 2009-12-29 14:44:02 +0100 (Di, 29 Dez 2009) $ 077 */ 078 public class LegendListener extends AbstractMapListener { 079 080 private static final ILogger LOG = LoggerFactory.getLogger( LegendListener.class ); 081 082 /** 083 * the method will be called if a zoomout action/event occurs. 084 */ 085 public void actionPerformed( FormEvent event ) { 086 087 super.actionPerformed( event ); 088 089 RPCWebEvent rpc = (RPCWebEvent) event; 090 RPCMethodCall mc = rpc.getRPCMethodCall(); 091 RPCParameter[] para = mc.getParameters(); 092 RPCStruct struct = (RPCStruct) para[0].getValue(); 093 HttpSession session = ( (HttpServletRequest) this.getRequest() ).getSession( true ); 094 ViewContext vc = (ViewContext) session.getAttribute( "DefaultMapContext" ); 095 096 Map<String, String>[] model = createWMSRequestModel( struct ); 097 098 try { 099 GetLegendGraphic legendParam = getLegendRequestParameter(); 100 Map<String, Object> symbols = setLegend( legendParam, model ); 101 Rectangle rect = calcLegendSize( symbols ); 102 BufferedImage bi = new BufferedImage( rect.width + 30, rect.height + 50, BufferedImage.TYPE_INT_RGB ); 103 bi = drawSymbolsToBI( symbols, bi ); 104 saveImage( vc, bi ); 105 } catch ( Exception e ) { 106 LOG.logError( "Error occurred in PrintListener: ", e ); 107 } 108 109 } 110 111 private Rectangle calcLegendSize( Map<String, Object> map ) { 112 113 String[] layers = (String[]) map.get( "NAMES" ); 114 BufferedImage[] legs = (BufferedImage[]) map.get( "IMAGES" ); 115 116 int w = 0; 117 int h = 0; 118 for ( int i = 0; i < layers.length; i++ ) { 119 h += legs[i].getHeight() + 6; 120 Graphics g = legs[i].getGraphics(); 121 Rectangle2D rect = g.getFontMetrics().getStringBounds( layers[i], g ); 122 g.dispose(); 123 if ( rect.getWidth() > w ) { 124 w = (int) rect.getWidth(); 125 } 126 } 127 w += 50; 128 129 return new Rectangle( w, h ); 130 } 131 132 private BufferedImage drawSymbolsToBI( Map<String, Object> map, BufferedImage bi ) { 133 134 Graphics g = bi.getGraphics(); 135 g.setColor( Color.WHITE ); 136 g.fillRect( 1, 1, bi.getWidth() - 2, bi.getHeight() - 2 ); 137 138 String[] layers = (String[]) map.get( "NAMES" ); 139 BufferedImage[] legs = (BufferedImage[]) map.get( "IMAGES" ); 140 int h = 5; 141 for ( int i = layers.length - 1; i >= 0; i-- ) { 142 g.drawImage( legs[i], 20, h, null ); 143 g.setColor( Color.BLACK ); 144 if ( legs[i].getHeight() < 50 ) { 145 g.drawString( layers[i], 30 + legs[i].getWidth(), h + (int) ( legs[i].getHeight() / 1.2 ) ); 146 } 147 h += legs[i].getHeight() + 5; 148 } 149 g.dispose(); 150 return bi; 151 } 152 153 @SuppressWarnings("unchecked") 154 private Map<String, String>[] createWMSRequestModel( RPCStruct struct ) { 155 156 RPCMember[] member = struct.getMembers(); 157 158 Map<String, String>[] getMR = new HashMap[member.length]; 159 for ( int i = 0; i < member.length; i++ ) { 160 String request = (String) member[i].getValue(); 161 getMR[i] = toMap( request ); 162 StringTokenizer st = new StringTokenizer( request, "?" ); 163 getMR[i].put( "URL", st.nextToken() ); 164 } 165 return getMR; 166 } 167 168 private void saveImage( ViewContext vc, BufferedImage bg ) { 169 170 GeneralExtension ge = vc.getGeneral().getExtension(); 171 IOSettings ios = ge.getIOSettings(); 172 String dir = ios.getPrintDirectory().getDirectoryName(); 173 String format = "jpeg"; 174 long l = IDGenerator.getInstance().generateUniqueID(); 175 String file = "legend" + l + '.' + format; 176 try { 177 FileOutputStream fos = new FileOutputStream( dir + "/" + file ); 178 179 ImageUtils.saveImage( bg, fos, format, 1 ); 180 181 fos.close(); 182 } catch ( Exception e ) { 183 LOG.logError( "Error occurred in saving legend image: ", e ); 184 } 185 int pos = dir.lastIndexOf( '/' ); 186 String access = "./" + dir.substring( pos + 1, dir.length() ) + "/" + file; 187 this.getRequest().setAttribute( "DYNLEGENDIMAGE", access ); 188 189 } 190 191 private GetLegendGraphic getLegendRequestParameter() 192 throws InconsistentRequestException { 193 194 HashMap<String, String> legend = toMap( "VERSION=1.1.1&REQUEST=GetLegendGraphic&FORMAT=image/jpeg&WIDTH=50&HEIGHT=50&" 195 + "EXCEPTIONS=application/vnd.ogc.se_inimage&LAYER=europe:major_rivers&STYLE=default&" 196 + "SLD=file:///styles.xml" ); 197 legend.put( "ID", "1" ); 198 return GetLegendGraphic.create( legend ); 199 200 } 201 202 /** 203 * creates legend 204 */ 205 private Map<String, Object> setLegend( GetLegendGraphic glr, Map<String, String>[] model ) 206 throws MalformedURLException, IOException { 207 208 ArrayList<String> list1 = new ArrayList<String>(); 209 ArrayList<BufferedImage> list2 = new ArrayList<BufferedImage>(); 210 211 StringTokenizer st = null; 212 String format = glr.getFormat(); 213 if ( format.equals( "image/jpg" ) ) 214 format = "image/jpeg"; 215 String legendURL = ""; 216 int lgHeight = 0; 217 for ( int i = 0; i < model.length; i++ ) { 218 219 String style = (String) model[i].get( "STYLE" ); 220 if ( style != null ) { 221 st = new StringTokenizer( style, "," ); 222 style = st.nextToken(); 223 } else 224 style = "default"; 225 st = new StringTokenizer( (String) model[i].get( "LAYERS" ), "," ); 226 227 while ( st.hasMoreTokens() ) { 228 String layer = st.nextToken(); 229 legendURL = setLegendURL( layer, style, format, glr, model[0] ); 230 lgHeight = lgHeight + 30; 231 BufferedImage legendGraphic = ImageUtils.loadImage( new URL( legendURL ) ); 232 list1.add( layer ); 233 list2.add( legendGraphic ); 234 } 235 } 236 237 String[] layers = list1.toArray( new String[list1.size()] ); 238 BufferedImage[] legs = list2.toArray( new BufferedImage[list2.size()] ); 239 Map<String, Object> map = new HashMap<String, Object>(); 240 map.put( "NAMES", layers ); 241 map.put( "IMAGES", legs ); 242 return map; 243 } 244 245 private String setLegendURL( String layer, String style, String format, GetLegendGraphic glr, 246 Map<String, String> model ) { 247 248 StringBuffer sb = new StringBuffer( 500 ); 249 sb.append( model.get( "URL" ) ).append( '?' ); 250 sb.append( "&VERSION=" ).append( glr.getVersion() ); 251 sb.append( "&REQUEST=GetLegendGraphic" ); 252 sb.append( "&FORMAT=" ).append( format ); 253 sb.append( "&WIDTH=15" ); 254 sb.append( "&HEIGHT=15&EXCEPTIONS=application/vnd.ogc.se_inimage" ); 255 sb.append( "&LAYER=" ).append( layer ); 256 sb.append( "&STYLE=" ).append( style ); 257 258 return sb.toString(); 259 } 260 261 }