#include <list>
#include "AutomatonIterationTest.h"

#include "automaton/transform/AutomatonIteration.h"
#include "automaton/transform/AutomatonIterationEpsilonTransition.h"

#include "automaton/simplify/MinimizeBrzozowski.h"
#include "automaton/simplify/Normalize.h"
#include "automaton/simplify/EpsilonRemoverIncoming.h"
#include "automaton/simplify/Trim.h"
#include "automaton/simplify/Total.h"
#include "automaton/determinize/Determinize.h"

#define CPPUNIT_IMPLY(x, y) CPPUNIT_ASSERT(!(x) || (y))

CPPUNIT_TEST_SUITE_REGISTRATION( AutomatonIterationTest );

void AutomatonIterationTest::setUp() {
}

void AutomatonIterationTest::tearDown() {
}

void AutomatonIterationTest::testAutomatonIteration() {

	// Melichar 2.83
	automaton::State q1(1), q2(2), q3(3);
	alphabet::Symbol a(alphabet::symbolFrom('a')), b(alphabet::symbolFrom('b'));

	automaton::DFA m1(q1);
	m1.setStates({q1, q2, q3});
	m1.addFinalState(q3);
	m1.setInputSymbols({a, b});
	m1.addTransition(q1, a, q2);
	m1.addTransition(q2, b, q2);
	m1.addTransition(q2, a, q3);

	automaton::NFA res(q1);
	res.setStates({q1, q2, q3});
	res.setInputSymbols({a, b});
	res.setFinalStates({q1, q3});
	res.addTransition(q1, a, q2);
	res.addTransition(q2, b, q2);
	res.addTransition(q2, a, q3);
	res.addTransition(q2, a, q1);

	automaton::Automaton i2 = automaton::transform::AutomatonIterationEpsilonTransition::iteration(automaton::Automaton(m1));
	automaton::Automaton i1 = automaton::transform::AutomatonIteration::iteration(automaton::Automaton(m1));

	automaton::Automaton mdfa1 (automaton::simplify::Normalize::normalize(automaton::simplify::Trim::trim(automaton::simplify::MinimizeBrzozowski::minimize(automaton::simplify::EpsilonRemoverIncoming::remove(i1)))));
	automaton::Automaton mdfa2 (automaton::simplify::Normalize::normalize(automaton::simplify::Trim::trim(automaton::simplify::MinimizeBrzozowski::minimize(automaton::simplify::EpsilonRemoverIncoming::remove(i2)))));
	automaton::Automaton mdfa3 (automaton::simplify::Normalize::normalize(automaton::simplify::Trim::trim(automaton::simplify::MinimizeBrzozowski::minimize(automaton::simplify::EpsilonRemoverIncoming::remove(res)))));

	CPPUNIT_ASSERT(mdfa1 == mdfa2);
	CPPUNIT_ASSERT(mdfa1 == mdfa3);
}