#include <iostream>
#include <iomanip>

#include "systemc.h"
#include "systemc"
#include <sysc/tracing/sc_trace.h>
#include <sysc/tracing/sc_vcd_trace.h>

#include "cpu_or.hpp"

SC_MODULE(test_cpu_or)
{
	sc_out<bool> a,b;
	sc_in<bool> clk;

	void test_cpu_or_stim()
	{
		wait();
		a.write(1);
		b.write(1);
		wait();

		a.write(0);
		b.write(0);
		wait();

		a.write(1);
		b.write(0);
		wait();

		a.write(0);
		b.write(1);
		wait();

		sc_stop();
	}

	SC_CTOR(test_cpu_or)
	{
		SC_THREAD(test_cpu_or_stim);
		sensitive << clk.pos();

	}
};

int sc_main(int argc, char **argv) {
	sc_signal<bool> sig_a, sig_b, sig_c;
	sc_clock TestClk("TestClk", 10, SC_NS, 0.5, 1, SC_NS);

	test_cpu_or Stim1("Stimulus");
	Stim1.a(sig_a);
	Stim1.b(sig_b);
	Stim1.clk(TestClk);

	cpu_or DUT("or");
	DUT.in_a(sig_a);
	DUT.in_b(sig_b);
	DUT.out_c(sig_c);

	sc_trace_file *Tf;
	Tf = sc_create_vcd_trace_file("trace_cpu_or.dat");
	sc_trace(Tf, sig_a, "IN_A");
	sc_trace(Tf, sig_b, "IN_B");
	sc_trace(Tf, sig_c, "OUT_C");

	sc_start();
	sc_close_vcd_trace_file(Tf);

	return(0);
}