Runtime/Misc/AsyncOperation.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

#include "UnityPrefix.h"
#include "AsyncOperation.h"

void AsyncOperation::SetCoroutineCallback (	DelayedCall* func, Object* coroutineBehaviour, void* userData, CleanupUserData* cleanup)
{
	m_CoroutineBehaviour = coroutineBehaviour;
	m_CoroutineDone = func;
	m_CoroutineCleanup = cleanup;
	m_CoroutineData = userData;
}

void AsyncOperation::InvokeCoroutine ()
{
	if (m_CoroutineDone != NULL)
	{
		Object* target = m_CoroutineBehaviour;
		if (target)
			m_CoroutineDone(target, m_CoroutineData);
		m_CoroutineCleanup(m_CoroutineData);
		m_CoroutineDone = NULL;
	}
}

void AsyncOperation::CleanupCoroutine ()
{
	if (m_CoroutineDone != NULL)
	{
		m_CoroutineCleanup(m_CoroutineData);
		m_CoroutineDone = NULL;
	}
}

AsyncOperation::~AsyncOperation ()
{
	AssertIf(m_CoroutineDone != NULL);
}

void AsyncOperation::Retain ()
{
	m_RefCount.Retain();
}

void AsyncOperation::Release ()
{
	if (m_RefCount.Release())
		delete this;
}