Subversion Repositories scarts

/* ConvolveOp.java --

   Copyright (C) 2004, 2005 Free Software Foundation -- ConvolveOp

This file is part of GNU Classpath.

GNU Classpath is free software; you can redistribute it and/or modify

it under the terms of the GNU General Public License as published by

the Free Software Foundation; either version 2, or (at your option)

any later version.

GNU Classpath is distributed in the hope that it will be useful, but

WITHOUT ANY WARRANTY; without even the implied warranty of

MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

General Public License for more details.

You should have received a copy of the GNU General Public License

along with GNU Classpath; see the file COPYING.  If not, write to the

Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA

02110-1301 USA.

Linking this library statically or dynamically with other modules is

making a combined work based on this library.  Thus, the terms and

conditions of the GNU General Public License cover the whole

combination.

As a special exception, the copyright holders of this library give you

permission to link this library with independent modules to produce an

executable, regardless of the license terms of these independent

modules, and to copy and distribute the resulting executable under

terms of your choice, provided that you also meet, for each linked

independent module, the terms and conditions of the license of that

module.  An independent module is a module which is not derived from

or based on this library.  If you modify this library, you may extend

this exception to your version of the library, but you are not

obligated to do so.  If you do not wish to do so, delete this

exception statement from your version. */

package java.awt.image;

import java.awt.Graphics2D;

import java.awt.RenderingHints;

import java.awt.geom.Point2D;

import java.awt.geom.Rectangle2D;

import java.util.Arrays;

/**

 * Convolution filter.

 *

 * ConvolveOp convolves the source image with a Kernel to generate a

 * destination image.  This involves multiplying each pixel and its neighbors

 * with elements in the kernel to compute a new pixel.

 *

 * Each band in a Raster is convolved and copied to the destination Raster.

 *

 * For BufferedImages, convolution is applied to all components.  If the

 * source is not premultiplied, the data will be premultiplied before

 * convolving.  Premultiplication will be undone if the destination is not

 * premultiplied.  Color conversion will be applied if needed.

 *

 * @author jlquinn@optonline.net

 */

public class ConvolveOp implements BufferedImageOp, RasterOp

{

  /** Edge pixels are set to 0. */

  public static final int EDGE_ZERO_FILL = 0;

  /** Edge pixels are copied from the source. */

  public static final int EDGE_NO_OP = 1;

  private Kernel kernel;

  private int edge;

  private RenderingHints hints;

  /**

   * Construct a ConvolveOp.

   *

   * The edge condition specifies that pixels outside the area that can be

   * filtered are either set to 0 or copied from the source image.

   *

   * @param kernel The kernel to convolve with.

   * @param edgeCondition Either EDGE_ZERO_FILL or EDGE_NO_OP.

   * @param hints Rendering hints for color conversion, or null.

   */

  public ConvolveOp(Kernel kernel,

                                int edgeCondition,

                                RenderingHints hints)

  {

    this.kernel = kernel;

    edge = edgeCondition;

    this.hints = hints;

  }

  /**

   * Construct a ConvolveOp.

   *

   * The edge condition defaults to EDGE_ZERO_FILL.

   *

   * @param kernel The kernel to convolve with.

   */

  public ConvolveOp(Kernel kernel)

  {

    this.kernel = kernel;

    edge = EDGE_ZERO_FILL;

    hints = null;

  }

  /* (non-Javadoc)

   * @see java.awt.image.BufferedImageOp#filter(java.awt.image.BufferedImage,

   * java.awt.image.BufferedImage)

   */

  public BufferedImage filter(BufferedImage src, BufferedImage dst)

  {

    if (src == dst)

      throw new IllegalArgumentException();

    if (dst == null)

      dst = createCompatibleDestImage(src, src.getColorModel());

    // Make sure source image is premultiplied

    BufferedImage src1 = src;

    if (!src.isPremultiplied)

    {

      src1 = createCompatibleDestImage(src, src.getColorModel());

      src.copyData(src1.getRaster());

      src1.coerceData(true);

    }

    BufferedImage dst1 = dst;

    if (!src.getColorModel().equals(dst.getColorModel()))

      dst1 = createCompatibleDestImage(src, src.getColorModel());

    filter(src1.getRaster(), dst1.getRaster());

    if (dst1 != dst)

    {

      // Convert between color models.

      // TODO Check that premultiplied alpha is handled correctly here.

      Graphics2D gg = dst.createGraphics();

      gg.setRenderingHints(hints);

      gg.drawImage(dst1, 0, 0, null);

      gg.dispose();

    }

    return dst;

  }

  /* (non-Javadoc)

   * @see

   * java.awt.image.BufferedImageOp#createCompatibleDestImage(java.awt.image.BufferedImage,

   * java.awt.image.ColorModel)

   */

  public BufferedImage createCompatibleDestImage(BufferedImage src,

                                                 ColorModel dstCM)

  {

    // FIXME: set properties to those in src

    return new BufferedImage(dstCM,

                             src.getRaster().createCompatibleWritableRaster(),

                             src.isPremultiplied, null);

  }

  /* (non-Javadoc)

   * @see java.awt.image.RasterOp#getRenderingHints()

   */

  public RenderingHints getRenderingHints()

  {

    return hints;

  }

  /**

   * @return The edge condition.

   */

  public int getEdgeCondition()

  {

    return edge;

  }

  /**

   * Returns (a clone of) the convolution kernel.

   *

   * @return The convolution kernel.

   */

  public Kernel getKernel()

  {

    return (Kernel) kernel.clone();

  }

  /* (non-Javadoc)

   * @see java.awt.image.RasterOp#filter(java.awt.image.Raster,

   * java.awt.image.WritableRaster)

   */

