1*a58d3d2aSXin Li /* Copyright (c) 2023 Amazon */ 2*a58d3d2aSXin Li /* 3*a58d3d2aSXin Li Redistribution and use in source and binary forms, with or without 4*a58d3d2aSXin Li modification, are permitted provided that the following conditions 5*a58d3d2aSXin Li are met: 6*a58d3d2aSXin Li 7*a58d3d2aSXin Li - Redistributions of source code must retain the above copyright 8*a58d3d2aSXin Li notice, this list of conditions and the following disclaimer. 9*a58d3d2aSXin Li 10*a58d3d2aSXin Li - Redistributions in binary form must reproduce the above copyright 11*a58d3d2aSXin Li notice, this list of conditions and the following disclaimer in the 12*a58d3d2aSXin Li documentation and/or other materials provided with the distribution. 13*a58d3d2aSXin Li 14*a58d3d2aSXin Li THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 15*a58d3d2aSXin Li ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 16*a58d3d2aSXin Li LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 17*a58d3d2aSXin Li A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR 18*a58d3d2aSXin Li CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 19*a58d3d2aSXin Li EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 20*a58d3d2aSXin Li PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 21*a58d3d2aSXin Li PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF 22*a58d3d2aSXin Li LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 23*a58d3d2aSXin Li NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 24*a58d3d2aSXin Li SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 25*a58d3d2aSXin Li */ 26*a58d3d2aSXin Li 27*a58d3d2aSXin Li #ifndef LOSSGEN_H 28*a58d3d2aSXin Li #define LOSSGEN_H 29*a58d3d2aSXin Li 30*a58d3d2aSXin Li 31*a58d3d2aSXin Li #include "lossgen_data.h" 32*a58d3d2aSXin Li 33*a58d3d2aSXin Li #define PITCH_MIN_PERIOD 32 34*a58d3d2aSXin Li #define PITCH_MAX_PERIOD 256 35*a58d3d2aSXin Li 36*a58d3d2aSXin Li #define NB_XCORR_FEATURES (PITCH_MAX_PERIOD-PITCH_MIN_PERIOD) 37*a58d3d2aSXin Li 38*a58d3d2aSXin Li 39*a58d3d2aSXin Li typedef struct { 40*a58d3d2aSXin Li LossGen model; 41*a58d3d2aSXin Li float gru1_state[LOSSGEN_GRU1_STATE_SIZE]; 42*a58d3d2aSXin Li float gru2_state[LOSSGEN_GRU2_STATE_SIZE]; 43*a58d3d2aSXin Li int last_loss; 44*a58d3d2aSXin Li int used; 45*a58d3d2aSXin Li } LossGenState; 46*a58d3d2aSXin Li 47*a58d3d2aSXin Li 48*a58d3d2aSXin Li void lossgen_init(LossGenState *st); 49*a58d3d2aSXin Li int lossgen_load_model(LossGenState *st, const void *data, int len); 50*a58d3d2aSXin Li 51*a58d3d2aSXin Li int sample_loss( 52*a58d3d2aSXin Li LossGenState *st, 53*a58d3d2aSXin Li float percent_loss); 54*a58d3d2aSXin Li 55*a58d3d2aSXin Li #endif 56