Luis Gonzalez
89 lines
3.2 KiB
C#
89 lines
3.2 KiB
C#
using Unity.Burst;
|
|
using Unity.Collections;
|
|
using Unity.Entities;
|
|
using Hash128 = Unity.Entities.Hash128;
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
namespace Rukhanka
|
|
{
|
|
[UpdateInGroup(typeof(InitializationSystemGroup), OrderLast = true)]
|
|
public partial struct BlobDatabaseUpdateSystem: ISystem
|
|
{
|
|
EntityQuery newBlobAssetsQuery;
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
[BurstCompile]
|
|
public void OnCreate(ref SystemState ss)
|
|
{
|
|
CreateDBSingleton(ref ss);
|
|
|
|
newBlobAssetsQuery = SystemAPI.QueryBuilder()
|
|
.WithAny
|
|
<
|
|
NewBlobAssetDatabaseRecord<AnimationClipBlob>,
|
|
NewBlobAssetDatabaseRecord<AvatarMaskBlob>
|
|
>()
|
|
.WithOptions(EntityQueryOptions.IncludePrefab)
|
|
.Build();
|
|
|
|
ss.RequireForUpdate(newBlobAssetsQuery);
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
void CreateDBSingleton(ref SystemState ss)
|
|
{
|
|
var blobDB = new BlobDatabaseSingleton()
|
|
{
|
|
animations = new (128, Allocator.Persistent),
|
|
avatarMasks = new (128, Allocator.Persistent),
|
|
};
|
|
ss.EntityManager.CreateSingleton(blobDB, "Rukhanka.BlobDatabaseSingleton");
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
[BurstCompile]
|
|
public void OnUpdate(ref SystemState ss)
|
|
{
|
|
if (!SystemAPI.TryGetSingletonRW<BlobDatabaseSingleton>(out var db))
|
|
return;
|
|
|
|
var ecbs = SystemAPI.GetSingleton<BeginSimulationEntityCommandBufferSystem.Singleton>();
|
|
var ecb = ecbs.CreateCommandBuffer(ss.WorldUnmanaged);
|
|
|
|
ProcessNewBlobs(ref ss, ref db.ValueRW.avatarMasks, ecb);
|
|
ProcessNewBlobs(ref ss, ref db.ValueRW.animations, ecb);
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
void ProcessNewBlobs<T>(ref SystemState ss, ref NativeHashMap<Hash128, BlobAssetReference<T>> db, EntityCommandBuffer ecb) where T: unmanaged
|
|
{
|
|
var componentTypeHandle = ss.EntityManager.GetBufferTypeHandle<NewBlobAssetDatabaseRecord<T>>(true);
|
|
var processNewBlobsJob = new ProcessNewBlobsJob<T>()
|
|
{
|
|
db = db,
|
|
componentTypeHandle = componentTypeHandle
|
|
};
|
|
ss.Dependency = processNewBlobsJob.Schedule(newBlobAssetsQuery, ss.Dependency);
|
|
ecb.RemoveComponent<NewBlobAssetDatabaseRecord<T>>(newBlobAssetsQuery, EntityQueryCaptureMode.AtPlayback);
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
[BurstCompile]
|
|
public void OnDestroy(ref SystemState ss)
|
|
{
|
|
if (SystemAPI.TryGetSingletonRW<BlobDatabaseSingleton>(out var animDB))
|
|
{
|
|
animDB.ValueRW.animations.Dispose();
|
|
animDB.ValueRW.avatarMasks.Dispose();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|