vector-quantizer/ImageVector.java at master · MostafaOmar98/vector-quantizer · GitHub

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
106
107
108
109
110
import java.util.ArrayList;

public class ImageVector {
    private int n, m;
    private int[][] data;
    ImageVector()
    {
        n = m = 0;
        data = null;
    }
    ImageVector(int n, int m)
    {
        this.n = n;
        this.m = m;
        data = new int[n][m];
    }

    ImageVector(int[][] other)
    {
        if (other != null) {
            n = other.length;
            m = other[0].length;
            data = other;
        }
        else
        {
            n = m = 0;
            data = null;
        }
    }

    public int get(int i, int j)
    {
        return data[i][j];
    }

    public void set(int i, int j, int v)
    {
        data[i][j] = v;
    }

    public static ImageVector getAverage(ArrayList<ImageVector> vectors)
    {
        int n = vectors.get(0).n, m = vectors.get(0).m; // handle error
        int[][] data = new int[n][m];
        for (ImageVector v : vectors)
        {
            for (int i = 0; i < v.n; ++i)
            {
                for (int j = 0; j < v.m; ++j)
                {
                    data[i][j] += v.get(i, j);
                }
            }
        }
        for (int i = 0; i < n; ++i)
        {
            for (int j = 0; j < m; ++j) {
                data[i][j] = Math.round((float)data[i][j]/vectors.size());
            }
        }

        ImageVector ret = new ImageVector(data);
        return ret;
    }

    int getNumberOfRows()
    {
        return n;
    }
    int getNumberOfColumns()
    {
        return m;
    }

    double getDistance(ImageVector other)
    {
        /*
        square(value(i,j) - otherValue(i,j)) for all i, j
         */
        double ret = 0;
        for (int i = 0; i < n; ++i)
        {
            for (int j = 0; j < m; ++j)
            {
                int d = this.get(i, j) - other.get(i, j);
                ret += d * d;
            }
        }
        return Math.sqrt(ret);
    }

    public String toString()
    {
        String ret = new String();
        for (int i = 0; i < n; ++i)
        {
            for (int j = 0; j < m; ++j)
            {
                ret += "" + data[i][j];
                if (j != m - 1)
                    ret += " ";
            }
            if (i != n - 1)
                ret += "\n";
        }
        return ret;
    }

}