package com.imegumii;

import java.util.*;

/**
 * Created by imegumii on 18/04/2017.
 */
public class Automata <T extends Comparable> extends Importable {

    protected Set<Transition<T>> transistions;

    protected SortedSet<T> states;
    protected SortedSet<T> beginStates;
    protected SortedSet<T> eindStates;

    protected SortedSet<Character> symbols;

    public Automata() {
        super(Type.ERROR);
        states = new TreeSet<T>();
        beginStates = new TreeSet<T>();
        eindStates = new TreeSet<T>();

        transistions = new TreeSet<>();

        this.symbols = new TreeSet<>();
    }

    public void addTransition (Transition<T> t) {
        if(t.symbol != Transition.EPSILON)
            symbols.add(t.symbol);

        transistions.add(t);
        states.add(t.vanState);
        states.add(t.naarState);
    }

    public void defineAsStartState(T t) {
        states.add(t);
        beginStates.add(t);
    }

    public void defineAsEndState(T t) {
        states.add(t);
        eindStates.add(t);
    }

    public void printTransitions () {
        transistions.forEach(System.out::println);
    }

    public String getTransitions()
    {
        StringBuilder s = new StringBuilder();
        s.append(type).append("\n");
        s.append('B');
        boolean first = true;
        for (T beginState : this.beginStates) {
            if (first) {
                first = false;
            } else {
                s.append(',');
            }
            s.append(beginState.toString());
        }
        s.append('\n');
        first = true;
        s.append('E');
        for (T eindState : this.eindStates) {
            if (first) {
                first = false;
            } else {
                s.append(',');
            }
            s.append(eindState.toString());
        }
        s.append('\n');

        for(Transition<T> t : transistions)
        {
            s.append(t.toString()).append("\n");
        }

        return s.toString();
    }

    public void print() {
        System.out.println("Transitions");
        this.printTransitions();
        System.out.println("Beginstates");
        System.out.println(beginStates);
        System.out.println("Eindstates");
        System.out.println(eindStates);
    }

    public void clearEindStates()
    {
        eindStates.clear();
    }

    public void clearBeginStates()
    {
        beginStates.clear();
    }

    @Override
    public String toString() {
        StringBuilder sb = new StringBuilder();
        for (T state : this.states) {
            sb.append(state);
        }
        return "Automata" + sb.toString();
    }

    @Override
    public boolean equals(Object obj) {
        if(!(obj instanceof Automata))
            return false;

        Automata<T> other = (Automata<T>) obj;

        DFA<String> dfa1;
        DFA<String> dfa2;

        if(this.type == Type.NDFA)
        {
            dfa1 = ((NDFA<String>)this).toDFA().minimaliseerHopcroft();
        }
        else
        {
            dfa1 = ((DFA<String>)this).minimaliseerHopcroft();
        }

        if(other.type == Type.NDFA)
        {
            dfa2 = ((NDFA<String>) obj).toDFA().minimaliseerHopcroft();
        }
        else
        {
            dfa2 = ((DFA<String>) obj).minimaliseerHopcroft();
        }

        DFA<String> result = dfa2.ontkenning().en(dfa1).minimaliseerHopcroft();

        if(result.states.size() <= 1)
            return true;

        return false;
    }
}