Add override annotations

Author: mialeska

src/main/java/aquality/selenium/core/visualization/ImageComparator.java

@@ -23,6 +23,12 @@ private int getComparisonWidth() {
         return visualConfiguration.getComparisonWidth();
     }
 
+    @Override
+    public float percentageDifference(Image thisOne, Image theOtherOne) {
+        return percentageDifference(thisOne, theOtherOne, visualConfiguration.getDefaultThreshold());
+    }
+
+    @Override
     public float percentageDifference(Image thisOne, Image theOtherOne, float threshold) {
         if (threshold < 0 || threshold > 1) {
             throw new IllegalArgumentException(String.format("Threshold should be between 0 and 1, but was [%s]", threshold));
@@ -32,10 +38,6 @@ public float percentageDifference(Image thisOne, Image theOtherOne, float thresh
         return percentageDifference(thisOne, theOtherOne, intThreshold);
     }
 
-    public float percentageDifference(Image thisOne, Image theOtherOne) {
-        return percentageDifference(thisOne, theOtherOne, visualConfiguration.getDefaultThreshold());
-    }
-
     protected float percentageDifference(Image thisOne, Image theOtherOne, int threshold) {
         int[][] differences = getDifferences(thisOne, theOtherOne);
 

src/main/java/aquality/selenium/core/visualization/ImageFunctions.java

@@ -5,13 +5,15 @@
 import org.openqa.selenium.OutputType;
 import org.openqa.selenium.remote.RemoteWebElement;
 
+import javax.imageio.IIOImage;
 import javax.imageio.ImageIO;
+import javax.imageio.ImageWriteParam;
+import javax.imageio.ImageWriter;
+import javax.imageio.stream.FileImageInputStream;
 import java.awt.*;
 import java.awt.image.BufferedImage;
 import java.awt.image.RenderedImage;
-import java.io.ByteArrayInputStream;
-import java.io.IOException;
-import java.io.InputStream;
+import java.io.*;
 
 import static java.awt.image.BufferedImage.TYPE_INT_RGB;
 
@@ -102,4 +104,26 @@ public static BufferedImage resize(Image image, int width, int height) {
         graphics.dispose();
         return resizedImage;
     }
+
+    /**
+     * Saves image in the highest quality.
+     * @param image source image.
+     * @param name target name without extension.
+     * @param format target format.
+     */
+    public static void save(RenderedImage image, String name, String format) {
+        final float highestQuality = 1.0F;
+        ImageWriter writer = ImageIO.getImageWritersByFormatName(format).next();
+        ImageWriteParam param = writer.getDefaultWriteParam();
+        param.setCompressionMode(ImageWriteParam.MODE_EXPLICIT);
+        param.setCompressionQuality(highestQuality);
+        String fileName = String.format("%s.%s", name, format.startsWith(".") ? format.substring(1) : format);
+        try {
+            writer.setOutput(new FileImageInputStream(new File(fileName)));
+            writer.write(null, new IIOImage(image, null, null), param);
+        } catch (IOException e) {
+            Logger.getInstance().fatal("Failed to save the image: " + e.getMessage(), e);
+            throw new UncheckedIOException(e);
+        }
+    }
 }

src/main/java/aquality/selenium/core/visualization/VisualStateProvider.java

@@ -9,7 +9,6 @@
 import java.util.function.Supplier;
 
 public class VisualStateProvider implements IVisualStateProvider {
-
     private final IImageComparator imageComparator;
     private final IElementActionRetrier actionRetrier;
     private final Supplier<RemoteWebElement> getElement;
@@ -22,20 +21,23 @@ public VisualStateProvider(IImageComparator imageComparator, IElementActionRetri
         this.stateLogger = stateLogger;
     }
 
+    @Override
     public Dimension getSize() {
         return getLoggedValue("size", element -> {
             final org.openqa.selenium.Dimension size = element.getSize();
             return new Dimension(size.getWidth(), size.getHeight());
         }, null);
     }
 
+    @Override
     public Point getLocation() {
         return getLoggedValue("location", element -> {
             final org.openqa.selenium.Point location = element.getLocation();
             return new Point(location.getX(), location.getY());
         }, null);
     }
 
+    @Override
     public Image getImage() {
         return getLoggedValue("image", ImageFunctions::getScreenshotAsImage,
                 image -> getStringValue(ImageFunctions.getSize(image)));