draw.h 2.74 KB
// See www.openfst.org for extensive documentation on this weighted
// finite-state transducer library.

#ifndef FST_SCRIPT_DRAW_H_
#define FST_SCRIPT_DRAW_H_

#include <ostream>

#include <fst/script/draw-impl.h>
#include <fst/script/fst-class.h>

namespace fst {
namespace script {

// Note: it is safe to pass these strings as references because
// this struct is only used to pass them deeper in the call graph.
// Be sure you understand why this is so before using this struct
// for anything else!
struct FstDrawerArgs {
  const FstClass &fst;
  const SymbolTable *isyms;
  const SymbolTable *osyms;
  const SymbolTable *ssyms;
  const bool accep;
  const string &title;
  const float width;
  const float height;
  const bool portrait;
  const bool vertical;
  const float ranksep;
  const float nodesep;
  const int fontsize;
  const int precision;
  const string &float_format;  // NOLINT
  const bool show_weight_one;
  std::ostream *ostrm;
  const string &dest;

  FstDrawerArgs(const FstClass &fst, const SymbolTable *isyms,
                const SymbolTable *osyms, const SymbolTable *ssyms, bool accep,
                const string &title, float width, float height, bool portrait,
                bool vertical, float ranksep, float nodesep, int fontsize,
                int precision, const string &float_format,
                bool show_weight_one, std::ostream *ostrm,  const string &dest)
      : fst(fst),
        isyms(isyms),
        osyms(osyms),
        ssyms(ssyms),
        accep(accep),
        title(title),
        width(width),
        height(height),
        portrait(portrait),
        vertical(vertical),
        ranksep(ranksep),
        nodesep(nodesep),
        fontsize(fontsize),
        precision(precision),
        float_format(float_format),
        show_weight_one(show_weight_one),
        ostrm(ostrm),
        dest(dest) {}
};

template <class Arc>
void DrawFst(FstDrawerArgs *args) {
  const Fst<Arc> &fst = *(args->fst.GetFst<Arc>());
  FstDrawer<Arc> fstdrawer(fst, args->isyms, args->osyms, args->ssyms,
      args->accep, args->title, args->width, args->height, args->portrait,
      args->vertical, args->ranksep, args->nodesep, args->fontsize,
      args->precision, args->float_format, args->show_weight_one);
  fstdrawer.Draw(args->ostrm, args->dest);
}

void DrawFst(const FstClass &fst, const SymbolTable *isyms,
             const SymbolTable *osyms, const SymbolTable *ssyms, bool accep,
             const string &title, float width, float height, bool portrait,
             bool vertical, float ranksep, float nodesep, int fontsize,
             int precision, const string &float_format, bool show_weight_one,
             std::ostream *ostrm, const string &dest);

}  // namespace script
}  // namespace fst

#endif  // FST_SCRIPT_DRAW_H_