Java图像处理类库 Java Image Filters.7z

/* Copyright 2006 Jerry Huxtable Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0 Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License. */ package com.jhlabs.image; import java.awt.image.*; import com.jhlabs.math.*; import com.jhlabs.vecmath.*; import java.awt.*; import java.io.*; import java.util.*; /** * A filter which produces lighting and embossing effects. */ public class LightFilter extends WholeImageFilter { /** * Take the output colors from the input image. */ public final static int COLORS_FROM_IMAGE = 0; /** * Use constant material color. */ public final static int COLORS_CONSTANT = 1; /** * Use the input image brightness as the bump map. */ public final static int BUMPS_FROM_IMAGE = 0; /** * Use the input image alpha as the bump map. */ public final static int BUMPS_FROM_IMAGE_ALPHA = 1; /** * Use a separate image alpha channel as the bump map. */ public final static int BUMPS_FROM_MAP = 2; /** * Use a custom function as the bump map. */ public final static int BUMPS_FROM_BEVEL = 3; private float bumpHeight; private float bumpSoftness; private int bumpShape; private float viewDistance = 10000.0f; Material material; private Vector lights; private int colorSource = COLORS_FROM_IMAGE; private int bumpSource = BUMPS_FROM_IMAGE; private Function2D bumpFunction; private Image environmentMap; private int[] envPixels; private int envWidth = 1, envHeight = 1; // Temporary variables used to avoid per-pixel memory allocation while filtering private Vector3f l; private Vector3f v; private Vector3f n; private Color4f shadedColor; private Color4f diffuse_color; private Color4f specular_color; private Vector3f tmpv, tmpv2; public LightFilter() { lights = new Vector(); addLight(new DistantLight()); bumpHeight = 1.0f; bumpSoftness = 5.0f; bumpShape = 0; material = new Material(); l = new Vector3f(); v = new Vector3f(); n = new Vector3f(); shadedColor = new Color4f(); diffuse_color = new Color4f(); specular_color = new Color4f(); tmpv = new Vector3f(); tmpv2 = new Vector3f(); } public void setMaterial( Material material ) { this.material = material; } public Material getMaterial() { return material; } public void setBumpFunction(Function2D bumpFunction) { this.bumpFunction = bumpFunction; } public Function2D getBumpFunction() { return bumpFunction; } public void setBumpHeight(float bumpHeight) { this.bumpHeight = bumpHeight; } public float getBumpHeight() { return bumpHeight; } public void setBumpSoftness(float bumpSoftness) { this.bumpSoftness = bumpSoftness; } public float getBumpSoftness() { return bumpSoftness; } public void setBumpShape(int bumpShape) { this.bumpShape = bumpShape; } public int getBumpShape() { return bumpShape; } public void setViewDistance(float viewDistance) { this.viewDistance = viewDistance; } public float getViewDistance() { return viewDistance; } public void setEnvironmentMap(BufferedImage environmentMap) { this.environmentMap = environmentMap; if (environmentMap != null) { envWidth = environmentMap.getWidth(); envHeight = environmentMap.getHeight(); envPixels = getRGB( environmentMap, 0, 0, envWidth, envHeight, null ); } else { envWidth = envHeight = 1; envPixels = null; } } public Image getEnvironmentMap() { return environmentMap; } public void setColorSource(int colorSource) { this.colorSource = colorSource; } public int getColorSource() { return colorSource; } public void setBumpSource(int bumpSource) { this.bumpSource = bumpSource; } public int getBumpSource() { return bumpSource; } public void setDiffuseColor(int diffuseColor) { material.diffuseColor = diffuseColor; } public int getDiffuseColor() { return material.diffuseColor; } public void addLight(Light light) { lights.addElement(light); } public void removeLight(Light light) { lights.removeElement(light); } public Vector getLights() { return lights; } protected final static float r255 = 1.0f/255.0f; protected void setFromRGB( Color4f c, int argb ) { c.set( ((argb >> 16) & 0xff) * r255, ((argb >> 8) & 0xff) * r255, (argb & 0xff) * r255, ((argb >> 24) & 0xff) * r255 ); } protected int[] filterPixels( int width, int height, int[] inPixels, Rectangle transformedSpace ) { int index = 0; int[] outPixels = new int[width * height]; float width45 = Math.abs(6.0f * bumpHeight); boolean invertBumps = bumpHeight < 0; Vector3f position = new Vector3f(0.0f, 0.0f, 0.0f); Vector3f viewpoint = new Vector3f((float)width / 2.0f, (float)height / 2.0f, viewDistance); Vector3f normal = new Vector3f(); Color4f envColor = new Color4f(); Color4f diffuseColor = new Color4f( new Color(material.diffuseColor) ); Color4f specularColor = new Color4f( new Color(material.specularColor) ); Function2D bump = bumpFunction; // Apply the bump softness if (bumpSource == BUMPS_FROM_IMAGE || bumpSource == BUMPS_FROM_IMAGE_ALPHA || bumpSource == BUMPS_FROM_MAP || bump == null) { if ( bumpSoftness != 0 ) { int bumpWidth = width; int bumpHeight = height; int[] bumpPixels = inPixels; if ( bumpSource == BUMPS_FROM_MAP && bumpFunction instanceof ImageFunction2D ) { ImageFunction2D if2d = (ImageFunction2D)bumpFunction; bumpWidth = if2d.getWidth(); bumpHeight = if2d.getHeight(); bumpPixels = if2d.getPixels(); } int [] tmpPixels = new int[bumpWidth * bumpHeight]; int [] softPixels = new int[bumpWidth * bumpHeight]; /* for (int i = 0; i < 3; i++ ) { BoxBlurFilter.blur( bumpPixels, tmpPixels, bumpWidth, bumpHeight, (int)bumpSoftness ); BoxBlurFilter.blur( tmpPixels, softPixels, bumpHeight, bumpWidth, (int)bumpSoftness ); } */ Kernel kernel = GaussianFilter.makeKernel( bumpSoftness ); GaussianFilter.convolveAndTranspose( kernel, bumpPixels, tmpPixels, bumpWidth, bumpHeight, true, false, false, GaussianFilter.WRAP_EDGES ); GaussianFilter.convolveAndTranspose( kernel, tmpPixels, softPixels, bumpHeight, bumpWidth, true, false, false, GaussianFilter.WRAP_EDGES ); bump = new ImageFunction2D(softPixels, bumpWidth, bumpHeight, ImageFunction2D.CLAMP, bumpSource == BUMPS_FROM_IMAGE_ALPHA); final Function2D bbump = bump; if ( bumpShape != 0 ) { bump = new Function2D() { private Function2D original = bbump; public float evaluate(float x, float y) { float v = original.evaluate( x, y ); switch ( bumpShape ) { case 1: // v = v > 0.5f ? 0.5f : v; v *= ImageMath.smoothStep( 0.45f, 0.55f, v ); break; case 2: v = v < 0.5f ? 0.5f : v; break; case 3: v = ImageMath.triangle( v ); break; case 4: v = ImageMath.circleDown( v ); break; case 5: v = ImageMath.gain( v, 0.75f ); break; } return v; } }; } } else if ( bumpSource != BUMPS_FROM_MAP ) bump = new ImageFunction2D(inPixels, width, height, ImageFunction2D.CLAMP, bumpSource == BUMPS_FROM_IMAGE_ALPHA); } float reflectivity = material.reflectivity; float areflectivity = (1-reflectivity); Vector3f v1 = new Vector3f(); Vector3f v2 = new Vector3f(); Vector3f n = new Vector3f(); Light[] lightsArray = new Light[lights.size()]; lights.copyInto(lightsArray); for (int i = 0; i < lightsArray.length; i++) lightsArray[i].prepare(width, height); float[][] heightWindow = new float[3][width]; for (int x = 0; x < width; x++) heightWindow[1][x] = width45*bump.evaluate(x, 0);