Decode RDO-VAE from stack-allocated state

dnn/dred_rdovae.c

@@ -35,6 +35,21 @@
 #include "dred_rdovae_dec.h"
 #include "dred_rdovae_stats_data.h"
 
+void DRED_rdovae_decode_all(float *features, const float *state, const float *latents, int nb_latents)
+{
+    int i;
+    RDOVAEDec dec;
+    memset(&dec, 0, sizeof(dec));
+    DRED_rdovae_dec_init_states(&dec, state);
+    for (i = 0; i < 2*nb_latents; i += 2)
+    {
+        DRED_rdovae_decode_qframe(
+            &dec,
+            &features[2*i*DRED_NUM_FEATURES],
+            &latents[(i/2)*DRED_LATENT_DIM]);
+    }
+}
+
 size_t DRED_rdovae_get_enc_size()
 {
     return sizeof(RDOVAEEnc);
@@ -117,4 +132,4 @@ const opus_uint16 * DRED_rdovae_get_r_pointer(void)
 const opus_uint16 * DRED_rdovae_get_quant_scales_pointer(void)
 {
     return &dred_quant_scales_q8[0];
-}
\ No newline at end of file
+}

dnn/include/dred_rdovae.h

@@ -33,6 +33,9 @@
 typedef struct RDOVAEDecStruct RDOVAEDec;
 typedef struct RDOVAEEncStruct RDOVAEEnc;
 
+void DRED_rdovae_decode_all(float *features, const float *state, const float *latents, int nb_latents);
+
+
 size_t DRED_rdovae_get_enc_size(void);
 
 size_t DRED_rdovae_get_dec_size(void);