Home | History | Annotate | Download | only in script 
      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 (at) 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 
     27 namespace fst {
     28 namespace 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!
     34 struct 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 
     78 template<class Arc>
     79 void 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 
     91 void 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