public static unsafe void RunJobChunk <T>(ref T jobData, EntityQuery query, JobChunkRunWithoutJobSystemDelegate functionPointer) where T : unmanaged, IJobChunk, IJobBase
{
var myIterator = query.GetArchetypeChunkIterator();
try
{
query._GetImpl()->_Access->DependencyManager->IsInForEachDisallowStructuralChange++;
var managedJobDataPtr = UnsafeUtility.AddressOf(ref jobData);
var unmanagedSize = jobData.GetUnmanagedJobSize_Gen();
if (unmanagedSize != -1)
{
const int kAlignment = 16;
int alignedSize = (unmanagedSize + kAlignment - 1) & ~(kAlignment - 1);
byte * unmanagedJobData = stackalloc byte[alignedSize];
byte * alignedUnmanagedJobData = (byte *)((UInt64)(unmanagedJobData + kAlignment - 1) & ~(UInt64)(kAlignment - 1));
// DOTS Runtime job marshalling code assumes the job is wrapped so create the wrapper and assign the jobData
JobChunkExtensions.JobChunkWrapper <T> jobChunkWrapper = default;
jobChunkWrapper.JobData = jobData;
byte *jobChunkDataPtr = (byte *)UnsafeUtility.AddressOf(ref jobChunkWrapper);
byte *dst = (byte *)alignedUnmanagedJobData;
byte *src = (byte *)jobChunkDataPtr;
var marshalToBurstFnPtr = JobMarshalFnLookup <T> .GetMarshalToBurstFn();
UnsafeUtility.EnterTempScope();
UnsafeUtility.CallFunctionPtr_pp(marshalToBurstFnPtr.ToPointer(), dst, src);
// Since we are running inline, normally the outer job scheduling code would
// reference jobWrapper.Data however we can't do that since if we are in this code it means
// we are dealing with a job/jobwrapper that is burst compiled and is non-blittable. Thus any
// type-safe offset we calculate here will be based on the managed data layout which is not useful.
// Instead we can at least know that for a sequential layout (which is what we know we must be using
// since we are burst compiled) our JobChunkData contains a safety field as its first member. Skipping over this will
// provide the necessary offset to jobChunkData.Data
var DataOffset = UnsafeUtility.SizeOf <JobChunkExtensions.EntitySafetyHandle>();
Assertions.Assert.AreEqual(jobChunkWrapper.safety.GetType(), typeof(JobChunkExtensions.EntitySafetyHandle));
functionPointer(&myIterator, alignedUnmanagedJobData + DataOffset);
// Since Run can capture locals for write back, we must write back the marshalled jobData after the job executes
var marshalFromBurstFnPtr = JobMarshalFnLookup <T> .GetMarshalFromBurstFn();
UnsafeUtility.CallFunctionPtr_pp(marshalFromBurstFnPtr.ToPointer(), src, dst);
UnsafeUtility.ExitTempScope();
jobData = jobChunkWrapper.JobData;
}
else
{
functionPointer(&myIterator, managedJobDataPtr);
}
}
finally
{
query._GetImpl()->_Access->DependencyManager->IsInForEachDisallowStructuralChange--;
}
}
public static unsafe void RunIJob <T>(ref T jobData, JobRunWithoutJobSystemDelegate functionPointer) where T : unmanaged, IJob, IJobBase
{
var managedJobDataPtr = UnsafeUtility.AddressOf(ref jobData);
var unmanagedSize = jobData.GetUnmanagedJobSize_Gen();
if (unmanagedSize != -1)
{
const int kAlignment = 16;
int alignedSize = (unmanagedSize + kAlignment - 1) & ~(kAlignment - 1);
byte * unmanagedJobData = stackalloc byte[alignedSize];
byte * alignedUnmanagedJobData = (byte *)((UInt64)(unmanagedJobData + kAlignment - 1) & ~(UInt64)(kAlignment - 1));
// DOTS Runtime job marshalling code assumes the job is wrapped so create the wrapper and assign the jobData
IJobExtensions.JobProducer <T> jobStructData = default;
jobStructData.JobData = jobData;
byte *jobStructDataPtr = (byte *)UnsafeUtility.AddressOf(ref jobStructData);
byte *dst = (byte *)alignedUnmanagedJobData;
byte *src = (byte *)jobStructDataPtr;
var marshalToBurstFnPtr = JobMarshalFnLookup <T> .GetMarshalToBurstFn();
UnsafeUtility.EnterTempScope();
try
{
UnsafeUtility.CallFunctionPtr_pp(marshalToBurstFnPtr.ToPointer(), dst, src);
// In the case of JobStruct we know the jobwrapper doesn't add
// anything to the jobData so just pass it along, no offset required unlike JobChunk
functionPointer(alignedUnmanagedJobData);
// Since Run can capture locals for write back, we must write back the marshalled jobData after the job executes
var marshalFromBurstFnPtr = JobMarshalFnLookup <T> .GetMarshalFromBurstFn();
UnsafeUtility.CallFunctionPtr_pp(marshalFromBurstFnPtr.ToPointer(), src, dst);
}
finally
{
UnsafeUtility.ExitTempScope();
}
jobData = jobStructData.JobData;
}
else
{
functionPointer(managedJobDataPtr);
}
}
