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 "resize2fs.h"
16 
17 struct ext2_sim_progress {
18 	FILE	*f;
19 	char	*label;
20 	int	labelwidth;
21 	int	barwidth;
22 	__u32	maxdone;
23 	__u32	current;
24 	int	shown;
25 	int	flags;
26 };
27 
ext2fs_progress_display(ext2_sim_progmeter prog)28 static errcode_t ext2fs_progress_display(ext2_sim_progmeter prog)
29 {
30 	int	i, width;
31 
32 	fputs(prog->label, prog->f);
33 	width = prog->labelwidth - strlen(prog->label);
34 	while (width-- > 0)
35 		putc(' ', prog->f);
36 	if (prog->labelwidth + prog->barwidth > 80) {
37 		fputs("\n", prog->f);
38 		for (width = prog->labelwidth; width > 0; width--)
39 			putc(' ', prog->f);
40 	}
41 	for (i=0; i < prog->barwidth; i++)
42 		putc('-', prog->f);
43 	for (i=0; i < prog->barwidth; i++)
44 		putc('\b', prog->f);
45 	fflush(prog->f);
46 	return 0;
47 }
48 
49 
ext2fs_progress_update(ext2_sim_progmeter prog,__u32 current)50 void ext2fs_progress_update(ext2_sim_progmeter prog, __u32 current)
51 {
52 	int		old_level, level, num, i;
53 
54 	level = prog->barwidth * current / prog->maxdone;
55 	old_level = prog->barwidth * prog->current / prog->maxdone;
56 	prog->current = current;
57 
58 	num = level - old_level;
59 	if (num == 0)
60 		return;
61 
62 	if (num > 0) {
63 		for (i=0; i < num; i++)
64 			putc('X', prog->f);
65 	} else {
66 		num = -num;
67 		for (i=0; i < num; i++)
68 			putc('\b', prog->f);
69 		for (i=0; i < num; i++)
70 			putc('-', prog->f);
71 		for (i=0; i < num; i++)
72 			putc('\b', prog->f);
73 	}
74 	fflush(prog->f);
75 }
76 
ext2fs_progress_init(ext2_sim_progmeter * ret_prog,const char * label,int labelwidth,int barwidth,__u32 maxdone,int flags)77 errcode_t ext2fs_progress_init(ext2_sim_progmeter *ret_prog,
78 			       const char *label,
79 			       int labelwidth, int barwidth,
80 			       __u32 maxdone, int flags)
81 {
82 	ext2_sim_progmeter	prog;
83 	errcode_t		retval;
84 
85 	retval = ext2fs_get_mem(sizeof(struct ext2_sim_progress), &prog);
86 	if (retval)
87 		return retval;
88 	memset(prog, 0, sizeof(struct ext2_sim_progress));
89 
90 	retval = ext2fs_get_mem(strlen(label)+1, &prog->label);
91 	if (retval) {
92 		free(prog);
93 		return retval;
94 	}
95 	strcpy(prog->label, label);
96 	prog->labelwidth = labelwidth;
97 	prog->barwidth = barwidth;
98 	prog->flags = flags;
99 	prog->maxdone = maxdone;
100 	prog->current = 0;
101 	prog->shown = 0;
102 	prog->f = stdout;
103 
104 	*ret_prog = prog;
105 
106 	return ext2fs_progress_display(prog);
107 }
108 
ext2fs_progress_close(ext2_sim_progmeter prog)109 void ext2fs_progress_close(ext2_sim_progmeter prog)
110 {
111 
112 	if (prog->label)
113 		ext2fs_free_mem(&prog->label);
114 	ext2fs_free_mem(&prog);
115 	printf("\n");
116 	return;
117 }
118