Package ca.eandb.jmist.framework.texture

Source Code of ca.eandb.jmist.framework.texture.RasterTexture2

/**
* Java Modular Image Synthesis Toolkit (JMIST)
* Copyright (C) 2008-2013 Bradley W. Kimmel
*
* Permission is hereby granted, free of charge, to any person
* obtaining a copy of this software and associated documentation
* files (the "Software"), to deal in the Software without
* restriction, including without limitation the rights to use,
* copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the
* Software is furnished to do so, subject to the following
* conditions:
*
* The above copyright notice and this permission notice shall be
* included in all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
* OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
* HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
* WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
* OTHER DEALINGS IN THE SOFTWARE.
*/
package ca.eandb.jmist.framework.texture;

import java.awt.image.BufferedImage;
import java.awt.image.Raster;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.net.URL;

import javax.imageio.ImageIO;
import javax.imageio.stream.ImageInputStream;

import ca.eandb.jmist.framework.PixelSpectrumFactory;
import ca.eandb.jmist.framework.Texture2;
import ca.eandb.jmist.framework.color.Color;
import ca.eandb.jmist.framework.color.ColorModel;
import ca.eandb.jmist.framework.color.WavelengthPacket;
import ca.eandb.jmist.math.MathUtil;
import ca.eandb.jmist.math.Point2;

/**
* A <code>Texture2</code> that is extrapolated from a <code>Raster</code>
* image.
* @author Brad Kimmel
*/
public final class RasterTexture2 implements Texture2 {

  /**
   * Serialization version ID.
   */
  private static final long serialVersionUID = -2712131011948642676L;

  /**
   * Creates a new <code>RasterTexture2</code>.
   * @param image The <code>BufferedImage</code> to use as the basis for the
   *     new <code>Texture2</code>.
   * @param factory The <code>PixelSpectrumFactory</code> to use to create
   *     spectra from <code>Pixel</code>s.
   * @param background The <code>Texture2</code> to render underneath if the
   *     image has an alpha channel.
   */
  public RasterTexture2(BufferedImage image, PixelSpectrumFactory factory, Texture2 background) {
    this.image = image;
    this.factory = factory;
    this.background = background;
  }

  public RasterTexture2(BufferedImage image, Texture2 background) {
    this(image, null, background);
  }

  public RasterTexture2(File file, Texture2 background) throws IOException {
    this(ImageIO.read(file), background);
  }

  public RasterTexture2(URL input, Texture2 background) throws IOException {
    this(ImageIO.read(input), background);
  }

  public RasterTexture2(ImageInputStream stream, Texture2 background) throws IOException {
    this(ImageIO.read(stream), background);
  }

  public RasterTexture2(InputStream input, Texture2 background) throws IOException {
    this(ImageIO.read(input), background);
  }

  /**
   * Creates a new <code>RasterTexture2</code>.
   * @param image The <code>BufferedImage</code> to use as the basis for the
   *     new <code>Texture2</code>.
   * @param factory The <code>PixelSpectrumFactory</code> to use to create
   *     spectra from <code>Pixel</code>s.
   */
  public RasterTexture2(BufferedImage image, PixelSpectrumFactory factory) {
    this(image, factory, Texture2.BLACK);
  }

  public RasterTexture2(BufferedImage image) {
    this(image, null, Texture2.BLACK);
  }

  public RasterTexture2(File file) throws IOException {
    this(ImageIO.read(file));
  }

  public RasterTexture2(URL input) throws IOException {
    this(ImageIO.read(input));
  }

  public RasterTexture2(ImageInputStream stream) throws IOException {
    this(ImageIO.read(stream));
  }

  public RasterTexture2(InputStream input) throws IOException {
    this(ImageIO.read(input));
  }

  private void writeObject(ObjectOutputStream oos) throws IOException {
    oos.defaultWriteObject();
    ImageIO.write(image, "png", oos);
  }

  private void readObject(ObjectInputStream ois)
      throws ClassNotFoundException, IOException {
    ois.defaultReadObject();
    image = ImageIO.read(ois);
    placeholder = new ThreadLocal<double[]>();
  }

  /* (non-Javadoc)
   * @see ca.eandb.jmist.framework.Texture2#evaluate(ca.eandb.jmist.math.Point2, ca.eandb.jmist.framework.color.WavelengthPacket)
   */
  public Color evaluate(Point2 p, WavelengthPacket lambda) {

    Raster    raster  = image.getRaster();
    double    u    = p.x() - Math.floor(p.x());
    double    v    = p.y() - Math.floor(p.y());
    int      w    = raster.getWidth();
    int      h    = raster.getHeight();
    int      x    = MathUtil.clamp((int) Math.floor(u * (double) w), 0, w - 1);
    int      y    = MathUtil.clamp((int) Math.floor(v * (double) h), 0, h - 1);
    double[]  pixel  = raster.getPixel(x, y, placeholder.get());
    ColorModel  cm    = lambda.getColorModel();

    placeholder.set(pixel);

    if (factory != null) {
      return cm.getContinuous(factory.createSpectrum(pixel)).sample(lambda);
    } else {
      switch (pixel.length) {
      case 1:
        return cm.getGray(pixel[0], lambda);
      case 3:
        return cm.fromRGB(pixel[0]/255.0, pixel[1]/255.0, pixel[2]/255.0).sample(lambda);
      case 4:
      {
        double alpha = pixel[3] / 255.0;
        if (MathUtil.equal(alpha, 1.0)) {
          return cm.fromRGB(pixel[0]/255.0, pixel[1]/255.0, pixel[2]/255.0).sample(lambda);
        } else if (MathUtil.equal(alpha, 0.0)) {
          return background.evaluate(p, lambda);
        } else {
          Color bg = background.evaluate(p, lambda);
          Color fg = cm.fromRGB(pixel[0]/255.0, pixel[1]/255.0, pixel[2]/255.0).sample(lambda);
          return fg.times(alpha).plus(bg.times(1.0 - alpha));
        }
      }
      default:
        throw new RuntimeException("Raster has unrecognized number of bands.");
      }
    }

  }

  /**
   * A place holder to hold a double array for the pixel, so that one is not
   * allocated on every call to {@link #evaluate(Point2, WavelengthPacket)}.
   */
  private transient ThreadLocal<double[]> placeholder = new ThreadLocal<double[]>();

  /**
   * The <code>BufferedImage</code> that serves as the basis for this
   * <code>Texture2</code>.
   */
  private transient BufferedImage image;

  /**
   * The <code>PixelSpectrumFactory</code> to use to extrapolate
   * <code>Pixel</code>s.
   */
  private final PixelSpectrumFactory factory;

  /**
   * The <code>Texture2</code> to render underneath this texture if the image
   * has an alpha channel.
   */
  private final Texture2 background;

}
TOP

Related Classes of ca.eandb.jmist.framework.texture.RasterTexture2

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.