package parser;

import java.util.*;

public class States {
    private Set<State> stateSet;
    private List<State> states;
    private int id = 0;

    public States() {
        this.stateSet = new HashSet<>();
        this.states = new ArrayList<>();
    }

    public int size() {
        return this.states.size();
    }

    public boolean addState(State state) {
        if (!this.stateSet.contains(state)) {
            this.stateSet.add(state);
            this.states.add(state);
            return true;
        }
        return false;
    }

    public List<State> getStateList() {
        return states;
    }

    public State getState(int name) {
        if (name >= states.size()) {
            return null;
        }
        return states.get(name);
    }

    @Override
    public String toString() {
        return states.toString();
    }

}