tickmarks.c

   1 #include <stdio.h>
   2 #include <math.h>
   3 #include <malloc.h>
   4 #include <string.h>
   5
   6 /*
   7  * adapted from Paul Heckbert's algorithm on p 657-659 of
   8  * Andrew S. Glassner's book, "Graphics Gems"
   9  * ISBN 0-12-286166-3
  10  *
  11  */
  12
  13 #include "tickmarks.h"
  14
  15 #define MAX(a, b) (((a) < (b)) ? (b) : (a))
  16
  17 static double nicenum(double x, int round)
  18 {
  19         int exp;        /* exponent of x */
  20         double f;       /* fractional part of x */
  21
  22         exp = floor(log10(x));
  23         f = x / pow(10.0, exp);
  24         if (round) {
  25                 if (f < 1.5)
  26                         return 1.0 * pow(10.0, exp);
  27                 if (f < 3.0)
  28                         return 2.0 * pow(10.0, exp);
  29                 if (f < 7.0)
  30                         return 5.0 * pow(10.0, exp);
  31                 return 10.0 * pow(10.0, exp);
  32         }
  33         if (f <= 1.0)
  34                 return 1.0 * pow(10.0, exp);
  35         if (f <= 2.0)
  36                 return 2.0 * pow(10.0, exp);
  37         if (f <= 5.0)
  38                 return 5.0 * pow(10.0, exp);
  39         return 10.0 * pow(10.0, exp);
  40 }
  41
  42 static void shorten(struct tickmark *tm, int nticks, int *power_of_ten,
  43                         int use_KMG_symbols, int base_offset)
  44 {
  45         const char shorten_chr[] = { 0, 'K', 'M', 'G', 'P', 'E', 0 };
  46         int i, l, minshorten, shorten_idx = 0;
  47         char *str;
  48
  49         minshorten = 100;
  50         for (i = 0; i < nticks; i++) {
  51                 str = tm[i].string;
  52                 l = strlen(str);
  53
  54                 if (strcmp(str, "0") == 0)
  55                         continue;
  56                 if (l > 9 && strcmp(&str[l - 9], "000000000") == 0) {
  57                         *power_of_ten = 9;
  58                         shorten_idx = 3;
  59                 } else if (6 < minshorten && l > 6 &&
  60                                 strcmp(&str[l - 6], "000000") == 0) {
  61                         *power_of_ten = 6;
  62                         shorten_idx = 2;
  63                 } else if (l > 3 && strcmp(&str[l - 3], "000") == 0) {
  64                         *power_of_ten = 3;
  65                         shorten_idx = 1;
  66                 } else {
  67                         *power_of_ten = 0;
  68                 }
  69
  70                 if (*power_of_ten < minshorten)
  71                         minshorten = *power_of_ten;
  72         }
  73
  74         if (minshorten == 0)
  75                 return;
  76         if (!use_KMG_symbols)
  77                 shorten_idx = 0;
  78         else if (base_offset)
  79                 shorten_idx += base_offset;
  80
  81         for (i = 0; i < nticks; i++) {
  82                 str = tm[i].string;
  83                 l = strlen(str);
  84                 str[l - minshorten] = shorten_chr[shorten_idx];
  85                 if (shorten_idx)
  86                         str[l - minshorten + 1] = '\0';
  87         }
  88 }
  89
  90 int calc_tickmarks(double min, double max, int nticks, struct tickmark **tm,
  91                 int *power_of_ten, int use_KMG_symbols, int base_offset)
  92 {
  93         char str[100];
  94         int nfrac;
  95         double d;       /* tick mark spacing */
  96         double graphmin, graphmax;      /* graph range min and max */
  97         double range, x;
  98         int count, i;
  99
 100         /* we expect min != max */
 101         range = nicenum(max - min, 0);
 102         d = nicenum(range / (nticks - 1), 1);
 103         graphmin = floor(min / d) * d;
 104         graphmax = ceil(max / d) * d;
 105         nfrac = MAX(-floor(log10(d)), 0);
 106         snprintf(str, sizeof(str)-1, "%%.%df", nfrac);
 107
 108         count = ((graphmax + 0.5 * d) - graphmin) / d + 1;
 109         *tm = malloc(sizeof(**tm) * count);
 110
 111         i = 0;
 112         for (x = graphmin; x < graphmax + 0.5 * d; x += d) {
 113                 (*tm)[i].value = x;
 114                 sprintf((*tm)[i].string, str, x);
 115                 i++;
 116         }
 117         shorten(*tm, i, power_of_ten, use_KMG_symbols, base_offset);
 118         return i;
 119 }
 120
 121 #if 0
 122
 123 static void test_range(double x, double y)
 124 {
 125         int nticks, i;
 126
 127         struct tickmark *tm = NULL;
 128         printf("Testing range %g - %g\n", x, y);
 129         nticks = calc_tickmarks(x, y, 10, &tm);
 130
 131         for (i = 0; i < nticks; i++) {
 132                 printf("   (%s) %g\n", tm[i].string, tm[i].value);
 133         }
 134         printf("\n\n");
 135         free(tm);
 136 }
 137
 138 int main(int argc, char *argv[])
 139 {
 140         test_range(0.0005, 0.008);
 141         test_range(0.5, 0.8);
 142         test_range(5.5, 8.8);
 143         test_range(50.5, 80.8);
 144         test_range(-20, 20.8);
 145         test_range(-30, 700.8);
 146 }
 147 #endif