Commit 097857e6 authored by Rusty Russell's avatar Rusty Russell

tally: don't add extra bucket unless buckets == 0

That way we match our comment about optimal tally_new() arg matching histogram height.
parent fdea8abe
...@@ -16,25 +16,28 @@ struct tally { ...@@ -16,25 +16,28 @@ struct tally {
size_t total[2]; size_t total[2];
/* This allows limited frequency analysis. */ /* This allows limited frequency analysis. */
unsigned buckets, step_bits; unsigned buckets, step_bits;
size_t counts[1 /* [buckets] */ ]; size_t counts[1 /* Actually: [buckets] */ ];
}; };
struct tally *tally_new(unsigned buckets) struct tally *tally_new(unsigned buckets)
{ {
struct tally *tally; struct tally *tally;
/* There is always 1 bucket. */
if (buckets == 0)
buckets = 1;
/* Check for overflow. */ /* Check for overflow. */
if (buckets && SIZE_MAX / buckets < sizeof(tally->counts[0])) if (buckets && SIZE_MAX / buckets < sizeof(tally->counts[0]))
return NULL; return NULL;
tally = malloc(sizeof(*tally) + sizeof(tally->counts[0])*buckets); tally = malloc(sizeof(*tally) + sizeof(tally->counts[0])*(buckets-1));
if (tally) { if (tally) {
tally->max = ((size_t)1 << (SIZET_BITS - 1)); tally->max = ((size_t)1 << (SIZET_BITS - 1));
tally->min = ~tally->max; tally->min = ~tally->max;
tally->total[0] = tally->total[1] = 0; tally->total[0] = tally->total[1] = 0;
/* There is always 1 bucket. */ tally->buckets = buckets;
tally->buckets = buckets+1;
tally->step_bits = 0; tally->step_bits = 0;
memset(tally->counts, 0, sizeof(tally->counts[0])*(buckets+1)); memset(tally->counts, 0, sizeof(tally->counts[0])*buckets);
} }
return tally; return tally;
} }
...@@ -422,7 +425,7 @@ char *tally_histogram(const struct tally *tally, ...@@ -422,7 +425,7 @@ char *tally_histogram(const struct tally *tally,
} else { } else {
/* We create a temporary then renormalize so < height. */ /* We create a temporary then renormalize so < height. */
/* FIXME: Antialias properly! */ /* FIXME: Antialias properly! */
tmp = tally_new(tally->buckets-1); tmp = tally_new(tally->buckets);
if (!tmp) if (!tmp)
return NULL; return NULL;
tmp->min = tally->min; tmp->min = tally->min;
......
...@@ -3,7 +3,7 @@ ...@@ -3,7 +3,7 @@
int main(void) int main(void)
{ {
struct tally *tally = tally_new(1); struct tally *tally = tally_new(2);
plan_tests(4); plan_tests(4);
tally->min = 0; tally->min = 0;
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment