Import libcds.

[SXSI/libcds.git] / src / static_sequence / wt_coder_huff.cpp

diff --git a/src/static_sequence/wt_coder_huff.cpp b/src/static_sequence/wt_coder_huff.cpp
new file mode 100644 (file)
index 0000000..5ba633b
--- /dev/null
+++ b/src/static_sequence/wt_coder_huff.cpp
@@ -0,0 +1,89 @@
+/* wt_coder_huff.cpp
+ * Copyright (C) 2008, Francisco Claude, all rights reserved.
+ *
+ * wt_coder_huff definition
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+ *
+ */
+ 
+#include <wt_coder_huff.h>
+
+wt_coder_huff::wt_coder_huff(uint * symbs, uint n, alphabet_mapper * am) {
+  for(uint i=0;i<n;i++)
+    symbs[i] = am->map(symbs[i]);
+       hc = new huffman_codes(symbs, n);
+  buffer = new uint[hc->max_length()/W+1]; 
+  s_len = 0; last_symbol = (uint)-1;
+  for(uint i=0;i<n;i++)
+    symbs[i] = am->unmap(symbs[i]);
+}
+
+wt_coder_huff::wt_coder_huff(uchar * symbs, uint n, alphabet_mapper * am) {
+  for(uint i=0;i<n;i++)
+    symbs[i] = (uchar)am->map((uint)symbs[i]);
+       hc = new huffman_codes(symbs, n);
+  buffer = new uint[hc->max_length()/W+1]; 
+  s_len = 0; last_symbol = (uint)-1;
+  for(uint i=0;i<n;i++)
+    symbs[i] = (uchar)am->unmap((uint)symbs[i]);
+}
+
+wt_coder_huff::wt_coder_huff() {}
+
+wt_coder_huff::~wt_coder_huff() {
+  delete hc;
+  delete [] buffer;
+}
+
+bool wt_coder_huff::is_set(uint symbol, uint l) {
+       if(symbol!=last_symbol) {
+    s_len = (uint)hc->encode(symbol, buffer, (ulong)0);
+    last_symbol = symbol;
+  }
+  return bitget(buffer,l);
+}
+
+bool wt_coder_huff::done(uint symbol, uint l) {
+  if(symbol!=last_symbol) {
+    s_len = (uint)hc->encode(symbol, buffer, (ulong)0);
+    last_symbol = symbol;
+  }
+  return l==s_len;
+}
+
+uint wt_coder_huff::size() {
+  return 2*sizeof(uint)+sizeof(wt_coder_huff)+hc->size()+(hc->max_length()/W+1)*sizeof(uint);
+}
+
+uint wt_coder_huff::save(FILE * fp) {
+  uint wr = WT_CODER_HUFF_HDR;
+  wr = fwrite(&wr,sizeof(uint),1,fp);
+  if(wr!=1) return 1;
+  if(hc->save(fp)) return 1;
+  //if(am->save(fp)) return 1;
+  return 0;
+}
+
+wt_coder_huff * wt_coder_huff::load(FILE *fp) {
+  uint rd;
+  if(fread(&rd,sizeof(uint),1,fp)!=1) return NULL;
+  if(rd!=WT_CODER_HUFF_HDR) return NULL;
+  wt_coder_huff * ret = new wt_coder_huff();
+  ret->hc = huffman_codes::load(fp);
+  ret->buffer = new uint[ret->hc->max_length()/W+1]; 
+  ret->s_len = 0; ret->last_symbol = (uint)-1;
+  return ret;
+}