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
|
#include <lib/png.h>
#include <stdio.h>
#include <png.h>
// Save bitmap to file
void save_png(struct bitmap_t* bitmap, char* fname)
{
FILE *fp;
png_structp png_ptr = NULL;
png_infop info_ptr = NULL;
uint8_t pixel_size = 3;
uint8_t pixel_depth = 8;
png_byte ** row_pointers = NULL;
size_t x, y;
fp = fopen(fname, "wb");
if (fp == NULL) {
fprintf(stderr, "Error opening file\n");
return;
}
png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
if (png_ptr == NULL) {
fprintf(stderr, "Error creating write structure\n");
fclose(fp);
return;
}
info_ptr = png_create_info_struct(png_ptr);
if (info_ptr == NULL) {
fprintf(stderr, "Error creating info structure\n");
fclose(fp);
return;
}
if (setjmp(png_jmpbuf(png_ptr))) {
fprintf(stderr, "Error setting jmp\n");
fclose(fp);
return;
}
png_set_IHDR(png_ptr, info_ptr,
bitmap->width, bitmap->height,
pixel_depth,
PNG_COLOR_TYPE_RGB, PNG_INTERLACE_NONE, PNG_COMPRESSION_TYPE_DEFAULT,
PNG_FILTER_TYPE_DEFAULT);
row_pointers = png_malloc(png_ptr, bitmap->height*sizeof(png_byte*));
for (y = 0; y < bitmap->height; y++) {
png_byte *row = png_malloc(png_ptr, sizeof(uint8_t)*bitmap->width*pixel_size);
row_pointers[y] = row;
for (x = 0; x < bitmap->width; x++) {
*row++ = bitmap->image_buffer[x + y*bitmap->width].red;
*row++ = bitmap->image_buffer[x + y*bitmap->width].green;
*row++ = bitmap->image_buffer[x + y*bitmap->width].blue;
}
}
png_init_io(png_ptr, fp);
png_set_rows(png_ptr, info_ptr, row_pointers);
png_write_png(png_ptr, info_ptr, PNG_TRANSFORM_IDENTITY, NULL);
for (y = 0; y < bitmap->height; y++) {
png_free(png_ptr, row_pointers[y]);
}
png_free(png_ptr, row_pointers);
png_destroy_write_struct(&png_ptr, &info_ptr);
fclose(fp);
}
|
