Search in sources :

Example 1 with GetLegendGraphicRequest

use of org.geoserver.wms.GetLegendGraphicRequest in project sldeditor by robward-scisys.

the class LegendManager method createLegend.

/**
 * Creates the legend.
 *
 * @param sld the sld
 * @param heading the heading
 * @param filename the filename
 * @param separateSymbolizers the separate symbolizers
 * @return the map
 */
public Map<String, BufferedImage> createLegend(StyledLayerDescriptor sld, String heading, String filename, boolean separateSymbolizers) {
    Map<String, BufferedImage> imageMap = new HashMap<String, BufferedImage>();
    // 
    // Set legend options
    // 
    Map<String, Object> legendOptions = new HashMap<String, Object>();
    if (heading != null) {
        legendOptions.put("heading", heading);
    }
    if (filename != null) {
        legendOptions.put("filename", filename);
    }
    if (legendOptionData == null) {
        legendOptionData = new LegendOptionData();
    }
    GetLegendGraphicRequest request = new GetLegendGraphicRequest();
    request.setWidth(legendOptionData.getImageWidth());
    request.setHeight(legendOptionData.getImageHeight());
    request.setTransparent(legendOptionData.isTransparent());
    request.setStrict(false);
    legendOptions.put("bgColor", ColourUtils.fromColour(legendOptionData.getBackgroundColour()));
    legendOptions.put("fontColor", ColourUtils.fromColour(legendOptionData.getLabelFontColour()));
    // 
    // Label Font
    // 
    Font font = legendOptionData.getLabelFont();
    legendOptions.put("fontName", font.getFontName());
    String styleValue = null;
    if ((font.getStyle() & java.awt.Font.BOLD) == java.awt.Font.BOLD) {
        styleValue = "bold";
    }
    if ((font.getStyle() & java.awt.Font.ITALIC) == java.awt.Font.ITALIC) {
        styleValue = "italic";
    }
    if (styleValue != null) {
        legendOptions.put("fontStyle", styleValue);
    }
    legendOptions.put("fontSize", String.valueOf(font.getSize()));
    legendOptions.put("dpi", Integer.valueOf(legendOptionData.getDpi()));
    legendOptions.put("fontAntiAliasing", getBooleanValueOnOff(legendOptionData.isFontAntiAliasing()));
    legendOptions.put("forceLabels", getBooleanValueOnOff(legendOptionData.isShowLabels()));
    legendOptions.put("forceTitles", getBooleanValueOnOff(legendOptionData.isShowTitle()));
    legendOptions.put("bandInfo", getBooleanValueTrueFalse(legendOptionData.isBandInformation()));
    legendOptions.put("border", getBooleanValueTrueFalse(legendOptionData.isBorder()));
    legendOptions.put("borderColor", ColourUtils.fromColour(legendOptionData.getBorderColour()));
    legendOptions.put("imageSizeFactor", String.valueOf(legendOptionData.getImageSize() / 100.0));
    request.setLegendOptions(legendOptions);
    if (sld != null) {
        Map<String, Style> styleMap = new LinkedHashMap<String, Style>();
        StyledLayer selectedStyledLayer = SelectedSymbol.getInstance().getStyledLayer();
        Style selectedStyle = SelectedSymbol.getInstance().getStyle();
        if (selectedStyle != null) {
            createSingleStyleLegend(styleMap, selectedStyledLayer, selectedStyle);
        } else {
            createMultipleStyleLegend(sld, styleMap, selectedStyledLayer);
        }
        // Merge symbolizers into 1 image
        if (!separateSymbolizers) {
            for (String key : styleMap.keySet()) {
                Style style = styleMap.get(key);
                if (!style.featureTypeStyles().isEmpty()) {
                    FeatureTypeStyle featureTypeStyle = style.featureTypeStyles().get(0);
                    if (featureTypeStyle != null) {
                        if (!featureTypeStyle.rules().isEmpty()) {
                            LegendRequest legendEntryRequest = request.new LegendRequest();
                            request.getLegends().add(legendEntryRequest);
                            legendEntryRequest.setTitle(key);
                            legendEntryRequest.setStyle(style);
                        }
                    }
                }
            }
            BufferedImage legendGraphic = null;
            try {
                legendGraphic = legendBuilder.buildLegendGraphic(request);
            } catch (Exception e) {
            // Ignore
            }
            imageMap.put("", legendGraphic);
        } else {
            for (String key : styleMap.keySet()) {
                request.getLegends().clear();
                LegendRequest legendEntryRequest = request.new LegendRequest();
                legendEntryRequest.setStyle(styleMap.get(key));
                legendEntryRequest.setStyleName(key);
                request.getLegends().add(legendEntryRequest);
                BufferedImage legendGraphic = null;
                try {
                    legendGraphic = legendBuilder.buildLegendGraphic(request);
                } catch (Exception e) {
                // Ignore
                }
                imageMap.put(key, legendGraphic);
            }
        }
    }
    return imageMap;
}
Also used : GetLegendGraphicRequest(org.geoserver.wms.GetLegendGraphicRequest) HashMap(java.util.HashMap) LinkedHashMap(java.util.LinkedHashMap) StyledLayer(org.geotools.styling.StyledLayer) LegendOptionData(com.sldeditor.ui.legend.option.LegendOptionData) BufferedImage(java.awt.image.BufferedImage) Font(java.awt.Font) IIOInvalidTreeException(javax.imageio.metadata.IIOInvalidTreeException) IOException(java.io.IOException) LinkedHashMap(java.util.LinkedHashMap) Style(org.geotools.styling.Style) FeatureTypeStyle(org.geotools.styling.FeatureTypeStyle) FeatureTypeStyle(org.geotools.styling.FeatureTypeStyle) LegendRequest(org.geoserver.wms.GetLegendGraphicRequest.LegendRequest)

Aggregations

LegendOptionData (com.sldeditor.ui.legend.option.LegendOptionData)1 Font (java.awt.Font)1 BufferedImage (java.awt.image.BufferedImage)1 IOException (java.io.IOException)1 HashMap (java.util.HashMap)1 LinkedHashMap (java.util.LinkedHashMap)1 IIOInvalidTreeException (javax.imageio.metadata.IIOInvalidTreeException)1 GetLegendGraphicRequest (org.geoserver.wms.GetLegendGraphicRequest)1 LegendRequest (org.geoserver.wms.GetLegendGraphicRequest.LegendRequest)1 FeatureTypeStyle (org.geotools.styling.FeatureTypeStyle)1 Style (org.geotools.styling.Style)1 StyledLayer (org.geotools.styling.StyledLayer)1