org.math.plot.plots.BoxPlot3D Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of JMathPlot Show documentation
The newest version!
package org.math.plot.plots;

import java.awt.*;

import javax.swing.JFrame;
import org.math.plot.FrameView;
import org.math.plot.Plot3DPanel;
import org.math.plot.PlotPanel;
import org.math.plot.render.*;
import org.math.plot.utils.Array;

public class BoxPlot3D extends Plot {

    double[] Xmin;
    double[] Xmax;
    double[] Ymin;
    double[] Ymax;
    double[] Zmin;
    double[] Zmax;
    double[][] widths;
    double[][] XY;

    public BoxPlot3D(double[][] _XY, double[][] w, Color c, String n) {
        super(n, c);
        XY = _XY;
        widths = w;

        // double[] datasMin = Array.min(XY);
        // double[] datasMax = Array.max(XY);
        // double[] widthsMax = Array.max(widths);
        // double[] min = { datasMin[0] - widthsMax[0] / 2, datasMin[1] -
        // widthsMax[1] / 2, datasMin[2] - widthsMax[2] / 2 };
        // double[] max = { datasMax[0] + widthsMax[0] / 2, datasMax[1] +
        // widthsMax[1] / 2, datasMax[2] + widthsMax[2] / 2 };
        // base.includeInBounds(min);
        // base.includeInBounds(max);

        Xmin = new double[XY.length];
        Xmax = new double[XY.length];
        Ymin = new double[XY.length];
        Ymax = new double[XY.length];
        Zmin = new double[XY.length];
        Zmax = new double[XY.length];
        for (int i = 0; i < XY.length; i++) {
            Xmin[i] = XY[i][0] - widths[i][0] / 2;
            Xmax[i] = XY[i][0] + widths[i][0] / 2;
            Ymin[i] = XY[i][1] - widths[i][1] / 2;
            Ymax[i] = XY[i][1] + widths[i][1] / 2;
            Zmin[i] = XY[i][2] - widths[i][2] / 2;
            Zmax[i] = XY[i][2] + widths[i][2] / 2;
        }
    }

    public void plot(AbstractDrawer draw, Color c) {
        if (!visible) {
            return;
        }

        draw.setColor(c);
        draw.setLineType(AbstractDrawer.CONTINOUS_LINE);
        for (int i = 0; i < XY.length; i++) {
            draw.drawLine(new double[]{Xmin[i], Ymin[i], Zmin[i]}, new double[]{Xmax[i], Ymin[i], Zmin[i]});
            draw.drawLine(new double[]{Xmax[i], Ymin[i], Zmin[i]}, new double[]{Xmax[i], Ymax[i], Zmin[i]});
            draw.drawLine(new double[]{Xmax[i], Ymax[i], Zmin[i]}, new double[]{Xmin[i], Ymax[i], Zmin[i]});
            draw.drawLine(new double[]{Xmin[i], Ymax[i], Zmin[i]}, new double[]{Xmin[i], Ymin[i], Zmin[i]});

            draw.drawLine(new double[]{Xmin[i], Ymin[i], Zmax[i]}, new double[]{Xmax[i], Ymin[i], Zmax[i]});
            draw.drawLine(new double[]{Xmax[i], Ymin[i], Zmax[i]}, new double[]{Xmax[i], Ymax[i], Zmax[i]});
            draw.drawLine(new double[]{Xmax[i], Ymax[i], Zmax[i]}, new double[]{Xmin[i], Ymax[i], Zmax[i]});
            draw.drawLine(new double[]{Xmin[i], Ymax[i], Zmax[i]}, new double[]{Xmin[i], Ymin[i], Zmax[i]});

            draw.drawLine(new double[]{Xmin[i], Ymin[i], Zmin[i]}, new double[]{Xmin[i], Ymin[i], Zmax[i]});
            draw.drawLine(new double[]{Xmax[i], Ymin[i], Zmin[i]}, new double[]{Xmax[i], Ymin[i], Zmax[i]});
            draw.drawLine(new double[]{Xmin[i], Ymax[i], Zmin[i]}, new double[]{Xmin[i], Ymax[i], Zmax[i]});
            draw.drawLine(new double[]{Xmax[i], Ymax[i], Zmin[i]}, new double[]{Xmax[i], Ymax[i], Zmax[i]});

            draw.drawDot(XY[i]);
        }
    }

    @Override
    public void setData(double[][] d) {
        datapanel = null;
        XY = d;
    }

    @Override
    public double[][] getData() {
        return XY;
    }

    @Override
    public double[][] getBounds() {
        return new double[][]{{Array.min(Xmin), Array.min(Ymin), Array.min(Zmin)}, {Array.max(Xmax), Array.max(Ymax), Array.max(Zmax)}};
    }

    public void setDataWidth(double[][] w) {
        widths = w;
    }

    public double[][] getDataWidth() {
        return widths;
    }

    public void setData(double[][] d, double[][] w) {
        setData(d);
        widths = w;
    }

    public double[] isSelected(int[] screenCoordTest, AbstractDrawer draw) {
        for (int i = 0; i < XY.length; i++) {
            int[] screenCoord = draw.project(XY[i]);

            if ((screenCoord[0] + note_precision > screenCoordTest[0]) && (screenCoord[0] - note_precision < screenCoordTest[0])
                    && (screenCoord[1] + note_precision > screenCoordTest[1]) && (screenCoord[1] - note_precision < screenCoordTest[1])) {
                return XY[i];
            }
        }
        return null;
    }

    public static void main(String[] args) {
        Plot3DPanel plotpanel = new Plot3DPanel();
        for (int i = 0; i < 1; i++) {
            double[][] receiverXYZ = new double[100][6];
            for (int j = 0; j < receiverXYZ.length; j++) {
                receiverXYZ[j][0] = /*1 + */ Math.random();
                receiverXYZ[j][1] = /*100 * */ Math.random();
                receiverXYZ[j][2] = /*100 * */ Math.random();
                receiverXYZ[j][3] = /*1 + */ Math.random() / 10;
                receiverXYZ[j][4] = /*100 * */ Math.random() / 10;
                receiverXYZ[j][5] = /*100 * */ Math.random() / 10;
            }
            int receiverPlotDataIndex = plotpanel.addBoxPlot("Receivers", Color.orange, receiverXYZ);
        }

        plotpanel.setLegendOrientation(PlotPanel.SOUTH);
        new FrameView(plotpanel).setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
    }
}