forked from magwyz/pastec
/
imageloader.cpp
74 lines (62 loc) · 2.01 KB
/
imageloader.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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
/*****************************************************************************
* Copyright (C) 2014 Visualink
*
* Authors: Adrien Maglo <adrien@visualink.io>
*
* This file is part of Pastec.
*
* Pastec is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* Pastec is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with Pastec. If not, see <http://www.gnu.org/licenses/>.
*****************************************************************************/
#include <iostream>
#include <vector>
#include <opencv2/highgui/highgui.hpp>
#include "imageloader.h"
#include "messages.h"
u_int32_t ImageLoader::loadImage(unsigned i_imgSize, char *p_imgData, Mat &img)
{
vector<char> imgData(i_imgSize);
memcpy(imgData.data(), p_imgData, i_imgSize);
try
{
img = imdecode(imgData, CV_LOAD_IMAGE_GRAYSCALE);
}
catch (cv::Exception& e) // The decoding of an image can raise an exception.
{
const char* err_msg = e.what();
cout << "Exception caught: " << err_msg << endl;
return IMAGE_NOT_DECODED;
}
if (!img.data)
{
cout << "Error reading the image." << std::endl;
return IMAGE_NOT_DECODED;
}
unsigned i_imgWidth = img.cols;
unsigned i_imgHeight = img.rows;
if (i_imgWidth > 1000
|| i_imgHeight > 1000)
{
cout << "Image too large." << endl;
return IMAGE_SIZE_TOO_BIG;
}
#if 1
if (i_imgWidth < 200
|| i_imgHeight < 200)
{
cout << "Image too small." << endl;
return IMAGE_SIZE_TOO_SMALL;
}
#endif
return OK;
}