318 lines
11 KiB
C#
318 lines
11 KiB
C#
// TProfilingSampler<TEnum>.samples should just be an array. Unfortunately, Enum cannot be converted to int without generating garbage.
|
|
// This could be worked around by using Unsafe but it's not available at the moment.
|
|
// So in the meantime we use a Dictionary with a perf hit...
|
|
//#define USE_UNSAFE
|
|
|
|
#if UNITY_2020_1_OR_NEWER
|
|
#define UNITY_USE_RECORDER
|
|
#endif
|
|
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using Unity.Profiling;
|
|
using UnityEngine.Profiling;
|
|
using UnityEngine.Rendering;
|
|
|
|
namespace HTraceWSGI.Scripts.Extensions
|
|
{
|
|
|
|
class ProfilingSamplerHTrace<TEnum> : ProfilingSamplerHTrace where TEnum : Enum
|
|
{
|
|
#if USE_UNSAFE
|
|
internal static TProfilingSampler<TEnum>[] samples;
|
|
#else
|
|
internal static Dictionary<TEnum, ProfilingSamplerHTrace<TEnum>> samples = new Dictionary<TEnum, ProfilingSamplerHTrace<TEnum>>();
|
|
#endif
|
|
static ProfilingSamplerHTrace()
|
|
{
|
|
var names = Enum.GetNames(typeof(TEnum));
|
|
#if USE_UNSAFE
|
|
var values = Enum.GetValues(typeof(TEnum)).Cast<int>().ToArray();
|
|
samples = new TProfilingSampler<TEnum>[values.Max() + 1];
|
|
#else
|
|
var values = Enum.GetValues(typeof(TEnum));
|
|
#endif
|
|
|
|
for (int i = 0; i < names.Length; i++)
|
|
{
|
|
var sample = new ProfilingSamplerHTrace<TEnum>(names[i]);
|
|
#if USE_UNSAFE
|
|
samples[values[i]] = sample;
|
|
#else
|
|
samples.Add((TEnum)values.GetValue(i), sample);
|
|
#endif
|
|
}
|
|
}
|
|
|
|
public ProfilingSamplerHTrace(string name)
|
|
: base(name, null)
|
|
{
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Wrapper around CPU and GPU profiling samplers.
|
|
/// Use this along ProfilingScope to profile a piece of code.
|
|
/// </summary>
|
|
#if !UNITY_2021
|
|
[IgnoredByDeepProfiler]
|
|
#endif
|
|
public class ProfilingSamplerHTrace
|
|
{
|
|
/// <summary>
|
|
/// Get the sampler for the corresponding enumeration value.
|
|
/// </summary>
|
|
/// <typeparam name="TEnum">Type of the enumeration.</typeparam>
|
|
/// <param name="marker">Enumeration value.</param>
|
|
/// <returns>The profiling sampler for the given enumeration value.</returns>
|
|
public static ProfilingSamplerHTrace Get<TEnum>(TEnum marker)
|
|
where TEnum : Enum
|
|
{
|
|
#if USE_UNSAFE
|
|
return TProfilingSamplerHTrace<TEnum>.samples[Unsafe.As<TEnum, int>(ref marker)];
|
|
#else
|
|
ProfilingSamplerHTrace<TEnum>.samples.TryGetValue(marker, out var sampler);
|
|
return sampler;
|
|
#endif
|
|
}
|
|
|
|
/// <summary>
|
|
/// Constructor.
|
|
/// </summary>
|
|
/// <param name="name">Name of the profiling sampler.</param>
|
|
/// <param name="parentName">Name of the PARENT profiling sampler.</param>
|
|
public ProfilingSamplerHTrace(string name, string parentName = null, int priority = 0)
|
|
{
|
|
// Caution: Name of sampler MUST not match name provide to cmd.BeginSample(), otherwise
|
|
// we get a mismatch of marker when enabling the profiler.
|
|
#if UNITY_USE_RECORDER
|
|
sampler = CustomSampler.Create(name, true); // Event markers, command buffer CPU profiling and GPU profiling
|
|
#else
|
|
// In this case, we need to use the BeginSample(string) API, since it creates a new sampler by that name under the hood,
|
|
// we need rename this sampler to not clash with the implicit one (it won't be used in this case)
|
|
sampler = CustomSampler.Create($"Dummy_{name}");
|
|
#endif
|
|
inlineSampler = CustomSampler.Create($"Inl_{name}"); // Profiles code "immediately"
|
|
this.name = name;
|
|
this.parentName = parentName;
|
|
this.priority = priority;
|
|
|
|
#if UNITY_USE_RECORDER
|
|
m_Recorder = sampler.GetRecorder();
|
|
m_Recorder.enabled = false;
|
|
m_InlineRecorder = inlineSampler.GetRecorder();
|
|
m_InlineRecorder.enabled = false;
|
|
#endif
|
|
}
|
|
|
|
/// <summary>
|
|
/// Begin the profiling block.
|
|
/// </summary>
|
|
/// <param name="cmd">Command buffer used by the profiling block.</param>
|
|
public void Begin(CommandBuffer cmd)
|
|
{
|
|
if (cmd != null)
|
|
#if UNITY_USE_RECORDER
|
|
if (sampler != null && sampler.isValid)
|
|
cmd.BeginSample(sampler);
|
|
else
|
|
cmd.BeginSample(name);
|
|
#else
|
|
cmd.BeginSample(name);
|
|
#endif
|
|
inlineSampler?.Begin();
|
|
}
|
|
|
|
/// <summary>
|
|
/// End the profiling block.
|
|
/// </summary>
|
|
/// <param name="cmd">Command buffer used by the profiling block.</param>
|
|
public void End(CommandBuffer cmd)
|
|
{
|
|
if (cmd != null)
|
|
#if UNITY_USE_RECORDER
|
|
if (sampler != null && sampler.isValid)
|
|
cmd.EndSample(sampler);
|
|
else
|
|
cmd.EndSample(name);
|
|
#else
|
|
m_Cmd.EndSample(name);
|
|
#endif
|
|
inlineSampler?.End();
|
|
}
|
|
|
|
internal bool IsValid() { return (sampler != null && inlineSampler != null); }
|
|
|
|
internal CustomSampler sampler { get; private set; }
|
|
internal CustomSampler inlineSampler { get; private set; }
|
|
/// <summary>
|
|
/// Name of the Profiling Sampler
|
|
/// </summary>
|
|
public string name { get; private set; }
|
|
/// <summary>
|
|
/// Name of the PARENT Profiling Sampler
|
|
/// </summary>
|
|
public string parentName { get; private set; }
|
|
/// <summary>
|
|
/// priority of Profiling Sampler
|
|
/// </summary>
|
|
public int priority { get; private set; }
|
|
|
|
#if UNITY_USE_RECORDER
|
|
Recorder m_Recorder;
|
|
Recorder m_InlineRecorder;
|
|
#endif
|
|
|
|
/// <summary>
|
|
/// Set to true to enable recording of profiling sampler timings.
|
|
/// </summary>
|
|
public bool enableRecording
|
|
{
|
|
set
|
|
{
|
|
#if UNITY_USE_RECORDER
|
|
m_Recorder.enabled = value;
|
|
m_InlineRecorder.enabled = value;
|
|
#endif
|
|
}
|
|
}
|
|
|
|
#if UNITY_USE_RECORDER
|
|
/// <summary>
|
|
/// GPU Elapsed time in milliseconds.
|
|
/// </summary>
|
|
public float gpuElapsedTime => m_Recorder.enabled ? m_Recorder.gpuElapsedNanoseconds / 1000000.0f : 0.0f;
|
|
/// <summary>
|
|
/// Number of times the Profiling Sampler has hit on the GPU
|
|
/// </summary>
|
|
public int gpuSampleCount => m_Recorder.enabled ? m_Recorder.gpuSampleBlockCount : 0;
|
|
/// <summary>
|
|
/// CPU Elapsed time in milliseconds (Command Buffer execution).
|
|
/// </summary>
|
|
public float cpuElapsedTime => m_Recorder.enabled ? m_Recorder.elapsedNanoseconds / 1000000.0f : 0.0f;
|
|
/// <summary>
|
|
/// Number of times the Profiling Sampler has hit on the CPU in the command buffer.
|
|
/// </summary>
|
|
public int cpuSampleCount => m_Recorder.enabled ? m_Recorder.sampleBlockCount : 0;
|
|
/// <summary>
|
|
/// CPU Elapsed time in milliseconds (Direct execution).
|
|
/// </summary>
|
|
public float inlineCpuElapsedTime => m_InlineRecorder.enabled ? m_InlineRecorder.elapsedNanoseconds / 1000000.0f : 0.0f;
|
|
/// <summary>
|
|
/// Number of times the Profiling Sampler has hit on the CPU.
|
|
/// </summary>
|
|
public int inlineCpuSampleCount => m_InlineRecorder.enabled ? m_InlineRecorder.sampleBlockCount : 0;
|
|
#else
|
|
/// <summary>
|
|
/// GPU Elapsed time in milliseconds.
|
|
/// </summary>
|
|
public float gpuElapsedTime => 0.0f;
|
|
/// <summary>
|
|
/// Number of times the Profiling Sampler has hit on the GPU
|
|
/// </summary>
|
|
public int gpuSampleCount => 0;
|
|
/// <summary>
|
|
/// CPU Elapsed time in milliseconds (Command Buffer execution).
|
|
/// </summary>
|
|
public float cpuElapsedTime => 0.0f;
|
|
/// <summary>
|
|
/// Number of times the Profiling Sampler has hit on the CPU in the command buffer.
|
|
/// </summary>
|
|
public int cpuSampleCount => 0;
|
|
/// <summary>
|
|
/// CPU Elapsed time in milliseconds (Direct execution).
|
|
/// </summary>
|
|
public float inlineCpuElapsedTime => 0.0f;
|
|
/// <summary>
|
|
/// Number of times the Profiling Sampler has hit on the CPU.
|
|
/// </summary>
|
|
public int inlineCpuSampleCount => 0;
|
|
#endif
|
|
// Keep the constructor private
|
|
ProfilingSamplerHTrace() { }
|
|
}
|
|
|
|
#if DEVELOPMENT_BUILD || UNITY_EDITOR
|
|
/// <summary>
|
|
/// Scoped Profiling markers
|
|
/// </summary>
|
|
#if !UNITY_2021
|
|
[IgnoredByDeepProfiler]
|
|
#endif
|
|
public struct HTraceProfilingScope : IDisposable
|
|
{
|
|
CommandBuffer m_Cmd;
|
|
bool m_Disposed;
|
|
ProfilingSamplerHTrace _mSamplerHTrace;
|
|
|
|
/// <summary>
|
|
/// Profiling Scope constructor
|
|
/// </summary>
|
|
/// <param name="cmd">Command buffer used to add markers and compute execution timings.</param>
|
|
/// <param name="samplerHTrace">Profiling Sampler to be used for this scope.</param>
|
|
public HTraceProfilingScope(CommandBuffer cmd, ProfilingSamplerHTrace samplerHTrace)
|
|
{
|
|
// NOTE: Do not mix with named CommandBuffers.
|
|
// Currently there's an issue which results in mismatched markers.
|
|
// The named CommandBuffer will close its "profiling scope" on execution.
|
|
// That will orphan ProfilingScope markers as the named CommandBuffer marker
|
|
// is their "parent".
|
|
// Resulting in following pattern:
|
|
// exec(cmd.start, scope.start, cmd.end) and exec(cmd.start, scope.end, cmd.end)
|
|
m_Cmd = cmd;
|
|
m_Disposed = false;
|
|
_mSamplerHTrace = samplerHTrace;
|
|
_mSamplerHTrace?.Begin(m_Cmd);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Dispose pattern implementation
|
|
/// </summary>
|
|
public void Dispose()
|
|
{
|
|
Dispose(true);
|
|
}
|
|
|
|
// Protected implementation of Dispose pattern.
|
|
void Dispose(bool disposing)
|
|
{
|
|
if (m_Disposed)
|
|
return;
|
|
|
|
// As this is a struct, it could have been initialized using an empty constructor so we
|
|
// need to make sure `cmd` isn't null to avoid a crash. Switching to a class would fix
|
|
// this but will generate garbage on every frame (and this struct is used quite a lot).
|
|
if (disposing)
|
|
{
|
|
_mSamplerHTrace?.End(m_Cmd);
|
|
}
|
|
|
|
m_Disposed = true;
|
|
}
|
|
}
|
|
#else
|
|
/// <summary>
|
|
/// Scoped Profiling markers
|
|
/// </summary>
|
|
public struct HTraceProfilingScope : IDisposable
|
|
{
|
|
/// <summary>
|
|
/// Profiling Scope constructor
|
|
/// </summary>
|
|
/// <param name="cmd">Command buffer used to add markers and compute execution timings.</param>
|
|
/// <param name="samplerHTrace">Profiling Sampler to be used for this scope.</param>
|
|
public HTraceProfilingScope(CommandBuffer cmd, ProfilingSamplerHTrace samplerHTrace)
|
|
{
|
|
}
|
|
|
|
/// <summary>
|
|
/// Dispose pattern implementation
|
|
/// </summary>
|
|
public void Dispose()
|
|
{
|
|
}
|
|
}
|
|
#endif
|
|
|
|
}
|