org.drugis.addis.gui.wizard.TreatmentCategorizationOverviewWizardStep.java Source code

Introduction

Here is the source code for org.drugis.addis.gui.wizard.TreatmentCategorizationOverviewWizardStep.java

Source

/*
 * This file is part of ADDIS (Aggregate Data Drug Information System).
 * ADDIS is distributed from http://drugis.org/.
 * Copyright  2009 Gert van Valkenhoef, Tommi Tervonen.
 * Copyright  2010 Gert van Valkenhoef, Tommi Tervonen, Tijs Zwinkels,
 * Maarten Jacobs, Hanno Koeslag, Florin Schimbinschi, Ahmad Kamal, Daniel
 * Reid.
 * Copyright  2011 Gert van Valkenhoef, Ahmad Kamal, Daniel Reid, Florin
 * Schimbinschi.
 * Copyright  2012 Gert van Valkenhoef, Daniel Reid, Jol Kuiper, Wouter
 * Reckman.
 * Copyright  2013 Gert van Valkenhoef, Jol Kuiper.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package org.drugis.addis.gui.wizard;

import java.awt.Color;
import java.awt.FontMetrics;
import java.awt.GridLayout;
import java.awt.Paint;
import java.awt.Shape;
import java.awt.geom.RoundRectangle2D;

import javax.swing.JPanel;

import org.apache.commons.collections15.Transformer;
import org.drugis.addis.entities.treatment.DecisionTree;
import org.drugis.addis.entities.treatment.DecisionTreeEdge;
import org.drugis.addis.entities.treatment.DecisionTreeNode;
import org.drugis.addis.entities.treatment.LeafNode;
import org.drugis.addis.presentation.wizard.TreatmentCategorizationWizardPresentation;

import edu.uci.ics.jung.algorithms.layout.Layout;
import edu.uci.ics.jung.algorithms.layout.TreeLayout;
import edu.uci.ics.jung.visualization.GraphZoomScrollPane;
import edu.uci.ics.jung.visualization.VisualizationViewer;
import edu.uci.ics.jung.visualization.decorators.ConstantDirectionalEdgeValueTransformer;
import edu.uci.ics.jung.visualization.decorators.ToStringLabeller;
import edu.uci.ics.jung.visualization.renderers.Renderer.VertexLabel.Position;

public class TreatmentCategorizationOverviewWizardStep extends AbstractTreatmentCategorizationWizardStep {
    private static final long serialVersionUID = -3991691781012756118L;

    public TreatmentCategorizationOverviewWizardStep(final TreatmentCategorizationWizardPresentation pm) {
        super(pm, "Overview", "Overview of created treatment.", null);
    }

    @Override
    protected void initialize() {
        d_pm.transformTree();
        setLayout(new GridLayout(1, 1));
        rebuildPanel();
    }

    @Override
    protected JPanel buildPanel() {
        return buildOverview(d_pm.getBean().getDecisionTree());
    }

    /**
     * Builds a graph overview using Jung's functionality.
     * @param tree
     * @return panel
     */
    public static JPanel buildOverview(final DecisionTree tree) {
        return new GraphZoomScrollPane(buildDecisionTreeView(tree));
    }

    public static VisualizationViewer<DecisionTreeNode, DecisionTreeEdge> buildDecisionTreeView(
            final DecisionTree tree) {
        // Crazy hack because sizes start at 600x600 by default.
        final Layout<DecisionTreeNode, DecisionTreeEdge> layout = new TreeLayout<DecisionTreeNode, DecisionTreeEdge>(
                new DecisionTree(new LeafNode()), 150, 75);
        layout.getSize().height = 1;
        layout.getSize().width = 1;
        layout.setGraph(tree);

        final VisualizationViewer<DecisionTreeNode, DecisionTreeEdge> vv = new VisualizationViewer<DecisionTreeNode, DecisionTreeEdge>(
                layout);

        vv.setVertexToolTipTransformer(new ToStringLabeller<DecisionTreeNode>());
        vv.getRenderContext().setVertexFillPaintTransformer(new Transformer<DecisionTreeNode, Paint>() {
            public Paint transform(final DecisionTreeNode node) {
                return (node instanceof LeafNode) ? new Color(0.55f, 0.55f, 1.0f) : Color.ORANGE;
            }
        });

        vv.getRenderContext().setVertexShapeTransformer(new Transformer<DecisionTreeNode, Shape>() {
            public Shape transform(final DecisionTreeNode input) {
                final FontMetrics fontMetrics = vv.getGraphics().getFontMetrics();
                final double width = fontMetrics.stringWidth(input.toString()) + 6;
                final double height = fontMetrics.getHeight() + 2;
                final double arc = 5;
                return new RoundRectangle2D.Double(-width / 2, -height / 2, width, height, arc, arc);
            }
        });

        vv.getRenderer().getVertexLabelRenderer().setPosition(Position.CNTR);
        vv.getRenderContext().setVertexLabelTransformer(new ToStringLabeller<DecisionTreeNode>());

        vv.getRenderContext().setEdgeLabelTransformer(new ToStringLabeller<DecisionTreeEdge>());
        vv.getRenderContext().getEdgeLabelRenderer().setRotateEdgeLabels(false);
        vv.getRenderContext().setEdgeLabelClosenessTransformer(
                new ConstantDirectionalEdgeValueTransformer<DecisionTreeNode, DecisionTreeEdge>(0.5, 0.4));
        return vv;
    }
}