package sample;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Map;

/**
 * Dfa 状态
 *
 * @author liufengkai
 * Created by liufengkai on 16/7/10.
 */
public class DfaState implements Comparable<DfaState> {

 private static int DFA_ID_COUNT = 0;
 /**
 * state id
 */
 private int stateId;
 /**
 * transition set
 * char / set of dfaState
 */
 private Map<Integer, DfaState> transitionSet;

 private DfaState parentState;

 private Integer parentInput;

 /**
 * 构造方法
 *
 * @param input 输入串
 * @param parentState 父节点
 */
 public DfaState(Integer input, DfaState parentState) {
 this.parentInput = input;
 this.parentState = parentState;
 this.stateId = DFA_ID_COUNT++;
 this.transitionSet = new HashMap<>();
 }

 /**
 * 添加一条转移语句
 *
 * @param input 输入字符
 * @param state 下一个状态
 * @return 返回添加状态
 */
 public DfaState addTransition(int input, DfaState state) {

 if (!transitionSet.containsKey(input)) {
 transitionSet.put(input, state);
 }

 return state;
 }

 public DfaState getTransitionInput(int input) {
 return getTransitionSet().get(input);
 }

 public int getStateId() {
 return stateId;
 }

 public static int getTotalNumber() {
 return DFA_ID_COUNT;
 }

 public Map<Integer, DfaState> getTransitionSet() {
 return transitionSet;
 }

 public DfaState getParentState() {
 return parentState;
 }

 @Override
 public int compareTo(DfaState o) {
 return 0;
 }

 public int getParentInput() {
 return parentInput;
 }

 /**
 * 打印状态
 */
 public void printState() {
 System.out.println("state : " + getStateId());
 for (Integer integer : transitionSet.keySet()) {
 System.out.println("symbol: " +
 (char) integer.intValue() + " to :" +
 transitionSet.get(integer).getStateId());
 transitionSet.get(integer).printState();
 }
 }

 /**
 * 返回结束状态
 *
 * @param list 传入结束状态
 */
 public void returnEndList(ArrayList<DfaState> list) {
 for (Integer key : transitionSet.keySet()) {
 DfaState cur = transitionSet.get(key);
 if (cur.getTransitionSet().isEmpty()) {
 list.add(cur);
 } else {
 cur.returnEndList(list);
 }
 }
 }
}