122 lines
3.9 KiB
OpenEdge ABL
122 lines
3.9 KiB
OpenEdge ABL
/* -*- c-basic-offset: 4; indent-tabs-mode: nil -*- */
|
|
/* ====================================================================
|
|
* Copyright (c) 2013 Carnegie Mellon University. All rights
|
|
* reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions
|
|
* are met:
|
|
*
|
|
* 1. Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
*
|
|
* 2. Redistributions in binary form must reproduce the above copyright
|
|
* notice, this list of conditions and the following disclaimer in
|
|
* the documentation and/or other materials provided with the
|
|
* distribution.
|
|
*
|
|
* This work was supported in part by funding from the Defense Advanced
|
|
* Research Projects Agency and the National Science Foundation of the
|
|
* United States of America, and the CMU Sphinx Speech Consortium.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY CARNEGIE MELLON UNIVERSITY ``AS IS'' AND
|
|
* ANY EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
|
|
* THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
|
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL CARNEGIE MELLON UNIVERSITY
|
|
* NOR ITS EMPLOYEES BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*
|
|
* ====================================================================
|
|
*
|
|
*/
|
|
|
|
|
|
%extend NGramModel {
|
|
|
|
static NGramModel *fromIter(ngram_model_set_iter_t *itor) {
|
|
const char *name;
|
|
return ngram_model_set_iter_model(itor, &name);
|
|
}
|
|
|
|
NGramModel(const char *path) {
|
|
logmath_t *lmath = logmath_init(1.0001, 0, 0);
|
|
ngram_model_t * model = ngram_model_read(NULL, path, NGRAM_AUTO, lmath);
|
|
logmath_free(lmath);
|
|
return model;
|
|
}
|
|
|
|
NGramModel(Config *config, LogMath *logmath, const char *path) {
|
|
return ngram_model_read(config, path, NGRAM_AUTO, logmath);
|
|
}
|
|
|
|
~NGramModel() {
|
|
ngram_model_free($self);
|
|
}
|
|
|
|
void write(const char *path, ngram_file_type_t ftype, int *errcode) {
|
|
*errcode = ngram_model_write($self, path, ftype);
|
|
}
|
|
|
|
ngram_file_type_t str_to_type(const char *str) {
|
|
return ngram_str_to_type(str);
|
|
}
|
|
|
|
const char * type_to_str(int type) {
|
|
return ngram_type_to_str(type);
|
|
}
|
|
|
|
void casefold(int kase, int *errcode) {
|
|
*errcode = ngram_model_casefold($self, kase);
|
|
}
|
|
|
|
int32 size() {
|
|
return ngram_model_get_size($self);
|
|
}
|
|
|
|
int32 add_word(const char *word, float32 weight) {
|
|
return ngram_model_add_word($self, word, weight);
|
|
}
|
|
|
|
int32 prob(size_t n, const char * const*ptr) {
|
|
return ngram_prob($self, ptr, n);
|
|
}
|
|
}
|
|
|
|
// TODO: shares ptr type with NGramModel, docstrings are not generated
|
|
%extend NGramModelSet {
|
|
NGramModelSet(Config *config, LogMath *logmath, const char *path) {
|
|
return ngram_model_set_read(config, path, logmath);
|
|
}
|
|
|
|
~NGramModelSet() {
|
|
ngram_model_free($self);
|
|
}
|
|
|
|
int32 count() {
|
|
return ngram_model_set_count($self);
|
|
}
|
|
|
|
NGramModel * add(
|
|
NGramModel *model, const char *name, float weight, bool reuse_widmap) {
|
|
return ngram_model_set_add($self, model, name, weight, reuse_widmap);
|
|
}
|
|
|
|
NGramModel * select(const char *name) {
|
|
return ngram_model_set_select($self, name);
|
|
}
|
|
|
|
NGramModel * lookup(const char *name) {
|
|
return ngram_model_set_lookup($self, name);
|
|
}
|
|
|
|
const char * current() {
|
|
return ngram_model_set_current($self);
|
|
}
|
|
}
|
|
|