gitweb: '--cc' for merges in 'commitdiff' view
[git] / progress.c
1 #include "git-compat-util.h"
2 #include "progress.h"
3
4 static volatile sig_atomic_t progress_update;
5
6 static void progress_interval(int signum)
7 {
8         progress_update = 1;
9 }
10
11 static void set_progress_signal(void)
12 {
13         struct sigaction sa;
14         struct itimerval v;
15
16         progress_update = 0;
17
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);
23
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);
28 }
29
30 static void clear_progress_signal(void)
31 {
32         struct itimerval v = {{0,},};
33         setitimer(ITIMER_REAL, &v, NULL);
34         signal(SIGALRM, SIG_IGN);
35         progress_update = 0;
36 }
37
38 int display_progress(struct progress *progress, unsigned n)
39 {
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;
52                         }
53                 }
54                 if (snprintf(buf, sizeof(buf),
55                              progress->delayed_title, progress->total))
56                         fprintf(stderr, "%s\n", buf);
57         }
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;
67                 }
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;
73         }
74         return 0;
75 }
76
77 void start_progress(struct progress *progress, const char *title,
78                     const char *prefix, unsigned total)
79 {
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();
89 }
90
91 void start_progress_delay(struct progress *progress, const char *title,
92                           const char *prefix, unsigned total,
93                           unsigned percent_treshold, unsigned delay)
94 {
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();
103 }
104
105 void stop_progress(struct progress *progress)
106 {
107         clear_progress_signal();
108         if (progress->need_lf)
109                 fputc('\n', stderr);
110 }