Class Profiler

src/main/java/pascal/taie/analysis/pta/plugin/Profiler.java:56–165

Profiler to help identify analysis hot spots in the analyzed program and assist performance tuning for pointer analysis.

 * and assist performance tuning for pointer analysis.
 */
public class Profiler implements Plugin {

    private static final Logger logger = LogManager.getLogger(Profiler.class);

    private static final String PROFILE_FILE = "pta-profile.txt";

    /**
     * Reports the results for top N elements.
     */
    private static final int TOP_N = 100;

    private Solver solver;

    private CSManager csManager;

    private final Map<CSVar, MutableInt> csVarVisited = Maps.newMap();

    private final Map<Var, MutableInt> varVisited = Maps.newMap();

    @Override
    public void setSolver(Solver solver) {
        this.solver = solver;
        this.csManager = solver.getCSManager();
    }

    @Override
    public void onNewPointsToSet(CSVar csVar, PointsToSet pts) {
        csVarVisited.computeIfAbsent(csVar, __ -> new MutableInt(0)).add(1);
        varVisited.computeIfAbsent(csVar.getVar(), __ -> new MutableInt(0)).add(1);
    }

    @Override
    public void onFinish() {
        File outFile = new File(World.get().getOptions().getOutputDir(), PROFILE_FILE);
        try (PrintStream out = new PrintStream(new FileOutputStream(outFile))) {
            logger.info("Dumping pointer analysis profile to {}",
                    outFile.getAbsolutePath());
            // report variables
            reportTop(out, "frequently-visited variables",
                    varVisited, v -> v.getMethod() + "/" + v.getName());
            reportTop(out, "frequently-visited CS variables",
                    csVarVisited, CSVar::toString);
            // count and report methods
            Map<JMethod, MutableInt> methodVarVisited = Maps.newMap();
            varVisited.forEach((v, times) ->
                    methodVarVisited.computeIfAbsent(v.getMethod(),
                                    __ -> new MutableInt(0))
                            .add(times.intValue()));
            reportTop(out, "method containers (of frequently-visited variables)",
                    methodVarVisited, JMethod::toString);
            Map<CSMethod, MutableInt> csMethodVarVisited = Maps.newMap();
            csVarVisited.forEach((v, times) -> {
                CSMethod method = csManager.getCSMethod(
                        v.getContext(), v.getVar().getMethod());
                csMethodVarVisited.computeIfAbsent(method,
                                __ -> new MutableInt(0))
                        .add(times.intValue());
            });