1/*
2 * sim_progress.c --- simple progress meter
3 *
4 * Copyright (C) 1997, 1998 by Theodore Ts'o and
5 * 	PowerQuest, Inc.
6 *
7 * Copyright (C) 1999, 2000 by Theosore Ts'o
8 *
9 * %Begin-Header%
10 * This file may be redistributed under the terms of the GNU Public
11 * License.
12 * %End-Header%
13 */
14
15#include "config.h"
16#include "resize2fs.h"
17
18struct ext2_sim_progress {
19	FILE	*f;
20	char	*label;
21	int	labelwidth;
22	int	barwidth;
23	__u32	maxdone;
24	__u32	current;
25	int	shown;
26	int	flags;
27};
28
29static errcode_t ext2fs_progress_display(ext2_sim_progmeter prog)
30{
31	int	i, width;
32
33	fputs(prog->label, prog->f);
34	width = prog->labelwidth - strlen(prog->label);
35	while (width-- > 0)
36		putc(' ', prog->f);
37	if (prog->labelwidth + prog->barwidth > 80) {
38		fputs("\n", prog->f);
39		for (width = prog->labelwidth; width > 0; width--)
40			putc(' ', prog->f);
41	}
42	for (i=0; i < prog->barwidth; i++)
43		putc('-', prog->f);
44	for (i=0; i < prog->barwidth; i++)
45		putc('\b', prog->f);
46	fflush(prog->f);
47	return 0;
48}
49
50
51void ext2fs_progress_update(ext2_sim_progmeter prog, __u32 current)
52{
53	int		old_level, level, num, i;
54
55	level = prog->barwidth * current / prog->maxdone;
56	old_level = prog->barwidth * prog->current / prog->maxdone;
57	prog->current = current;
58
59	num = level - old_level;
60	if (num == 0)
61		return;
62
63	if (num > 0) {
64		for (i=0; i < num; i++)
65			putc('X', prog->f);
66	} else {
67		num = -num;
68		for (i=0; i < num; i++)
69			putc('\b', prog->f);
70		for (i=0; i < num; i++)
71			putc('-', prog->f);
72		for (i=0; i < num; i++)
73			putc('\b', prog->f);
74	}
75	fflush(prog->f);
76}
77
78errcode_t ext2fs_progress_init(ext2_sim_progmeter *ret_prog,
79			       const char *label,
80			       int labelwidth, int barwidth,
81			       __u32 maxdone, int flags)
82{
83	ext2_sim_progmeter	prog;
84	errcode_t		retval;
85
86	retval = ext2fs_get_mem(sizeof(struct ext2_sim_progress), &prog);
87	if (retval)
88		return retval;
89	memset(prog, 0, sizeof(struct ext2_sim_progress));
90
91	retval = ext2fs_get_mem(strlen(label)+1, &prog->label);
92	if (retval) {
93		free(prog);
94		return retval;
95	}
96	strcpy(prog->label, label);
97	prog->labelwidth = labelwidth;
98	prog->barwidth = barwidth;
99	prog->flags = flags;
100	prog->maxdone = maxdone;
101	prog->current = 0;
102	prog->shown = 0;
103	prog->f = stdout;
104
105	*ret_prog = prog;
106
107	return ext2fs_progress_display(prog);
108}
109
110void ext2fs_progress_close(ext2_sim_progmeter prog)
111{
112
113	if (prog->label)
114		ext2fs_free_mem(&prog->label);
115	ext2fs_free_mem(&prog);
116	printf("\n");
117	return;
118}
119