// Copyright 2005-2024 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the 'License');
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an 'AS IS' BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
// 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 <string>

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

namespace fst {
namespace script {

// Note: it is safe to pass these `string_view`s 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 FstDrawArgs {
  const FstClass& fst;
  const SymbolTable* isyms;
  const SymbolTable* osyms;
  const SymbolTable* ssyms;
  const bool accep;
  const std::string_view 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 std::string_view float_format;
  const bool show_weight_one;
  const std::string_view format;
  std::ostream& ostrm;
  const std::string_view dest;
};

template <class Arc>
void Draw(FstDrawArgs* 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, args->format);
  fstdrawer.Draw(args->ostrm, args->dest);
}

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

}  // namespace script
}  // namespace fst

#endif  // FST_SCRIPT_DRAW_H_
