refine code

3 files changed, 16 insertions, 6 deletions

diff --git a/src/storage/flexible_ngram.h b/src/storage/flexible_ngram.h
index f437b80..9bcfdc3 100644
--- a/src/storage/flexible_ngram.h
+++ b/src/storage/flexible_ngram.h
@@ -19,6 +19,12 @@ private:
     MemoryChunk m_chunk;
     FlexibleSingleGram(void * buffer, size_t length);
 public:
+    /* item typedefs */
+    typedef struct{
+        phrase_token_t m_token;
+        ArrayItem m_item;
+    } ArrayItemWithToken;
+
     /* Null Constructor */
     FlexibleSingleGram();
     /* retrieve all items */
@@ -73,6 +79,10 @@ public:
     bool store(phrase_token_t index, FlexibleSingleGram * & single_gram);
     /* array of phrase_token_t items, for parameter estimation. */
     bool get_all_items(GArray * items);
+
+    /* get/set magic header. */
+    bool get_magic_header(MagicHeader & header);
+    bool set_magic_header(const MagicHeader & header);
 };
 
 #endif
diff --git a/src/storage/ngram.cpp b/src/storage/ngram.cpp
index 8ea8334..c6b1736 100644
--- a/src/storage/ngram.cpp
+++ b/src/storage/ngram.cpp
@@ -38,15 +38,15 @@ SingleGram::SingleGram(void * buffer, size_t length){
     m_chunk.set_chunk(buffer, length, NULL);
 }
 
-bool SingleGram::set_total_freq(guint32 m_total){
+bool SingleGram::set_total_freq(guint32 total){
     char * buf_begin = (char *)m_chunk.begin();
-    *((guint32 *)buf_begin) = m_total;
+    *((guint32 *)buf_begin) = total;
     return true;
 }
 
-bool SingleGram::get_total_freq(guint32 & m_total){
+bool SingleGram::get_total_freq(guint32 & total){
     char * buf_begin = (char *)m_chunk.begin();
-    m_total = *((guint32 *)buf_begin);
+    total = *((guint32 *)buf_begin);
     return true;
 }
 
diff --git a/src/storage/ngram.h b/src/storage/ngram.h
index c4fdf7b..c5e7bc8 100644
--- a/src/storage/ngram.h
+++ b/src/storage/ngram.h
@@ -66,12 +66,12 @@ public:
     /* get_total_freq method
      * used in user bigram table
      */
-    bool get_total_freq(guint32 & m_total);
+    bool get_total_freq(guint32 & total);
 
     /* set_total_freq method
      * used in user bigram table
      */
-    bool set_total_freq(guint32 m_total);
+    bool set_total_freq(guint32 total);
     
     /* prune one method
      * only used in training