com.talanlabs.avatargenerator.utils.AvatarUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of avatar-generator Show documentation
The newest version!
package com.talanlabs.avatargenerator.utils;

import javax.imageio.ImageIO;
import java.awt.AlphaComposite;
import java.awt.Color;
import java.awt.Graphics2D;
import java.awt.Image;
import java.awt.RenderingHints;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.Arrays;
import java.util.List;

public class AvatarUtils {

    public static List defaultColors = Arrays
            .asList(new Color(0x6e1e78), new Color(0x82be00), new Color(0xa1006b), new Color(0x009aa6), new Color(0xcd0037), new Color(0x0088ce), new Color(0xe05206), new Color(0xd52b1e),
                    new Color(0xffb612), new Color(0xd2e100));

    private AvatarUtils() {
        super();
    }

    /**
     * Active antialiasing
     */
    public static void activeAntialiasing(Graphics2D g2) {
        g2.setRenderingHint(RenderingHints.KEY_INTERPOLATION, RenderingHints.VALUE_INTERPOLATION_BICUBIC);
        g2.setRenderingHint(RenderingHints.KEY_ALPHA_INTERPOLATION, RenderingHints.VALUE_ALPHA_INTERPOLATION_QUALITY);
        g2.setRenderingHint(RenderingHints.KEY_RENDERING, RenderingHints.VALUE_RENDER_QUALITY);
        g2.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
        g2.setRenderingHint(RenderingHints.KEY_COLOR_RENDERING, RenderingHints.VALUE_COLOR_RENDER_QUALITY);
    }

    /**
     * Resize Image with aspect ratio
     *
     * @param src    image source
     * @param width  new width
     * @param height new height
     * @return new image
     */
    public static BufferedImage resizeImage(BufferedImage src, int width, int height) {
        int original_width = src.getWidth();
        int original_height = src.getHeight();

        if (original_width == width && original_height == height) {
            return src;
        }

        double diffComponent = width / (double) height;
        double diffImage = original_width / (double) original_height;

        double diff = diffImage / diffComponent;
        int w = width;
        int h = height;
        if (diff >= 1.0) {
            h = (int) (h / diff);
        } else {
            w = (int) (w * diff);
        }

        BufferedImage dest = new BufferedImage(width, height, BufferedImage.TYPE_INT_ARGB);
        Graphics2D g2 = dest.createGraphics();
        AvatarUtils.activeAntialiasing(g2);
        g2.drawImage(src.getScaledInstance(w, h, Image.SCALE_SMOOTH), (width - w) / 2, (height - h) / 2, null);
        g2.dispose();
        return dest;
    }

    /**
     * Colorize image
     *
     * @param src   image source
     * @param color color
     * @return image with color
     */
    public static BufferedImage fillColorImage(BufferedImage src, Color color) {
        int original_width = src.getWidth();
        int original_height = src.getHeight();

        BufferedImage dest = new BufferedImage(original_width, original_height, BufferedImage.TYPE_INT_ARGB);
        Graphics2D g2 = dest.createGraphics();
        g2.setColor(color);
        g2.fillRect(0, 0, original_width, original_height);
        g2.setComposite(AlphaComposite.DstIn);
        g2.drawImage(src, 0, 0, null);
        g2.dispose();
        return dest;
    }

    /**
     * Tint image
     *
     * @param src   image source
     * @param color color
     * @return image tint with color
     */
    public static BufferedImage tintImage(BufferedImage src, Color color) {
        int w = src.getWidth();
        int h = src.getHeight();
        BufferedImage dest = new BufferedImage(w, h, BufferedImage.TYPE_INT_ARGB);
        Graphics2D g = dest.createGraphics();
        g.drawImage(src, 0, 0, null);
        g.setComposite(AlphaComposite.SrcAtop);
        g.setColor(color);
        g.fillRect(0, 0, w, h);
        g.dispose();
        return dest;
    }

    /**
     * Resize planImage
     *
     * @param src    image source
     * @param width  new width
     * @param height new height
     * @return new image
     */
    public static BufferedImage planImage(BufferedImage src, int width, int height) {
        int original_width = src.getWidth();
        int original_height = src.getHeight();
        if (original_width == width && original_height == height) {
            return src;
        }

        BufferedImage dest = new BufferedImage(width, height, BufferedImage.TYPE_INT_ARGB);
        Graphics2D g2 = dest.createGraphics();
        g2.drawImage(src, (width - original_width) / 2, (height - original_height) / 2, null);
        g2.dispose();
        return dest;
    }

    /**
     * Complement color
     *
     * @param color      current color
     * @param blackColor color dark
     * @param whiteColor color light
     * @return blackColor or whiteColor
     */
    public static Color getComplementColor(Color color, Color blackColor, Color whiteColor) {
        float[] rgba1 = color.getComponents(null);
        double l = 0.2126 * rgba1[0] + 0.7152 * rgba1[1] + 0.0722 * rgba1[2];
        double ratio = (l + 0.05) / 0.05;
        return ratio > 7 ? blackColor : whiteColor;
    }

    /**
     * Convert image to argb image
     *
     * @param src image
     * @return new image type argb
     */
    public static BufferedImage toARGBImage(Image src) {
        BufferedImage dest = new BufferedImage(src.getWidth(null), src.getHeight(null), BufferedImage.TYPE_INT_ARGB);
        Graphics2D g2 = dest.createGraphics();
        g2.drawImage(src, 0, 0, null);
        g2.dispose();
        return dest;
    }

    /**
     * Save image in temp path
     *
     * @param src image
     * @return path
     */
    public static Path saveImageInTemp(BufferedImage src) {
        try {
            Path path = Files.createTempFile("image", ".png");
            ImageIO.write(src, "png", path.toFile());
            return path;
        } catch (IOException e) {
            throw new RuntimeException("Failed to save image", e);
        }
    }

    private static String addZeroes(String ns, int nd) {
        for (int i = 0; i < nd - ns.length(); i++) {
            ns = "0" + ns;
        }
        return ns;
    }

    /**
     * Extract color in code
     *
     * @param code current code
     * @return color
     */
    public static Color extractColor(long code) {
        String n = addZeroes(Long.toHexString(code), 6);
        return new Color(Integer.parseInt(n.substring(0, 6), 16));
    }

    /**
     * Get distance between two colors
     *
     * @param c1 color 1
     * @param c2 color 2
     * @return distance
     */
    public static float getColorDistance(Color c1, Color c2) {
        float dx = c1.getRed() - c2.getRed();
        float dy = c1.getGreen() - c2.getGreen();
        float dz = c1.getBlue() - c2.getBlue();
        return (float) Math.sqrt(dx * dx + dy * dy + dz * dz);
    }

    /**
     * Returns complementary color.
     *
     * @param color
     * @return
     */
    public static Color getComplementaryColor(Color color) {
        return new Color(color.getRGB() ^ 0x00FFFFFF);
    }
}