+/* -*-c-*-
+ *
+ * Progress indicators for command-line tools
+ *
+ * (c) 2011 Straylight/Edgeware
+ */
+
+/*----- Licensing notice --------------------------------------------------*
+ *
+ * This file is part of Catacomb.
+ *
+ * Catacomb is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU Library General Public License as
+ * published by the Free Software Foundation; either version 2 of the
+ * License, or (at your option) any later version.
+ *
+ * Catacomb is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU Library General Public License for more details.
+ *
+ * You should have received a copy of the GNU Library General Public
+ * License along with Catacomb; if not, write to the Free
+ * Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
+ * MA 02111-1307, USA.
+ */
+
+/*----- Header files ------------------------------------------------------*/
+
+#define _FILE_OFFSET_BITS 64
+
+#include "config.h"
+
+#include "cc.h"
+
+#ifndef PATHSEP
+# if defined(__riscos)
+# define PATHSEP '.'
+# elif defined(__unix) || defined(unix)
+# define PATHSEP '/'
+# else
+# define PATHSEP '\\'
+# endif
+#endif
+
+/*----- Static data -------------------------------------------------------*/
+
+static const char baton[] = "-\\|/";
+
+/*----- Human-friendly unit printing --------------------------------------*/
+
+struct unit {
+ const char *name;
+ int m;
+};
+
+/* --- @prhuman_time@ --- *
+ *
+ * Arguments: @FILE *fp@ = stream to print on
+ * @unsigned long n@ = time in seconds to print
+ *
+ * Returns: ---
+ *
+ * Use: Prints a time in some reasonable format. The time takes up
+ * PRHUMAN_TIMEWD character spaces.
+ */
+
+#define PRHUMAN_TIMEWD 7
+
+static void prhuman_time(FILE *fp, unsigned long n)
+{
+ const static struct unit utime[] = {
+ { "s", 60 }, { "m", 60 }, { "h", 24 }, { "d", 0 }
+ };
+
+ unsigned long m = 0;
+ const struct unit *u = utime;
+
+ while (u[1].m && n > u[0].m*u[1].m) { n /= u->m; u++; }
+ m = n / u[1].m; n %= u[0].m;
+ if (m) fprintf(fp, "%3lu%s%02lu%s", m, u[1].name, n, u[0].name);
+ else fprintf(fp, " %2lu%s", n, u[0].name);
+}
+
+/* --- @prhuman_data@ --- *
+ *
+ * Arguments: @FILE *fp@ = file to print on
+ * @off_t n@ = size to be printed
+ *
+ * Returns: ---
+ *
+ * Use: Prints a data size in some reasonable format. The data size
+ * takes up PRHUMAN_DATAWD character spaces.
+ */
+
+#define PRHUMAN_DATAWD 7
+
+static void prhuman_data(FILE *fp, off_t n)
+{
+ const static struct unit udata[] = {
+ { " ", 1024 }, { "k", 1024 }, { "M", 1024 }, { "G", 1024 },
+ { "T", 1024 }, { "P", 1024 }, { "E", 1024 }, { "Z", 1024 },
+ { "Y", 0 }
+ };
+
+ double x = n;
+ const struct unit *u = udata;
+
+ while (u->m && x >= u->m) { x /= u->m; u++; }
+ fprintf(fp, "%6.1f%s", x, u->name);
+}
+
+/*----- Main code ---------------------------------------------------------*/
+
+#define BARWD 16
+
+/* --- @fprogress_init@ --- *
+ *
+ * Arguments: @fprogress *f@ = progress context to be initialized
+ * @const char *name@ = file name string to show
+ * @FILE *fp@ = file we're reading from
+ *
+ * Returns: Zero on success, nonzero if the file's state is now broken.
+ *
+ * Use: Initializes a progress context. Nothing is actually
+ * displayed yet.
+ */
+
+int fprogress_init(fprogress *f, const char *name, FILE *fp)
+{
+ const char *p;
+ off_t o, sz = -1;
+ size_t n;
+
+ /* --- Set up the offset --- */
+
+ if ((o = ftello(fp)) >= 0 &&
+ fseeko(fp, 0, SEEK_END) >= 0 &&
+ (sz = ftello(fp),
+ fseeko(fp, o, SEEK_SET) < 0))
+ return (-1);
+ if (o != -1 && sz != -1) sz -= o;
+ f->o = f->olast = 0; f->sz = sz;
+
+ /* --- Set up the file name --- */
+
+ n = strlen(name);
+ if (n < sizeof(f->name))
+ strcpy(f->name, name);
+ else if ((p = strchr(name + n - sizeof(f->name) + 4, PATHSEP)) != 0)
+ sprintf(f->name, "...%s", p);
+ else {
+ p = strrchr(name, PATHSEP);
+ if (!p) sprintf(f->name, "%.*s...", (int)sizeof(f->name) - 4, name);
+ else sprintf(f->name, "...%.*s...", (int)sizeof(f->name) - 7, p);
+ }
+
+ /* --- Set up some other stuff --- */
+
+ f->start = f->last = time(0);
+ f->bp = baton;
+
+ /* --- Done --- */
+
+ return (0);
+}
+
+/* --- @fprogress_clear@ --- *
+ *
+ * Arguments: @fprogress *f@ = progress context
+ *
+ * Returns: ---
+ *
+ * Use: Clears the progress display from the screen.
+ */
+
+void fprogress_clear(fprogress *f)
+{
+ fprintf(stderr, "\r%*s\r",
+ sizeof(f->name) + 2*PRHUMAN_DATAWD + PRHUMAN_TIMEWD + BARWD + 16,
+ "");
+}
+
+/* --- @fprogress_update@ --- *
+ *
+ * Arguments: @fprogress *f@ = progress context
+ * @size_t n@ = how much progress has been made
+ *
+ * Returns: ---
+ *
+ * Use: Maybe updates the display to show that some progress has been
+ * made.
+ */
+
+void fprogress_update(fprogress *f, size_t sz)
+{
+ time_t now = time(0);
+ int i, n;
+
+ /* --- See if there's anything to do --- */
+
+ f->o += sz;
+ if (difftime(now, f->last) < 1) return;
+ f->last = now;
+
+ /* --- See if we're going to lose the ETA and percentage indicators --- */
+
+ if (f->olast < f->sz && f->o > f->sz) fprogress_clear(f);
+ f->olast = f->o;
+
+ /* --- Do the initial display --- */
+
+ fprintf(stderr, "\r%-*s%c ",
+ (int)sizeof(f->name), f->name,
+ *f->bp++);
+ if (!*f->bp) f->bp = baton;
+ prhuman_data(stderr, f->o);
+
+ /* --- More complicated display if we have a set size --- */
+
+ if (f->sz > f->o) {
+ fputc('/', stderr);
+ prhuman_data(stderr, f->sz);
+ fputs(" [", stderr);
+ n = (f->o*BARWD + f->sz/2)/f->sz;
+ for (i = 0; i < n; i++) fputc('.', stderr);
+ fprintf(stderr, "%*s] %3d%% ETA ", BARWD - n, "",
+ (int)((f->o*100 + 50)/f->sz));
+ prhuman_time(stderr, difftime(now, f->start)*(f->sz - f->o)/f->o);
+ }
+}
+
+/* --- @fprogress_done@ --- *
+ *
+ * Arguments: @fprogress *f@ = progress context
+ *
+ * Returns: ---
+ *
+ * Use: Clear up the progress context and removes any display.
+ */
+
+void fprogress_done(fprogress *f) { fprogress_clear(f); }
+
+/*----- That's all, folks -------------------------------------------------*/