fractol.c

/* ************************************************************************** */
/*                                                                            */
/*                                                        :::      ::::::::   */
/*   fractol.c                                          :+:      :+:    :+:   */
/*                                                    +:+ +:+         +:+     */
/*   By: bbellatr <marvin@42.fr>                    +#+  +:+       +#+        */
/*                                                +#+#+#+#+#+   +#+           */
/*   Created: 2022/01/24 16:56:19 by bbellatr          #+#    #+#             */
/*   Updated: 2022/01/24 16:56:22 by bbellatr         ###   ########.fr       */
/*                                                                            */
/* ************************************************************************** */

#include "fractol.h"

void	res_params(t_params *params)
{
	params->image.length = WIN_HEIGHT;
	params->image.width = WIN_WIDTH;
	params->max_iteration = 100;
	if (params->argc == 4)
	{
		params->julia_k.re = ft_double(params->argv[2]);
		params->julia_k.im = ft_double(params->argv[3]);
	}
	else
	{
		params->julia_k.re = -0.8;
		params->julia_k.im = 0.0;
	}
	params->min.im = -2;
	params->min.re = -2;
	params->max.im = 2;
	params->max.re = 2;
	params->color_shift = 0;
}

static int	(*get_formula(char *name)) (int x, int y, t_params *params)
{
	size_t				i;
	int					(*formula)(int x, int y, t_params *vars);
	static t_formula	formulas[3];

	formulas[0].name = "mandelbrot";
	formulas[0].formula = &mandelbrot;
	formulas[1].name = "julia";
	formulas[1].formula = &julia;
	formulas[2].name = "burning_ship";
	formulas[2].formula = &burning_ship;
	i = 0;
	formula = NULL;
	while (i < (sizeof(formulas) / sizeof(t_formula)))
	{
		if (!ft_strncmp(name, formulas[i].name, ft_strlen(formulas[i].name)))
			formula = formulas[i].formula;
		i++;
	}
	return (formula);
}

void	init_image(t_params *params)
{
	params->mlx_data.img = mlx_new_image(params->mlx_data.mlx,
			params->image.length, params->image.width);
	params->mlx_data.addr = mlx_get_data_addr(params->mlx_data.img,
			&params->mlx_data.bits_per_pixel,
			&params->mlx_data.line_length,
			&params->mlx_data.endian);
	params->mlx_data.win = mlx_new_window(params->mlx_data.mlx,
			params->image.length,
			params->image.width,
			params->argv[1]);
	mlx_loop_hook(params->mlx_data.mlx, &draw_fractal, params);
	mlx_hook(params->mlx_data.win, BUTTON_PRESS,
		1L << 2, scroll_handler, params);
	mlx_hook(params->mlx_data.win, KEY_PRESS,
		1L << 0, press_key, params);
	mlx_hook(params->mlx_data.win, DESTROY_NOTIFY,
		1L << 17, end_p, params);
	mlx_hook(params->mlx_data.win, MOTION_NOTIFY,
		1L << 6, pointer_handler, params);
	mlx_loop(params->mlx_data.mlx);
}

int	init_params(t_params *params, char **argv)
{
	params->argv = argv;
	params->mlx_data.win = NULL;
	params->mlx_data.img = NULL;
	params->mlx_data.addr = NULL;
	params->formula = get_formula(argv[1]);
	if (params->formula == NULL)
		return (0);
	if (params->argc == 4)
	{
		params->julia_k.re = ft_double(params->argv[2]);
		params->julia_k.im = ft_double(params->argv[3]);
		if (params->julia_k.im > 1.0 || params->julia_k.im < -1.0
			|| params->julia_k.re > 1.0 || params->julia_k.re < -1.0)
			return (0);
	}
	res_params(params);
	params->mlx_data.mlx = mlx_init();
	return (1);
}

int	main(int argc, char **argv)
{
	t_params	params;

	params.argc = argc;
	if (argc != 2 && argc != 4)
	{
		u_tip();
		exit(EXIT_FAILURE);
	}
	if (argc == 4)
	{
		if (!check(argv[2]) || !check(argv[3]))
		{
			u_tip();
			exit(EXIT_FAILURE);
		}
	}
	if (!init_params(&params, argv))
	{
		u_tip();
		end_p(&params);
	}
	init_image(&params);
}