ProfilerInvocationStrategy.java

package net.mostlyoriginal.plugin.profiler;

import com.artemis.BaseSystem;
import com.artemis.SystemInvocationStrategy;
import com.artemis.utils.Bag;
import com.artemis.utils.ImmutableBag;

/**
 * {@link SystemInvocationStrategy} that will create a profiler for all systems that don't already have one
 * Can be used in addition to or instead of {@link com.artemis.annotations.Profile} annotation
 * <p>
 * In addition creates {@link SystemProfiler} with name "Frame" for total frame time
 * It can be accessed with {@link SystemProfiler#get(String)}
 *
 * @author piotr-j
 * @author Daan van Yperen
 */
public class ProfilerInvocationStrategy extends SystemInvocationStrategy {
    private boolean initialized = false;

    protected SystemProfiler frameProfiler;
    protected SystemProfiler[] profilers;

    @Override
    protected void process() {

        if (!initialized) {
            initialize();
            initialized = true;
        }

        frameProfiler.start();
        processProfileSystems(systems);
        frameProfiler.stop();
    }

    private void processProfileSystems(Bag<BaseSystem> systems) {
        final Object[] systemsData = systems.getData();
        for (int i = 0, s = systems.size(); s > i; i++) {
            if (disabled.get(i))
                continue;

            updateEntityStates();
            processProfileSystem(profilers[i], (BaseSystem) systemsData[i]);
        }

        updateEntityStates();
    }

    private void processProfileSystem(SystemProfiler profiler, BaseSystem system) {
        if (profiler != null) profiler.start();
        system.process();
        if (profiler != null) profiler.stop();
    }

    protected void initialize() {
        createFrameProfiler();
        createSystemProfilers();
    }

    private void createSystemProfilers() {
        final ImmutableBag<BaseSystem> systems = world.getSystems();
        profilers = new SystemProfiler[systems.size()];
        for (int i = 0; i < systems.size(); i++) {
            profilers[i] = createSystemProfiler(systems.get(i));
        }
    }

    private SystemProfiler createSystemProfiler(BaseSystem system) {
        SystemProfiler old = SystemProfiler.getFor(system);
        if (old == null) {
            old = SystemProfiler.createFor(system, world);
        }
        return old;
    }

    private void createFrameProfiler() {
        frameProfiler = SystemProfiler.create("Frame");
        frameProfiler.setColor(1, 1, 1, 1);
    }
}