#include "cfapass.h"
#include "analysis/cfagraph.h"

#include <boost/range/iterator_range_core.hpp>

using namespace std;
using namespace xreate;

void
CFAPass::initSignatures(){
	auto range = man->root->__interfacesData.equal_range(CFA);
	for (auto i = range.first; i!= range.second; ++i){
		__signatures.emplace(i->second.op, i->second);
	}
}

void CFAPass::run(){
	initSignatures();

	return AbstractPass::run();
}

void
CFAPass::finish()
{
    man->clasp->setCFAData(move(__context.graph));

    return AbstractPass::finish();
}

void
CFAPass::processFnCall(ManagedFnPtr function, PassContext context)
{
	ClaspLayer* clasp = man->clasp;
	__context.graph->addCallConnection(clasp->pack(context.scope), function->getName());

	return AbstractPass::processFnCall(function, context);
}

void
CFAPass::processFnCallUncertain(ManagedFnPtr function, PassContext context){
	ClaspLayer* clasp = man->clasp;
	__context.graph->addCallConnection(clasp->pack(context.scope), function->getName());

	return AbstractPass::processFnCallUncertain(function, context);
}

void
CFAPass::process(CodeScope* scope, PassContext context, const std::string& hintBlockDecl){
	ClaspLayer* clasp = man->clasp;

	CodeScope* scopeParent = context.scope;
	ScopePacked scopeId = clasp->pack(scope);

	if (scopeParent){
		__context.graph->addParentConnection(scopeId, clasp->pack(scopeParent));
	} else {
		__context.graph->addParentConnection(scopeId, context.function->getName());
	}

		//TOTEST scope annotations
		//SECTIONTAG context gather scope annotations
	__context.graph->addScopeAnnotations(scopeId, scope->tags);

	__context.graph->addContextRules(scopeId, scope->contextRules);

	return AbstractPass::process(scope, context, hintBlockDecl);
}

//TOTEST scope annotations via scheme
void
CFAPass::process(const Expression& expression, PassContext context, const std::string& varDecl){
	ClaspLayer* clasp = man->clasp;

	if (expression.__state == Expression::COMPOUND){
		Operator  op= expression.op;

		if (__signatures.count(op)) {
			assert(expression.blocks.size());

			for (const auto& scheme: boost::make_iterator_range(__signatures.equal_range(expression.op))) {
					__context.graph->addScopeAnnotations(clasp->pack(expression.blocks.front()), scheme.second.getOperands());
			}
		}
	}

	return AbstractPass::process(expression, context, varDecl);
}

void
CFAPass::process(ManagedFnPtr function)
{
	__context.graph->addFunctionAnnotations(function->getName(), function->getTags());
	return AbstractPass::process(function);
}

CFAPass::CFAPass(PassManager* manager)
	: AbstractPass(manager)
          , __context{new xreate::analysis::CFAGraph(manager->clasp)}
{}