jamie@214: /*
jamie@214:  * Copyright (C) 2012 Jamie Bullock
jamie@214:  *
jamie@214:  * Permission is hereby granted, free of charge, to any person obtaining a copy
jamie@214:  * of this software and associated documentation files (the "Software"), to
jamie@214:  * deal in the Software without restriction, including without limitation the
jamie@214:  * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
jamie@214:  * sell copies of the Software, and to permit persons to whom the Software is
jamie@214:  * furnished to do so, subject to the following conditions:
jamie@214:  *
jamie@214:  * The above copyright notice and this permission notice shall be included in
jamie@214:  * all copies or substantial portions of the Software.
jamie@214:  *
jamie@214:  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
jamie@214:  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
jamie@214:  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
jamie@214:  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
jamie@214:  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
jamie@214:  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
jamie@214:  * IN THE SOFTWARE.
jamie@214:  *
jamie@214:  */
jamie@214: 
jamie@214: /* stateful.c: declares functions that extract features that require stateful data to be retained between frames */
jamie@214: 
jamie@259: #include "xtract/xtract_stateful.h"
jamie@259: #include "xtract/libxtract.h"
jamie@214: 
jamie@214: #include "c-ringbuf/ringbuf.h"
jamie@214: 
jamie@214: #include <stdlib.h>
jamie@214: #include <stdio.h>
jamie@214: #include <float.h>
jamie@214: 
jamie@214: struct xtract_last_n_state_
jamie@214: {
jamie@214:     ringbuf_t ringbuf;
jamie@214: };
jamie@214: 
jamie@214: 
jamie@214: xtract_last_n_state *xtract_last_n_state_new(size_t N)
jamie@214: {
jamie@214:     xtract_last_n_state *last_n_state = malloc(sizeof(xtract_last_n_state));
jamie@214:     
jamie@214:     if (last_n_state == NULL)
jamie@214:     {
jamie@214:         perror("could not allocate memory for xtract_last_n_state");
jamie@214:         return NULL;
jamie@214:     }
jamie@214:     
jamie@214:     last_n_state->ringbuf = ringbuf_new(N * sizeof(double));
jamie@214:     
jamie@242:     if (last_n_state->ringbuf == NULL)
jamie@214:     {
jamie@214:         perror("could not allocate memory for xtract_last_n_state->ringbuf");
jamie@214:     }
jamie@214:         
jamie@214:     return last_n_state;
jamie@214: }
jamie@214: 
jamie@214: void xtract_last_n_state_delete(xtract_last_n_state *last_n_state)
jamie@214: {
jamie@214:     ringbuf_free(&last_n_state->ringbuf);
jamie@214:     free(last_n_state);
jamie@214: }
jamie@214: 
jamie@214: int xtract_last_n(const xtract_last_n_state *state, const double *data, const int N, const void *argv, double *result)
jamie@214: {
jamie@214:     size_t N_bytes = N * sizeof(double);
jamie@214:     
jamie@214:     if (N_bytes != ringbuf_capacity(state->ringbuf))
jamie@214:     {
jamie@214:         fprintf(stderr, "libxtract: error: xtract_last_n(): inconsitent size");
jamie@214:         return XTRACT_BAD_STATE;
jamie@214:     }
jamie@214:     
jamie@242:     ringbuf_memcpy_into(state->ringbuf, data, sizeof(double));
jamie@214:     size_t used = ringbuf_bytes_used(state->ringbuf);
jamie@214:     ringbuf_memcpy_from(result, state->ringbuf, used, false);
jamie@214:     
jamie@214:     if (used < N_bytes)
jamie@214:     {
jamie@214:         /* zero pad */
jamie@214:         for (size_t n = used / sizeof(double); n < N; ++n)
jamie@214:         {
jamie@214:             result[n] = 0.0;
jamie@214:         }
jamie@214:     }
jamie@214:     
jamie@214:     return XTRACT_SUCCESS;
jamie@214: }
jamie@214: