void ReturnByRef::returnByRefCollectCalls(RefMap& calls)
{
RefMap::iterator iter;
forv_Vec(CallExpr, call, gCallExprs)
{
if (FnSymbol* fn = theTransformableFunction(call))
{
RefMap::iterator iter = calls.find(fn->id);
ReturnByRef* info = NULL;
if (iter == calls.end())
{
info = new ReturnByRef(fn);
calls[fn->id] = info;
}
else
{
info = iter->second;
}
info->addCall(call);
}
}
}
void ReturnByRef::returnByRefCollectCalls(RefMap& calls, FnSet& fns)
{
RefMap::iterator iter;
forv_Vec(CallExpr, call, gCallExprs)
{
// Only transform calls that are still in the AST tree
// (defer statement bodies have been removed at this point
// in this pass)
if (call->inTree()) {
// Only transform calls to transformable functions
// The common case is a user-level call to a resolved function
// Also handle the PRIMOP for a virtual method call
if (FnSymbol* fn = call->resolvedOrVirtualFunction()) {
TransformationKind tfKind = transformableFunctionKind(fn);
if (tfKind == TF_FULL)
{
RefMap::iterator iter = calls.find(fn->id);
ReturnByRef* info = NULL;
if (iter == calls.end())
{
info = new ReturnByRef(fn);
calls[fn->id] = info;
}
else
{
info = iter->second;
}
info->addCall(call);
}
else if (tfKind == TF_ASGN)
{
fns.insert(fn);
}
}
}
}
}
