| |
| package com.android.gallery3d.filtershow.filters; |
| |
| import java.util.Arrays; |
| |
| public class ColorSpaceMatrix { |
| private final float[] mMatrix = new float[16]; |
| private static final float RLUM = 0.3086f; |
| private static final float GLUM = 0.6094f; |
| private static final float BLUM = 0.0820f; |
| |
| public ColorSpaceMatrix() { |
| identity(); |
| } |
| |
| /** |
| * Copy constructor |
| * |
| * @param matrix |
| */ |
| public ColorSpaceMatrix(ColorSpaceMatrix matrix) { |
| System.arraycopy(matrix.mMatrix, 0, mMatrix, 0, matrix.mMatrix.length); |
| } |
| |
| /** |
| * get the matrix |
| * |
| * @return the internal matrix |
| */ |
| public float[] getMatrix() { |
| return mMatrix; |
| } |
| |
| /** |
| * set matrix to identity |
| */ |
| public void identity() { |
| Arrays.fill(mMatrix, 0); |
| mMatrix[0] = mMatrix[5] = mMatrix[10] = mMatrix[15] = 1; |
| } |
| |
| public void convertToLuminance() { |
| mMatrix[0] = mMatrix[1] = mMatrix[2] = 0.3086f; |
| mMatrix[4] = mMatrix[5] = mMatrix[6] = 0.6094f; |
| mMatrix[8] = mMatrix[9] = mMatrix[10] = 0.0820f; |
| } |
| |
| private void multiply(float[] a) |
| { |
| int x, y; |
| float[] temp = new float[16]; |
| |
| for (y = 0; y < 4; y++) { |
| int y4 = y * 4; |
| for (x = 0; x < 4; x++) { |
| temp[y4 + x] = mMatrix[y4 + 0] * a[x] |
| + mMatrix[y4 + 1] * a[4 + x] |
| + mMatrix[y4 + 2] * a[8 + x] |
| + mMatrix[y4 + 3] * a[12 + x]; |
| } |
| } |
| for (int i = 0; i < 16; i++) |
| mMatrix[i] = temp[i]; |
| } |
| |
| private void xRotateMatrix(float rs, float rc) |
| { |
| ColorSpaceMatrix c = new ColorSpaceMatrix(); |
| float[] tmp = c.mMatrix; |
| |
| tmp[5] = rc; |
| tmp[6] = rs; |
| tmp[9] = -rs; |
| tmp[10] = rc; |
| |
| multiply(tmp); |
| } |
| |
| private void yRotateMatrix(float rs, float rc) |
| { |
| ColorSpaceMatrix c = new ColorSpaceMatrix(); |
| float[] tmp = c.mMatrix; |
| |
| tmp[0] = rc; |
| tmp[2] = -rs; |
| tmp[8] = rs; |
| tmp[10] = rc; |
| |
| multiply(tmp); |
| } |
| |
| private void zRotateMatrix(float rs, float rc) |
| { |
| ColorSpaceMatrix c = new ColorSpaceMatrix(); |
| float[] tmp = c.mMatrix; |
| |
| tmp[0] = rc; |
| tmp[1] = rs; |
| tmp[4] = -rs; |
| tmp[5] = rc; |
| multiply(tmp); |
| } |
| |
| private void zShearMatrix(float dx, float dy) |
| { |
| ColorSpaceMatrix c = new ColorSpaceMatrix(); |
| float[] tmp = c.mMatrix; |
| |
| tmp[2] = dx; |
| tmp[6] = dy; |
| multiply(tmp); |
| } |
| |
| /** |
| * sets the transform to a shift in Hue |
| * |
| * @param rot rotation in degrees |
| */ |
| public void setHue(float rot) |
| { |
| float mag = (float) Math.sqrt(2.0); |
| float xrs = 1 / mag; |
| float xrc = 1 / mag; |
| xRotateMatrix(xrs, xrc); |
| mag = (float) Math.sqrt(3.0); |
| float yrs = -1 / mag; |
| float yrc = (float) Math.sqrt(2.0) / mag; |
| yRotateMatrix(yrs, yrc); |
| |
| float lx = getRedf(RLUM, GLUM, BLUM); |
| float ly = getGreenf(RLUM, GLUM, BLUM); |
| float lz = getBluef(RLUM, GLUM, BLUM); |
| float zsx = lx / lz; |
| float zsy = ly / lz; |
| zShearMatrix(zsx, zsy); |
| |
| float zrs = (float) Math.sin(rot * Math.PI / 180.0); |
| float zrc = (float) Math.cos(rot * Math.PI / 180.0); |
| zRotateMatrix(zrs, zrc); |
| zShearMatrix(-zsx, -zsy); |
| yRotateMatrix(-yrs, yrc); |
| xRotateMatrix(-xrs, xrc); |
| } |
| |
| /** |
| * set it to a saturation matrix |
| * |
| * @param s |
| */ |
| public void changeSaturation(float s) { |
| mMatrix[0] = (1 - s) * RLUM + s; |
| mMatrix[1] = (1 - s) * RLUM; |
| mMatrix[2] = (1 - s) * RLUM; |
| mMatrix[4] = (1 - s) * GLUM; |
| mMatrix[5] = (1 - s) * GLUM + s; |
| mMatrix[6] = (1 - s) * GLUM; |
| mMatrix[8] = (1 - s) * BLUM; |
| mMatrix[9] = (1 - s) * BLUM; |
| mMatrix[10] = (1 - s) * BLUM + s; |
| } |
| |
| /** |
| * Transform RGB value |
| * |
| * @param r red pixel value |
| * @param g green pixel value |
| * @param b blue pixel value |
| * @return computed red pixel value |
| */ |
| public float getRed(int r, int g, int b) { |
| return r * mMatrix[0] + g * mMatrix[4] + b * mMatrix[8] + mMatrix[12]; |
| } |
| |
| /** |
| * Transform RGB value |
| * |
| * @param r red pixel value |
| * @param g green pixel value |
| * @param b blue pixel value |
| * @return computed green pixel value |
| */ |
| public float getGreen(int r, int g, int b) { |
| return r * mMatrix[1] + g * mMatrix[5] + b * mMatrix[9] + mMatrix[13]; |
| } |
| |
| /** |
| * Transform RGB value |
| * |
| * @param r red pixel value |
| * @param g green pixel value |
| * @param b blue pixel value |
| * @return computed blue pixel value |
| */ |
| public float getBlue(int r, int g, int b) { |
| return r * mMatrix[2] + g * mMatrix[6] + b * mMatrix[10] + mMatrix[14]; |
| } |
| |
| private float getRedf(float r, float g, float b) { |
| return r * mMatrix[0] + g * mMatrix[4] + b * mMatrix[8] + mMatrix[12]; |
| } |
| |
| private float getGreenf(float r, float g, float b) { |
| return r * mMatrix[1] + g * mMatrix[5] + b * mMatrix[9] + mMatrix[13]; |
| } |
| |
| private float getBluef(float r, float g, float b) { |
| return r * mMatrix[2] + g * mMatrix[6] + b * mMatrix[10] + mMatrix[14]; |
| } |
| |
| } |