source: autoquest-core-usability-evaluation-test/src/main/java/de/ugoe/cs/autoquest/usability/sandboxapp/CreateHeatmapImageFromTaskTreeDemoApp.java @ 1030

package de.ugoe.cs.autoquest.usability.sandboxapp;

import static de.ugoe.cs.autoquest.usability.tasktree.filter.EventTypeFilter.MOUSE_BUTTON_INTERACTION;

import java.io.File;
import java.io.IOException;

import org.tc33.jheatchart.HeatChart;

import de.ugoe.cs.autoquest.eventcore.gui.MouseButtonInteraction;
import de.ugoe.cs.autoquest.tasktrees.treeifc.IEventTask;
import de.ugoe.cs.autoquest.tasktrees.treeifc.ITaskTree;
import de.ugoe.cs.autoquest.tasktrees.treeifc.ITaskTreeNode;
import de.ugoe.cs.autoquest.usability.tasktree.filter.FilterStatistic;
import de.ugoe.cs.autoquest.usability.tasktree.filter.IterativeDFSFilterStrategy;
import de.ugoe.cs.autoquest.usability.tasktree.filter.TaskTreeFilter;
import de.ugoe.cs.autoquest.usability.testutil.GenerateTaskTreeUtil;

public class CreateHeatmapImageFromTaskTreeDemoApp {

    private static final String fileToParse = "javatrace.xml";
    
    private static final String heatmapFilename = "click-heatmap.png";

    /**
     * <p>
     * TODO: comment
     * </p>
     * 
     * @param args
     * @throws IOException
     */
    public static void main(String[] args) throws IOException {
        ITaskTree taskTree = GenerateTaskTreeUtil.getTaskTreeFromFile(fileToParse);
        FilterStatistic mouseInteractions = filterMouseInteractions(taskTree);
        double[][] data = fillHeatmapDataMatrix(mouseInteractions);
        createAndSaveHeatmapImage(mouseInteractions, data);
    }

    private static FilterStatistic filterMouseInteractions(ITaskTree taskTree) {
        return new TaskTreeFilter(new IterativeDFSFilterStrategy())
            .filterByEventType(MOUSE_BUTTON_INTERACTION).from(taskTree);
    }

    private static double[][] fillHeatmapDataMatrix(FilterStatistic mouseInteractions) {
        double data[][] = new double[1024][768];
        for (ITaskTreeNode node : mouseInteractions.nodesMatchedFilter()) {
            MouseButtonInteraction interaction =
                (MouseButtonInteraction) ((IEventTask) node).getEventType();
            int x = interaction.getX();
            int y = interaction.getY();
            data[x][y]++;
            boolean x_has_left = x - 1 >= 0;
            boolean x_has_right = x + 1 <= 800;
            boolean y_has_top = y - 1 >= 0;
            boolean y_has_bottem = y + 1 <= 800;
            if (x_has_left) {
                data[x - 1][y]++; // left
                if (y_has_top)
                    data[x - 1][y + 1]++; // topleft
                if (y_has_bottem)
                    data[x - 1][y - 1]++; // bottomleft
            }
            if (x_has_right) {
                data[x + 1][y]++; // right
                if (y_has_top)
                    data[x + 1][y + 1]++; // topright
                if (y_has_bottem)
                    data[x + 1][y - 1]++; // bottomright
            }
            if (y_has_top) {
                data[x][y + 1]++; // top
            }
            if (y_has_bottem) {
                data[x][y - 1]++; // bottom
            }
        }
        return data;
    }

    private static void createAndSaveHeatmapImage(FilterStatistic mouseInteractions, double[][] data)
        throws IOException
    {
        HeatChart hc = new HeatChart(data);
        hc.setTitle(mouseInteractions.nodesMatchedFilter().size() + "Mouse interactions");
        // Cautious, took some time!
        // Also you have to run the JVM with extended heapspace, like -Xms512m -Xmx4096m
        hc.saveToFile(new File(heatmapFilename));
    }

}