Plex-FAWE/worldedit-core/src/main/java/com/boydti/fawe/beta/implementation/filter/ArrayImageMask.java

package com.boydti.fawe.beta.implementation.filter;

import com.boydti.fawe.beta.implementation.filter.block.FilterBlock;
import com.boydti.fawe.beta.FilterBlockMask;
import java.awt.image.BufferedImage;
import java.util.concurrent.ThreadLocalRandom;

public class ArrayImageMask implements FilterBlockMask {

    private final ThreadLocalRandom random;
    private final boolean white;
    private final BufferedImage image;

    public ArrayImageMask(BufferedImage image, boolean white) {
        this.image = image;
        this.white = white;
        this.random = ThreadLocalRandom.current();
    }

    @Override
    public boolean applyBlock(FilterBlock block) {
        int height = image.getRGB(block.getX(), block.getZ()) & 0xFF;
        return height == 255 || height > 0 && !white && random.nextInt(256) <= height;
    }
}