1
2// Licensed under the Apache License, Version 2.0 (the "License");
3// you may not use this file except in compliance with the License.
4// You may obtain a copy of the License at
5//
6//     http://www.apache.org/licenses/LICENSE-2.0
7//
8// Unless required by applicable law or agreed to in writing, software
9// distributed under the License is distributed on an "AS IS" BASIS,
10// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
11// See the License for the specific language governing permissions and
12// limitations under the License.
13//
14// Copyright 2005-2010 Google, Inc.
15// Author: jpr@google.com (Jake Ratkiewicz)
16
17#ifndef FST_SCRIPT_DRAW_H_
18#define FST_SCRIPT_DRAW_H_
19
20#include <fst/script/arg-packs.h>
21#include <fst/script/fst-class.h>
22#include <fst/script/draw-impl.h>
23#include <iostream>
24#include <fstream>
25#include <sstream>
26
27namespace fst {
28namespace script {
29
30// Note: it is safe to pass these strings as references because
31// this struct is only used to pass them deeper in the call graph.
32// Be sure you understand why this is so before using this struct
33// for anything else!
34struct FstDrawerArgs {
35  const FstClass &fst;
36  const SymbolTable *isyms;
37  const SymbolTable *osyms;
38  const SymbolTable *ssyms;
39  const bool accep;
40  const string& title;
41  const float width;
42  const float height;
43  const bool portrait;
44  const bool vertical;
45  const float ranksep;
46  const float nodesep;
47  const int fontsize;
48  const int precision;
49  const bool show_weight_one;
50  ostream *ostrm;
51  const string &dest;
52
53  FstDrawerArgs(const FstClass &fst,
54                const SymbolTable *isyms,
55                const SymbolTable *osyms,
56                const SymbolTable *ssyms,
57                bool accep,
58                const string &title,
59                float width,
60                float height,
61                bool portrait,
62                bool vertical,
63                float ranksep,
64                float nodesep,
65                int fontsize,
66                int precision,
67                bool show_weight_one,
68                ostream *ostrm,
69                const string &dest) :
70      fst(fst), isyms(isyms), osyms(osyms), ssyms(ssyms), accep(accep),
71      title(title), width(width), height(height), portrait(portrait),
72      vertical(vertical), ranksep(ranksep), nodesep(nodesep),
73      fontsize(fontsize), precision(precision),
74      show_weight_one(show_weight_one), ostrm(ostrm), dest(dest) { }
75};
76
77
78template<class Arc>
79void DrawFst(FstDrawerArgs *args) {
80  const Fst<Arc> &fst = *(args->fst.GetFst<Arc>());
81
82  FstDrawer<Arc> fstdrawer(fst, args->isyms, args->osyms, args->ssyms,
83                           args->accep, args->title, args->width,
84                           args->height, args->portrait,
85                           args->vertical, args->ranksep,
86                           args->nodesep, args->fontsize,
87                           args->precision, args->show_weight_one);
88  fstdrawer.Draw(args->ostrm, args->dest);
89}
90
91void DrawFst(const FstClass &fst,
92             const SymbolTable *isyms,
93             const SymbolTable *osyms,
94             const SymbolTable *ssyms,
95             bool accep,
96             const string &title,
97             float width,
98             float height,
99             bool portrait,
100             bool vertical,
101             float ranksep,
102             float nodesep,
103             int fontsize,
104             int precision,
105             bool show_weight_one,
106             ostream *ostrm,
107             const string &dest);
108
109}  // namespace script
110}  // namespace fst
111
112
113
114#endif  // FST_SCRIPT_DRAW_H_
115