thirdparty.marvin.Convolution Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of scrimage-filters Show documentation
JVM Image Library
There is a newer version: 4.3.0
package thirdparty.marvin;

import thirdparty.marvin.image.MarvinAbstractImagePlugin;
import thirdparty.marvin.image.MarvinAttributes;
import thirdparty.marvin.image.MarvinImage;
import thirdparty.marvin.image.MarvinImageMask;

/**
 * @author Gabriel Ambr�sio Archanjo
 */
public class Convolution extends MarvinAbstractImagePlugin {

    public double[][] matrix = new double[3][3];

    public void process(MarvinImage imageIn,
                        MarvinImage imageOut,
                        MarvinAttributes attributesOut,
                        MarvinImageMask mask,
                        boolean previewMode) {

        for (int y = 0; y < imageIn.getHeight(); y++) {
            for (int x = 0; x < imageIn.getWidth(); x++) {
                applyMatrix(x, y, matrix, imageIn, imageOut);
            }
        }
    }

    private void applyMatrix(int x,
                             int y,
                             double[][] matrix,
                             MarvinImage imageIn,
                             MarvinImage imageOut) {

        int nx, ny;
        double resultRed = 0;
        double resultGreen = 0;
        double resultBlue = 0;

        int xC = matrix[0].length / 2;
        int yC = matrix.length / 2;

        for (int i = 0; i < matrix.length; i++) {
            for (int j = 0; j < matrix[0].length; j++) {

                if (matrix[i][j] != 0) {
                    nx = x + (j - xC);
                    ny = y + (i - yC);

                    if (nx >= 0 && nx < imageOut.getWidth() && ny >= 0 && ny < imageOut.getHeight()) {

                        resultRed += (matrix[i][j] * (imageIn.getIntComponent0(nx, ny)));
                        resultGreen += (matrix[i][j] * (imageIn.getIntComponent1(nx, ny)));
                        resultBlue += (matrix[i][j] * (imageIn.getIntComponent2(nx, ny)));
                    }


                }


            }
        }

        resultRed = Math.abs(resultRed);
        resultGreen = Math.abs(resultGreen);
        resultBlue = Math.abs(resultBlue);

        // allow the combination of multiple appications
        resultRed += imageOut.getIntComponent0(x, y);
        resultGreen += imageOut.getIntComponent1(x, y);
        resultBlue += imageOut.getIntComponent2(x, y);

        resultRed = Math.min(resultRed, 255);
        resultGreen = Math.min(resultGreen, 255);
        resultBlue = Math.min(resultBlue, 255);

        resultRed = Math.max(resultRed, 0);
        resultGreen = Math.max(resultGreen, 0);
        resultBlue = Math.max(resultBlue, 0);

        imageOut.setIntColor(x, y, imageIn.getAlphaComponent(x, y), (int) resultRed, (int) resultGreen, (int) resultBlue);
    }

}