Loading...
Searching...
No Matches
wfst_run_main.cc
/*************************************************************************/
/* */
/* Centre for Speech Technology Research */
/* University of Edinburgh, UK */
/* Copyright (c) 1996,1997 */
/* All Rights Reserved. */
/* */
/* Permission is hereby granted, free of charge, to use and distribute */
/* this software and its documentation without restriction, including */
/* without limitation the rights to use, copy, modify, merge, publish, */
/* distribute, sublicense, and/or sell copies of this work, and to */
/* permit persons to whom this work is furnished to do so, subject to */
/* the following conditions: */
/* 1. The code must retain the above copyright notice, this list of */
/* conditions and the following disclaimer. */
/* 2. Any modifications must be clearly marked as such. */
/* 3. Original authors' names are not deleted. */
/* 4. The authors' names are not used to endorse or promote products */
/* derived from this software without specific prior written */
/* permission. */
/* */
/* THE UNIVERSITY OF EDINBURGH AND THE CONTRIBUTORS TO THIS WORK */
/* DISCLAIM ALL WARRANTIES WITH REGARD TO THIS SOFTWARE, INCLUDING */
/* ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS, IN NO EVENT */
/* SHALL THE UNIVERSITY OF EDINBURGH NOR THE CONTRIBUTORS BE LIABLE */
/* FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES */
/* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN */
/* AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, */
/* ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF */
/* THIS SOFTWARE. */
/* */
/*************************************************************************/
/* Author : Alan W Black */
/* Date : December 1997 */
/*-----------------------------------------------------------------------*/
/* Run a WFST on some data, either as a recognizer or as a transducer */
/* */
/*=======================================================================*/
#include <cstdlib>
#include <cstdio>
#include <iostream>
#include <fstream>
#include <cstring>
#include "EST.h"
#include "EST_simplestats.h"
#include "EST_WFST.h"
static int wfst_run_main(int argc, char **argv);
/** @name <command>wfst_run</command> <emphasis>Run a weighted finite-state transducer</emphasis>
@id wfst-run-manual
* @toc
*/
//@{
/**@name Synopsis
*/
//@{
//@synopsis
/**
This program runs a WFST on some given data. It works in either
recognize mode where both inputs and output are specified, but also
in transduction mode where an input is transduced to the output.
*/
//@}
/**@name OPTIONS
*/
//@{
//@options
//@}
int main(int argc, char **argv)
{
wfst_run_main(argc,argv);
exit(0);
return 0;
}
static int wfst_run_main(int argc, char **argv)
{
// recognize/transduce
EST_Option al;
EST_StrList files;
EST_Litem *f;
EST_String wfstfile;
FILE *ofd;
int r;
EST_SuffStats R;
float sumlogp=0,isumlogp;
float count=0,icount;
parse_command_line
(argc, argv,
EST_String("[WFSTFILE] [input file0] ... [-o output file]\n")+
"Summary: Recognize/transduce using a WFST on data\n"+
"-wfst <ifile> The WFST to use\n"+
"-transduce Transduce input to output (default)\n"+
"-recog Recognize input consists of pairs\n"+
"-cumulate_into <ofile>\n"+
" Cumulate transitions to give new weights\n"+
" save new WFST into ofile\n"+
"-itype <string> char or token\n"+
"-quiet No extraneous messages\n"+
"-perplexity Calculate perplexity on given data set\n"+
"-heap <int> {210000}\n"+
" Set size of Lisp heap, needed for large wfsts\n"+
"-o <ofile> Output file for transduced forms\n",
files, al);
if (al.present("-o"))
{
if ((ofd=fopen(al.val("-o"),"w")) == NULL)
EST_error("can't open output file for writing \"%s\"",
(const char *)al.val("-o"));
}
else
ofd = stdout;
if (al.present("-wfst"))
wfstfile = al.val("-wfst");
else
EST_error("no WFST specified");
siod_init(al.ival("-heap"));
EST_WFST wfst;
EST_TokenStream ts;
if (wfst.load(wfstfile) != format_ok)
EST_error("failed to read WFST from \"%s\"",
(const char *)wfstfile);
if (al.present("-cumulate_into"))
wfst.start_cumulate();
for (f=files.head(); f != 0; f=f->next())
{
if (files(f) == "-")
ts.open(stdin,FALSE);
else
if (ts.open(files(f)) != 0)
EST_error("failed to read WFST data file from \"%s\"",
(const char *)files(f));
// Not the best way to input things but will do the the present
while(!ts.eof())
{
EST_StrList ostrs,istrs;
do
istrs.append(ts.get());
while((!ts.eof()) && (!ts.eoln()));
if (al.present("-recog"))
{
if (al.present("-perplexity"))
{
r = recognize_for_perplexity(wfst,istrs,
al.present("-quiet"),
icount,
isumlogp);
if (r)
{
count += icount;
sumlogp += isumlogp;
}
}
else
r = recognize(wfst,istrs,al.present("-quiet"));
}
else
{
r = transduce(wfst,istrs,ostrs);
if (r)
{
cout << ostrs;
cout << endl;
}
}
R += r;
if (!al.present("-quiet"))
{
if (r)
cout << "OK." << endl;
else
cout << "failed." << endl;
}
}
ts.close();
}
if (al.present("-cumulate_into"))
{
wfst.stop_cumulate();
if (wfst.save(al.val("-cumulate_into")) != write_ok)
EST_error("failed to write cumulated WFST to \"%s\"",
(const char *)al.val("-cumulate_into"));
}
printf("total %d OK %f%% failed %f%%\n",
(int)R.samples(),R.mean()*100,(1-R.mean())*100);
if (al.present("-perplexity"))
{
printf("perplexity is %f\n", pow(float(2.0),float(-1.0 * (sumlogp/count))));
}
if (ofd != stdout)
fclose(ofd);
if (R.mean() == 1) // true is *all* files were recognized
return 0;
else
return -1;
}
EST_Option
Definition EST_Option.h:50
EST_Option::ival
int ival(const EST_String &rkey, int m=1) const
Definition EST_Option.cc:76
EST_String
Definition EST_String.h:70
EST_TKVL::val
const V & val(const K &rkey, bool m=0) const
return value according to key (const)
Definition EST_TKVL.cc:145
EST_TKVL::present
const int present(const K &rkey) const
Returns true if key is present.
Definition EST_TKVL.cc:222
EST_TList::append
void append(const T &item)
add item onto end of list
Definition EST_TList.h:191
EST_TokenStream
Definition EST_Token.h:235
EST_TokenStream::eof
int eof()
end of file
Definition EST_Token.h:356
EST_TokenStream::eoln
int eoln()
end of line
Definition EST_Token.cc:818
EST_TokenStream::close
void close(void)
Close stream.
Definition EST_Token.cc:406
EST_TokenStream::open
int open(const EST_String &filename)
open a \Ref{EST_TokenStream} for a file.
Definition EST_Token.cc:200
EST_TokenStream::get
EST_TokenStream & get(EST_Token &t)
get next token in stream
Definition EST_Token.cc:486
EST_WFST
Definition EST_WFST.h:150
EST_WFST::start_cumulate
void start_cumulate()
Clear and start cumulation.
Definition EST_WFST.cc:670
EST_WFST::save
EST_write_status save(const EST_String &filename, const EST_String type="ascii")
?
Definition EST_WFST.cc:349
EST_WFST::stop_cumulate
void stop_cumulate()
Stop cumulation and calculate probabilities on transitions.
Definition EST_WFST.cc:685
EST_WFST::load
EST_read_status load(const EST_String &filename)
?
Definition EST_WFST.cc:508