summaryrefslogtreecommitdiff
path: root/2004/i/nono/src/image/png_writer.cc
blob: 347eca32f1df845d51091e932be99b9a5a0539af (plain)
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
// png_writer.cc
// nono - programme du robot 2004. {{{
//
// Copyright (C) 2004 Nicolas Schodet
//
// Robot APB Team/Efrei 2004.
//        Web: http://assos.efrei.fr/robot/
//      Email: robot AT efrei DOT fr
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or
// (at your option) any later version.
// 
// This program 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 General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
//
// }}}
#include "png_writer.h"
#include "utils/errno_exception.h"

#include <png.h>
#include <stdexcept>

/// Constructeur.
PngWriter::PngWriter (const std::string &filename)
    : filename_ (filename)
{
}

/// �crit une image, retourne le nombre d'octets �crits.
int
PngWriter::write (void *buf, int size, int width, int height) const
{
    int y;
    unsigned char *p;
    FILE *fd;
    png_infop info_ptr;
    png_structp png_ptr;
    int depth = 3;
    // Ouvre le fichier.
    fd = fopen(filename_.c_str (), "w+");
    if (!fd)
	throw std::runtime_error (filename_);
    // Cr�e le structure PNG.
    png_ptr = png_create_write_struct (PNG_LIBPNG_VER_STRING, 0, 0, 0);
    if (!png_ptr)
	return 0;
    info_ptr = png_create_info_struct (png_ptr);
    if (!info_ptr)
      {
	png_destroy_write_struct (&png_ptr, 0);
	return 0;
      }
    // Initialise.
    png_init_io (png_ptr, fd);
    png_set_IHDR (png_ptr, info_ptr, width, height, 8, PNG_COLOR_TYPE_RGB,
		  PNG_INTERLACE_NONE, PNG_COMPRESSION_TYPE_DEFAULT,
		  PNG_FILTER_TYPE_DEFAULT);
    png_set_bgr (png_ptr);
    png_write_info (png_ptr, info_ptr);
    // �crit l'image.
    p = reinterpret_cast<unsigned char *> (buf);
    for (y = 0; y < height; y++) {
	png_write_row (png_ptr, p);
	p += width * depth;
    }
    // Fin.
    png_write_end (png_ptr, info_ptr);
    png_destroy_write_struct (&png_ptr, &info_ptr);
    fclose (fd);
    return size;
}