src/slm/tslmendian/slm_file.h

   1 // -*- mode: c++ -*-
   2 /*
   3  * Copyright (c) 2009 Kov Chai <tchaikov@gmail.com>
   4  *
   5  * The contents of this file are subject to the terms of either the GNU Lesser
   6  * General Public License Version 2.1 only ("LGPL") or the Common Development and
   7  * Distribution License ("CDDL")(collectively, the "License"). You may not use this
   8  * file except in compliance with the License. You can obtain a copy of the CDDL at
   9  * http://www.opensource.org/licenses/cddl1.php and a copy of the LGPLv2.1 at
  10  * http://www.opensource.org/licenses/lgpl-license.php. See the License for the
  11  * specific language governing permissions and limitations under the License. When
  12  * distributing the software, include this License Header Notice in each file and
  13  * include the full text of the License in the License file as well as the
  14  * following notice:
  15  *
  16  * NOTICE PURSUANT TO SECTION 9 OF THE COMMON DEVELOPMENT AND DISTRIBUTION LICENSE
  17  * (CDDL)
  18  * For Covered Software in this distribution, this License shall be governed by the
  19  * laws of the State of California (excluding conflict-of-law provisions).
  20  * Any litigation relating to this License shall be subject to the jurisdiction of
  21  * the Federal Courts of the Northern District of California and the state courts
  22  * of the State of California, with venue lying in Santa Clara County, California.
  23  *
  24  * Contributor(s):
  25  *
  26  * If you wish your version of this file to be governed by only the CDDL or only
  27  * the LGPL Version 2.1, indicate your decision by adding "[Contributor]" elects to
  28  * include this software in this distribution under the [CDDL or LGPL Version 2.1]
  29  * license." If you don't indicate a single choice of license, a recipient has the
  30  * option to distribute your version of this file under either the CDDL or the LGPL
  31  * Version 2.1, or to extend the choice of license to its licensees as provided
  32  * above. However, if you add LGPL Version 2.1 code and therefore, elected the LGPL
  33  * Version 2 license, then the option applies only if the new code is made subject
  34  * to such option by the copyright holder.
  35  */
  36
  37 /*
  38  * convert the threaded lm binary file from big-endian to small-endian or vice versa
  39  */
  40 #ifndef SLM_FILE_H
  41 #define SLM_FILE_H
  42
  43 #include <stdint.h>
  44
  45 #include "../slm.h"
  46
  47 // TODO: may need consolidate this class with CThreadSlm
  48 class CThreadSlmFile
  49 {
  50 public:
  51     CThreadSlmFile();
  52     ~CThreadSlmFile();
  53     /**
  54      * read from a lm_sc.t3g file and convert it to host byte order
  55      * @param fname the filenam
  56      * @return true if the given file is read successfully, false otherwise
  57      */
  58     bool load(const char* fname);
  59     /**
  60      * write the language model to a given file
  61      * @param fname filenam of output file
  62      * @param endian should be BIG_ENDIAN or LITTLE_ENDIAN
  63      * @return the number of bytes wrote
  64      */
  65     size_t save(const char* fname, int endian);
  66     /**
  67      * return the size of the whole file
  68      * @return size in byte
  69      */
  70     size_t size() const;
  71     int getEndian() const;
  72     static int getHostEndian();
  73
  74 private:
  75     CThreadSlmFile(const CThreadSlmFile&);
  76
  77 private:
  78     size_t m_bufLen;
  79     char*     m_buf;
  80     uint32_t m_N;
  81     uint32_t m_usingLogPr;
  82     uint32_t* m_levelSizes;
  83     float*    m_prTable;
  84     float*    m_bowTable;
  85
  86     size_t m_nnode;
  87     CThreadSlm::TNode* m_nodes;
  88     size_t m_nleaf;
  89     CThreadSlm::TLeaf* m_leafs;
  90 };
  91
  92 #endif //SLM_FILE_H