edu.cmu.tetradapp.editor.ScatterplotPanel Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of tetrad-gui Show documentation
There is a newer version: 7.6.6
package edu.cmu.tetradapp.editor;

import org.apache.commons.math3.util.FastMath;

import javax.swing.*;
import javax.swing.border.LineBorder;
import java.awt.*;
import java.awt.geom.Point2D;
import java.text.NumberFormat;
import java.util.Vector;

/**
 * This view draws the ScatterPlot using the information from the ScatterPlot class. It draws the ScatterPlot line,
 * axes, labels and the statistical values.
 *
 * @author Adrian Tang
 * @author josephramsey
 */
class ScatterplotPanel extends JPanel {
    private final NumberFormat nf;
    private final boolean removeZeroPointsPerPlot;
    private ScatterPlot scatterPlot;
    private boolean drawAxes = false;
    private int pointSize = 5;

    public ScatterplotPanel(ScatterPlot ScatterPlot) {
        this(ScatterPlot, false);
    }

    /**
     * Constructor.
     */
    public ScatterplotPanel(ScatterPlot ScatterPlot, boolean removeZeroPointsPerPlot) {
        this.scatterPlot = ScatterPlot;
        this.removeZeroPointsPerPlot = removeZeroPointsPerPlot;

        setBorder(BorderFactory.createLineBorder(Color.DARK_GRAY));

        this.nf = NumberFormat.getNumberInstance();
        this.nf.setMinimumFractionDigits(2);
        this.nf.setMaximumFractionDigits(2);

        setBorder(new LineBorder(Color.BLACK));
    }

    /**
     * Renders the view.
     */
    public void paintComponent(Graphics graphics) {
        double xmin = this.scatterPlot.getXmin();// - 0.000001;
        double xmax = this.scatterPlot.getXmax();// + 0.000001;
        double ymin = this.scatterPlot.getYmin();// - 0.000001;
        double ymax = this.scatterPlot.getYmax();// + 0.000001;

        Graphics2D g = (Graphics2D) graphics;

        g.setColor(Color.white);
        g.setFont(new Font("Dialog", Font.PLAIN, 11));
        g.fillRect(0, 0, getSize().width, getSize().height);

        int chartWidth = getSize().width;
        int chartHeight = getSize().height;

        final int xStringMin = 10;
        int xMin = drawAxes ? 50 : 4;
        int xMax = drawAxes ? chartWidth - 20 : chartWidth - 4;
        int xRange = xMax - xMin;
        int yMin = drawAxes ? 30 : 4;
        int yMax = drawAxes ? chartHeight - 20 : chartHeight - 4;
        int yRange = yMax - yMin;

        // Draw axis lines.
        if (drawAxes) {
            g.setPaint(Color.black);
            g.drawLine(xMin, yMax, xMax, yMax);
            g.drawLine(xMin, yMin, xMin, yMax);

            /* draws the labels for the corresponding experiment and sample names */
            g.setFont(g.getFont().deriveFont(11f).deriveFont(Font.BOLD));

            /* draws axis labels and scale */
            g.drawString(this.nf.format(ymax), 2 + xStringMin, yMin + 7);
            g.drawString(this.nf.format(ymin), 2 + xStringMin, yMax);
            g.drawString(this.nf.format(xmax), xMax - 20, yMax + 14);
            g.drawString(this.nf.format(xmin), 20 + 30, yMax + 14);
            g.drawString(this.scatterPlot.getXvar(), xMin + (xRange / 2) - 10, yMax + 14);
            g.translate(xMin - 7, yMin + (yRange / 2) + 10);
            g.rotate(-FastMath.PI / 2.0);
            g.drawString(this.scatterPlot.getYvar(), xStringMin, 0);
            g.rotate(FastMath.PI / 2.0);
            g.translate(-(xMin - 7), -(yMin + (yRange / 2) + 10));
        }

        // Draw ScatterPlot of the values.
        Vector pts = this.scatterPlot.getSievedValues();
        double _xRange = xmax - xmin;
        double _yRange = ymax - ymin;
        int x, y;

        g.setColor(Color.RED.darker());
        for (Point2D.Double _pt : pts) {
            if (Double.isNaN(_pt.getX()) || Double.isNaN(_pt.getY())) continue;

            if (removeZeroPointsPerPlot) {
                if (_pt.getX() == 0 || _pt.getY() == 0) continue;
            }

            x = (int) (((_pt.getX() - xmin) / _xRange) * xRange + xMin);
            y = (int) (((ymax - _pt.getY()) / _yRange) * yRange + yMin);
            g.fillOval(x - pointSize / 2, y - pointSize / 2, pointSize, pointSize);
        }

        /* draws best-fit line */
        if (this.scatterPlot.isIncludeLine()) {
            double a = this.scatterPlot.getRegressionCoeff();
            double b = this.scatterPlot.getRegressionIntercept();

            double x1, y1 = 0;

            for (x1 = xmin; x1 <= xmax; x1 += 0.01) {
                y1 = a * x1 + b;
                if (y1 >= ymin && y1 <= ymax) {
                    break;
                }
            }

            double x2, y2 = 0;

            for (x2 = xmax; x2 >= xmin; x2 -= 0.01) {
                y2 = a * x2 + b;
                if (y2 >= ymin && y2 <= ymax) {
                    break;
                }
            }

            int xa = (int) (((x1 - xmin) / _xRange) * xRange + xMin);
            int ya = (int) (((ymax - y1) / _yRange) * yRange + yMin);

            int xb = (int) (((x2 - xmin) / _xRange) * xRange + xMin);
            int yb = (int) (((ymax - y2) / _yRange) * yRange + yMin);

            g.setColor(Color.DARK_GRAY);
            g.setStroke(new BasicStroke(2));
            g.drawLine(xa, ya, xb, yb);
        }

        /* draws statistical values */
        if (this.scatterPlot.isIncludeLine()) {
            g.setColor(Color.black);
            this.nf.setMinimumFractionDigits(3);
            this.nf.setMaximumFractionDigits(3);
            double r = this.scatterPlot.getCorrelationCoeff();
            double p = this.scatterPlot.getCorrelationPValue();

            if (drawAxes) {
                g.setStroke(new BasicStroke(3));
                g.drawString("Correlation coef = " + this.nf.format(r)
                        + "  (p=" + this.nf.format(p) + ")", 70, 21);
            }
        }
    }

    public void setDrawAxes(boolean drawAxes) {
        this.drawAxes = drawAxes;
    }

    public void setPointSize(int pointSize) {
        this.pointSize = pointSize;
    }
}