resize/sim_progress.c

   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 Theodore 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
  18 struct 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
  29 static 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
  51 void 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
  78 errcode_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
 110 void 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 }