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    }