printf/format.c at master · nsterk/printf · GitHub

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
134
135
136
137
138
/* ************************************************************************** */
/*                                                                            */
/*                                                        ::::::::            */
/*   format.c                                           :+:    :+:            */
/*                                                     +:+                    */
/*   By: nsterk <nsterk@student.codam.nl>             +#+                     */
/*                                                   +#+                      */
/*   Created: 2020/11/19 10:23:15 by nsterk        #+#    #+#                 */
/*   Updated: 2021/01/07 19:01:24 by nsterk        ########   odam.nl         */
/*                                                                            */
/* ************************************************************************** */

#include "libftprintf.h"

static char		*make_precision_padding(t_tab *tab)
{
	int		len;
	char	*padding;

	len = tab->precision - (int)ft_strlen(tab->argument);
	if (tab->hash)
		len += 4;
	else if (tab->plus)
		len += 2;
	padding = ft_calloc(len + 1, sizeof(*padding));
	if (!padding)
		return (NULL);
	ft_memset(padding, '0', len);
	if (tab->plus)
		padding[0] = tab->argument[0];
	else if (tab->hash)
		padding[1] = tab->argument[1];
	return (padding);
}

static char		*make_width_padding(t_tab *tab)
{
	int		len;
	char	*padding;

	len = tab->width - (int)ft_strlen(tab->argument);
	if (tab->plus && tab->zero)
		len++;
	if (tab->hash && tab->zero)
		len += 2;
	padding = ft_calloc(len + 1, sizeof(*padding));
	if (!padding)
		return (NULL);
	if (tab->zero)
	{
		ft_memset(padding, '0', len);
		if (tab->plus)
			padding[0] = tab->argument[0];
		else if (tab->space)
			padding[0] = ' ';
		else if (tab->hash)
			padding[1] = tab->argument[1];
	}
	else
		ft_memset(padding, ' ', len);
	return (padding);
}

static int		format_precision(t_tab *tab)
{
	char	*padding;
	char	*temp;

	temp = ft_strdup(tab->argument);
	padding = make_precision_padding(tab);
	free(tab->argument);
	if (!temp || !padding)
		return (-1);
	if (tab->hash)
		tab->argument = ft_strjoin(padding, temp + 2);
	else if (tab->plus)
		tab->argument = ft_strjoin(padding, temp + 1);
	else
		tab->argument = ft_strjoin(padding, temp);
	tab->plus = 0;
	free(padding);
	free(temp);
	if (!tab->argument)
		return (-1);
	return (tab->ret);
}

static int		format_width(t_tab *tab)
{
	char	*temp;
	char	*padding;

	temp = ft_strdup(tab->argument);
	padding = make_width_padding(tab);
	free(tab->argument);
	if (!temp || !padding)
		return (-1);
	if (tab->left_justify)
		tab->argument = ft_strjoin(temp, padding);
	else
	{
		if (tab->plus && tab->zero)
			tab->argument = ft_strjoin(padding, temp + 1);
		else if (tab->hash && tab->zero)
			tab->argument = ft_strjoin(padding, temp + 2);
		else
			tab->argument = ft_strjoin(padding, temp);
	}
	free(padding);
	free(temp);
	if (!tab->argument)
		return (-1);
	return (tab->ret);
}

int				format(t_tab *tab)
{
	if (tab->precision_bool)
	{
		if (!tab->precision)
		{
			if (tab->plus && !ft_strcmp(tab->argument + 1, "0"))
				tab->argument[1] = '\0';
			else if (!ft_strcmp(tab->argument, "0"))
				tab->argument[0] = '\0';
		}
		else if (tab->precision >= (int)ft_strlen(tab->argument))
		{
			if (format_precision(tab) < 0)
				return (-1);
		}
	}
	if ((int)ft_strlen(tab->argument) < tab->width && tab->specifier != 'c')
		if (format_width(tab) < 0)
			return (-1);
	print_argument(tab);
	return (tab->ret);
}