tzh
2024-08-22 c7d0944258c7d0943aa7b2211498fd612971ce27
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
/*
 * sim_progress.c --- simple progress meter
 *
 * Copyright (C) 1997, 1998 by Theodore Ts'o and
 *     PowerQuest, Inc.
 *
 * Copyright (C) 1999, 2000 by Theodore Ts'o
 *
 * %Begin-Header%
 * This file may be redistributed under the terms of the GNU Public
 * License.
 * %End-Header%
 */
 
#include "config.h"
#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;
   errcode_t        retval;
 
   retval = ext2fs_get_mem(sizeof(struct ext2_sim_progress), &prog);
   if (retval)
       return retval;
   memset(prog, 0, sizeof(struct ext2_sim_progress));
 
   retval = ext2fs_get_mem(strlen(label)+1, &prog->label);
   if (retval) {
       free(prog);
       return retval;
   }
   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)
       ext2fs_free_mem(&prog->label);
   ext2fs_free_mem(&prog);
   printf("\n");
   return;
}