org.math.plot.plots.CloudPlot3D 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!
/*
 * Created on 13 juil. 07 by richet
 */
package org.math.plot.plots;

import java.awt.Color;

import javax.swing.JFrame;

import org.math.plot.FrameView;
import org.math.plot.Plot3DPanel;
import org.math.plot.PlotPanel;
import org.math.plot.render.AbstractDrawer;
import org.math.plot.utils.Array;

public class CloudPlot3D extends Plot {

    double[][] topNW;
    double[][] topNE;
    double[][] topSW;
    double[][] topSE;
    double[][] botNW;
    double[][] botNE;
    double[][] botSW;
    double[][] botSE;
    double[] width_constant = {-1, -1, -1};
    double[][] XY;
    float[] f;
    boolean fill_shape = true;

    public CloudPlot3D(String n, Color c, double[][] _XYcard, double wX, double wY, double wZ) {
        super(n, c);
        splitXYf(_XYcard);
        width_constant = new double[]{wX, wY, wZ};

        build();
    }

    private void splitXYf(double[][] xycard) {
        XY = new double[xycard.length][3];
        f = new float[xycard.length];
        float normf = 0;
        for (int i = 0; i < xycard.length; i++) {
            XY[i][0] = xycard[i][0];
            XY[i][1] = xycard[i][1];
            XY[i][2] = xycard[i][2];
            f[i] = (float) xycard[i][3];
            normf += f[i];//Math.max(normf, f[i]);
        }

        for (int i = 0; i < f.length; i++) {
            f[i] = f[i] / normf;
        }

    }

    private void build() {
        if (width_constant[0] > 0) {
            topNW = new double[XY.length][];
            topNE = new double[XY.length][];
            topSW = new double[XY.length][];
            topSE = new double[XY.length][];
            botNW = new double[XY.length][];
            botNE = new double[XY.length][];
            botSW = new double[XY.length][];
            botSE = new double[XY.length][];
            for (int i = 0; i < XY.length; i++) {
                topNW[i] = new double[]{XY[i][0] - width_constant[0] / 2, XY[i][1] + width_constant[1] / 2, XY[i][2] + width_constant[2] / 2};
                topNE[i] = new double[]{XY[i][0] + width_constant[0] / 2, XY[i][1] + width_constant[1] / 2, XY[i][2] + width_constant[2] / 2};
                topSW[i] = new double[]{XY[i][0] - width_constant[0] / 2, XY[i][1] - width_constant[1] / 2, XY[i][2] + width_constant[2] / 2};
                topSE[i] = new double[]{XY[i][0] + width_constant[0] / 2, XY[i][1] - width_constant[1] / 2, XY[i][2] + width_constant[2] / 2};
                botNW[i] = new double[]{XY[i][0] - width_constant[0] / 2, XY[i][1] + width_constant[1] / 2, XY[i][2] - width_constant[2] / 2};
                botNE[i] = new double[]{XY[i][0] + width_constant[0] / 2, XY[i][1] + width_constant[1] / 2, XY[i][2] - width_constant[2] / 2};
                botSW[i] = new double[]{XY[i][0] - width_constant[0] / 2, XY[i][1] - width_constant[1] / 2, XY[i][2] - width_constant[2] / 2};
                botSE[i] = new double[]{XY[i][0] + width_constant[0] / 2, XY[i][1] - width_constant[1] / 2, XY[i][2] - width_constant[2] / 2};
            }
        }
    }

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

        draw.canvas.includeInBounds(botSW[0]);
        draw.canvas.includeInBounds(topNE[XY.length - 1]);

        draw.setColor(c);
        draw.setLineType(AbstractDrawer.CONTINOUS_LINE);
        for (int i = 0; i < XY.length; i++) {
            if (f[i] > 0) {
                draw.fillPolygon(f[i], topNW[i], topNE[i], topSE[i], topSW[i]);
                draw.fillPolygon(f[i], botNW[i], botNE[i], botSE[i], botSW[i]);

                draw.fillPolygon(f[i], botNW[i], botNE[i], topNE[i], topNW[i]);
                draw.fillPolygon(f[i], botSW[i], botSE[i], topSE[i], topSW[i]);

                draw.fillPolygon(f[i], botNW[i], botSW[i], topSW[i], topNW[i]);
                draw.fillPolygon(f[i], botNE[i], botSE[i], topSE[i], topNE[i]);
            }
        }
    }

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

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

    @Override
    public double[][] getBounds() {
        double[][] b = new double[][]{Array.min(XY), Array.max(XY)};
        for (int i = 0; i < b[0].length; i++) {
            b[0][i] = b[0][i] - width_constant[i] / 2;
            b[1][i] = b[1][i] + width_constant[i] / 2;
        }
        return b;
    }

    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 p = new Plot3DPanel();

        //triangular random cloud (as sum of two uniform random numbers)
        double[][] cloud = new double[100][3];
        for (int i = 0; i < cloud.length; i++) {
            cloud[i][0] = Math.random() + Math.random();
            cloud[i][1] = Math.random() + Math.random();
            cloud[i][2] = Math.random() + Math.random();
        }
        p.addCloudPlot("cloud", Color.RED, cloud, 3, 3, 3);

        double[][] cloud2 = new double[100][3];
        for (int i = 0; i < cloud.length; i++) {
            cloud2[i][0] = 2 + Math.random() + Math.random();
            cloud2[i][1] = 2 + Math.random() + Math.random();
            cloud2[i][2] = 2 + Math.random() + Math.random();
        }
        p.addCloudPlot("cloud2", Color.RED, cloud2, 3, 3, 3);

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