58 lines
1.9 KiB
Java
58 lines
1.9 KiB
Java
package net.minecraft.util.profiling;
|
|
|
|
import com.mojang.jtracy.TracyClient;
|
|
import java.util.Objects;
|
|
import java.util.concurrent.atomic.AtomicInteger;
|
|
|
|
public final class Profiler {
|
|
private static final ThreadLocal<TracyZoneFiller> TRACY_FILLER = ThreadLocal.withInitial(TracyZoneFiller::new);
|
|
private static final ThreadLocal<ProfilerFiller> ACTIVE = new ThreadLocal<>();
|
|
private static final AtomicInteger ACTIVE_COUNT = new AtomicInteger();
|
|
|
|
private Profiler() {
|
|
}
|
|
|
|
public static Profiler.Scope use(ProfilerFiller p_361779_) {
|
|
startUsing(p_361779_);
|
|
return Profiler::stopUsing;
|
|
}
|
|
|
|
private static void startUsing(ProfilerFiller p_367831_) {
|
|
if (ACTIVE.get() != null) {
|
|
throw new IllegalStateException("Profiler is already active");
|
|
} else {
|
|
ProfilerFiller profilerfiller = decorateFiller(p_367831_);
|
|
ACTIVE.set(profilerfiller);
|
|
ACTIVE_COUNT.incrementAndGet();
|
|
profilerfiller.startTick();
|
|
}
|
|
}
|
|
|
|
private static void stopUsing() {
|
|
ProfilerFiller profilerfiller = ACTIVE.get();
|
|
if (profilerfiller == null) {
|
|
throw new IllegalStateException("Profiler was not active");
|
|
} else {
|
|
ACTIVE.remove();
|
|
ACTIVE_COUNT.decrementAndGet();
|
|
profilerfiller.endTick();
|
|
}
|
|
}
|
|
|
|
private static ProfilerFiller decorateFiller(ProfilerFiller p_364301_) {
|
|
return ProfilerFiller.combine(getDefaultFiller(), p_364301_);
|
|
}
|
|
|
|
public static ProfilerFiller get() {
|
|
return ACTIVE_COUNT.get() == 0 ? getDefaultFiller() : Objects.requireNonNullElseGet(ACTIVE.get(), Profiler::getDefaultFiller);
|
|
}
|
|
|
|
private static ProfilerFiller getDefaultFiller() {
|
|
return (ProfilerFiller)(TracyClient.isAvailable() ? TRACY_FILLER.get() : InactiveProfiler.INSTANCE);
|
|
}
|
|
|
|
public interface Scope extends AutoCloseable {
|
|
@Override
|
|
void close();
|
|
}
|
|
} |