aboutsummaryrefslogtreecommitdiff
path: root/src/capture.c
blob: 89455936defc1f2c24b52a94d5f3838f9e58437b (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
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
/* ************************************************************************** */
/*                                                                            */
/*                                                        :::      ::::::::   */
/*   capture.c                                          :+:      :+:    :+:   */
/*                                                    +:+ +:+         +:+     */
/*   By: cacharle <marvin@42.fr>                    +#+  +:+       +#+        */
/*                                                +#+#+#+#+#+   +#+           */
/*   Created: 2020/01/11 13:15:11 by cacharle          #+#    #+#             */
/*   Updated: 2020/02/27 14:46:01 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)].rgb.r;
			bmp_data[3 * j + 1] = image->data[4 * (i * image->width + j)].rgb.g;
			bmp_data[3 * j + 2] = image->data[4 * (i * image->width + j)].rgb.b;
		}
		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);
	return (bmp_write(&state->window, file_header, info_header, filename));
}