WordPixelPadder.java

package wordcloud.padding;

import wordcloud.Word;
import wordcloud.collide.Vector2d;
import wordcloud.image.CollisionRaster;

import java.awt.*;
import java.util.HashSet;
import java.util.Set;

/**
 * Created by kenny on 7/1/14.
 * Add pixel padding around the numbers
 */
public class WordPixelPadder implements Padder {

    // TODO as CollisionRaster changes to use boolean states rgb is not really needed or makes sense for padding
    // it used to actually changed the buffered image that the word's text is written into.
    private static final Color PAD_COLOR = Color.BLACK;

    private RectanglePadder rectanglePadder = new RectanglePadder();

    public void pad(final Word word, final int padding) {
        if(padding <= 0) { return; }
        rectanglePadder.pad(word, padding);

        final CollisionRaster collisionRaster = word.getCollisionRaster();

        final Set<Vector2d> toPad = new HashSet<>();
        final int width = collisionRaster.getWidth();
        final int height = collisionRaster.getHeight();

        for(int y = 0; y < height; y++) {
            for(int x = 0; x < width; x++) {
                if(shouldPad(collisionRaster, x, y, padding)) {
                    toPad.add(new Vector2d(x, y));
                }
            }
        }
        for(Vector2d padPoint : toPad) {
            collisionRaster.setRGB(padPoint.getX(), padPoint.getY(), PAD_COLOR.getRGB());
        }
    }

    private boolean shouldPad(final CollisionRaster collisionRaster, final int cx, final int cy, final int padding) {
        if(!collisionRaster.isTransparent(cx, cy)) { return false; }

        for(int y = cy - padding; y <= cy + padding; y++) {
            for(int x = cx - padding; x <= cx + padding; x++) {
                if(x == cx && y == cy) { continue; }
                if(inBounds(collisionRaster, x, y)) {
                    if(!collisionRaster.isTransparent(x, y)) {
                        return true;
                    }
                }
            }
        }
        return false;
    }

    private boolean inBounds(final CollisionRaster collisionRaster, final int x, final int y) {
        return x >= 0
                && y >= 0
                && x < collisionRaster.getWidth()
                && y < collisionRaster.getHeight();
    }

}