summaryrefslogtreecommitdiffstats
path: root/Algorithms/Part-II/2-SeamCarving/SCUtility.java
blob: bcd95f3a42f93adf9c90bf989c0552576fc1bf92 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
import java.awt.Color;

public class SCUtility {


    // create random W-by-H array of tiles
    public static Picture randomPicture(int W, int H) {
        Picture p = new Picture(W, H);
        for (int i = 0; i < W; i++)
            for (int j = 0; j < H; j++) {
                int r = StdRandom.uniform(255);
                int g = StdRandom.uniform(255);
                int b = StdRandom.uniform(255);
                Color c = new Color(r, g, b);
                p.set(i, j, c);
            }
        return p;
    }


    public static double[][] toEnergyMatrix(SeamCarver sc)
    {
        double[][] returnDouble = new double[sc.width()][sc.height()];
        for (int i = 0; i < sc.width(); i++)
            for (int j = 0; j < sc.height(); j++)
                returnDouble[i][j] = sc.energy(i, j);

        return returnDouble;
    }

    // displays grayvalues as energy (converts to picture, calls show)
    public static void showEnergy(SeamCarver sc)
    {
        doubleToPicture(toEnergyMatrix(sc)).show();
    }

    public static Picture toEnergyPicture(SeamCarver sc)
    {
        double[][] energyMatrix = toEnergyMatrix(sc);
        return doubleToPicture(energyMatrix);
    }

    // converts a double matrix of values into a normalized picture
    // values are normalized by the maximum grayscale value
    public static Picture doubleToPicture(double[][] grayValues)
    {

        //each 1D array in the matrix represents a single column, so number
        //of 1D arrays is the width, and length of each array is the height
        int width = grayValues.length;
        int height = grayValues[0].length;

        Picture p = new Picture(width, height);

        double maxVal = 0;
        for (int i = 0; i < width; i++)
            for (int j = 0; j < height; j++)
                if (grayValues[i][j] > maxVal)
                    maxVal = grayValues[i][j];

        if (maxVal == 0)
            return p; //return black picture

        for (int i = 0; i < width; i++)
            for (int j = 0; j < height; j++)
            {
                float normalizedGrayValue =
                    (float) grayValues[i][j] / (float) maxVal;
                p.set(i, j, new Color(normalizedGrayValue,
                            normalizedGrayValue, normalizedGrayValue));
            }

        return p;
    }


    // This method is useful for debugging seams. It overlays red
    // pixels over the calculate seam. Due to the lack of a copy
    // constructor, it also alters the original picture.

    public static Picture seamOverlay(Picture p, boolean horizontal,
            int[] seamIndices)
    {
        Picture overlaid = new Picture(p.width(), p.height());

        for (int i = 0; i < p.width(); i++)
            for (int j = 0; j < p.height(); j++)
                overlaid.set(i, j, p.get(i, j));

        int width = p.width();
        int height = p.height();

        //if horizontal seam, then set one pixel in every column
        if (horizontal)
            for (int i = 0; i < width; i++)
                overlaid.set(i, seamIndices[i], new Color(255, 0, 0));
        else //if vertical, put one pixel in every row
            for (int j = 0; j < height; j++)
                overlaid.set(seamIndices[j], j, new Color(255, 0, 0));

        return overlaid;
    }


}