average.c 1.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162
  1. // SPDX-License-Identifier: GPL-3.0-or-later
  2. #include "average.h"
  3. // ----------------------------------------------------------------------------
  4. // average
  5. struct grouping_average {
  6. calculated_number sum;
  7. size_t count;
  8. };
  9. void *grouping_create_average(RRDR *r) {
  10. (void)r;
  11. return callocz(1, sizeof(struct grouping_average));
  12. }
  13. // resets when switches dimensions
  14. // so, clear everything to restart
  15. void grouping_reset_average(RRDR *r) {
  16. struct grouping_average *g = (struct grouping_average *)r->internal.grouping_data;
  17. g->sum = 0;
  18. g->count = 0;
  19. }
  20. void grouping_free_average(RRDR *r) {
  21. freez(r->internal.grouping_data);
  22. r->internal.grouping_data = NULL;
  23. }
  24. void grouping_add_average(RRDR *r, calculated_number value) {
  25. if(!isnan(value)) {
  26. struct grouping_average *g = (struct grouping_average *)r->internal.grouping_data;
  27. g->sum += value;
  28. g->count++;
  29. }
  30. }
  31. calculated_number grouping_flush_average(RRDR *r, RRDR_VALUE_FLAGS *rrdr_value_options_ptr) {
  32. struct grouping_average *g = (struct grouping_average *)r->internal.grouping_data;
  33. calculated_number value;
  34. if(unlikely(!g->count)) {
  35. value = 0.0;
  36. *rrdr_value_options_ptr |= RRDR_VALUE_EMPTY;
  37. }
  38. else {
  39. if(unlikely(r->internal.resampling_group != 1)) {
  40. if (unlikely(r->result_options & RRDR_RESULT_OPTION_VARIABLE_STEP))
  41. value = g->sum / g->count / r->internal.resampling_divisor;
  42. else
  43. value = g->sum / r->internal.resampling_divisor;
  44. } else
  45. value = g->sum / g->count;
  46. }
  47. g->sum = 0.0;
  48. g->count = 0;
  49. return value;
  50. }