/* Copyright (C) 2022-2023 Salvatore Sanfilippo -- All Rights Reserved
 * See the LICENSE file for information about the license. */

#include <inttypes.h>
#include <furi/core/string.h>
#include <furi.h>
#include <furi_hal.h>
#include "app_buffer.h"

/* Allocate and initialize a samples buffer. */
RawSamplesBuffer *raw_samples_alloc(void) {
    RawSamplesBuffer *buf = malloc(sizeof(*buf));
    buf->mutex = furi_mutex_alloc(FuriMutexTypeNormal);
    raw_samples_reset(buf);
    return buf;
}

/* Free a sample buffer. Should be called when the mutex is released. */
void raw_samples_free(RawSamplesBuffer *s) {
    furi_mutex_free(s->mutex);
    free(s);
}

/* This just set all the samples to zero and also resets the internal
 * index. There is no need to call it after raw_samples_alloc(), but only
 * when one wants to reset the whole buffer of samples. */
void raw_samples_reset(RawSamplesBuffer *s) {
    furi_mutex_acquire(s->mutex,FuriWaitForever);
    s->total = RAW_SAMPLES_NUM;
    s->idx = 0;
    s->short_pulse_dur = 0;
    memset(s->level,0,sizeof(s->level));
    memset(s->dur,0,sizeof(s->dur));
    furi_mutex_release(s->mutex);
}

/* Set the raw sample internal index so that what is currently at
 * offset 'offset', will appear to be at 0 index. */
void raw_samples_center(RawSamplesBuffer *s, uint32_t offset) {
    s->idx = (s->idx+offset) % RAW_SAMPLES_NUM;
}

/* Add the specified sample in the circular buffer. */
void raw_samples_add(RawSamplesBuffer *s, bool level, uint32_t dur) {
    furi_mutex_acquire(s->mutex,FuriWaitForever);
    s->level[s->idx] = level;
    s->dur[s->idx] = dur;
    s->idx = (s->idx+1) % RAW_SAMPLES_NUM;
    furi_mutex_release(s->mutex);
}

/* Get the sample from the buffer. It is possible to use out of range indexes
 * as 'idx' because the modulo operation will rewind back from the start. */
void raw_samples_get(RawSamplesBuffer *s, uint32_t idx, bool *level, uint32_t *dur)
{
    furi_mutex_acquire(s->mutex,FuriWaitForever);
    idx = (s->idx + idx) % RAW_SAMPLES_NUM;
    *level = s->level[idx];
    *dur = s->dur[idx];
    furi_mutex_release(s->mutex);
}

/* Copy one buffer to the other, including current index. */
void raw_samples_copy(RawSamplesBuffer *dst, RawSamplesBuffer *src) {
    furi_mutex_acquire(src->mutex,FuriWaitForever);
    furi_mutex_acquire(dst->mutex,FuriWaitForever);
    dst->idx = src->idx;
    dst->short_pulse_dur = src->short_pulse_dur;
    memcpy(dst->level,src->level,sizeof(dst->level));
    memcpy(dst->dur,src->dur,sizeof(dst->dur));
    furi_mutex_release(src->mutex);
    furi_mutex_release(dst->mutex);
}