From 59791f6534bb0198319a7c6c1c6d840b96f26a26 Mon Sep 17 00:00:00 2001
From: Mathis Rosenhauer <rosenhauer@dkrz.de>
Date: Wed, 5 Dec 2012 15:28:45 +0100
Subject: [PATCH] Update total_* only once per aec_encode()

---
 src/encode.c           |  8 ++++----
 src/encode_accessors.c | 15 ---------------
 2 files changed, 4 insertions(+), 19 deletions(-)

diff --git a/src/encode.c b/src/encode.c
index 5f75eae..44ffcaf 100644
--- a/src/encode.c
+++ b/src/encode.c
@@ -469,7 +469,6 @@ static int m_flush_block_resumable(struct aec_stream *strm)
 
         *strm->next_out++ = state->cds_buf[state->i];
         strm->avail_out--;
-        strm->total_out++;
         state->i++;
     }
     state->mode = m_get_block;
@@ -490,7 +489,6 @@ static int m_flush_block(struct aec_stream *strm)
         n = state->cds - strm->next_out;
         strm->next_out += n;
         strm->avail_out -= n;
-        strm->total_out += n;
         state->mode = m_get_block;
         return M_CONTINUE;
     }
@@ -679,7 +677,6 @@ static int m_get_rsi_resumable(struct aec_stream *strm)
                         if (!state->direct_out)
                             *strm->next_out++ = *state->cds;
                         strm->avail_out--;
-                        strm->total_out++;
                     }
                     return M_EXIT;
                 }
@@ -870,6 +867,8 @@ int aec_encode(struct aec_stream *strm, int flush)
     struct internal_state *state = strm->state;
 
     state->flush = flush;
+    strm->total_in += strm->avail_in;
+    strm->total_out += strm->avail_out;
 
     while (state->mode(strm) == M_CONTINUE);
 
@@ -877,12 +876,13 @@ int aec_encode(struct aec_stream *strm, int flush)
         n = state->cds - strm->next_out;
         strm->next_out += n;
         strm->avail_out -= n;
-        strm->total_out += n;
 
         *state->cds_buf = *state->cds;
         state->cds = state->cds_buf;
         state->direct_out = 0;
     }
+    strm->total_in -= strm->avail_in;
+    strm->total_out -= strm->avail_out;
     return AEC_OK;
 }
 
diff --git a/src/encode_accessors.c b/src/encode_accessors.c
index 3bc68d8..2b8ccd5 100644
--- a/src/encode_accessors.c
+++ b/src/encode_accessors.c
@@ -64,7 +64,6 @@
 uint32_t aec_get_8(struct aec_stream *strm)
 {
     strm->avail_in--;
-    strm->total_in++;
     return *strm->next_in++;
 }
 
@@ -76,7 +75,6 @@ uint32_t aec_get_lsb_16(struct aec_stream *strm)
         | (uint32_t)strm->next_in[0];
 
     strm->next_in += 2;
-    strm->total_in += 2;
     strm->avail_in -= 2;
     return data;
 }
@@ -89,7 +87,6 @@ uint32_t aec_get_msb_16(struct aec_stream *strm)
         | (uint32_t)strm->next_in[1];
 
     strm->next_in += 2;
-    strm->total_in += 2;
     strm->avail_in -= 2;
     return data;
 }
@@ -103,7 +100,6 @@ uint32_t aec_get_lsb_24(struct aec_stream *strm)
         | (uint32_t)strm->next_in[0];
 
     strm->next_in += 3;
-    strm->total_in += 3;
     strm->avail_in -= 3;
     return data;
 }
@@ -117,7 +113,6 @@ uint32_t aec_get_msb_24(struct aec_stream *strm)
         | (uint32_t)strm->next_in[2];
 
     strm->next_in += 3;
-    strm->total_in += 3;
     strm->avail_in -= 3;
     return data;
 }
@@ -132,7 +127,6 @@ uint32_t aec_get_lsb_32(struct aec_stream *strm)
         | (uint32_t)strm->next_in[0];
 
     strm->next_in += 4;
-    strm->total_in += 4;
     strm->avail_in -= 4;
     return data;
 }
@@ -147,7 +141,6 @@ uint32_t aec_get_msb_32(struct aec_stream *strm)
         | (uint32_t)strm->next_in[3];
 
     strm->next_in += 4;
-    strm->total_in += 4;
     strm->avail_in -= 4;
     return data;
 }
@@ -159,7 +152,6 @@ void aec_get_rsi_8(struct aec_stream *strm)
     int rsi = strm->rsi * strm->block_size;
 
     strm->next_in += rsi;
-    strm->total_in += rsi;
     strm->avail_in -= rsi;
 
     while (rsi) {
@@ -184,7 +176,6 @@ void aec_get_rsi_lsb_16(struct aec_stream *strm)
     int rsi = strm->rsi * strm->block_size;
 
     strm->next_in += 2 * rsi;
-    strm->total_in += 2 * rsi;
     strm->avail_in -= 2 * rsi;
 
     while (rsi) {
@@ -217,7 +208,6 @@ void aec_get_rsi_msb_16(struct aec_stream *strm)
     int rsi = strm->rsi * strm->block_size;
 
     strm->next_in += 2 * rsi;
-    strm->total_in += 2 * rsi;
     strm->avail_in -= 2 * rsi;
 
     while (rsi) {
@@ -250,7 +240,6 @@ void aec_get_rsi_lsb_24(struct aec_stream *strm)
     int rsi = strm->rsi * strm->block_size;
 
     strm->next_in += 3 * rsi;
-    strm->total_in += 3 * rsi;
     strm->avail_in -= 3 * rsi;
 
     while (rsi) {
@@ -291,7 +280,6 @@ void aec_get_rsi_msb_24(struct aec_stream *strm)
     int rsi = strm->rsi * strm->block_size;
 
     strm->next_in += 3 * rsi;
-    strm->total_in += 3 * rsi;
     strm->avail_in -= 3 * rsi;
 
     while (rsi) {
@@ -332,7 +320,6 @@ void aec_get_rsi_msb_24(struct aec_stream *strm)
         memcpy(strm->state->data_raw,               \
                strm->next_in, 4 * rsi);             \
         strm->next_in += 4 * rsi;                   \
-        strm->total_in += 4 * rsi;                  \
         strm->avail_in -= 4 * rsi;                  \
     }
 
@@ -344,7 +331,6 @@ void aec_get_rsi_lsb_32(struct aec_stream *strm)
     int rsi = strm->rsi * strm->block_size;
 
     strm->next_in += 4 * rsi;
-    strm->total_in += 4 * rsi;
     strm->avail_in -= 4 * rsi;
 
     while (rsi) {
@@ -396,7 +382,6 @@ void aec_get_rsi_msb_32(struct aec_stream *strm)
     int rsi = strm->rsi * strm->block_size;
 
     strm->next_in += 4 * rsi;
-    strm->total_in += 4 * rsi;
     strm->avail_in -= 4 * rsi;
 
     while (rsi) {
-- 
2.7.4