lattice-copy-backoff.cc File Reference

#include "base/kaldi-common.h"
#include "util/common-utils.h"
#include "fstext/fstext-lib.h"
#include "lat/kaldi-lattice.h"

Include dependency graph for lattice-copy-backoff.cc:

Go to the source code of this file.

Functions
int	main (int argc, char *argv[])

Function Documentation

main()

int main	(	int	argc,
		char *	argv[]
	)

Definition at line 26 of file lattice-copy-backoff.cc.

References SequentialTableReader< Holder >::Done(), ParseOptions::GetArg(), RandomAccessTableReader< Holder >::HasKey(), KALDI_LOG, KALDI_VLOG, SequentialTableReader< Holder >::Key(), SequentialTableReader< Holder >::Next(), ParseOptions::NumArgs(), ParseOptions::PrintUsage(), ParseOptions::Read(), RandomAccessTableReader< Holder >::Value(), SequentialTableReader< Holder >::Value(), and TableWriter< Holder >::Write().

                                 {
  using namespace kaldi;
  typedef kaldi::int32 int32;
  try {
    const char *usage =
        "Copy a table of lattices (1st argument), but for any keys that appear\n"
        "in the table from the 2nd argument, use the one from the 2nd argument.\n"
        "If the sets of keys are identical, this is equivalent to copying the 2nd\n"
        "table.  Note: the arguments are in this order due to the convention that\n"
        "sequential access is always over the 1st argument.\n"
        "\n"
        "Usage: lattice-copy-backoff [options] <lat-rspecifier1> "
        "<lat-rspecifier2> <lat-wspecifier>\n"
        " e.g.: lattice-copy-backoff ark:bad_but_complete.lat "
        "ark:good_but_incomplete.lat ark:out.lat\n";
      
    ParseOptions po(usage);
    
    po.Read(argc, argv);

    if (po.NumArgs() != 3) {
      po.PrintUsage();
      exit(1);
    }

    std::string lats_rspecifier1 = po.GetArg(1),
        lats_rspecifier2 = po.GetArg(2),
        lats_wspecifier = po.GetArg(3);
    
    int32 n_done = 0, n_backed_off = 0;
    
    SequentialCompactLatticeReader lattice_reader1(lats_rspecifier1);
    RandomAccessCompactLatticeReader lattice_reader2(lats_rspecifier2);
    CompactLatticeWriter lattice_writer(lats_wspecifier);
    for (; !lattice_reader1.Done(); lattice_reader1.Next(), n_done++) {
      const std::string &key = lattice_reader1.Key();
      if (lattice_reader2.HasKey(key)) {
        lattice_writer.Write(key, lattice_reader2.Value(key));
      } else {
        lattice_writer.Write(key, lattice_reader1.Value());
        KALDI_VLOG(1) << "Backed off to 1st archive for key " << key;
        n_backed_off++;
      }
    }
    KALDI_LOG << "Done copying " << n_done << " lattices; backed off to 1st "
              << "archive for " << n_backed_off << " of those.";
    return (n_done != 0 ? 0 : 1);
  } catch(const std::exception &e) {
    std::cerr << e.what();
    return -1;
  }
}