LinguistStats.java

/*
 * Copyright 1999-2003 Carnegie Mellon University.  
 * Portions Copyright 2003 Sun Microsystems, Inc.  
 * Portions Copyright 2003 Mitsubishi Electric Research Laboratories.
 * All Rights Reserved.  Use is subject to license terms.
 * 
 * See the file "license.terms" for information on usage and
 * redistribution of this file, and for a DISCLAIMER OF ALL 
 * WARRANTIES.
 *
 */
package edu.cmu.sphinx.linguist.util;

import edu.cmu.sphinx.linguist.*;

import java.util.*;

/** A linguist processor that dumps out statistics about the search space */
public class LinguistStats extends LinguistProcessor {

    private final Map<Class<? extends SearchState>, Integer> stateCountByType = new HashMap<Class<? extends SearchState>, Integer>();

    public LinguistStats( Linguist linguist ) {
        super(linguist);
    }

    public LinguistStats() {
        
    }

    /** Dumps the statistics of the linguist */
    @Override
    public void run() {
        Linguist linguist = getLinguist();
        List<SearchState> queue = new LinkedList<SearchState>();
        Set<SearchState> visitedStates = new HashSet<SearchState>();
        int stateCount = 0;
        queue.add(linguist.getSearchGraph().getInitialState());
        while (!queue.isEmpty()) {
            SearchState state = queue.remove(0);
            if (!visitedStates.contains(state)) {
                stateCount++;
                incrementStateTypeCount(state);
                visitedStates.add(state);
                SearchStateArc[] arcs = state.getSuccessors();
                for (int i = arcs.length - 1; i >= 0; i--) {
                    SearchState nextState = arcs[i].getState();
                    queue.add(nextState);
                }
                // Dump out the unique word states for debugging
                // if (state instanceof WordSearchState) {
                //    System.out.println("WS: " + state);
                // }
                // if (false && state instanceof HMMSearchState) {
                //    System.out.println("HS: " + state);
                // }
            }
        }
        System.out.println("# ----------- linguist stats ------------ ");
        System.out.println("# Total states: " + stateCount);
        dumpStateTypeCounts();
    }


    /**
     * Keeps track of state counts by class
     *
     * @param state the state to track
     */
    private void incrementStateTypeCount(SearchState state) {
        Integer count = stateCountByType.get(state.getClass());
        if (count == null) {
            count = 0;
        }
        count = count + 1;
        stateCountByType.put(state.getClass(), count);
    }


    /** Dumps all of the class counts */
    private void dumpStateTypeCounts() {
        for (Map.Entry<Class<? extends SearchState>, Integer> entry : stateCountByType.entrySet())
            System.out.println("# " + entry.getKey() + ": " + entry.getValue());
    }
}