-
Notifications
You must be signed in to change notification settings - Fork 2
/
Picture.cpp
50 lines (44 loc) · 1.08 KB
/
Picture.cpp
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
//
// Created by lazycal on 2017/6/15.
//
#include "Picture.h"
Picture::Picture() {data = NULL;}
Picture::Picture(const Mat &mat)
{
operator=(mat);
}
Picture::~Picture()
{
if (data != NULL)
delete[] data;
}
Picture &Picture::operator=(const Mat &mat)
{
if (data != NULL)
delete[] data;
data = new Vec3b[mat.rows * mat.cols];
rows = mat.rows; cols = mat.cols;
for (int i = 0; i < rows; ++i) {
for (int j = 0; j < cols; ++j) {
Vec3b tmp = mat.at<Vec3b>(i, j);
std::swap(tmp[0], tmp[2]);
data[i * cols + j] = tmp;
}
}
return *this;
}
Color Picture::getColor(double x, double y) const
{
x = min(max(x, 0.), 1.);
y = min(max(y, 0.), 1.);
y = (1 - y) * (rows - 1);
x = x * (cols - 1);
Color res;
for (int i = 0; i < 2; ++i)
for (int j = 0; j < 2; ++j) {
int ny = std::min(int(y) + i, this->rows);
int nx = std::min(int(x) + j, this->cols);
res += (1 - fabs(y - ny)) * (1 - fabs(x - nx)) * (*this)[ny][nx];
}
return res / 255;
}