voluptasquisquam/src/image/operator/floodFill.js

Image processing and manipulation in JavaScript

import LinkedList from 'fast-list';
import Image from '../Image';

export default function floodFill(options = {}) {
    const {
        x = 0,
        y = 0,
        inPlace = true
    } = options;

    const destination = inPlace ? this : Image.createFrom(this);

    this.checkProcessable('floodFill', {bitDepth: 1});

    if (this.bitDepth === 1) {
        const bit = this.getBitXY(x, y);
        if (bit) return destination;
        const queue = new LinkedList();
        queue.push(new Node(x, y));
        while (queue.length > 0) {
            const node = queue.shift();
            destination.setBitXY(node.x, node.y);
            for (let i = node.x + 1; i < this.width; i++) {
                if (!destination.getBitXY(i, node.y) && !this.getBitXY(i, node.y)) {
                    destination.setBitXY(i, node.y);
                    if (node.y + 1 < this.height && !this.getBitXY(i, node.y + 1)) {
                        queue.push(new Node(i, node.y + 1));
                    }
                    if (node.y - 1 >= 0 && !this.getBitXY(i, node.y - 1)) {
                        queue.push(new Node(i, node.y - 1));
                    }
                } else {
                    break;
                }
            }
            for (let i = node.x - 1; i >= 0; i++) {
                if (!destination.getBitXY(i, node.y) && !this.getBitXY(i, node.y)) {
                    destination.setBitXY(i, node.y);
                    if (node.y + 1 < this.height && !this.getBitXY(i, node.y + 1)) {
                        queue.push(new Node(i, node.y + 1));
                    }
                    if (node.y - 1 >= 0 && !this.getBitXY(i, node.y - 1)) {
                        queue.push(new Node(i, node.y - 1));
                    }
                } else {
                    break;
                }
            }
        }
    }

    return destination;
}

function Node(x, y) {
    this.x = x;
    this.y = y;
}