Merge branch 'rs/archive'
[git/builtin-gsoc.git] / progress.c
blob4344f4eed5e46c4e013018af8ef9ab062f740d8f
1 #include "git-compat-util.h"
2 #include "progress.h"
4 static volatile sig_atomic_t progress_update;
6 static void progress_interval(int signum)
8 progress_update = 1;
11 static void set_progress_signal(void)
13 struct sigaction sa;
14 struct itimerval v;
16 progress_update = 0;
18 memset(&sa, 0, sizeof(sa));
19 sa.sa_handler = progress_interval;
20 sigemptyset(&sa.sa_mask);
21 sa.sa_flags = SA_RESTART;
22 sigaction(SIGALRM, &sa, NULL);
24 v.it_interval.tv_sec = 1;
25 v.it_interval.tv_usec = 0;
26 v.it_value = v.it_interval;
27 setitimer(ITIMER_REAL, &v, NULL);
30 static void clear_progress_signal(void)
32 struct itimerval v = {{0,},};
33 setitimer(ITIMER_REAL, &v, NULL);
34 signal(SIGALRM, SIG_IGN);
35 progress_update = 0;
38 int display_progress(struct progress *progress, unsigned n)
40 if (progress->delay) {
41 char buf[80];
42 if (!progress_update || --progress->delay)
43 return 0;
44 if (progress->total) {
45 unsigned percent = n * 100 / progress->total;
46 if (percent > progress->delayed_percent_treshold) {
47 /* inhibit this progress report entirely */
48 clear_progress_signal();
49 progress->delay = -1;
50 progress->total = 0;
51 return 0;
54 if (snprintf(buf, sizeof(buf),
55 progress->delayed_title, progress->total))
56 fprintf(stderr, "%s\n", buf);
58 if (progress->total) {
59 unsigned percent = n * 100 / progress->total;
60 if (percent != progress->last_percent || progress_update) {
61 progress->last_percent = percent;
62 fprintf(stderr, "%s%4u%% (%u/%u) done\r",
63 progress->prefix, percent, n, progress->total);
64 progress_update = 0;
65 progress->need_lf = 1;
66 return 1;
68 } else if (progress_update) {
69 fprintf(stderr, "%s%u\r", progress->prefix, n);
70 progress_update = 0;
71 progress->need_lf = 1;
72 return 1;
74 return 0;
77 void start_progress(struct progress *progress, const char *title,
78 const char *prefix, unsigned total)
80 char buf[80];
81 progress->prefix = prefix;
82 progress->total = total;
83 progress->last_percent = -1;
84 progress->delay = 0;
85 progress->need_lf = 0;
86 if (snprintf(buf, sizeof(buf), title, total))
87 fprintf(stderr, "%s\n", buf);
88 set_progress_signal();
91 void start_progress_delay(struct progress *progress, const char *title,
92 const char *prefix, unsigned total,
93 unsigned percent_treshold, unsigned delay)
95 progress->prefix = prefix;
96 progress->total = total;
97 progress->last_percent = -1;
98 progress->delayed_percent_treshold = percent_treshold;
99 progress->delayed_title = title;
100 progress->delay = delay;
101 progress->need_lf = 0;
102 set_progress_signal();
105 void stop_progress(struct progress *progress)
107 clear_progress_signal();
108 if (progress->need_lf)
109 fputc('\n', stderr);