Package thirdparty.jhlabs.math

Examples of thirdparty.jhlabs.math.FFT


        int h = rows;

        tileWidth = w;
        tileHeight = h;//FIXME-tileWidth, w, and cols are always all the same

        FFT fft = new FFT(Math.max(log2rows, log2cols));

        int[] rgb = new int[w * h];
        float[][] mask = new float[2][w * h];
        float[][] gb = new float[2][w * h];
        float[][] ar = new float[2][w * h];

        // Create the kernel
        double polyAngle = Math.PI / sides;
        double polyScale = 1.0f / Math.cos(polyAngle);
        double r2 = radius * radius;
        float angle = 0;
        double rangle = Math.toRadians(angle);
        float total = 0;
        int i = 0;
        for (int y = 0; y < h; y++) {
            for (int x = 0; x < w; x++) {
                double dx = x - w / 2f;
                double dy = y - h / 2f;
                double r = dx * dx + dy * dy;
                double f = r < r2 ? 1 : 0;
                if (f != 0) {
                    r = Math.sqrt(r);
                    if (sides != 0) {
                        double a = Math.atan2(dy, dx) + rangle;
                        a = ImageMath.mod(a, polyAngle * 2) - polyAngle;
                        f = Math.cos(a) * polyScale;
                    } else
                        f = 1;
                    f = f * r < radius ? 1 : 0;
                }
                total += (float) f;

                mask[0][i] = (float) f;
                mask[1][i] = 0;
                i++;
            }
        }

        // Normalize the kernel
        i = 0;
        for (int y = 0; y < h; y++) {
            for (int x = 0; x < w; x++) {
                mask[0][i] /= total;
                i++;
            }
        }

        fft.transform2D(mask[0], mask[1], w, h, true);

        for (int tileY = -iradius; tileY < height; tileY += tileHeight - 2 * iradius) {
            for (int tileX = -iradius; tileX < width; tileX += tileWidth - 2 * iradius) {
//                System.out.println("Tile: "+tileX+" "+tileY+" "+tileWidth+" "+tileHeight);

                // Clip the tile to the image bounds
                int tx = tileX, ty = tileY, tw = tileWidth, th = tileHeight;
                int fx = 0, fy = 0;
                if (tx < 0) {
                    tw += tx;
                    fx -= tx;
                    tx = 0;
                }
                if (ty < 0) {
                    th += ty;
                    fy -= ty;
                    ty = 0;
                }
                if (tx + tw > width)
                    tw = width - tx;
                if (ty + th > height)
                    th = height - ty;
                src.getRGB(tx, ty, tw, th, rgb, fy * w + fx, w);

                // Create a float array from the pixels. Any pixels off the edge of the source image get duplicated from the edge.
                i = 0;
                for (int y = 0; y < h; y++) {
                    int imageY = y + tileY;
                    int j;
                    if (imageY < 0)
                        j = fy;
                    else if (imageY > height)
                        j = fy + th - 1;
                    else
                        j = y;
                    j *= w;
                    for (int x = 0; x < w; x++) {
                        int imageX = x + tileX;
                        int k;
                        if (imageX < 0)
                            k = fx;
                        else if (imageX > width)
                            k = fx + tw - 1;
                        else
                            k = x;
                        k += j;

                        ar[0][i] = ((rgb[k] >> 24) & 0xff);
                        float r = ((rgb[k] >> 16) & 0xff);
                        float g = ((rgb[k] >> 8) & 0xff);
                        float b = (rgb[k] & 0xff);

                        // Bloom...
                        if (r > bloomThreshold)
                            r *= bloom;
//              r = bloomThreshold + (r-bloomThreshold) * bloom;
                        if (g > bloomThreshold)
                            g *= bloom;
//              g = bloomThreshold + (g-bloomThreshold) * bloom;
                        if (b > bloomThreshold)
                            b *= bloom;
//              b = bloomThreshold + (b-bloomThreshold) * bloom;

                        ar[1][i] = r;
                        gb[0][i] = g;
                        gb[1][i] = b;

                        i++;
                        k++;
                    }
                }

                // Transform into frequency space
                fft.transform2D(ar[0], ar[1], cols, rows, true);
                fft.transform2D(gb[0], gb[1], cols, rows, true);

                // Multiply the transformed pixels by the transformed kernel
                i = 0;
                for (int y = 0; y < h; y++) {
                    for (int x = 0; x < w; x++) {
                        float re = ar[0][i];
                        float im = ar[1][i];
                        float rem = mask[0][i];
                        float imm = mask[1][i];
                        ar[0][i] = re * rem - im * imm;
                        ar[1][i] = re * imm + im * rem;

                        re = gb[0][i];
                        im = gb[1][i];
                        gb[0][i] = re * rem - im * imm;
                        gb[1][i] = re * imm + im * rem;
                        i++;
                    }
                }

                // Transform back
                fft.transform2D(ar[0], ar[1], cols, rows, false);
                fft.transform2D(gb[0], gb[1], cols, rows, false);

                // Convert back to RGB pixels, with quadrant remapping
                int row_flip = w >> 1;
                int col_flip = h >> 1;
                int index = 0;
View Full Code Here

TOP

Related Classes of thirdparty.jhlabs.math.FFT

Copyright © 2018 www.massapicom. 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.