您最多选择25个主题
主题必须以中文或者字母或数字开头,可以包含连字符 (-),并且长度不得超过35个字符
118 行
4.1 KiB
118 行
4.1 KiB
using Unity.Collections;
|
|
using Unity.Entities;
|
|
using Unity.Jobs;
|
|
using Unity.Burst;
|
|
using Unity.Mathematics;
|
|
using UnityEngine.Jobs;
|
|
|
|
namespace Unity.Transforms
|
|
{
|
|
[UpdateBefore(typeof(EndFrameTransformSystem))]
|
|
public class CopyInitialTransformFromGameObjectSystem : JobComponentSystem
|
|
{
|
|
[Inject] ComponentDataFromEntity<Position> m_Positions;
|
|
[Inject] ComponentDataFromEntity<Rotation> m_Rotations;
|
|
|
|
struct TransformStash
|
|
{
|
|
public float3 position;
|
|
public quaternion rotation;
|
|
}
|
|
|
|
[BurstCompile]
|
|
struct StashTransforms : IJobParallelForTransform
|
|
{
|
|
public NativeArray<TransformStash> transformStashes;
|
|
|
|
public void Execute(int index, TransformAccess transform)
|
|
{
|
|
transformStashes[index] = new TransformStash
|
|
{
|
|
rotation = transform.rotation,
|
|
position = transform.position,
|
|
};
|
|
}
|
|
}
|
|
|
|
// [BurstCompile]
|
|
struct CopyTransforms : IJobParallelFor
|
|
{
|
|
[NativeDisableParallelForRestriction] public ComponentDataFromEntity<Position> positions;
|
|
[NativeDisableParallelForRestriction] public ComponentDataFromEntity<Rotation> rotations;
|
|
[ReadOnly]
|
|
public EntityArray entities;
|
|
[DeallocateOnJobCompletion] public NativeArray<TransformStash> transformStashes;
|
|
|
|
public void Execute(int index)
|
|
{
|
|
var transformStash = transformStashes[index];
|
|
var entity = entities[index];
|
|
if (positions.Exists(entity))
|
|
{
|
|
positions[entity] = new Position { Value = transformStash.position };
|
|
}
|
|
if (rotations.Exists(entity))
|
|
{
|
|
rotations[entity] = new Rotation { Value = transformStash.rotation };
|
|
}
|
|
}
|
|
}
|
|
|
|
struct RemoveCopyInitialTransformFromGameObjectComponent : IJob
|
|
{
|
|
[ReadOnly]
|
|
public EntityArray entities;
|
|
public EntityCommandBuffer entityCommandBuffer;
|
|
|
|
public void Execute()
|
|
{
|
|
for (int i = 0; i < entities.Length; i++)
|
|
{
|
|
entityCommandBuffer.RemoveComponent<CopyInitialTransformFromGameObject>(entities[i]);
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
[Inject] private EndFrameBarrier m_EndFrameBarrier;
|
|
|
|
ComponentGroup m_InitialTransformGroup;
|
|
|
|
protected override void OnCreateManager(int capacity)
|
|
{
|
|
m_InitialTransformGroup = GetComponentGroup(ComponentType.ReadOnly(typeof(CopyInitialTransformFromGameObject)),typeof(UnityEngine.Transform));
|
|
}
|
|
|
|
protected override JobHandle OnUpdate(JobHandle inputDeps)
|
|
{
|
|
var transforms = m_InitialTransformGroup.GetTransformAccessArray();
|
|
var entities = m_InitialTransformGroup.GetEntityArray();
|
|
|
|
var transformStashes = new NativeArray<TransformStash>(transforms.length, Allocator.TempJob);
|
|
var stashTransformsJob = new StashTransforms
|
|
{
|
|
transformStashes = transformStashes
|
|
};
|
|
|
|
var stashTransformsJobHandle = stashTransformsJob.Schedule(transforms, inputDeps);
|
|
|
|
var copyTransformsJob = new CopyTransforms
|
|
{
|
|
positions = m_Positions,
|
|
rotations = m_Rotations,
|
|
transformStashes = transformStashes,
|
|
entities = entities
|
|
};
|
|
|
|
var copyTransformsJobHandle = copyTransformsJob.Schedule(transformStashes.Length,64,stashTransformsJobHandle);
|
|
|
|
var removeComponentsJob = new RemoveCopyInitialTransformFromGameObjectComponent
|
|
{
|
|
entities = entities,
|
|
entityCommandBuffer = m_EndFrameBarrier.CreateCommandBuffer()
|
|
};
|
|
var removeComponentsJobHandle = removeComponentsJob.Schedule(copyTransformsJobHandle);
|
|
return removeComponentsJobHandle;
|
|
}
|
|
}
|
|
}
|