1 files changed, 28 insertions, 26 deletions

diff --git a/apps/codecs/wav_enc.c b/apps/codecs/wav_enc.c
index ef1a88ec23..e4afeaf93c 100644
--- a/apps/codecs/wav_enc.c
+++ b/apps/codecs/wav_enc.c
@@ -345,40 +345,42 @@ static bool init_encoder(void)
     return true;
 } /* init_encoder */
 
-/* main codec entry point */
-enum codec_status codec_main(void)
+/* this is the codec entry point */
+enum codec_status codec_main(enum codec_entry_call_reason reason)
 {
-    if (!init_encoder())
-        return CODEC_ERROR;
+    if (reason == CODEC_LOAD) {
+        if (!init_encoder())
+            return CODEC_ERROR;
+    }
+    else if (reason == CODEC_UNLOAD) {
+        /* reset parameters to initial state */
+        ci->enc_set_parameters(NULL);
+    }
 
+    return CODEC_OK;
+}
+
+/* this is called for each file to process */
+enum codec_status codec_run(void)
+{
     /* main encoding loop */
-    while(!ci->stop_codec)
+    while(ci->get_command(NULL) != CODEC_ACTION_HALT)
     {
-        uint32_t *src;
+        uint32_t *src = (uint32_t *)ci->enc_get_pcm_data(PCM_CHUNK_SIZE);
+        struct enc_chunk_hdr *chunk;
 
-        while ((src = (uint32_t *)ci->enc_get_pcm_data(PCM_CHUNK_SIZE)) != NULL)
-        {
-            struct enc_chunk_hdr *chunk;
-
-            if (ci->stop_codec)
-                break;
+        if(src == NULL)
+            continue;
 
-            chunk           = ci->enc_get_chunk();
-            chunk->enc_size = enc_size;
-            chunk->num_pcm  = PCM_SAMP_PER_CHUNK;
-            chunk->enc_data = ENC_CHUNK_SKIP_HDR(chunk->enc_data, chunk);
+        chunk           = ci->enc_get_chunk();
+        chunk->enc_size = enc_size;
+        chunk->num_pcm  = PCM_SAMP_PER_CHUNK;
+        chunk->enc_data = ENC_CHUNK_SKIP_HDR(chunk->enc_data, chunk);
 
-            chunk_to_wav_format(src, (uint32_t *)chunk->enc_data);
+        chunk_to_wav_format(src, (uint32_t *)chunk->enc_data);
 
-            ci->enc_finish_chunk();
-            ci->yield();
-        }
-
-        ci->yield();
+        ci->enc_finish_chunk();
     }
 
-    /* reset parameters to initial state */
-    ci->enc_set_parameters(NULL);
-
     return CODEC_OK;
-} /* codec_start */
+}