Index: java/org/chromium/distiller/webdocument/WebDocument.java |
diff --git a/java/org/chromium/distiller/webdocument/WebDocument.java b/java/org/chromium/distiller/webdocument/WebDocument.java |
index 021851baa3f4a138c09382fc7682b5030e4df40a..0b8f482e517d754a8e9f3a3387d3b3f8bed4d9d7 100644 |
--- a/java/org/chromium/distiller/webdocument/WebDocument.java |
+++ b/java/org/chromium/distiller/webdocument/WebDocument.java |
@@ -41,11 +41,12 @@ public class WebDocument { |
return elements; |
} |
- public List<WebImage> getContentImages() { |
- List<WebImage> images = new ArrayList<>(); |
+ public List<String> getImageUrls() { |
+ List<String> images = new ArrayList<>(); |
for (WebElement e : elements) { |
- if (e instanceof WebImage && e.getIsContent()) { |
- images.add((WebImage) e); |
+ if (!e.getIsContent()) continue; |
+ if (e instanceof WebImage) { |
+ images.addAll(((WebImage) e).getUrlList()); |
} |
} |
return images; |