projects / thirdparty / e2fsprogs.git / blame
| Commit | Line | Data |
|---|---|---|
| c762c8e6 TT |
1 | /* |
| 2 | * sim_progress.c --- simple progress meter | |
| 0cee8a5c TT |
3 | * |
| 4 | * Copyright (C) 1997, 1998 by Theodore Ts'o and | |
| 5 | * PowerQuest, Inc. | |
| 6 | * | |
| 055866d8 | 7 | * Copyright (C) 1999, 2000 by Theodore Ts'o |
| efc6f628 | 8 | * |
| 0cee8a5c TT |
9 | * %Begin-Header% |
| 10 | * This file may be redistributed under the terms of the GNU Public | |
| 11 | * License. | |
| 12 | * %End-Header% | |
| c762c8e6 TT |
13 | */ |
| 14 | ||
| d1154eb4 | 15 | #include "config.h" |
| c762c8e6 TT |
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; | |
| efc6f628 | 62 | |
| c762c8e6 TT |
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 | { | |
| ca8abba7 TT |
83 | ext2_sim_progmeter prog; |
| 84 | errcode_t retval; | |
| c762c8e6 | 85 | |
| c4e3d3f3 | 86 | retval = ext2fs_get_mem(sizeof(struct ext2_sim_progress), &prog); |
| 520a1772 | 87 | if (retval) |
| ca8abba7 | 88 | return retval; |
| c762c8e6 TT |
89 | memset(prog, 0, sizeof(struct ext2_sim_progress)); |
| 90 | ||
| de8f3a76 | 91 | retval = ext2fs_get_mem(strlen(label)+1, &prog->label); |
| ca8abba7 | 92 | if (retval) { |
| c762c8e6 | 93 | free(prog); |
| ca8abba7 | 94 | return retval; |
| c762c8e6 TT |
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; | |
| efc6f628 | 104 | |
| c762c8e6 TT |
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) | |
| c4e3d3f3 TT |
114 | ext2fs_free_mem(&prog->label); |
| 115 | ext2fs_free_mem(&prog); | |
| c762c8e6 TT |
116 | printf("\n"); |
| 117 | return; | |
| 118 | } |