source: josm/trunk/src/org/openstreetmap/josm/tools/ImageProvider.java@ 11516

Last change on this file since 11516 was 11399, checked in by Don-vip, 7 years ago

sonar - squid:S2095 - Resources should be closed

  • Property svn:eol-style set to native
File size: 73.9 KB
Line 
1// License: GPL. For details, see LICENSE file.
2package org.openstreetmap.josm.tools;
3
4import static org.openstreetmap.josm.tools.I18n.tr;
5
6import java.awt.Color;
7import java.awt.Cursor;
8import java.awt.Dimension;
9import java.awt.Graphics;
10import java.awt.Graphics2D;
11import java.awt.GraphicsEnvironment;
12import java.awt.Image;
13import java.awt.Point;
14import java.awt.RenderingHints;
15import java.awt.Toolkit;
16import java.awt.Transparency;
17import java.awt.image.BufferedImage;
18import java.awt.image.ColorModel;
19import java.awt.image.FilteredImageSource;
20import java.awt.image.ImageFilter;
21import java.awt.image.ImageProducer;
22import java.awt.image.RGBImageFilter;
23import java.awt.image.WritableRaster;
24import java.io.ByteArrayInputStream;
25import java.io.File;
26import java.io.IOException;
27import java.io.InputStream;
28import java.io.StringReader;
29import java.net.URI;
30import java.net.URL;
31import java.nio.charset.StandardCharsets;
32import java.util.ArrayList;
33import java.util.Arrays;
34import java.util.Base64;
35import java.util.Collection;
36import java.util.HashMap;
37import java.util.Hashtable;
38import java.util.Iterator;
39import java.util.LinkedList;
40import java.util.List;
41import java.util.Map;
42import java.util.TreeSet;
43import java.util.concurrent.CompletableFuture;
44import java.util.concurrent.ExecutorService;
45import java.util.concurrent.Executors;
46import java.util.regex.Matcher;
47import java.util.regex.Pattern;
48import java.util.zip.ZipEntry;
49import java.util.zip.ZipFile;
50
51import javax.imageio.IIOException;
52import javax.imageio.ImageIO;
53import javax.imageio.ImageReadParam;
54import javax.imageio.ImageReader;
55import javax.imageio.metadata.IIOMetadata;
56import javax.imageio.stream.ImageInputStream;
57import javax.swing.ImageIcon;
58import javax.xml.parsers.ParserConfigurationException;
59
60import org.openstreetmap.josm.Main;
61import org.openstreetmap.josm.data.osm.DataSet;
62import org.openstreetmap.josm.data.osm.OsmPrimitive;
63import org.openstreetmap.josm.data.osm.OsmPrimitiveType;
64import org.openstreetmap.josm.gui.mappaint.MapPaintStyles;
65import org.openstreetmap.josm.gui.mappaint.Range;
66import org.openstreetmap.josm.gui.mappaint.StyleElementList;
67import org.openstreetmap.josm.gui.mappaint.styleelement.MapImage;
68import org.openstreetmap.josm.gui.mappaint.styleelement.NodeElement;
69import org.openstreetmap.josm.gui.mappaint.styleelement.StyleElement;
70import org.openstreetmap.josm.gui.tagging.presets.TaggingPreset;
71import org.openstreetmap.josm.gui.tagging.presets.TaggingPresets;
72import org.openstreetmap.josm.gui.util.GuiSizesHelper;
73import org.openstreetmap.josm.io.CachedFile;
74import org.openstreetmap.josm.plugins.PluginHandler;
75import org.w3c.dom.Element;
76import org.w3c.dom.Node;
77import org.w3c.dom.NodeList;
78import org.xml.sax.Attributes;
79import org.xml.sax.InputSource;
80import org.xml.sax.SAXException;
81import org.xml.sax.XMLReader;
82import org.xml.sax.helpers.DefaultHandler;
83
84import com.kitfox.svg.SVGDiagram;
85import com.kitfox.svg.SVGException;
86import com.kitfox.svg.SVGUniverse;
87
88/**
89 * Helper class to support the application with images.
90 *
91 * How to use:
92 *
93 * <code>ImageIcon icon = new ImageProvider(name).setMaxSize(ImageSizes.MAP).get();</code>
94 * (there are more options, see below)
95 *
96 * short form:
97 * <code>ImageIcon icon = ImageProvider.get(name);</code>
98 *
99 * @author imi
100 */
101public class ImageProvider {
102
103 // CHECKSTYLE.OFF: SingleSpaceSeparator
104 private static final String HTTP_PROTOCOL = "http://";
105 private static final String HTTPS_PROTOCOL = "https://";
106 private static final String WIKI_PROTOCOL = "wiki://";
107 // CHECKSTYLE.ON: SingleSpaceSeparator
108
109 /**
110 * Supported image types
111 */
112 public enum ImageType {
113 /** Scalable vector graphics */
114 SVG,
115 /** Everything else, e.g. png, gif (must be supported by Java) */
116 OTHER
117 }
118
119 /**
120 * Supported image sizes
121 * @since 7687
122 */
123 public enum ImageSizes {
124 /** SMALL_ICON value of an Action */
125 SMALLICON(Main.pref.getInteger("iconsize.smallicon", 16)),
126 /** LARGE_ICON_KEY value of an Action */
127 LARGEICON(Main.pref.getInteger("iconsize.largeicon", 24)),
128 /** map icon */
129 MAP(Main.pref.getInteger("iconsize.map", 16)),
130 /** map icon maximum size */
131 MAPMAX(Main.pref.getInteger("iconsize.mapmax", 48)),
132 /** cursor icon size */
133 CURSOR(Main.pref.getInteger("iconsize.cursor", 32)),
134 /** cursor overlay icon size */
135 CURSOROVERLAY(CURSOR),
136 /** menu icon size */
137 MENU(SMALLICON),
138 /** menu icon size in popup menus
139 * @since 8323
140 */
141 POPUPMENU(LARGEICON),
142 /** Layer list icon size
143 * @since 8323
144 */
145 LAYER(Main.pref.getInteger("iconsize.layer", 16)),
146 /** Toolbar button icon size
147 * @since 9253
148 */
149 TOOLBAR(LARGEICON),
150 /** Side button maximum height
151 * @since 9253
152 */
153 SIDEBUTTON(Main.pref.getInteger("iconsize.sidebutton", 20)),
154 /** Settings tab icon size
155 * @since 9253
156 */
157 SETTINGS_TAB(Main.pref.getInteger("iconsize.settingstab", 48)),
158 /**
159 * The default image size
160 * @since 9705
161 */
162 DEFAULT(Main.pref.getInteger("iconsize.default", 24)),
163 /**
164 * Splash dialog logo size
165 * @since 10358
166 */
167 SPLASH_LOGO(128, 129),
168 /**
169 * About dialog logo size
170 * @since 10358
171 */
172 ABOUT_LOGO(256, 258);
173
174 private final int virtualWidth;
175 private final int virtualHeight;
176
177 ImageSizes(int imageSize) {
178 this.virtualWidth = imageSize;
179 this.virtualHeight = imageSize;
180 }
181
182 ImageSizes(int width, int height) {
183 this.virtualWidth = width;
184 this.virtualHeight = height;
185 }
186
187 ImageSizes(ImageSizes that) {
188 this.virtualWidth = that.virtualWidth;
189 this.virtualHeight = that.virtualHeight;
190 }
191
192 /**
193 * Returns the image width in virtual pixels
194 * @return the image width in virtual pixels
195 * @since 9705
196 */
197 public int getVirtualWidth() {
198 return virtualWidth;
199 }
200
201 /**
202 * Returns the image height in virtual pixels
203 * @return the image height in virtual pixels
204 * @since 9705
205 */
206 public int getVirtualHeight() {
207 return virtualHeight;
208 }
209
210 /**
211 * Returns the image width in pixels to use for display
212 * @return the image width in pixels to use for display
213 * @since 10484
214 */
215 public int getAdjustedWidth() {
216 return GuiSizesHelper.getSizeDpiAdjusted(virtualWidth);
217 }
218
219 /**
220 * Returns the image height in pixels to use for display
221 * @return the image height in pixels to use for display
222 * @since 10484
223 */
224 public int getAdjustedHeight() {
225 return GuiSizesHelper.getSizeDpiAdjusted(virtualHeight);
226 }
227
228 /**
229 * Returns the image size as dimension
230 * @return the image size as dimension
231 * @since 9705
232 */
233 public Dimension getImageDimension() {
234 return new Dimension(virtualWidth, virtualHeight);
235 }
236 }
237
238 /**
239 * Property set on {@code BufferedImage} returned by {@link #makeImageTransparent}.
240 * @since 7132
241 */
242 public static final String PROP_TRANSPARENCY_FORCED = "josm.transparency.forced";
243
244 /**
245 * Property set on {@code BufferedImage} returned by {@link #read} if metadata is required.
246 * @since 7132
247 */
248 public static final String PROP_TRANSPARENCY_COLOR = "josm.transparency.color";
249
250 /** directories in which images are searched */
251 protected Collection<String> dirs;
252 /** caching identifier */
253 protected String id;
254 /** sub directory the image can be found in */
255 protected String subdir;
256 /** image file name */
257 protected String name;
258 /** archive file to take image from */
259 protected File archive;
260 /** directory inside the archive */
261 protected String inArchiveDir;
262 /** virtual width of the resulting image, -1 when original image data should be used */
263 protected int virtualWidth = -1;
264 /** virtual height of the resulting image, -1 when original image data should be used */
265 protected int virtualHeight = -1;
266 /** virtual maximum width of the resulting image, -1 for no restriction */
267 protected int virtualMaxWidth = -1;
268 /** virtual maximum height of the resulting image, -1 for no restriction */
269 protected int virtualMaxHeight = -1;
270 /** In case of errors do not throw exception but return <code>null</code> for missing image */
271 protected boolean optional;
272 /** <code>true</code> if warnings should be suppressed */
273 protected boolean suppressWarnings;
274 /** list of class loaders to take images from */
275 protected Collection<ClassLoader> additionalClassLoaders;
276 /** ordered list of overlay images */
277 protected List<ImageOverlay> overlayInfo;
278 /** <code>true</code> if icon must be grayed out */
279 protected boolean isDisabled;
280
281 private static SVGUniverse svgUniverse;
282
283 /**
284 * The icon cache
285 */
286 private static final Map<String, ImageResource> cache = new HashMap<>();
287
288 /**
289 * Caches the image data for rotated versions of the same image.
290 */
291 private static final Map<Image, Map<Long, ImageResource>> ROTATE_CACHE = new HashMap<>();
292
293 private static final ExecutorService IMAGE_FETCHER =
294 Executors.newSingleThreadExecutor(Utils.newThreadFactory("image-fetcher-%d", Thread.NORM_PRIORITY));
295
296 /**
297 * Constructs a new {@code ImageProvider} from a filename in a given directory.
298 * @param subdir subdirectory the image lies in
299 * @param name the name of the image. If it does not end with '.png' or '.svg',
300 * both extensions are tried.
301 */
302 public ImageProvider(String subdir, String name) {
303 this.subdir = subdir;
304 this.name = name;
305 }
306
307 /**
308 * Constructs a new {@code ImageProvider} from a filename.
309 * @param name the name of the image. If it does not end with '.png' or '.svg',
310 * both extensions are tried.
311 */
312 public ImageProvider(String name) {
313 this.name = name;
314 }
315
316 /**
317 * Constructs a new {@code ImageProvider} from an existing one.
318 * @param image the existing image provider to be copied
319 * @since 8095
320 */
321 public ImageProvider(ImageProvider image) {
322 this.dirs = image.dirs;
323 this.id = image.id;
324 this.subdir = image.subdir;
325 this.name = image.name;
326 this.archive = image.archive;
327 this.inArchiveDir = image.inArchiveDir;
328 this.virtualWidth = image.virtualWidth;
329 this.virtualHeight = image.virtualHeight;
330 this.virtualMaxWidth = image.virtualMaxWidth;
331 this.virtualMaxHeight = image.virtualMaxHeight;
332 this.optional = image.optional;
333 this.suppressWarnings = image.suppressWarnings;
334 this.additionalClassLoaders = image.additionalClassLoaders;
335 this.overlayInfo = image.overlayInfo;
336 this.isDisabled = image.isDisabled;
337 }
338
339 /**
340 * Directories to look for the image.
341 * @param dirs The directories to look for.
342 * @return the current object, for convenience
343 */
344 public ImageProvider setDirs(Collection<String> dirs) {
345 this.dirs = dirs;
346 return this;
347 }
348
349 /**
350 * Set an id used for caching.
351 * If name starts with <tt>http://</tt> Id is not used for the cache.
352 * (A URL is unique anyway.)
353 * @param id the id for the cached image
354 * @return the current object, for convenience
355 */
356 public ImageProvider setId(String id) {
357 this.id = id;
358 return this;
359 }
360
361 /**
362 * Specify a zip file where the image is located.
363 *
364 * (optional)
365 * @param archive zip file where the image is located
366 * @return the current object, for convenience
367 */
368 public ImageProvider setArchive(File archive) {
369 this.archive = archive;
370 return this;
371 }
372
373 /**
374 * Specify a base path inside the zip file.
375 *
376 * The subdir and name will be relative to this path.
377 *
378 * (optional)
379 * @param inArchiveDir path inside the archive
380 * @return the current object, for convenience
381 */
382 public ImageProvider setInArchiveDir(String inArchiveDir) {
383 this.inArchiveDir = inArchiveDir;
384 return this;
385 }
386
387 /**
388 * Add an overlay over the image. Multiple overlays are possible.
389 *
390 * @param overlay overlay image and placement specification
391 * @return the current object, for convenience
392 * @since 8095
393 */
394 public ImageProvider addOverlay(ImageOverlay overlay) {
395 if (overlayInfo == null) {
396 overlayInfo = new LinkedList<>();
397 }
398 overlayInfo.add(overlay);
399 return this;
400 }
401
402 /**
403 * Set the dimensions of the image.
404 *
405 * If not specified, the original size of the image is used.
406 * The width part of the dimension can be -1. Then it will only set the height but
407 * keep the aspect ratio. (And the other way around.)
408 * @param size final dimensions of the image
409 * @return the current object, for convenience
410 */
411 public ImageProvider setSize(Dimension size) {
412 this.virtualWidth = size.width;
413 this.virtualHeight = size.height;
414 return this;
415 }
416
417 /**
418 * Set the dimensions of the image.
419 *
420 * If not specified, the original size of the image is used.
421 * @param size final dimensions of the image
422 * @return the current object, for convenience
423 * @since 7687
424 */
425 public ImageProvider setSize(ImageSizes size) {
426 return setSize(size.getImageDimension());
427 }
428
429 /**
430 * Set the dimensions of the image.
431 *
432 * @param width final width of the image
433 * @param height final height of the image
434 * @return the current object, for convenience
435 * @since 10358
436 */
437 public ImageProvider setSize(int width, int height) {
438 this.virtualWidth = width;
439 this.virtualHeight = height;
440 return this;
441 }
442
443 /**
444 * Set image width
445 * @param width final width of the image
446 * @return the current object, for convenience
447 * @see #setSize
448 */
449 public ImageProvider setWidth(int width) {
450 this.virtualWidth = width;
451 return this;
452 }
453
454 /**
455 * Set image height
456 * @param height final height of the image
457 * @return the current object, for convenience
458 * @see #setSize
459 */
460 public ImageProvider setHeight(int height) {
461 this.virtualHeight = height;
462 return this;
463 }
464
465 /**
466 * Limit the maximum size of the image.
467 *
468 * It will shrink the image if necessary, but keep the aspect ratio.
469 * The given width or height can be -1 which means this direction is not bounded.
470 *
471 * 'size' and 'maxSize' are not compatible, you should set only one of them.
472 * @param maxSize maximum image size
473 * @return the current object, for convenience
474 */
475 public ImageProvider setMaxSize(Dimension maxSize) {
476 this.virtualMaxWidth = maxSize.width;
477 this.virtualMaxHeight = maxSize.height;
478 return this;
479 }
480
481 /**
482 * Limit the maximum size of the image.
483 *
484 * It will shrink the image if necessary, but keep the aspect ratio.
485 * The given width or height can be -1 which means this direction is not bounded.
486 *
487 * This function sets value using the most restrictive of the new or existing set of
488 * values.
489 *
490 * @param maxSize maximum image size
491 * @return the current object, for convenience
492 * @see #setMaxSize(Dimension)
493 */
494 public ImageProvider resetMaxSize(Dimension maxSize) {
495 if (this.virtualMaxWidth == -1 || maxSize.width < this.virtualMaxWidth) {
496 this.virtualMaxWidth = maxSize.width;
497 }
498 if (this.virtualMaxHeight == -1 || maxSize.height < this.virtualMaxHeight) {
499 this.virtualMaxHeight = maxSize.height;
500 }
501 return this;
502 }
503
504 /**
505 * Limit the maximum size of the image.
506 *
507 * It will shrink the image if necessary, but keep the aspect ratio.
508 * The given width or height can be -1 which means this direction is not bounded.
509 *
510 * 'size' and 'maxSize' are not compatible, you should set only one of them.
511 * @param size maximum image size
512 * @return the current object, for convenience
513 * @since 7687
514 */
515 public ImageProvider setMaxSize(ImageSizes size) {
516 return setMaxSize(size.getImageDimension());
517 }
518
519 /**
520 * Convenience method, see {@link #setMaxSize(Dimension)}.
521 * @param maxSize maximum image size
522 * @return the current object, for convenience
523 */
524 public ImageProvider setMaxSize(int maxSize) {
525 return this.setMaxSize(new Dimension(maxSize, maxSize));
526 }
527
528 /**
529 * Limit the maximum width of the image.
530 * @param maxWidth maximum image width
531 * @return the current object, for convenience
532 * @see #setMaxSize
533 */
534 public ImageProvider setMaxWidth(int maxWidth) {
535 this.virtualMaxWidth = maxWidth;
536 return this;
537 }
538
539 /**
540 * Limit the maximum height of the image.
541 * @param maxHeight maximum image height
542 * @return the current object, for convenience
543 * @see #setMaxSize
544 */
545 public ImageProvider setMaxHeight(int maxHeight) {
546 this.virtualMaxHeight = maxHeight;
547 return this;
548 }
549
550 /**
551 * Decide, if an exception should be thrown, when the image cannot be located.
552 *
553 * Set to true, when the image URL comes from user data and the image may be missing.
554 *
555 * @param optional true, if JOSM should <b>not</b> throw a RuntimeException
556 * in case the image cannot be located.
557 * @return the current object, for convenience
558 */
559 public ImageProvider setOptional(boolean optional) {
560 this.optional = optional;
561 return this;
562 }
563
564 /**
565 * Suppresses warning on the command line in case the image cannot be found.
566 *
567 * In combination with setOptional(true);
568 * @param suppressWarnings if <code>true</code> warnings are suppressed
569 * @return the current object, for convenience
570 */
571 public ImageProvider setSuppressWarnings(boolean suppressWarnings) {
572 this.suppressWarnings = suppressWarnings;
573 return this;
574 }
575
576 /**
577 * Add a collection of additional class loaders to search image for.
578 * @param additionalClassLoaders class loaders to add to the internal list
579 * @return the current object, for convenience
580 */
581 public ImageProvider setAdditionalClassLoaders(Collection<ClassLoader> additionalClassLoaders) {
582 this.additionalClassLoaders = additionalClassLoaders;
583 return this;
584 }
585
586 /**
587 * Set, if image must be filtered to grayscale so it will look like disabled icon.
588 *
589 * @param disabled true, if image must be grayed out for disabled state
590 * @return the current object, for convenience
591 * @since 10428
592 */
593 public ImageProvider setDisabled(boolean disabled) {
594 this.isDisabled = disabled;
595 return this;
596 }
597
598 /**
599 * Execute the image request and scale result.
600 * @return the requested image or null if the request failed
601 */
602 public ImageIcon get() {
603 ImageResource ir = getResource();
604
605 if (ir == null) {
606 return null;
607 }
608 if (virtualMaxWidth != -1 || virtualMaxHeight != -1)
609 return ir.getImageIconBounded(new Dimension(virtualMaxWidth, virtualMaxHeight));
610 else
611 return ir.getImageIcon(new Dimension(virtualWidth, virtualHeight));
612 }
613
614 /**
615 * Load the image in a background thread.
616 *
617 * This method returns immediately and runs the image request asynchronously.
618 *
619 * @return the future of the requested image
620 * @since 10714
621 */
622 public CompletableFuture<ImageIcon> getAsync() {
623 return name.startsWith(HTTP_PROTOCOL) || name.startsWith(WIKI_PROTOCOL)
624 ? CompletableFuture.supplyAsync(this::get, IMAGE_FETCHER)
625 : CompletableFuture.completedFuture(get());
626 }
627
628 /**
629 * Execute the image request.
630 *
631 * @return the requested image or null if the request failed
632 * @since 7693
633 */
634 public ImageResource getResource() {
635 ImageResource ir = getIfAvailableImpl(additionalClassLoaders);
636 if (ir == null) {
637 if (!optional) {
638 String ext = name.indexOf('.') != -1 ? "" : ".???";
639 throw new JosmRuntimeException(
640 tr("Fatal: failed to locate image ''{0}''. This is a serious configuration problem. JOSM will stop working.",
641 name + ext));
642 } else {
643 if (!suppressWarnings) {
644 Main.error(tr("Failed to locate image ''{0}''", name));
645 }
646 return null;
647 }
648 }
649 if (overlayInfo != null) {
650 ir = new ImageResource(ir, overlayInfo);
651 }
652 if (isDisabled) {
653 ir.setDisabled(true);
654 }
655 return ir;
656 }
657
658 /**
659 * Load the image in a background thread.
660 *
661 * This method returns immediately and runs the image request asynchronously.
662 *
663 * @return the future of the requested image
664 * @since 10714
665 */
666 public CompletableFuture<ImageResource> getResourceAsync() {
667 return name.startsWith(HTTP_PROTOCOL) || name.startsWith(WIKI_PROTOCOL)
668 ? CompletableFuture.supplyAsync(this::getResource, IMAGE_FETCHER)
669 : CompletableFuture.completedFuture(getResource());
670 }
671
672 /**
673 * Load an image with a given file name.
674 *
675 * @param subdir subdirectory the image lies in
676 * @param name The icon name (base name with or without '.png' or '.svg' extension)
677 * @return The requested Image.
678 * @throws RuntimeException if the image cannot be located
679 */
680 public static ImageIcon get(String subdir, String name) {
681 return new ImageProvider(subdir, name).get();
682 }
683
684 /**
685 * Load an image with a given file name.
686 *
687 * @param name The icon name (base name with or without '.png' or '.svg' extension)
688 * @return the requested image or null if the request failed
689 * @see #get(String, String)
690 */
691 public static ImageIcon get(String name) {
692 return new ImageProvider(name).get();
693 }
694
695 /**
696 * Load an image from directory with a given file name and size.
697 *
698 * @param subdir subdirectory the image lies in
699 * @param name The icon name (base name with or without '.png' or '.svg' extension)
700 * @param size Target icon size
701 * @return The requested Image.
702 * @throws RuntimeException if the image cannot be located
703 * @since 10428
704 */
705 public static ImageIcon get(String subdir, String name, ImageSizes size) {
706 return new ImageProvider(subdir, name).setSize(size).get();
707 }
708
709 /**
710 * Load an empty image with a given size.
711 *
712 * @param size Target icon size
713 * @return The requested Image.
714 * @since 10358
715 */
716 public static ImageIcon getEmpty(ImageSizes size) {
717 Dimension iconRealSize = GuiSizesHelper.getDimensionDpiAdjusted(size.getImageDimension());
718 return new ImageIcon(new BufferedImage(iconRealSize.width, iconRealSize.height,
719 BufferedImage.TYPE_INT_ARGB));
720 }
721
722 /**
723 * Load an image with a given file name, but do not throw an exception
724 * when the image cannot be found.
725 *
726 * @param subdir subdirectory the image lies in
727 * @param name The icon name (base name with or without '.png' or '.svg' extension)
728 * @return the requested image or null if the request failed
729 * @see #get(String, String)
730 */
731 public static ImageIcon getIfAvailable(String subdir, String name) {
732 return new ImageProvider(subdir, name).setOptional(true).get();
733 }
734
735 /**
736 * Load an image with a given file name and size.
737 *
738 * @param name The icon name (base name with or without '.png' or '.svg' extension)
739 * @param size Target icon size
740 * @return the requested image or null if the request failed
741 * @see #get(String, String)
742 * @since 10428
743 */
744 public static ImageIcon get(String name, ImageSizes size) {
745 return new ImageProvider(name).setSize(size).get();
746 }
747
748 /**
749 * Load an image with a given file name, but do not throw an exception
750 * when the image cannot be found.
751 *
752 * @param name The icon name (base name with or without '.png' or '.svg' extension)
753 * @return the requested image or null if the request failed
754 * @see #getIfAvailable(String, String)
755 */
756 public static ImageIcon getIfAvailable(String name) {
757 return new ImageProvider(name).setOptional(true).get();
758 }
759
760 /**
761 * {@code data:[<mediatype>][;base64],<data>}
762 * @see <a href="http://tools.ietf.org/html/rfc2397">RFC2397</a>
763 */
764 private static final Pattern dataUrlPattern = Pattern.compile(
765 "^data:([a-zA-Z]+/[a-zA-Z+]+)?(;base64)?,(.+)$");
766
767 /**
768 * Clears the internal image cache.
769 * @since 11021
770 */
771 public static void clearCache() {
772 synchronized (cache) {
773 cache.clear();
774 }
775 }
776
777 /**
778 * Internal implementation of the image request.
779 *
780 * @param additionalClassLoaders the list of class loaders to use
781 * @return the requested image or null if the request failed
782 */
783 private ImageResource getIfAvailableImpl(Collection<ClassLoader> additionalClassLoaders) {
784 synchronized (cache) {
785 // This method is called from different thread and modifying HashMap concurrently can result
786 // for example in loops in map entries (ie freeze when such entry is retrieved)
787 // Yes, it did happen to me :-)
788 if (name == null)
789 return null;
790
791 String prefix = "";
792 if (isDisabled)
793 prefix = "dis:"+prefix;
794 if (name.startsWith("data:")) {
795 String url = name;
796 ImageResource ir = cache.get(prefix+url);
797 if (ir != null) return ir;
798 ir = getIfAvailableDataUrl(url);
799 if (ir != null) {
800 cache.put(prefix+url, ir);
801 }
802 return ir;
803 }
804
805 ImageType type = Utils.hasExtension(name, "svg") ? ImageType.SVG : ImageType.OTHER;
806
807 if (name.startsWith(HTTP_PROTOCOL) || name.startsWith(HTTPS_PROTOCOL)) {
808 String url = name;
809 ImageResource ir = cache.get(prefix+url);
810 if (ir != null) return ir;
811 ir = getIfAvailableHttp(url, type);
812 if (ir != null) {
813 cache.put(prefix+url, ir);
814 }
815 return ir;
816 } else if (name.startsWith(WIKI_PROTOCOL)) {
817 ImageResource ir = cache.get(prefix+name);
818 if (ir != null) return ir;
819 ir = getIfAvailableWiki(name, type);
820 if (ir != null) {
821 cache.put(prefix+name, ir);
822 }
823 return ir;
824 }
825
826 if (subdir == null) {
827 subdir = "";
828 } else if (!subdir.isEmpty() && !subdir.endsWith("/")) {
829 subdir += '/';
830 }
831 String[] extensions;
832 if (name.indexOf('.') != -1) {
833 extensions = new String[] {""};
834 } else {
835 extensions = new String[] {".png", ".svg"};
836 }
837 final int typeArchive = 0;
838 final int typeLocal = 1;
839 for (int place : new Integer[] {typeArchive, typeLocal}) {
840 for (String ext : extensions) {
841
842 if (".svg".equals(ext)) {
843 type = ImageType.SVG;
844 } else if (".png".equals(ext)) {
845 type = ImageType.OTHER;
846 }
847
848 String fullName = subdir + name + ext;
849 String cacheName = prefix + fullName;
850 /* cache separately */
851 if (dirs != null && !dirs.isEmpty()) {
852 cacheName = "id:" + id + ':' + fullName;
853 if (archive != null) {
854 cacheName += ':' + archive.getName();
855 }
856 }
857
858 switch (place) {
859 case typeArchive:
860 if (archive != null) {
861 cacheName = "zip:"+archive.hashCode()+':'+cacheName;
862 ImageResource ir = cache.get(cacheName);
863 if (ir != null) return ir;
864
865 ir = getIfAvailableZip(fullName, archive, inArchiveDir, type);
866 if (ir != null) {
867 cache.put(cacheName, ir);
868 return ir;
869 }
870 }
871 break;
872 case typeLocal:
873 ImageResource ir = cache.get(cacheName);
874 if (ir != null) return ir;
875
876 // getImageUrl() does a ton of "stat()" calls and gets expensive
877 // and redundant when you have a whole ton of objects. So,
878 // index the cache by the name of the icon we're looking for
879 // and don't bother to create a URL unless we're actually
880 // creating the image.
881 URL path = getImageUrl(fullName, dirs, additionalClassLoaders);
882 if (path == null) {
883 continue;
884 }
885 ir = getIfAvailableLocalURL(path, type);
886 if (ir != null) {
887 cache.put(cacheName, ir);
888 return ir;
889 }
890 break;
891 }
892 }
893 }
894 return null;
895 }
896 }
897
898 /**
899 * Internal implementation of the image request for URL's.
900 *
901 * @param url URL of the image
902 * @param type data type of the image
903 * @return the requested image or null if the request failed
904 */
905 private static ImageResource getIfAvailableHttp(String url, ImageType type) {
906 try (CachedFile cf = new CachedFile(url).setDestDir(new File(Main.pref.getCacheDirectory(), "images").getPath());
907 InputStream is = cf.getInputStream()) {
908 switch (type) {
909 case SVG:
910 SVGDiagram svg = null;
911 synchronized (getSvgUniverse()) {
912 URI uri = getSvgUniverse().loadSVG(is, Utils.fileToURL(cf.getFile()).toString());
913 svg = getSvgUniverse().getDiagram(uri);
914 }
915 return svg == null ? null : new ImageResource(svg);
916 case OTHER:
917 BufferedImage img = null;
918 try {
919 img = read(Utils.fileToURL(cf.getFile()), false, false);
920 } catch (IOException e) {
921 Main.warn(e, "IOException while reading HTTP image:");
922 }
923 return img == null ? null : new ImageResource(img);
924 default:
925 throw new AssertionError("Unsupported type: " + type);
926 }
927 } catch (IOException e) {
928 Main.debug(e);
929 return null;
930 }
931 }
932
933 /**
934 * Internal implementation of the image request for inline images (<b>data:</b> urls).
935 *
936 * @param url the data URL for image extraction
937 * @return the requested image or null if the request failed
938 */
939 private static ImageResource getIfAvailableDataUrl(String url) {
940 Matcher m = dataUrlPattern.matcher(url);
941 if (m.matches()) {
942 String base64 = m.group(2);
943 String data = m.group(3);
944 byte[] bytes;
945 if (";base64".equals(base64)) {
946 bytes = Base64.getDecoder().decode(data);
947 } else {
948 try {
949 bytes = Utils.decodeUrl(data).getBytes(StandardCharsets.UTF_8);
950 } catch (IllegalArgumentException ex) {
951 Main.warn(ex, "Unable to decode URL data part: "+ex.getMessage() + " (" + data + ')');
952 return null;
953 }
954 }
955 String mediatype = m.group(1);
956 if ("image/svg+xml".equals(mediatype)) {
957 String s = new String(bytes, StandardCharsets.UTF_8);
958 SVGDiagram svg;
959 synchronized (getSvgUniverse()) {
960 URI uri = getSvgUniverse().loadSVG(new StringReader(s), Utils.encodeUrl(s));
961 svg = getSvgUniverse().getDiagram(uri);
962 }
963 if (svg == null) {
964 Main.warn("Unable to process svg: "+s);
965 return null;
966 }
967 return new ImageResource(svg);
968 } else {
969 try {
970 // See #10479: for PNG files, always enforce transparency to be sure tNRS chunk is used even not in paletted mode
971 // This can be removed if someday Oracle fixes https://bugs.openjdk.java.net/browse/JDK-6788458
972 // CHECKSTYLE.OFF: LineLength
973 // hg.openjdk.java.net/jdk8u/jdk8u/jdk/file/dc4322602480/src/share/classes/com/sun/imageio/plugins/png/PNGImageReader.java#l656
974 // CHECKSTYLE.ON: LineLength
975 Image img = read(new ByteArrayInputStream(bytes), false, true);
976 return img == null ? null : new ImageResource(img);
977 } catch (IOException e) {
978 Main.warn(e, "IOException while reading image:");
979 }
980 }
981 }
982 return null;
983 }
984
985 /**
986 * Internal implementation of the image request for wiki images.
987 *
988 * @param name image file name
989 * @param type data type of the image
990 * @return the requested image or null if the request failed
991 */
992 private static ImageResource getIfAvailableWiki(String name, ImageType type) {
993 final Collection<String> defaultBaseUrls = Arrays.asList(
994 "https://wiki.openstreetmap.org/w/images/",
995 "https://upload.wikimedia.org/wikipedia/commons/",
996 "https://wiki.openstreetmap.org/wiki/File:"
997 );
998 final Collection<String> baseUrls = Main.pref.getCollection("image-provider.wiki.urls", defaultBaseUrls);
999
1000 final String fn = name.substring(name.lastIndexOf('/') + 1);
1001
1002 ImageResource result = null;
1003 for (String b : baseUrls) {
1004 String url;
1005 if (b.endsWith(":")) {
1006 url = getImgUrlFromWikiInfoPage(b, fn);
1007 if (url == null) {
1008 continue;
1009 }
1010 } else {
1011 final String fnMD5 = Utils.md5Hex(fn);
1012 url = b + fnMD5.substring(0, 1) + '/' + fnMD5.substring(0, 2) + '/' + fn;
1013 }
1014 result = getIfAvailableHttp(url, type);
1015 if (result != null) {
1016 break;
1017 }
1018 }
1019 return result;
1020 }
1021
1022 /**
1023 * Internal implementation of the image request for images in Zip archives.
1024 *
1025 * @param fullName image file name
1026 * @param archive the archive to get image from
1027 * @param inArchiveDir directory of the image inside the archive or <code>null</code>
1028 * @param type data type of the image
1029 * @return the requested image or null if the request failed
1030 */
1031 private static ImageResource getIfAvailableZip(String fullName, File archive, String inArchiveDir, ImageType type) {
1032 try (ZipFile zipFile = new ZipFile(archive, StandardCharsets.UTF_8)) {
1033 if (inArchiveDir == null || ".".equals(inArchiveDir)) {
1034 inArchiveDir = "";
1035 } else if (!inArchiveDir.isEmpty()) {
1036 inArchiveDir += '/';
1037 }
1038 String entryName = inArchiveDir + fullName;
1039 ZipEntry entry = zipFile.getEntry(entryName);
1040 if (entry != null) {
1041 int size = (int) entry.getSize();
1042 int offs = 0;
1043 byte[] buf = new byte[size];
1044 try (InputStream is = zipFile.getInputStream(entry)) {
1045 switch (type) {
1046 case SVG:
1047 SVGDiagram svg = null;
1048 synchronized (getSvgUniverse()) {
1049 URI uri = getSvgUniverse().loadSVG(is, entryName);
1050 svg = getSvgUniverse().getDiagram(uri);
1051 }
1052 return svg == null ? null : new ImageResource(svg);
1053 case OTHER:
1054 while (size > 0) {
1055 int l = is.read(buf, offs, size);
1056 offs += l;
1057 size -= l;
1058 }
1059 BufferedImage img = null;
1060 try {
1061 img = read(new ByteArrayInputStream(buf), false, false);
1062 } catch (IOException e) {
1063 Main.warn(e);
1064 }
1065 return img == null ? null : new ImageResource(img);
1066 default:
1067 throw new AssertionError("Unknown ImageType: "+type);
1068 }
1069 }
1070 }
1071 } catch (IOException e) {
1072 Main.warn(e, tr("Failed to handle zip file ''{0}''. Exception was: {1}", archive.getName(), e.toString()));
1073 }
1074 return null;
1075 }
1076
1077 /**
1078 * Internal implementation of the image request for local images.
1079 *
1080 * @param path image file path
1081 * @param type data type of the image
1082 * @return the requested image or null if the request failed
1083 */
1084 private static ImageResource getIfAvailableLocalURL(URL path, ImageType type) {
1085 switch (type) {
1086 case SVG:
1087 SVGDiagram svg;
1088 synchronized (getSvgUniverse()) {
1089 URI uri = getSvgUniverse().loadSVG(path);
1090 svg = getSvgUniverse().getDiagram(uri);
1091 }
1092 return svg == null ? null : new ImageResource(svg);
1093 case OTHER:
1094 BufferedImage img = null;
1095 try {
1096 // See #10479: for PNG files, always enforce transparency to be sure tNRS chunk is used even not in paletted mode
1097 // This can be removed if someday Oracle fixes https://bugs.openjdk.java.net/browse/JDK-6788458
1098 // hg.openjdk.java.net/jdk8u/jdk8u/jdk/file/dc4322602480/src/share/classes/com/sun/imageio/plugins/png/PNGImageReader.java#l656
1099 img = read(path, false, true);
1100 if (Main.isDebugEnabled() && isTransparencyForced(img)) {
1101 Main.debug("Transparency has been forced for image "+path.toExternalForm());
1102 }
1103 } catch (IOException e) {
1104 Main.warn(e);
1105 }
1106 return img == null ? null : new ImageResource(img);
1107 default:
1108 throw new AssertionError();
1109 }
1110 }
1111
1112 private static URL getImageUrl(String path, String name, Collection<ClassLoader> additionalClassLoaders) {
1113 if (path != null && path.startsWith("resource://")) {
1114 String p = path.substring("resource://".length());
1115 Collection<ClassLoader> classLoaders = new ArrayList<>(PluginHandler.getResourceClassLoaders());
1116 if (additionalClassLoaders != null) {
1117 classLoaders.addAll(additionalClassLoaders);
1118 }
1119 for (ClassLoader source : classLoaders) {
1120 URL res;
1121 if ((res = source.getResource(p + name)) != null)
1122 return res;
1123 }
1124 } else {
1125 File f = new File(path, name);
1126 if ((path != null || f.isAbsolute()) && f.exists())
1127 return Utils.fileToURL(f);
1128 }
1129 return null;
1130 }
1131
1132 private static URL getImageUrl(String imageName, Collection<String> dirs, Collection<ClassLoader> additionalClassLoaders) {
1133 URL u;
1134
1135 // Try passed directories first
1136 if (dirs != null) {
1137 for (String name : dirs) {
1138 try {
1139 u = getImageUrl(name, imageName, additionalClassLoaders);
1140 if (u != null)
1141 return u;
1142 } catch (SecurityException e) {
1143 Main.warn(e, tr(
1144 "Failed to access directory ''{0}'' for security reasons. Exception was: {1}",
1145 name, e.toString()));
1146 }
1147
1148 }
1149 }
1150 // Try user-data directory
1151 if (Main.pref != null) {
1152 String dir = new File(Main.pref.getUserDataDirectory(), "images").getAbsolutePath();
1153 try {
1154 u = getImageUrl(dir, imageName, additionalClassLoaders);
1155 if (u != null)
1156 return u;
1157 } catch (SecurityException e) {
1158 Main.warn(e, tr(
1159 "Failed to access directory ''{0}'' for security reasons. Exception was: {1}", dir, e
1160 .toString()));
1161 }
1162 }
1163
1164 // Absolute path?
1165 u = getImageUrl(null, imageName, additionalClassLoaders);
1166 if (u != null)
1167 return u;
1168
1169 // Try plugins and josm classloader
1170 u = getImageUrl("resource://images/", imageName, additionalClassLoaders);
1171 if (u != null)
1172 return u;
1173
1174 // Try all other resource directories
1175 if (Main.pref != null) {
1176 for (String location : Main.pref.getAllPossiblePreferenceDirs()) {
1177 u = getImageUrl(location + "images", imageName, additionalClassLoaders);
1178 if (u != null)
1179 return u;
1180 u = getImageUrl(location, imageName, additionalClassLoaders);
1181 if (u != null)
1182 return u;
1183 }
1184 }
1185
1186 return null;
1187 }
1188
1189 /** Quit parsing, when a certain condition is met */
1190 private static class SAXReturnException extends SAXException {
1191 private final String result;
1192
1193 SAXReturnException(String result) {
1194 this.result = result;
1195 }
1196
1197 public String getResult() {
1198 return result;
1199 }
1200 }
1201
1202 /**
1203 * Reads the wiki page on a certain file in html format in order to find the real image URL.
1204 *
1205 * @param base base URL for Wiki image
1206 * @param fn filename of the Wiki image
1207 * @return image URL for a Wiki image or null in case of error
1208 */
1209 private static String getImgUrlFromWikiInfoPage(final String base, final String fn) {
1210 try {
1211 final XMLReader parser = Utils.newSafeSAXParser().getXMLReader();
1212 parser.setContentHandler(new DefaultHandler() {
1213 @Override
1214 public void startElement(String uri, String localName, String qName, Attributes atts) throws SAXException {
1215 if ("img".equalsIgnoreCase(localName)) {
1216 String val = atts.getValue("src");
1217 if (val.endsWith(fn))
1218 throw new SAXReturnException(val); // parsing done, quit early
1219 }
1220 }
1221 });
1222
1223 parser.setEntityResolver((publicId, systemId) -> new InputSource(new ByteArrayInputStream(new byte[0])));
1224
1225 try (CachedFile cf = new CachedFile(base + fn).setDestDir(new File(Main.pref.getUserDataDirectory(), "images").getPath());
1226 InputStream is = cf.getInputStream()) {
1227 parser.parse(new InputSource(is));
1228 }
1229 } catch (SAXReturnException r) {
1230 Main.trace(r);
1231 return r.getResult();
1232 } catch (IOException | SAXException | ParserConfigurationException e) {
1233 Main.warn("Parsing " + base + fn + " failed:\n" + e);
1234 return null;
1235 }
1236 Main.warn("Parsing " + base + fn + " failed: Unexpected content.");
1237 return null;
1238 }
1239
1240 /**
1241 * Load a cursor with a given file name, optionally decorated with an overlay image.
1242 *
1243 * @param name the cursor image filename in "cursor" directory
1244 * @param overlay optional overlay image
1245 * @return cursor with a given file name, optionally decorated with an overlay image
1246 */
1247 public static Cursor getCursor(String name, String overlay) {
1248 ImageIcon img = get("cursor", name);
1249 if (overlay != null) {
1250 img = new ImageProvider("cursor", name).setMaxSize(ImageSizes.CURSOR)
1251 .addOverlay(new ImageOverlay(new ImageProvider("cursor/modifier/" + overlay)
1252 .setMaxSize(ImageSizes.CURSOROVERLAY))).get();
1253 }
1254 if (GraphicsEnvironment.isHeadless()) {
1255 if (Main.isDebugEnabled()) {
1256 Main.debug("Cursors are not available in headless mode. Returning null for '"+name+'\'');
1257 }
1258 return null;
1259 }
1260 return Toolkit.getDefaultToolkit().createCustomCursor(img.getImage(),
1261 "crosshair".equals(name) ? new Point(10, 10) : new Point(3, 2), "Cursor");
1262 }
1263
1264 /** 90 degrees in radians units */
1265 private static final double DEGREE_90 = 90.0 * Math.PI / 180.0;
1266
1267 /**
1268 * Creates a rotated version of the input image.
1269 *
1270 * @param img the image to be rotated.
1271 * @param rotatedAngle the rotated angle, in degree, clockwise. It could be any double but we
1272 * will mod it with 360 before using it. More over for caching performance, it will be rounded to
1273 * an entire value between 0 and 360.
1274 *
1275 * @return the image after rotating.
1276 * @since 6172
1277 */
1278 public static Image createRotatedImage(Image img, double rotatedAngle) {
1279 return createRotatedImage(img, rotatedAngle, ImageResource.DEFAULT_DIMENSION);
1280 }
1281
1282 /**
1283 * Creates a rotated version of the input image, scaled to the given dimension.
1284 *
1285 * @param img the image to be rotated.
1286 * @param rotatedAngle the rotated angle, in degree, clockwise. It could be any double but we
1287 * will mod it with 360 before using it. More over for caching performance, it will be rounded to
1288 * an entire value between 0 and 360.
1289 * @param dimension The requested dimensions. Use (-1,-1) for the original size
1290 * and (width, -1) to set the width, but otherwise scale the image proportionally.
1291 * @return the image after rotating and scaling.
1292 * @since 6172
1293 */
1294 public static Image createRotatedImage(Image img, double rotatedAngle, Dimension dimension) {
1295 CheckParameterUtil.ensureParameterNotNull(img, "img");
1296
1297 // convert rotatedAngle to an integer value from 0 to 360
1298 Long originalAngle = Math.round(rotatedAngle % 360);
1299 if (rotatedAngle != 0 && originalAngle == 0) {
1300 originalAngle = 360L;
1301 }
1302
1303 ImageResource imageResource;
1304
1305 synchronized (ROTATE_CACHE) {
1306 Map<Long, ImageResource> cacheByAngle = ROTATE_CACHE.get(img);
1307 if (cacheByAngle == null) {
1308 cacheByAngle = new HashMap<>();
1309 ROTATE_CACHE.put(img, cacheByAngle);
1310 }
1311
1312 imageResource = cacheByAngle.get(originalAngle);
1313
1314 if (imageResource == null) {
1315 // convert originalAngle to a value from 0 to 90
1316 double angle = originalAngle % 90;
1317 if (originalAngle != 0 && angle == 0) {
1318 angle = 90.0;
1319 }
1320
1321 double radian = Math.toRadians(angle);
1322
1323 new ImageIcon(img); // load completely
1324 int iw = img.getWidth(null);
1325 int ih = img.getHeight(null);
1326 int w;
1327 int h;
1328
1329 if ((originalAngle >= 0 && originalAngle <= 90) || (originalAngle > 180 && originalAngle <= 270)) {
1330 w = (int) (iw * Math.sin(DEGREE_90 - radian) + ih * Math.sin(radian));
1331 h = (int) (iw * Math.sin(radian) + ih * Math.sin(DEGREE_90 - radian));
1332 } else {
1333 w = (int) (ih * Math.sin(DEGREE_90 - radian) + iw * Math.sin(radian));
1334 h = (int) (ih * Math.sin(radian) + iw * Math.sin(DEGREE_90 - radian));
1335 }
1336 Image image = new BufferedImage(w, h, BufferedImage.TYPE_INT_ARGB);
1337 imageResource = new ImageResource(image);
1338 cacheByAngle.put(originalAngle, imageResource);
1339 Graphics g = image.getGraphics();
1340 Graphics2D g2d = (Graphics2D) g.create();
1341
1342 // calculate the center of the icon.
1343 int cx = iw / 2;
1344 int cy = ih / 2;
1345
1346 // move the graphics center point to the center of the icon.
1347 g2d.translate(w / 2, h / 2);
1348
1349 // rotate the graphics about the center point of the icon
1350 g2d.rotate(Math.toRadians(originalAngle));
1351
1352 g2d.setRenderingHint(RenderingHints.KEY_INTERPOLATION, RenderingHints.VALUE_INTERPOLATION_BICUBIC);
1353 g2d.drawImage(img, -cx, -cy, null);
1354
1355 g2d.dispose();
1356 new ImageIcon(image); // load completely
1357 }
1358 return imageResource.getImageIcon(dimension).getImage();
1359 }
1360 }
1361
1362 /**
1363 * Creates a scaled down version of the input image to fit maximum dimensions. (Keeps aspect ratio)
1364 *
1365 * @param img the image to be scaled down.
1366 * @param maxSize the maximum size in pixels (both for width and height)
1367 *
1368 * @return the image after scaling.
1369 * @since 6172
1370 */
1371 public static Image createBoundedImage(Image img, int maxSize) {
1372 return new ImageResource(img).getImageIconBounded(new Dimension(maxSize, maxSize)).getImage();
1373 }
1374
1375 /**
1376 * Replies the icon for an OSM primitive type
1377 * @param type the type
1378 * @return the icon
1379 */
1380 public static ImageIcon get(OsmPrimitiveType type) {
1381 CheckParameterUtil.ensureParameterNotNull(type, "type");
1382 return get("data", type.getAPIName());
1383 }
1384
1385 /**
1386 * @param primitive Object for which an icon shall be fetched. The icon is chosen based on tags.
1387 * @param iconSize Target size of icon. Icon is padded if required.
1388 * @return Icon for {@code primitive} that fits in cell.
1389 * @since 8903
1390 */
1391 public static ImageIcon getPadded(OsmPrimitive primitive, Dimension iconSize) {
1392 // Check if the current styles have special icon for tagged nodes.
1393 if (primitive instanceof org.openstreetmap.josm.data.osm.Node) {
1394 Pair<StyleElementList, Range> nodeStyles;
1395 DataSet ds = primitive.getDataSet();
1396 if (ds != null) {
1397 ds.getReadLock().lock();
1398 }
1399 try {
1400 nodeStyles = MapPaintStyles.getStyles().generateStyles(primitive, 100, false);
1401 } finally {
1402 if (ds != null) {
1403 ds.getReadLock().unlock();
1404 }
1405 }
1406 for (StyleElement style : nodeStyles.a) {
1407 if (style instanceof NodeElement) {
1408 NodeElement nodeStyle = (NodeElement) style;
1409 MapImage icon = nodeStyle.mapImage;
1410 if (icon != null) {
1411 int backgroundRealWidth = GuiSizesHelper.getSizeDpiAdjusted(iconSize.width);
1412 int backgroundRealHeight = GuiSizesHelper.getSizeDpiAdjusted(iconSize.height);
1413 int iconRealWidth = icon.getWidth();
1414 int iconRealHeight = icon.getHeight();
1415 BufferedImage image = new BufferedImage(backgroundRealWidth, backgroundRealHeight,
1416 BufferedImage.TYPE_INT_ARGB);
1417 double scaleFactor = Math.min(backgroundRealWidth / (double) iconRealWidth, backgroundRealHeight
1418 / (double) iconRealHeight);
1419 BufferedImage iconImage = icon.getImage(false);
1420 Image scaledIcon;
1421 final int scaledWidth;
1422 final int scaledHeight;
1423 if (scaleFactor < 1) {
1424 // Scale icon such that it fits on background.
1425 scaledWidth = (int) (iconRealWidth * scaleFactor);
1426 scaledHeight = (int) (iconRealHeight * scaleFactor);
1427 scaledIcon = iconImage.getScaledInstance(scaledWidth, scaledHeight, Image.SCALE_SMOOTH);
1428 } else {
1429 // Use original size, don't upscale.
1430 scaledWidth = iconRealWidth;
1431 scaledHeight = iconRealHeight;
1432 scaledIcon = iconImage;
1433 }
1434 image.getGraphics().drawImage(scaledIcon, (backgroundRealWidth - scaledWidth) / 2,
1435 (backgroundRealHeight - scaledHeight) / 2, null);
1436
1437 return new ImageIcon(image);
1438 }
1439 }
1440 }
1441 }
1442
1443 // Check if the presets have icons for nodes/relations.
1444 if (!OsmPrimitiveType.WAY.equals(primitive.getType())) {
1445 final Collection<TaggingPreset> presets = new TreeSet<>((o1, o2) -> {
1446 final int o1TypesSize = o1.types == null || o1.types.isEmpty() ? Integer.MAX_VALUE : o1.types.size();
1447 final int o2TypesSize = o2.types == null || o2.types.isEmpty() ? Integer.MAX_VALUE : o2.types.size();
1448 return Integer.compare(o1TypesSize, o2TypesSize);
1449 });
1450 presets.addAll(TaggingPresets.getMatchingPresets(primitive));
1451 for (final TaggingPreset preset : presets) {
1452 if (preset.getIcon() != null) {
1453 return preset.getIcon();
1454 }
1455 }
1456 }
1457
1458 // Use generic default icon.
1459 return ImageProvider.get(primitive.getDisplayType());
1460 }
1461
1462 /**
1463 * Constructs an image from the given SVG data.
1464 * @param svg the SVG data
1465 * @param dim the desired image dimension
1466 * @return an image from the given SVG data at the desired dimension.
1467 */
1468 public static BufferedImage createImageFromSvg(SVGDiagram svg, Dimension dim) {
1469 if (Main.isTraceEnabled()) {
1470 Main.trace(String.format("createImageFromSvg: %s %s", svg.getXMLBase(), dim));
1471 }
1472 float sourceWidth = svg.getWidth();
1473 float sourceHeight = svg.getHeight();
1474 int realWidth = Math.round(GuiSizesHelper.getSizeDpiAdjusted(sourceWidth));
1475 int realHeight = Math.round(GuiSizesHelper.getSizeDpiAdjusted(sourceHeight));
1476 Double scaleX, scaleY;
1477 if (dim.width != -1) {
1478 realWidth = dim.width;
1479 scaleX = (double) realWidth / sourceWidth;
1480 if (dim.height == -1) {
1481 scaleY = scaleX;
1482 realHeight = (int) Math.round(sourceHeight * scaleY);
1483 } else {
1484 realHeight = dim.height;
1485 scaleY = (double) realHeight / sourceHeight;
1486 }
1487 } else if (dim.height != -1) {
1488 realHeight = dim.height;
1489 scaleX = scaleY = (double) realHeight / sourceHeight;
1490 realWidth = (int) Math.round(sourceWidth * scaleX);
1491 } else {
1492 scaleX = scaleY = (double) realHeight / sourceHeight;
1493 }
1494
1495 if (realWidth == 0 || realHeight == 0) {
1496 return null;
1497 }
1498 BufferedImage img = new BufferedImage(realWidth, realHeight, BufferedImage.TYPE_INT_ARGB);
1499 Graphics2D g = img.createGraphics();
1500 g.setClip(0, 0, realWidth, realHeight);
1501 g.scale(scaleX, scaleY);
1502 g.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
1503 try {
1504 synchronized (getSvgUniverse()) {
1505 svg.render(g);
1506 }
1507 } catch (SVGException ex) {
1508 Main.error(ex, "Unable to load svg:");
1509 return null;
1510 }
1511 return img;
1512 }
1513
1514 private static synchronized SVGUniverse getSvgUniverse() {
1515 if (svgUniverse == null) {
1516 svgUniverse = new SVGUniverse();
1517 }
1518 return svgUniverse;
1519 }
1520
1521 /**
1522 * Returns a <code>BufferedImage</code> as the result of decoding
1523 * a supplied <code>File</code> with an <code>ImageReader</code>
1524 * chosen automatically from among those currently registered.
1525 * The <code>File</code> is wrapped in an
1526 * <code>ImageInputStream</code>. If no registered
1527 * <code>ImageReader</code> claims to be able to read the
1528 * resulting stream, <code>null</code> is returned.
1529 *
1530 * <p> The current cache settings from <code>getUseCache</code>and
1531 * <code>getCacheDirectory</code> will be used to control caching in the
1532 * <code>ImageInputStream</code> that is created.
1533 *
1534 * <p> Note that there is no <code>read</code> method that takes a
1535 * filename as a <code>String</code>; use this method instead after
1536 * creating a <code>File</code> from the filename.
1537 *
1538 * <p> This method does not attempt to locate
1539 * <code>ImageReader</code>s that can read directly from a
1540 * <code>File</code>; that may be accomplished using
1541 * <code>IIORegistry</code> and <code>ImageReaderSpi</code>.
1542 *
1543 * @param input a <code>File</code> to read from.
1544 * @param readMetadata if {@code true}, makes sure to read image metadata to detect transparency color, if any.
1545 * In that case the color can be retrieved later through {@link #PROP_TRANSPARENCY_COLOR}.
1546 * Always considered {@code true} if {@code enforceTransparency} is also {@code true}
1547 * @param enforceTransparency if {@code true}, makes sure to read image metadata and, if the image does not
1548 * provide an alpha channel but defines a {@code TransparentColor} metadata node, that the resulting image
1549 * has a transparency set to {@code TRANSLUCENT} and uses the correct transparent color.
1550 *
1551 * @return a <code>BufferedImage</code> containing the decoded
1552 * contents of the input, or <code>null</code>.
1553 *
1554 * @throws IllegalArgumentException if <code>input</code> is <code>null</code>.
1555 * @throws IOException if an error occurs during reading.
1556 * @see BufferedImage#getProperty
1557 * @since 7132
1558 */
1559 public static BufferedImage read(File input, boolean readMetadata, boolean enforceTransparency) throws IOException {
1560 CheckParameterUtil.ensureParameterNotNull(input, "input");
1561 if (!input.canRead()) {
1562 throw new IIOException("Can't read input file!");
1563 }
1564
1565 ImageInputStream stream = ImageIO.createImageInputStream(input);
1566 if (stream == null) {
1567 throw new IIOException("Can't create an ImageInputStream!");
1568 }
1569 BufferedImage bi = read(stream, readMetadata, enforceTransparency);
1570 if (bi == null) {
1571 stream.close();
1572 }
1573 return bi;
1574 }
1575
1576 /**
1577 * Returns a <code>BufferedImage</code> as the result of decoding
1578 * a supplied <code>InputStream</code> with an <code>ImageReader</code>
1579 * chosen automatically from among those currently registered.
1580 * The <code>InputStream</code> is wrapped in an
1581 * <code>ImageInputStream</code>. If no registered
1582 * <code>ImageReader</code> claims to be able to read the
1583 * resulting stream, <code>null</code> is returned.
1584 *
1585 * <p> The current cache settings from <code>getUseCache</code>and
1586 * <code>getCacheDirectory</code> will be used to control caching in the
1587 * <code>ImageInputStream</code> that is created.
1588 *
1589 * <p> This method does not attempt to locate
1590 * <code>ImageReader</code>s that can read directly from an
1591 * <code>InputStream</code>; that may be accomplished using
1592 * <code>IIORegistry</code> and <code>ImageReaderSpi</code>.
1593 *
1594 * <p> This method <em>does not</em> close the provided
1595 * <code>InputStream</code> after the read operation has completed;
1596 * it is the responsibility of the caller to close the stream, if desired.
1597 *
1598 * @param input an <code>InputStream</code> to read from.
1599 * @param readMetadata if {@code true}, makes sure to read image metadata to detect transparency color for non translucent images, if any.
1600 * In that case the color can be retrieved later through {@link #PROP_TRANSPARENCY_COLOR}.
1601 * Always considered {@code true} if {@code enforceTransparency} is also {@code true}
1602 * @param enforceTransparency if {@code true}, makes sure to read image metadata and, if the image does not
1603 * provide an alpha channel but defines a {@code TransparentColor} metadata node, that the resulting image
1604 * has a transparency set to {@code TRANSLUCENT} and uses the correct transparent color.
1605 *
1606 * @return a <code>BufferedImage</code> containing the decoded
1607 * contents of the input, or <code>null</code>.
1608 *
1609 * @throws IllegalArgumentException if <code>input</code> is <code>null</code>.
1610 * @throws IOException if an error occurs during reading.
1611 * @since 7132
1612 */
1613 public static BufferedImage read(InputStream input, boolean readMetadata, boolean enforceTransparency) throws IOException {
1614 CheckParameterUtil.ensureParameterNotNull(input, "input");
1615
1616 ImageInputStream stream = ImageIO.createImageInputStream(input);
1617 BufferedImage bi = read(stream, readMetadata, enforceTransparency);
1618 if (bi == null) {
1619 stream.close();
1620 }
1621 return bi;
1622 }
1623
1624 /**
1625 * Returns a <code>BufferedImage</code> as the result of decoding
1626 * a supplied <code>URL</code> with an <code>ImageReader</code>
1627 * chosen automatically from among those currently registered. An
1628 * <code>InputStream</code> is obtained from the <code>URL</code>,
1629 * which is wrapped in an <code>ImageInputStream</code>. If no
1630 * registered <code>ImageReader</code> claims to be able to read
1631 * the resulting stream, <code>null</code> is returned.
1632 *
1633 * <p> The current cache settings from <code>getUseCache</code>and
1634 * <code>getCacheDirectory</code> will be used to control caching in the
1635 * <code>ImageInputStream</code> that is created.
1636 *
1637 * <p> This method does not attempt to locate
1638 * <code>ImageReader</code>s that can read directly from a
1639 * <code>URL</code>; that may be accomplished using
1640 * <code>IIORegistry</code> and <code>ImageReaderSpi</code>.
1641 *
1642 * @param input a <code>URL</code> to read from.
1643 * @param readMetadata if {@code true}, makes sure to read image metadata to detect transparency color for non translucent images, if any.
1644 * In that case the color can be retrieved later through {@link #PROP_TRANSPARENCY_COLOR}.
1645 * Always considered {@code true} if {@code enforceTransparency} is also {@code true}
1646 * @param enforceTransparency if {@code true}, makes sure to read image metadata and, if the image does not
1647 * provide an alpha channel but defines a {@code TransparentColor} metadata node, that the resulting image
1648 * has a transparency set to {@code TRANSLUCENT} and uses the correct transparent color.
1649 *
1650 * @return a <code>BufferedImage</code> containing the decoded
1651 * contents of the input, or <code>null</code>.
1652 *
1653 * @throws IllegalArgumentException if <code>input</code> is <code>null</code>.
1654 * @throws IOException if an error occurs during reading.
1655 * @since 7132
1656 */
1657 public static BufferedImage read(URL input, boolean readMetadata, boolean enforceTransparency) throws IOException {
1658 CheckParameterUtil.ensureParameterNotNull(input, "input");
1659
1660 InputStream istream = null;
1661 try {
1662 istream = input.openStream();
1663 } catch (IOException e) {
1664 throw new IIOException("Can't get input stream from URL!", e);
1665 }
1666 ImageInputStream stream = ImageIO.createImageInputStream(istream);
1667 BufferedImage bi;
1668 try {
1669 bi = read(stream, readMetadata, enforceTransparency);
1670 if (bi == null) {
1671 stream.close();
1672 }
1673 } finally {
1674 istream.close();
1675 }
1676 return bi;
1677 }
1678
1679 /**
1680 * Returns a <code>BufferedImage</code> as the result of decoding
1681 * a supplied <code>ImageInputStream</code> with an
1682 * <code>ImageReader</code> chosen automatically from among those
1683 * currently registered. If no registered
1684 * <code>ImageReader</code> claims to be able to read the stream,
1685 * <code>null</code> is returned.
1686 *
1687 * <p> Unlike most other methods in this class, this method <em>does</em>
1688 * close the provided <code>ImageInputStream</code> after the read
1689 * operation has completed, unless <code>null</code> is returned,
1690 * in which case this method <em>does not</em> close the stream.
1691 *
1692 * @param stream an <code>ImageInputStream</code> to read from.
1693 * @param readMetadata if {@code true}, makes sure to read image metadata to detect transparency color for non translucent images, if any.
1694 * In that case the color can be retrieved later through {@link #PROP_TRANSPARENCY_COLOR}.
1695 * Always considered {@code true} if {@code enforceTransparency} is also {@code true}
1696 * @param enforceTransparency if {@code true}, makes sure to read image metadata and, if the image does not
1697 * provide an alpha channel but defines a {@code TransparentColor} metadata node, that the resulting image
1698 * has a transparency set to {@code TRANSLUCENT} and uses the correct transparent color.
1699 *
1700 * @return a <code>BufferedImage</code> containing the decoded
1701 * contents of the input, or <code>null</code>.
1702 *
1703 * @throws IllegalArgumentException if <code>stream</code> is <code>null</code>.
1704 * @throws IOException if an error occurs during reading.
1705 * @since 7132
1706 */
1707 public static BufferedImage read(ImageInputStream stream, boolean readMetadata, boolean enforceTransparency) throws IOException {
1708 CheckParameterUtil.ensureParameterNotNull(stream, "stream");
1709
1710 Iterator<ImageReader> iter = ImageIO.getImageReaders(stream);
1711 if (!iter.hasNext()) {
1712 return null;
1713 }
1714
1715 ImageReader reader = iter.next();
1716 ImageReadParam param = reader.getDefaultReadParam();
1717 reader.setInput(stream, true, !readMetadata && !enforceTransparency);
1718 BufferedImage bi;
1719 try {
1720 bi = reader.read(0, param);
1721 if (bi.getTransparency() != Transparency.TRANSLUCENT && (readMetadata || enforceTransparency)) {
1722 Color color = getTransparentColor(bi.getColorModel(), reader);
1723 if (color != null) {
1724 Hashtable<String, Object> properties = new Hashtable<>(1);
1725 properties.put(PROP_TRANSPARENCY_COLOR, color);
1726 bi = new BufferedImage(bi.getColorModel(), bi.getRaster(), bi.isAlphaPremultiplied(), properties);
1727 if (enforceTransparency) {
1728 if (Main.isTraceEnabled()) {
1729 Main.trace("Enforcing image transparency of "+stream+" for "+color);
1730 }
1731 bi = makeImageTransparent(bi, color);
1732 }
1733 }
1734 }
1735 } finally {
1736 reader.dispose();
1737 stream.close();
1738 }
1739 return bi;
1740 }
1741
1742 // CHECKSTYLE.OFF: LineLength
1743
1744 /**
1745 * Returns the {@code TransparentColor} defined in image reader metadata.
1746 * @param model The image color model
1747 * @param reader The image reader
1748 * @return the {@code TransparentColor} defined in image reader metadata, or {@code null}
1749 * @throws IOException if an error occurs during reading
1750 * @see <a href="http://docs.oracle.com/javase/8/docs/api/javax/imageio/metadata/doc-files/standard_metadata.html">javax_imageio_1.0 metadata</a>
1751 * @since 7499
1752 */
1753 public static Color getTransparentColor(ColorModel model, ImageReader reader) throws IOException {
1754 // CHECKSTYLE.ON: LineLength
1755 try {
1756 IIOMetadata metadata = reader.getImageMetadata(0);
1757 if (metadata != null) {
1758 String[] formats = metadata.getMetadataFormatNames();
1759 if (formats != null) {
1760 for (String f : formats) {
1761 if ("javax_imageio_1.0".equals(f)) {
1762 Node root = metadata.getAsTree(f);
1763 if (root instanceof Element) {
1764 NodeList list = ((Element) root).getElementsByTagName("TransparentColor");
1765 if (list.getLength() > 0) {
1766 Node item = list.item(0);
1767 if (item instanceof Element) {
1768 // Handle different color spaces (tested with RGB and grayscale)
1769 String value = ((Element) item).getAttribute("value");
1770 if (!value.isEmpty()) {
1771 String[] s = value.split(" ");
1772 if (s.length == 3) {
1773 return parseRGB(s);
1774 } else if (s.length == 1) {
1775 int pixel = Integer.parseInt(s[0]);
1776 int r = model.getRed(pixel);
1777 int g = model.getGreen(pixel);
1778 int b = model.getBlue(pixel);
1779 return new Color(r, g, b);
1780 } else {
1781 Main.warn("Unable to translate TransparentColor '"+value+"' with color model "+model);
1782 }
1783 }
1784 }
1785 }
1786 }
1787 break;
1788 }
1789 }
1790 }
1791 }
1792 } catch (IIOException | NumberFormatException e) {
1793 // JAI doesn't like some JPEG files with error "Inconsistent metadata read from stream" (see #10267)
1794 Main.warn(e);
1795 }
1796 return null;
1797 }
1798
1799 private static Color parseRGB(String ... s) {
1800 int[] rgb = new int[3];
1801 try {
1802 for (int i = 0; i < 3; i++) {
1803 rgb[i] = Integer.parseInt(s[i]);
1804 }
1805 return new Color(rgb[0], rgb[1], rgb[2]);
1806 } catch (IllegalArgumentException e) {
1807 Main.error(e);
1808 return null;
1809 }
1810 }
1811
1812 /**
1813 * Returns a transparent version of the given image, based on the given transparent color.
1814 * @param bi The image to convert
1815 * @param color The transparent color
1816 * @return The same image as {@code bi} where all pixels of the given color are transparent.
1817 * This resulting image has also the special property {@link #PROP_TRANSPARENCY_FORCED} set to {@code color}
1818 * @see BufferedImage#getProperty
1819 * @see #isTransparencyForced
1820 * @since 7132
1821 */
1822 public static BufferedImage makeImageTransparent(BufferedImage bi, Color color) {
1823 // the color we are looking for. Alpha bits are set to opaque
1824 final int markerRGB = color.getRGB() | 0xFF000000;
1825 ImageFilter filter = new RGBImageFilter() {
1826 @Override
1827 public int filterRGB(int x, int y, int rgb) {
1828 if ((rgb | 0xFF000000) == markerRGB) {
1829 // Mark the alpha bits as zero - transparent
1830 return 0x00FFFFFF & rgb;
1831 } else {
1832 return rgb;
1833 }
1834 }
1835 };
1836 ImageProducer ip = new FilteredImageSource(bi.getSource(), filter);
1837 Image img = Toolkit.getDefaultToolkit().createImage(ip);
1838 ColorModel colorModel = ColorModel.getRGBdefault();
1839 WritableRaster raster = colorModel.createCompatibleWritableRaster(img.getWidth(null), img.getHeight(null));
1840 String[] names = bi.getPropertyNames();
1841 Hashtable<String, Object> properties = new Hashtable<>(1 + (names != null ? names.length : 0));
1842 if (names != null) {
1843 for (String name : names) {
1844 properties.put(name, bi.getProperty(name));
1845 }
1846 }
1847 properties.put(PROP_TRANSPARENCY_FORCED, Boolean.TRUE);
1848 BufferedImage result = new BufferedImage(colorModel, raster, false, properties);
1849 Graphics2D g2 = result.createGraphics();
1850 g2.drawImage(img, 0, 0, null);
1851 g2.dispose();
1852 return result;
1853 }
1854
1855 /**
1856 * Determines if the transparency of the given {@code BufferedImage} has been enforced by a previous call to {@link #makeImageTransparent}.
1857 * @param bi The {@code BufferedImage} to test
1858 * @return {@code true} if the transparency of {@code bi} has been enforced by a previous call to {@code makeImageTransparent}.
1859 * @see #makeImageTransparent
1860 * @since 7132
1861 */
1862 public static boolean isTransparencyForced(BufferedImage bi) {
1863 return bi != null && !bi.getProperty(PROP_TRANSPARENCY_FORCED).equals(Image.UndefinedProperty);
1864 }
1865
1866 /**
1867 * Determines if the given {@code BufferedImage} has a transparent color determined by a previous call to {@link #read}.
1868 * @param bi The {@code BufferedImage} to test
1869 * @return {@code true} if {@code bi} has a transparent color determined by a previous call to {@code read}.
1870 * @see #read
1871 * @since 7132
1872 */
1873 public static boolean hasTransparentColor(BufferedImage bi) {
1874 return bi != null && !bi.getProperty(PROP_TRANSPARENCY_COLOR).equals(Image.UndefinedProperty);
1875 }
1876
1877 /**
1878 * Shutdown background image fetcher.
1879 * @param now if {@code true}, attempts to stop all actively executing tasks, halts the processing of waiting tasks.
1880 * if {@code false}, initiates an orderly shutdown in which previously submitted tasks are executed, but no new tasks will be accepted
1881 * @since 8412
1882 */
1883 public static void shutdown(boolean now) {
1884 if (now) {
1885 IMAGE_FETCHER.shutdownNow();
1886 } else {
1887 IMAGE_FETCHER.shutdown();
1888 }
1889 }
1890}
Note: See TracBrowser for help on using the repository browser.