/**
   * Creates the grammar from the language model. This Grammar contains one word per grammar node.
   * Each word (and grammar node) is connected to all other words with the given probability
   *
   * @return the initial grammar node
   */
  @Override
  protected GrammarNode createGrammar() throws IOException {
    if (lattice == null) {
      return createGrammarNode("<s>");
    }

    lattice.removeFillers();

    GrammarNode firstNode = null;
    HashMap<Node, GrammarNode> nodeMap = new HashMap<Node, GrammarNode>();
    for (Node n : lattice.getNodes()) {
      String word = n.getWord().toString();
      GrammarNode node = createGrammarNode(word);
      if (n.equals(lattice.getInitialNode())) firstNode = node;
      if (n.equals(lattice.getTerminalNode())) node.setFinalNode(true);
      nodeMap.put(n, node);
    }
    if (firstNode == null) {
      throw new Error("No lattice start found");
    }

    for (Edge e : lattice.getEdges()) {
      float logProbability = (float) e.getLMScore();
      GrammarNode prevNode = nodeMap.get(e.getFromNode());
      GrammarNode toNode = nodeMap.get(e.getToNode());
      prevNode.add(toNode, logProbability);
    }

    return firstNode;
  }
Example #2
0
  /**
   * Main method for running the LatticeCompTest demo.
   *
   * @throws IOException
   * @throws UnsupportedAudioFileException
   */
  @Test
  public void testLatticeComp() throws UnsupportedAudioFileException, IOException {

    URL audioFileURL = new File("src/test/edu/cmu/sphinx/result/test/green.wav").toURI().toURL();
    URL configURL = new File("src/test/edu/cmu/sphinx/result/test/config.xml").toURI().toURL();

    ConfigurationManager cm = new ConfigurationManager(configURL);

    Recognizer recognizer = (Recognizer) cm.lookup("recognizer");

    StreamDataSource reader = (StreamDataSource) cm.lookup("streamDataSource");

    AudioInputStream ais = AudioSystem.getAudioInputStream(audioFileURL);

    /* set the stream data source to read from the audio file */
    reader.setInputStream(ais, audioFileURL.getFile());

    /* allocate the resource necessary for the recognizer */
    recognizer.allocate();

    /* decode the audio file */
    Result result = recognizer.recognize();

    /* print out the results */
    Lattice lattice = new Lattice(result);
    lattice.dumpAISee("lattice.gdl", "lattice");

    recognizer.deallocate();

    cm = new ConfigurationManager(configURL);

    ConfigurationManagerUtils.setProperty(cm, "keepAllTokens", "true");

    recognizer = (Recognizer) cm.lookup("recognizer");
    recognizer.allocate();

    reader = (StreamDataSource) cm.lookup("streamDataSource");
    reader.setInputStream(AudioSystem.getAudioInputStream(audioFileURL), audioFileURL.getFile());

    Result allResult = recognizer.recognize();

    Lattice allLattice = new Lattice(allResult);
    allLattice.dumpAISee("allLattice.gdl", "All Lattice");

    assertTrue(lattice.isEquivalent(allLattice));
  }