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

#include "automaton/transform/AutomataConcatenation.h"
#include "automaton/transform/AutomataConcatenationEpsilonTransition.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( AutomataConcatenationTest, "automaton" );
CPPUNIT_TEST_SUITE_REGISTRATION( AutomataConcatenationTest );

void AutomataConcatenationTest::setUp() {
}

void AutomataConcatenationTest::tearDown() {
}

void AutomataConcatenationTest::testAutomataConcatenation() {
	// based on Melichar, 2.79

	label::Label q1a = label::Label("1");
	label::Label q2a = label::Label("2");
	label::Label q0a = label::Label("0");
	label::Label q1b = label::Label("1'");
	label::Label q2b = label::Label("2'");
	label::Label q0b = label::Label("0'");
	label::Label q0102 = label::Label("q0102");
	alphabet::Symbol a(alphabet::Symbol('a')), b(alphabet::Symbol('b'));

	automaton::DFA<> m1(q1a);
	automaton::DFA<> m2(q1b);
	automaton::NFA < >  m3(q1a);

	m1.setInputAlphabet({a, b});
	m1.setStates({q1a, q2a, q0a});
	m1.addTransition(q1a, a, q2a);
	m1.addTransition(q1a, b, q0a);
	m1.addTransition(q2a, a, q2a);
	m1.addTransition(q2a, b, q0a);
	m1.addTransition(q0a, a, q0a);
	m1.addTransition(q0a, b, q0a);
	m1.addFinalState(q2a);

	m2.setInputAlphabet({a, b});
	m2.setStates({q1b, q2b});
	m2.addTransition(q1b, b, q2b);
	m2.addTransition(q2b, b, q2b);
	m2.addFinalState(q2b);

	m3.setInputAlphabet({a, b});
	m3.setStates({q1a, q1b, q2a, q2b, q0a, q0b, q0102});
	m3.addTransition(q1a, a, q2a);
	m3.addTransition(q1a, a, q1b);
	m3.addTransition(q1a, b, q0a);
	m3.addTransition(q2a, a, q2a);
	m3.addTransition(q2a, a, q1b);
	m3.addTransition(q2a, b, q0a);
	m3.addTransition(q0a, a, q0a);
	m3.addTransition(q0a, b, q0a);
	m3.addTransition(q1b, a, q0b);
	m3.addTransition(q1b, b, q2b);
	m3.addTransition(q2b, a, q0b);
	m3.addTransition(q2b, b, q2b);
	m3.addTransition(q0b, a, q0b);
	m3.addTransition(q0b, b, q0b);
	m3.setFinalStates({q2b});

	auto u11 = automaton::transform::AutomataConcatenationEpsilonTransition::concatenation(automaton::Automaton(automaton::DFA<>(m1)), automaton::Automaton(automaton::NFA < > (m2)));
	auto u12 = automaton::transform::AutomataConcatenationEpsilonTransition::concatenation(automaton::Automaton(automaton::DFA<>(m1)), automaton::Automaton(automaton::DFA<>(m2)));
	auto u21 = automaton::transform::AutomataConcatenation::concatenation(automaton::Automaton(automaton::DFA<>(m1)), automaton::Automaton(automaton::NFA < > (m2)));
	auto u22 = automaton::transform::AutomataConcatenation::concatenation(automaton::Automaton(automaton::DFA<>(m1)), automaton::Automaton(automaton::DFA<>(m2)));

	automaton::Automaton umdfa  (automaton::simplify::Normalize::normalize(automaton::simplify::Trim::trim(automaton::simplify::MinimizeBrzozowski::minimize(automaton::simplify::EpsilonRemoverIncoming::remove(m3)))));
	automaton::Automaton umdfa11(automaton::simplify::Normalize::normalize(automaton::simplify::Trim::trim(automaton::simplify::MinimizeBrzozowski::minimize(automaton::simplify::EpsilonRemoverIncoming::remove(u11)))));
	automaton::Automaton umdfa12(automaton::simplify::Normalize::normalize(automaton::simplify::Trim::trim(automaton::simplify::MinimizeBrzozowski::minimize(automaton::simplify::EpsilonRemoverIncoming::remove(u12)))));
	automaton::Automaton umdfa21(automaton::simplify::Normalize::normalize(automaton::simplify::Trim::trim(automaton::simplify::MinimizeBrzozowski::minimize(automaton::simplify::EpsilonRemoverIncoming::remove(u21)))));
	automaton::Automaton umdfa22(automaton::simplify::Normalize::normalize(automaton::simplify::Trim::trim(automaton::simplify::MinimizeBrzozowski::minimize(automaton::simplify::EpsilonRemoverIncoming::remove(u22)))));

	CPPUNIT_ASSERT(umdfa11 == umdfa);
	CPPUNIT_ASSERT(umdfa12 == umdfa);
	CPPUNIT_ASSERT(umdfa21 == umdfa);
	CPPUNIT_ASSERT(umdfa22 == umdfa);

}