#!/bin/bash

# script showing use of nnet3_to_dot.py
# Copyright 2015  Johns Hopkins University (Author: Vijayaditya Peddinti).

# Begin configuration section.
component_attributes="name,type"
node_prefixes=""
info_bin=nnet3-am-info
echo "$0 $@"  # Print the command line for logging

[ -f ./path.sh ] && . ./path.sh; # source the path.
. parse_options.sh || exit 1;

if [ $# != 3 ]; then
  echo "Usage: $0 [opts] <nnet3-mdl-file> <output-dot-file> <output-png-file>"
  echo " e.g.: $0 exp/sdm1/nnet3/lstm_sp/0.mdl lstm.dot lstm.png"
  echo ""
  echo "Main options (for others, see top of script file)"
  echo "  --info-bin <nnet3-am-info|nnet3-info>        # Name of the binary to generate the nnet3 file"
  echo "  --component-attributes <string|name,type>     # attributes to be printed in nnet3 components"
  echo "  --node-prefixes <string|Lstm1,Lstm2>          # list of prefixes. Nnet3 components/component-nodes with the same prefix"
  echo "                                                # will be clustered together in the dot-graph"


  exit 1;
fi

model=$1
dot_file=$2
output_file=$3

attr=${node_prefixes:+ --node-prefixes "$node_prefixes"}
$info_bin $model | \
  steps/nnet3/dot/nnet3_to_dot.py \
    --component-attributes "$component_attributes" \
    $attr $dot_file
echo "Generated the dot file $dot_file"

command -v dot >/dev/null 2>&1 || { echo >&2 "This script requires dot but it's not installed. Please compile $dot_file with dot"; exit 1; }
dot -Tpdf $dot_file -o $output_file