#include <UECS/UECS.hpp>

#include <iostream>

using namespace Ubpa::UECS;
using namespace Ubpa;
using namespace std;

struct S { float value; };
struct A { float value; };
struct B { float value; };

constexpr float dt = 0.003f;

struct SAB_System {
	static void OnUpdate(Schedule& schedule) {
		ArchetypeFilter filter;
		filter.all = { AccessTypeID_of<Write<S>> };
		filter.any = { AccessTypeID_of<Latest<A>>, AccessTypeID_of<Latest<B>> };

		schedule.RegisterChunkJob([](ChunkView chunk) {
			auto arrayS = chunk->GetCmptArray<S>();
			auto arrayA = chunk->GetCmptArray<A>();
			auto arrayB = chunk->GetCmptArray<B>();
			bool containsA = !arrayA.empty();
			bool containsB = !arrayB.empty();
			bool containsAB = containsA && containsB;

			if (containsAB) {
				cout << "[AB]" << endl;
				for (std::size_t i = 0; i < chunk->EntityNum(); i++) {
					arrayS[i].value += arrayA[i].value * arrayB[i].value;
				}
			}
			else if (containsA) {
				cout << "[A]" << endl;
				for (std::size_t i = 0; i < chunk->EntityNum(); i++) {
					arrayS[i].value += arrayA[i].value;
				}
			}
			else { // containsB
				cout << "[B]" << endl;
				for (std::size_t i = 0; i < chunk->EntityNum(); i++) {
					arrayS[i].value += arrayB[i].value;
				}
			}
		}, "SAB_System", filter);
	}
};

int main() {
	World w;
	w.entityMngr.cmptTraits.Register<S, A, B>();
	w.systemMngr.RegisterAndActivate<SAB_System>();

	w.entityMngr.Create(Ubpa::TypeIDs_of<S>);
	w.entityMngr.Create(Ubpa::TypeIDs_of<S, A>);
	w.entityMngr.Create(Ubpa::TypeIDs_of<S, B>);
	w.entityMngr.Create(Ubpa::TypeIDs_of<S, A, B>);

	w.Update();

	cout << w.GenUpdateFrameGraph().Dump() << endl;

	return 0;
}