src/slm/ids2ngram/idngram.h

   1 // -*- mode: c++ -*-
   2 /*
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
   4  *
   5  * Copyright (c) 2007 Sun Microsystems, Inc. All Rights Reserved.
   6  *
   7  * The contents of this file are subject to the terms of either the GNU Lesser
   8  * General Public License Version 2.1 only ("LGPL") or the Common Development and
   9  * Distribution License ("CDDL")(collectively, the "License"). You may not use this
  10  * file except in compliance with the License. You can obtain a copy of the CDDL at
  11  * http://www.opensource.org/licenses/cddl1.php and a copy of the LGPLv2.1 at
  12  * http://www.opensource.org/licenses/lgpl-license.php. See the License for the
  13  * specific language governing permissions and limitations under the License. When
  14  * distributing the software, include this License Header Notice in each file and
  15  * include the full text of the License in the License file as well as the
  16  * following notice:
  17  *
  18  * NOTICE PURSUANT TO SECTION 9 OF THE COMMON DEVELOPMENT AND DISTRIBUTION LICENSE
  19  * (CDDL)
  20  * For Covered Software in this distribution, this License shall be governed by the
  21  * laws of the State of California (excluding conflict-of-law provisions).
  22  * Any litigation relating to this License shall be subject to the jurisdiction of
  23  * the Federal Courts of the Northern District of California and the state courts
  24  * of the State of California, with venue lying in Santa Clara County, California.
  25  *
  26  * Contributor(s):
  27  *
  28  * If you wish your version of this file to be governed by only the CDDL or only
  29  * the LGPL Version 2.1, indicate your decision by adding "[Contributor]" elects to
  30  * include this software in this distribution under the [CDDL or LGPL Version 2.1]
  31  * license." If you don't indicate a single choice of license, a recipient has the
  32  * option to distribute your version of this file under either the CDDL or the LGPL
  33  * Version 2.1, or to extend the choice of license to its licensees as provided
  34  * above. However, if you add LGPL Version 2.1 code and therefore, elected the LGPL
  35  * Version 2 license, then the option applies only if the new code is made subject
  36  * to such option by the copyright holder.
  37  */
  38
  39 #ifndef _SIM_IDNGRAM_H
  40 #define _SIM_IDNGRAM_H
  41
  42 #include "../../portability.h"
  43 #include "../sim_fmerge.h"
  44
  45 template<int N>
  46 class CSIM_Idngram {
  47 public:
  48     TSIMWordId ids[N];
  49 public:
  50     CSIM_Idngram()
  51     { for (int i = 0; i < N; ++i) ids[i] = 0; }
  52     CSIM_Idngram & operator=(const CSIM_Idngram& r){
  53         for (int i = 0; i < N; ++i)
  54             ids[i] = r.ids[i];
  55         return *this;
  56     }
  57     bool operator<(const CSIM_Idngram& r) const {
  58         for (int i = 0; i < N; ++i) {
  59             if (ids[i] < r.ids[i])
  60                 return true;
  61             else if (ids[i] > r.ids[i])
  62                 return false;
  63         }
  64         return false;
  65     }
  66
  67     bool operator>(const CSIM_Idngram& r) const {
  68         for (int i = 0; i < N; ++i) {
  69             if (ids[i] > r.ids[i])
  70                 return true;
  71             else if (ids[i] < r.ids[i])
  72                 return false;
  73         }
  74         return false;
  75     }
  76
  77     bool operator==(const CSIM_Idngram& r) const {
  78         for (int i = 0; i < N; ++i)
  79             if (ids[i] != r.ids[i]) return false;
  80         return true;
  81     }
  82 };
  83
  84
  85 template<int N>
  86 class CSIM_IdngramFreq : public CSIM_Idngram<N> {
  87 public:
  88     CSIM_IdngramFreq()
  89         : CSIM_Idngram<N>(), freq(0) { }
  90     CSIM_IdngramFreq& operator=(const CSIM_IdngramFreq& r){
  91         for (int i = 0; i < N; ++i)
  92             this->ids[i] = r.ids[i];
  93         freq = r.freq;
  94         return *this;
  95     }
  96     bool read(FILE *fp, size_t& start_offset, size_t last_offset){
  97         if (start_offset + size() <= last_offset) {
  98             fseek(fp, start_offset, SEEK_SET);
  99             if (fread(this->ids, sizeof(TSIMWordId), N,
 100                       fp) == N &&
 101                 fread(&freq, sizeof(unsigned int), 1, fp) == 1) {
 102                 start_offset += size();
 103                 return true;
 104             }
 105         }
 106         return false;
 107     }
 108
 109 protected:
 110     size_t size() { return N * sizeof(TSIMWordId) + sizeof(unsigned int); }
 111
 112 public:
 113     unsigned int freq;
 114 };
 115
 116 #endif