1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
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;
}
|