Function AddReturnNode

NULLC/Callbacks.cpp:1045–1105

}

void AddReturnNode(const char* pos, bool yield)
{
	bool localReturn = currDefinedFunc.size() != 0;

	// If new function is returned, convert it to pointer
	ConvertFunctionToPointer(pos, currDefinedFunc.size() ? currDefinedFunc.back()->retType : NULL);
	
	// Type that is being returned
	TypeInfo *realRetType = CodeInfo::nodeList.back()->typeInfo;

	// Type that should be returned
	TypeInfo *expectedType = NULL;
	if(currDefinedFunc.size() != 0)
	{
		// If return type is auto, set it to type that is being returned
		if(!currDefinedFunc.back()->retType)
		{
			currDefinedFunc.back()->retType = realRetType;
			currDefinedFunc.back()->funcType = CodeInfo::GetFunctionType(currDefinedFunc.back()->retType, currDefinedFunc.back()->firstParam, currDefinedFunc.back()->paramCount);
		}else{
			ConvertArrayToUnsized(pos, currDefinedFunc.back()->retType);
			HandlePointerToObject(pos, currDefinedFunc.back()->retType);
			realRetType = CodeInfo::nodeList.back()->typeInfo;
		}
		// Take expected return type
		expectedType = currDefinedFunc.back()->retType;
		if(expectedType->refLevel || (expectedType->arrLevel && expectedType->arrSize == TypeInfo::UNSIZED_ARRAY))
		{
			if(CodeInfo::nodeList.back()->nodeType != typeNodeZeroOp)
			{
				CodeInfo::nodeList.push_back(new NodeUnaryOp(cmdCheckedRet, expectedType->arrLevel ? expectedType->typeIndex : expectedType->subType->typeIndex));
				((NodeUnaryOp*)CodeInfo::nodeList.back())->SetParentFunc(currDefinedFunc.back());
			}
		}
		// Check for errors
		if(((expectedType->type == TypeInfo::TYPE_COMPLEX || realRetType->type == TypeInfo::TYPE_COMPLEX || expectedType->firstVariable || realRetType->firstVariable) && expectedType != realRetType) || expectedType->subType != realRetType->subType)
			ThrowError(pos, "ERROR: function returns %s but supposed to return %s", realRetType->GetFullTypeName(), expectedType->GetFullTypeName());
		if(expectedType == typeVoid && realRetType != typeVoid)
			ThrowError(pos, "ERROR: 'void' function returning a value");
		if(expectedType != typeVoid && realRetType == typeVoid)
			ThrowError(pos, "ERROR: function should return %s", expectedType->GetFullTypeName());
		if(yield && currDefinedFunc.back()->type != FunctionInfo::COROUTINE)
			ThrowError(pos, "ERROR: yield can only be used inside a coroutine");
#ifdef NULLC_ENABLE_C_TRANSLATION
		if(yield)
			currDefinedFunc.back()->yieldCount++;
#endif
		currDefinedFunc.back()->explicitlyReturned = true;
	}else{
		// Check for errors
		if(realRetType == typeVoid)
			ThrowError(pos, "ERROR: global return cannot accept void");
		else if(realRetType->type == TypeInfo::TYPE_COMPLEX)
			ThrowError(pos, "ERROR: global return cannot accept complex types");
		// Global return return type is auto, so we take type that is being returned
		expectedType = realRetType;
	}
	// Add node and link source to instruction