-using System.IO;

-using System.Runtime.CompilerServices;

-using System.Runtime.InteropServices;

- private const ulong ShaderCodeGenVersion = 1717;

- TransformFeedbackDescriptor[] tfd = ReadTransformationFeedbackInformations(ref guestProgramReadOnlySpan, fileHeader);

- byte[] programCode = null;

- GuestShaderCacheEntryHeader[] shaderCacheEntries = null;

- GetProgramInformations(null, shaderContexts, out programCode, out programCodeHash, out shaderCacheEntries);

- _cacheManager.SaveProgram(ref programCodeHash, CreateGuestProgramDump(programCode, shaderCacheEntries, null), hostProgramBinary);

- byte[] programCode = null;

- GuestShaderCacheEntryHeader[] shaderCacheEntries = null;

- GetProgramInformations(tfd, shaderContexts, out programCode, out programCodeHash, out shaderCacheEntries);

- _cacheManager.SaveProgram(ref programCodeHash, CreateGuestProgramDump(programCode, shaderCacheEntries, tfd), hostProgramBinary);

-

- /// <summary>

- /// Create a guest shader program.

- /// </summary>

- /// <param name="programCode">The program code of the shader code</param>

- /// <param name="shaderCacheEntries">The resulting guest shader entries header</param>

- /// <param name="tfd">The transform feedback descriptors in use</param>

- /// <returns>The resulting guest shader program</returns>

- private static byte[] CreateGuestProgramDump(ReadOnlySpan<byte> programCode, GuestShaderCacheEntryHeader[] shaderCacheEntries, TransformFeedbackDescriptor[] tfd)

- {

- using (MemoryStream resultStream = new MemoryStream())

- {

- BinaryWriter resultStreamWriter = new BinaryWriter(resultStream);

-

- byte transformFeedbackCount = 0;

-

- if (tfd != null)

- {

- transformFeedbackCount = (byte)tfd.Length;

- }

-

- // Header

- resultStreamWriter.WriteStruct(new GuestShaderCacheHeader((byte)shaderCacheEntries.Length, transformFeedbackCount));

-

- // Write all entries header

- foreach (GuestShaderCacheEntryHeader entry in shaderCacheEntries)

- {

- resultStreamWriter.WriteStruct(entry);

- }

-

- // Finally, write all program code and all transform feedback information.

- resultStreamWriter.Write(programCode);

-

- return resultStream.ToArray();

- }

- }

-

- /// <summary>

- /// Write transform feedback guest information to the given stream.

- /// </summary>

- /// <param name="stream">The stream to write data to</param>

- /// <param name="tfd">The current transform feedback descriptors used</param>

- private static void WriteTransformationFeedbackInformation(Stream stream, TransformFeedbackDescriptor[] tfd)

- {

- if (tfd != null)

- {

- BinaryWriter writer = new BinaryWriter(stream);

-

- foreach (TransformFeedbackDescriptor transform in tfd)

- {

- writer.WriteStruct(new GuestShaderCacheTransformFeedbackHeader(transform.BufferIndex, transform.Stride, transform.VaryingLocations.Length));

- writer.Write(transform.VaryingLocations);

- }

- }

- }

-

- /// <summary>

- /// Read transform feedback descriptors from guest.

- /// </summary>

- /// <param name="data">The raw guest transform feedback descriptors</param>

- /// <param name="header">The guest shader program header</param>

- /// <returns>The transform feedback descriptors read from guest</returns>

- private static TransformFeedbackDescriptor[] ReadTransformationFeedbackInformations(ref ReadOnlySpan<byte> data, GuestShaderCacheHeader header)

- {

- if (header.TransformFeedbackCount != 0)

- {

- TransformFeedbackDescriptor[] result = new TransformFeedbackDescriptor[header.TransformFeedbackCount];

-

- for (int i = 0; i < result.Length; i++)

- {

- GuestShaderCacheTransformFeedbackHeader feedbackHeader = MemoryMarshal.Read<GuestShaderCacheTransformFeedbackHeader>(data);

-

- result[i] = new TransformFeedbackDescriptor(feedbackHeader.BufferIndex, feedbackHeader.Stride, data.Slice(Unsafe.SizeOf<GuestShaderCacheTransformFeedbackHeader>(), feedbackHeader.VaryingLocationsLength).ToArray());

-

- data = data.Slice(Unsafe.SizeOf<GuestShaderCacheTransformFeedbackHeader>() + feedbackHeader.VaryingLocationsLength);

- }

-

- return result;

- }

-

- return null;

- }

