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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
/* ************************************************************************** */
/* */
/* ::: :::::::: */
/* capture.c :+: :+: :+: */
/* +:+ +:+ +:+ */
/* By: cacharle <marvin@42.fr> +#+ +:+ +#+ */
/* +#+#+#+#+#+ +#+ */
/* Created: 2020/01/11 13:15:11 by cacharle #+# #+# */
/* Updated: 2020/02/26 18:13:10 by cacharle ### ########.fr */
/* */
/* ************************************************************************** */
#include "fractol.h"
#define IMG_DEPTH 3
#define FILE_HEADER_SIZE 14
#define INFO_HEADER_SIZE 40
/*
** bmp file format:
** header:
** file_header:
** 2: signature = "BM"
** 4: file size
** 4: reserved
** 4: offset to pixel array
** info_header:
** 4: header size
** 4: image width
** 4: image height
** 2: number of color planes
** 2: bits per pixel
** 4: compression
** 4: image size
** 4: horizontal resolution
** 4: vertical resolution
** 4: colors in color table
** 4: important color count
** data:
** pixel in rgb format (without alpha component)
** padding added at the end of each pixel row
** so the length of the row is a multiple of 4
*/
static void bmp_write_pixels(int fd, t_image *image, uint8_t *bmp_data)
{
int i;
int j;
uint8_t padding[3];
int padding_size;
ft_bzero(padding, 3);
padding_size = (4 - (image->width * IMG_DEPTH) % 4) % 4;
i = image->height;
while (--i >= 0)
{
j = -1;
while (++j < image->width)
{
bmp_data[3 * j + 0] = image->data[4 * (i * image->width + j) + 0];
bmp_data[3 * j + 1] = image->data[4 * (i * image->width + j) + 1];
bmp_data[3 * j + 2] = image->data[4 * (i * image->width + j) + 2];
}
write(fd, bmp_data, image->width * 3);
write(fd, padding, padding_size);
}
}
static void bmp_fill_header(t_image *image, uint8_t file_header[FILE_HEADER_SIZE],
uint8_t info_header[INFO_HEADER_SIZE])
{
int file_size;
file_size = FILE_HEADER_SIZE + INFO_HEADER_SIZE + (IMG_DEPTH * image->width
+ ((4 - (image->width * IMG_DEPTH) % 4) % 4)) * image->height;
ft_bzero(file_header, FILE_HEADER_SIZE);
ft_bzero(info_header, INFO_HEADER_SIZE);
file_header[0] = (uint8_t)('B');
file_header[1] = (uint8_t)('M');
file_header[2] = (uint8_t)(file_size);
file_header[3] = (uint8_t)(file_size >> 8);
file_header[4] = (uint8_t)(file_size >> 16);
file_header[5] = (uint8_t)(file_size >> 24);
file_header[10] = (uint8_t)(FILE_HEADER_SIZE + INFO_HEADER_SIZE);
info_header[0] = (uint8_t)(INFO_HEADER_SIZE);
info_header[4] = (uint8_t)(image->width);
info_header[5] = (uint8_t)(image->width >> 8);
info_header[6] = (uint8_t)(image->width >> 16);
info_header[7] = (uint8_t)(image->width >> 24);
info_header[8] = (uint8_t)(image->height);
info_header[9] = (uint8_t)(image->height >> 8);
info_header[10] = (uint8_t)(image->height >> 16);
info_header[11] = (uint8_t)(image->height >> 24);
info_header[12] = (uint8_t)(1);
info_header[14] = (uint8_t)(IMG_DEPTH * 8);
}
static bool bmp_write(t_image *image, uint8_t file_header[FILE_HEADER_SIZE],
uint8_t info_header[INFO_HEADER_SIZE], char *filename)
{
int fd;
uint8_t *bmp_data;
if ((fd = open(filename, O_WRONLY | O_CREAT, 0644)) < 0)
return (false);
if ((bmp_data = malloc(sizeof(unsigned char) *
(image->width * IMG_DEPTH))) == NULL)
{
close(fd);
return (false);
}
write(fd, file_header, FILE_HEADER_SIZE);
write(fd, info_header, INFO_HEADER_SIZE);
bmp_write_pixels(fd, image, bmp_data);
close(fd);
return (true);
}
int capture(t_state *state, char *filename)
{
uint8_t file_header[FILE_HEADER_SIZE];
uint8_t info_header[INFO_HEADER_SIZE];
bmp_fill_header(&state->window, file_header, info_header);
if (!bmp_write(&state->window, file_header, info_header, filename))
{
/* state_destroy(state); */
return (1);
}
/* state_destroy(state); */
return (0);
}
|
