automatic commit
[folded-ctf.git] / rgb_image_subpixel.cc
1 /*
2  *  folded-ctf is an implementation of the folded hierarchy of
3  *  classifiers for object detection, developed by Francois Fleuret
4  *  and Donald Geman.
5  *
6  *  Copyright (c) 2008 Idiap Research Institute, http://www.idiap.ch/
7  *  Written by Francois Fleuret <francois.fleuret@idiap.ch>
8  *
9  *  This file is part of folded-ctf.
10  *
11  *  folded-ctf is free software: you can redistribute it and/or modify
12  *  it under the terms of the GNU General Public License as published
13  *  by the Free Software Foundation, either version 3 of the License,
14  *  or (at your option) any later version.
15  *
16  *  folded-ctf is distributed in the hope that it will be useful, but
17  *  WITHOUT ANY WARRANTY; without even the implied warranty of
18  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
19  *  General Public License for more details.
20  *
21  *  You should have received a copy of the GNU General Public License
22  *  along with folded-ctf.  If not, see <http://www.gnu.org/licenses/>.
23  *
24  */
25
26 #include "rgb_image_subpixel.h"
27
28 RGBImageSubpixel::RGBImageSubpixel(int width, int height) : RGBImage(width * scale, height* scale) { }
29
30 RGBImageSubpixel::RGBImageSubpixel(RGBImage *image) : RGBImage(image->width() * scale, image->height() * scale) {
31   for(int y = 0; y < _height; y++) {
32     for(int x = 0; x < _width; x++) {
33       set_pixel(x, y,
34                 image->pixel(x / scale, y / scale, RGBImage::RED),
35                 image->pixel(x / scale, y / scale, RGBImage::GREEN),
36                 image->pixel(x / scale, y / scale, RGBImage::BLUE));
37     }
38   }
39 }
40
41 RGBImageSubpixel::~RGBImageSubpixel() { }
42
43 void RGBImageSubpixel::read_png(const char *filename) {
44   abort();
45 }
46
47 void RGBImageSubpixel::write_png(const char *filename) {
48   RGBImage tmp(_width / scale, _height / scale);
49   for(int y = 0; y < _height / scale; y++) {
50     for(int x = 0; x < _width / scale; x++) {
51       int sr = 0, sg = 0, sb = 0;
52       for(int yy = y * scale; yy < (y + 1) * scale; yy++) {
53         for(int xx = x * scale; xx < (x + 1) * scale; xx++) {
54           sr += int(_bit_plans[RED][yy][xx]);
55           sg += int(_bit_plans[GREEN][yy][xx]);
56           sb += int(_bit_plans[BLUE][yy][xx]);
57         }
58       }
59
60       tmp.set_pixel(x, y,
61                     sr / (scale * scale), sg / (scale * scale), sb / (scale * scale));
62     }
63   }
64   tmp.write_png(filename);
65 }
66
67
68 void RGBImageSubpixel::read_jpg(const char *filename) {
69   abort();
70 }
71
72 void RGBImageSubpixel::write_jpg(const char *filename, int quality) {
73   RGBImage tmp(_width / scale, _height / scale);
74
75   for(int y = 0; y < _height / scale; y++) {
76     for(int x = 0; x < _width / scale; x++) {
77       int sr = 0, sg = 0, sb = 0;
78       for(int yy = y * scale; yy < (y + 1) * scale; yy++) {
79         for(int xx = x * scale; xx < (x + 1) * scale; xx++) {
80           sr += int(_bit_plans[RED][yy][xx]);
81           sg += int(_bit_plans[GREEN][yy][xx]);
82           sb += int(_bit_plans[BLUE][yy][xx]);
83         }
84       }
85
86       tmp.set_pixel(x, y,
87                     sr / (scale * scale), sg / (scale * scale), sb / (scale * scale));
88     }
89   }
90
91   tmp.write_jpg(filename, quality);
92 }
93
94
95 void RGBImageSubpixel::draw_line(int thickness,
96                                  unsigned char r, unsigned char g, unsigned char b,
97                                  scalar_t x0, scalar_t y0, scalar_t x1, scalar_t y1) {
98   RGBImage::draw_line(int(thickness * scale),
99                       r, g, b,
100                       x0 * scale + scale/2, y0 * scale + scale/2,
101                       x1 * scale + scale/2, y1 * scale + scale/2);
102 }