e2fsprogs/resize/sim_progress.c

107 lines
2.0 KiB
C

/*
* sim_progress.c --- simple progress meter
*/
#include "resize2fs.h"
struct ext2_sim_progress {
FILE *f;
char *label;
int labelwidth;
int barwidth;
__u32 maxdone;
__u32 current;
int shown;
int flags;
};
static errcode_t ext2fs_progress_display(ext2_sim_progmeter prog)
{
int i, width;
fputs(prog->label, prog->f);
width = prog->labelwidth - strlen(prog->label);
while (width-- > 0)
putc(' ', prog->f);
if (prog->labelwidth + prog->barwidth > 80) {
fputs("\n", prog->f);
for (width = prog->labelwidth; width > 0; width--)
putc(' ', prog->f);
}
for (i=0; i < prog->barwidth; i++)
putc('-', prog->f);
for (i=0; i < prog->barwidth; i++)
putc('\b', prog->f);
fflush(prog->f);
return 0;
}
void ext2fs_progress_update(ext2_sim_progmeter prog, __u32 current)
{
int old_level, level, num, i;
level = prog->barwidth * current / prog->maxdone;
old_level = prog->barwidth * prog->current / prog->maxdone;
prog->current = current;
num = level - old_level;
if (num == 0)
return;
if (num > 0) {
for (i=0; i < num; i++)
putc('X', prog->f);
} else {
num = -num;
for (i=0; i < num; i++)
putc('\b', prog->f);
for (i=0; i < num; i++)
putc('-', prog->f);
for (i=0; i < num; i++)
putc('\b', prog->f);
}
fflush(prog->f);
}
errcode_t ext2fs_progress_init(ext2_sim_progmeter *ret_prog,
const char *label,
int labelwidth, int barwidth,
__u32 maxdone, int flags)
{
ext2_sim_progmeter prog;
prog = malloc(sizeof(struct ext2_sim_progress));
if (!prog)
return ENOMEM;
memset(prog, 0, sizeof(struct ext2_sim_progress));
prog->label = malloc(strlen(label)+1);
if (!prog->label) {
free(prog);
return ENOMEM;
}
strcpy(prog->label, label);
prog->labelwidth = labelwidth;
prog->barwidth = barwidth;
prog->flags = flags;
prog->maxdone = maxdone;
prog->current = 0;
prog->shown = 0;
prog->f = stdout;
*ret_prog = prog;
return ext2fs_progress_display(prog);
}
void ext2fs_progress_close(ext2_sim_progmeter prog)
{
if (prog->label)
free(prog->label);
free(prog);
printf("\n");
return;
}