Visionary/util.cpp at master · vitalight/Visionary · 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
111
112
113
114
115
116
117
#include "util.h"
using namespace std;

static const double PI = 4.0*atan(1.0);

int U_getKernelSum(U_Kernel_i kernel)
{
    int sum = 0;
    for (unsigned int i=0; i<kernel.size(); i++)
    {
        for (unsigned int j = 0; j < kernel.size(); j++)
        {
            sum += kernel[i][j];
        }
    }
    return sum;
}

double U_getKernelSum(U_Kernel_d kernel)
{
    double sum = 0;
    for (unsigned int i=0; i<kernel.size(); i++)
    {
        for (unsigned int j = 0; j < kernel.size(); j++)
        {
            sum += kernel[i][j];
        }
    }
    return sum;
}

vector<vector<double>> U_getGaussianKernel(int size, double sigma)
{
    vector<vector<double>> kernel;
    int center = size/2;
    double sum = 0, val;
    for (int i=0; i<size; i++)
    {
        vector<double> line;
        for (int j=0; j<size; j++)
        {
            val = (1/(2*PI*sigma*sigma))*exp(-((i-center)*(i-center)+(j-center)*(j-center))/(2*sigma*sigma));
            line.push_back(val);
            sum += val;
        }
        kernel.push_back(line);
    }

    for (int i=0; i<size; i++)
    {
        for (int j=0; j<size; j++)
        {
            kernel[i][j]/=sum;
        }
    }
    return kernel;
}

vector<vector<double>> U_getFlatKernel_d(int size, double val)
{
    vector<vector<double>> kernel;

    for (int i=0; i<size; i++)
    {
        vector<double> line(size, val);
        kernel.push_back(line);
    }

    return kernel;
}

vector<vector<int>> U_getFlatKernel_i(int size, int val)
{
    vector<vector<int>> kernel;

    for (int i=0; i<size; i++)
    {
        vector<int> line(size, val);
        kernel.push_back(line);
    }

    return kernel;
}

int U_round(double f)
{
    return (int)(f+0.5);
}

bool U_in(int low, int value, int high)
{
    return value>=low && value<high;
}

// change from int to double
bool U_legal(int width, int height, double x, double y)
{
    return x>=0 && x<width && y>=0 && y<height;
}

void U_colorBound(int &r, int &g, int &b)
{
    if (r>255)
        r = 255;
    else if (r<0)
        r = 0;

    if (g>255)
        g = 255;
    else if (g<0)
        g = 0;

    if (b>255)
        b = 255;
    else if (b<0)
        b = 0;
}