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;
}
Aggregations