You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
70 lines
1.1 KiB
70 lines
1.1 KiB
#include <stdint.h>
|
|
#include <malloc.h>
|
|
|
|
#include "meanbuf.h"
|
|
#include "malloc_safe.h"
|
|
|
|
|
|
struct meanbuf_struct {
|
|
float * buf; // buffer (allocated at init)
|
|
size_t cap; // capacity
|
|
size_t nw; // next write index
|
|
float mean; // updated on write
|
|
};
|
|
|
|
|
|
/** Init a buffer */
|
|
MeanBuf *meanbuf_create(size_t size)
|
|
{
|
|
MeanBuf *mb = malloc_s(sizeof(MeanBuf));
|
|
|
|
if (size < 1) size = 1;
|
|
|
|
mb->buf = calloc_s(size, sizeof(float)); // calloc, so it starts with zeros.
|
|
mb->cap = size;
|
|
mb->nw = 0;
|
|
mb->mean = 0;
|
|
|
|
// clean buffer
|
|
for (uint16_t i = 0; i < size; i++) {
|
|
mb->buf[i] = 0;
|
|
}
|
|
|
|
return mb;
|
|
}
|
|
|
|
|
|
void meanbuf_destroy(MeanBuf *mb)
|
|
{
|
|
if (mb == NULL) return;
|
|
|
|
if (mb->buf != NULL) {
|
|
free(mb->buf);
|
|
}
|
|
|
|
free(mb);
|
|
}
|
|
|
|
|
|
/** Add a value to the buffer. Returns current mean. */
|
|
float meanbuf_add(MeanBuf *mb, float f)
|
|
{
|
|
// add sample
|
|
mb->buf[mb->nw++] = f;
|
|
if (mb->nw == mb->cap) mb->nw = 0;
|
|
|
|
// calculate average
|
|
float acc = 0;
|
|
for (size_t i = 0; i < mb->cap; i++) {
|
|
acc += mb->buf[i];
|
|
}
|
|
|
|
acc /= mb->cap;
|
|
|
|
return mb->mean = acc;
|
|
}
|
|
|
|
float meanbuf_current(MeanBuf *mb)
|
|
{
|
|
return mb->mean;
|
|
}
|
|
|