1 files changed, 125 insertions, 0 deletions
diff --git a/gfsm/gfsm/src/programs/gfsmviterbi_main.c b/gfsm/gfsm/src/programs/gfsmviterbi_main.c
new file mode 100644
index 0000000..33bb01a
--- /dev/null
+++ b/gfsm/gfsm/src/programs/gfsmviterbi_main.c
@@ -0,0 +1,125 @@
+/*
+   gfsm-utils : finite state automaton utilities
+   Copyright (C) 2005 by Bryan Jurish <moocow@ling.uni-potsdam.de>
+
+   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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+*/
+
+#ifdef HAVE_CONFIG_H
+# include <config.h>
+#endif
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <errno.h>
+#include <ctype.h>
+
+#include <gfsm.h>
+
+#include "gfsmviterbi_cmdparser.h"
+
+/*--------------------------------------------------------------------------
+ * Globals
+ *--------------------------------------------------------------------------*/
+char *progname = "gfsmviterbi";
+
+//-- options
+struct gengetopt_args_info args;
+
+//-- files
+const char *fstfilename = "-";
+const char *outfilename = "-";
+
+//-- global structs
+gfsmAutomaton *fst = NULL;
+gfsmError     *err = NULL;
+
+/*--------------------------------------------------------------------------
+ * Option Processing
+ *--------------------------------------------------------------------------*/
+void get_my_options(int argc, char **argv)
+{
+  if (cmdline_parser(argc, argv, &args) != 0)
+    exit(1);
+
+  //-- load environmental defaults
+  //cmdline_parser_envdefaults(&args);
+
+  //-- filenames
+  if (args.fst_given) fstfilename = args.fst_arg;
+  outfilename = args.output_arg;
+
+  //-- load FST
+  fst = gfsm_automaton_new();
+  if (!gfsm_automaton_load_bin_filename(fst, fstfilename, &err)) {
+    g_printerr("%s: load failed for FST file '%s': %s\n", progname, fstfilename, err->message);
+    exit(255);
+  }
+}
+
+/*--------------------------------------------------------------------------
+ * viterbi_labels()
+ */
+gfsmAutomaton *viterbi_labels(gfsmAutomaton *fst, int argc, char **argv)
+{
+  gfsmLabelVector *vec = g_ptr_array_sized_new(argc);
+  char            *s=NULL, *tail=NULL;
+  gfsmLabelVal     lab;
+  gfsmAutomaton   *trellis = NULL;
+  int              i;
+
+  //-- fill input vector
+  for (i=0; i < argc; i++) {
+    for (s=argv[i], lab=strtol(s,&tail,0); s != tail; s=tail, lab=strtol(s,&tail,0)) {
+      g_ptr_array_add(vec, (gpointer)lab);
+    }
+  }
+
+  //-- actual viterbi lookup
+  trellis = gfsm_automaton_lookup_viterbi(fst, vec, trellis);
+
+  //-- cleanup
+  g_ptr_array_free(vec,TRUE);
+
+  return trellis;
+}
+
+/*--------------------------------------------------------------------------
+ * MAIN
+ *--------------------------------------------------------------------------*/
+int main (int argc, char **argv)
+{
+  gfsmAutomaton *trellis;
+
+  GFSM_INIT
+  get_my_options(argc,argv);
+
+  //-- process input
+  trellis = viterbi_labels(fst, args.inputs_num, args.inputs);
+
+  //-- save output
+  if (!gfsm_automaton_save_bin_filename(trellis,outfilename,args.compress_arg,&err)) {
+    g_printerr("%s: store failed to '%s': %s\n", progname, outfilename, err->message);
+    exit(4);
+  }
+
+  //-- cleanup
+  if (fst)     gfsm_automaton_free(fst);
+  if (trellis) gfsm_automaton_free(trellis);
+
+  GFSM_FINISH
+  return 0;
+}