  public WritableRaster filter(Raster src, WritableRaster dest) {

    if (src == dest)

      throw new IllegalArgumentException();

    if (src.getWidth() < kernel.getWidth() ||

        src.getHeight() < kernel.getHeight())

      throw new ImagingOpException(null);

    if (dest == null)

      dest = createCompatibleDestRaster(src);

    else if (src.numBands != dest.numBands)

      throw new ImagingOpException(null);

    // Deal with bottom edge

    if (edge == EDGE_ZERO_FILL)

    {

      float[] zeros = new float[src.getNumBands() * src.getWidth()

                                * (kernel.getYOrigin() - 1)];

      Arrays.fill(zeros, 0);

      dest.setPixels(src.getMinX(), src.getMinY(), src.getWidth(),

                     kernel.getYOrigin() - 1, zeros);

    }

    else

    {

      float[] vals = new float[src.getNumBands() * src.getWidth()

                               * (kernel.getYOrigin() - 1)];

      src.getPixels(src.getMinX(), src.getMinY(), src.getWidth(),

                    kernel.getYOrigin() - 1, vals);

      dest.setPixels(src.getMinX(), src.getMinY(), src.getWidth(),

                     kernel.getYOrigin() - 1, vals);

    }

    // Handle main section

    float[] kvals = kernel.getKernelData(null);

    float[] tmp = new float[kernel.getWidth() * kernel.getHeight()];

    for (int y = src.getMinY() + kernel.getYOrigin();

         y < src.getMinY() + src.getHeight() - kernel.getYOrigin() / 2; y++)

    {

      // Handle unfiltered edge pixels at start of line

      float[] t1 = new float[(kernel.getXOrigin() - 1) * src.getNumBands()];

      if (edge == EDGE_ZERO_FILL)

        Arrays.fill(t1, 0);

      else

        src.getPixels(src.getMinX(), y, kernel.getXOrigin() - 1, 1, t1);

      dest.setPixels(src.getMinX(), y, kernel.getXOrigin() - 1, 1, t1);

      for (int x = src.getMinX(); x < src.getWidth() + src.getMinX(); x++)

      {

        // FIXME: This needs a much more efficient implementation

        for (int b = 0; b < src.getNumBands(); b++)

        {

          float v = 0;

          src.getSamples(x, y, kernel.getWidth(), kernel.getHeight(), b, tmp);

          for (int i=0; i < tmp.length; i++)

            v += tmp[i] * kvals[i];

          dest.setSample(x, y, b, v);

        }

      }

      // Handle unfiltered edge pixels at end of line

      float[] t2 = new float[(kernel.getWidth() / 2) * src.getNumBands()];

      if (edge == EDGE_ZERO_FILL)

        Arrays.fill(t2, 0);

      else

        src.getPixels(src.getMinX() + src.getWidth()

                      - (kernel.getWidth() / 2),

                      y, kernel.getWidth() / 2, 1, t2);

      dest.setPixels(src.getMinX() + src.getWidth() - (kernel.getWidth() / 2),

                     y, kernel.getWidth() / 2, 1, t2);

    }

    for (int y = src.getMinY(); y < src.getHeight() + src.getMinY(); y++)

      for (int x = src.getMinX(); x< src.getWidth() + src.getMinX(); x++)

      {

      }

    for (int y = src.getMinY(); y < src.getHeight() + src.getMinY(); y++)

      for (int x = src.getMinX(); x< src.getWidth() + src.getMinX(); x++)

      {

      }

    // Handle top edge

    if (edge == EDGE_ZERO_FILL)

    {

      float[] zeros = new float[src.getNumBands() * src.getWidth() *

                                (kernel.getHeight() / 2)];

      Arrays.fill(zeros, 0);

      dest.setPixels(src.getMinX(),

          src.getHeight() + src.getMinY() - (kernel.getHeight() / 2),

          src.getWidth(), kernel.getHeight() / 2, zeros);

    }

    else

    {

      float[] vals = new float[src.getNumBands() * src.getWidth() *

                               (kernel.getHeight() / 2)];

      src.getPixels(src.getMinX(),

                    src.getHeight() + src.getMinY()

                    - (kernel.getHeight() / 2),

                    src.getWidth(), kernel.getHeight() / 2, vals);

      dest.setPixels(src.getMinX(),

                     src.getHeight() + src.getMinY()

                     - (kernel.getHeight() / 2),

                     src.getWidth(), kernel.getHeight() / 2, vals);

    }

    return dest;

  }

  /* (non-Javadoc)

   * @see java.awt.image.RasterOp#createCompatibleDestRaster(java.awt.image.Raster)

   */

  public WritableRaster createCompatibleDestRaster(Raster src)

  {

    return src.createCompatibleWritableRaster();

  }

  /* (non-Javadoc)

   * @see java.awt.image.BufferedImageOp#getBounds2D(java.awt.image.BufferedImage)

   */

  public Rectangle2D getBounds2D(BufferedImage src)

  {

    return src.getRaster().getBounds();

  }

  /* (non-Javadoc)

   * @see java.awt.image.RasterOp#getBounds2D(java.awt.image.Raster)

   */

  public Rectangle2D getBounds2D(Raster src)

  {

    return src.getBounds();

  }

  /** Return corresponding destination point for source point.

   *

   * ConvolveOp will return the value of src unchanged.

   * @param src The source point.

   * @param dst The destination point.

   * @see java.awt.image.RasterOp#getPoint2D(java.awt.geom.Point2D,

   * java.awt.geom.Point2D)

   */

  public Point2D getPoint2D(Point2D src, Point2D dst)

  {

    if (dst == null) return (Point2D)src.clone();

    dst.setLocation(src);

    return dst;

  }

}