-

- /// <summary>

- /// Create a new instance of <see cref="GuestGpuAccessorHeader"/> from an gpu accessor.

- /// </summary>

- /// <param name="gpuAccessor">The gpu accessor</param>

- /// <returns>a new instance of <see cref="GuestGpuAccessorHeader"/></returns>

- private static GuestGpuAccessorHeader CreateGuestGpuAccessorCache(IGpuAccessor gpuAccessor)

- {

- return new GuestGpuAccessorHeader

- {

- ComputeLocalSizeX = gpuAccessor.QueryComputeLocalSizeX(),

- ComputeLocalSizeY = gpuAccessor.QueryComputeLocalSizeY(),

- ComputeLocalSizeZ = gpuAccessor.QueryComputeLocalSizeZ(),

- ComputeLocalMemorySize = gpuAccessor.QueryComputeLocalMemorySize(),

- ComputeSharedMemorySize = gpuAccessor.QueryComputeSharedMemorySize(),

- PrimitiveTopology = gpuAccessor.QueryPrimitiveTopology(),

- };

- }

-

- /// <summary>

- /// Write the guest GpuAccessor informations to the given stream.

- /// </summary>

- /// <param name="stream">The stream to write the guest GpuAcessor</param>

- /// <param name="shaderContext">The shader tranlator context in use</param>

- /// <returns>The guest gpu accessor header</returns>

- private static GuestGpuAccessorHeader WriteGuestGpuAccessorCache(Stream stream, TranslatorContext shaderContext)

- {

- BinaryWriter writer = new BinaryWriter(stream);

-

- GuestGpuAccessorHeader header = CreateGuestGpuAccessorCache(shaderContext.GpuAccessor);

-

- // If we have a full gpu accessor, cache textures descriptors

- if (shaderContext.GpuAccessor is GpuAccessor gpuAccessor)

- {

- HashSet<int> textureHandlesInUse = shaderContext.TextureHandlesForCache;

-

- header.TextureDescriptorCount = textureHandlesInUse.Count;

-

- foreach (int textureHandle in textureHandlesInUse)

- {

- GuestTextureDescriptor textureDescriptor = ((Image.TextureDescriptor)gpuAccessor.GetTextureDescriptor(textureHandle)).ToCache();

-

- textureDescriptor.Handle = (uint)textureHandle;

-

- writer.WriteStruct(textureDescriptor);

- }

- }

-

- return header;

- }

-

- /// <summary>

- /// Get the shader program information for use on the shader cache.

- /// </summary>

- /// <param name="tfd">The current transform feedback descriptors used</param>

- /// <param name="shaderContexts">The shader translators context in use</param>

- /// <param name="programCode">The resulting raw shader program code</param>

- /// <param name="programCodeHash">The resulting raw shader program code hash</param>

- /// <param name="entries">The resulting guest shader entries header</param>

- private void GetProgramInformations(TransformFeedbackDescriptor[] tfd, ReadOnlySpan<TranslatorContext> shaderContexts, out byte[] programCode, out Hash128 programCodeHash, out GuestShaderCacheEntryHeader[] entries)

- {

- GuestShaderCacheEntryHeader ComputeStage(Stream stream, TranslatorContext context)

- {

- if (context == null)

- {

- return new GuestShaderCacheEntryHeader();

- }

-

- ReadOnlySpan<byte> data = _context.MemoryManager.GetSpan(context.Address, context.Size);

-

- stream.Write(data);

-

- int size = data.Length;

- int sizeA = 0;

-

- if (context.AddressA != 0)

- {

- data = _context.MemoryManager.GetSpan(context.AddressA, context.SizeA);

-

- sizeA = data.Length;

-

- stream.Write(data);

- }

-

- GuestGpuAccessorHeader gpuAccessorHeader = WriteGuestGpuAccessorCache(stream, context);

-

- return new GuestShaderCacheEntryHeader(context.Stage, size, sizeA, gpuAccessorHeader);

- }

-

- entries = new GuestShaderCacheEntryHeader[shaderContexts.Length];

-

- using (MemoryStream stream = new MemoryStream())

- {

- for (int i = 0; i < shaderContexts.Length; i++)

- {

- entries[i] = ComputeStage(stream, shaderContexts[i]);

- }

-

- WriteTransformationFeedbackInformation(stream, tfd);

-

- programCode = stream.ToArray();

- programCodeHash = _cacheManager.ComputeHash(programCode);

- }

- }