#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_NAMED_REGISTRATION( AutomatonIterationTest, "automaton" );
CPPUNIT_TEST_SUITE_REGISTRATION( AutomatonIterationTest );

void AutomatonIterationTest::setUp() {
}

void AutomatonIterationTest::tearDown() {
}

void AutomatonIterationTest::testAutomatonIteration() {

	// Melichar 2.83
	label::Label q0 = label::Label(0);
	label::Label q1 = label::Label(1);
	label::Label q2 = label::Label(2);
	label::Label q3 = label::Label(3);
	alphabet::Symbol a = alphabet::Symbol('a');
	alphabet::Symbol b = alphabet::Symbol('b');

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

	automaton::EpsilonNFA < > res(q0);
	res.setStates({q0, q1, q2, q3});
	res.setInputAlphabet({a, b});
	res.setFinalStates({q0, q3});

	res.addTransition(q1, a, q2);
	res.addTransition(q2, b, q2);
	res.addTransition(q2, a, q3);
	res.addTransition(q3, a, q1);

	res.addTransition(q0, q1);
	res.addTransition(q3, 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);
}