import javax.swing.*;
import org.math.plot.*;
 
import static java.lang.Math.*;
 
import static org.math.array.DoubleArray.*;
 
public class GridPlotsExample {
	public static void main(String[] args) {
 
		// define your data
		double[] x = increment(0.0, 0.1, 1.0); // x = 0.0:0.1:1.0
		double[] y = increment(0.0, 0.05, 1.0);// y = 0.0:0.05:1.0
		double[][] z1 = f1(x, y);
		double[][] z2 = f2(x, y);
 
		// create your PlotPanel (you can use it as a JPanel) with a legend at SOUTH
		Plot3DPanel plot = new Plot3DPanel("SOUTH");
 
		// add grid plot to the PlotPanel
		plot.addGridPlot("z=cos(PI*x)*sin(PI*y)", x, y, z1);
		plot.addGridPlot("z=sin(PI*x)*cos(PI*y)", x, y, z2);
 
		// put the PlotPanel in a JFrame like a JPanel
		JFrame frame = new JFrame("a plot panel");
		frame.setSize(600, 600);
		frame.setContentPane(plot);
		frame.setVisible(true);
 
	}
 
	// function definition: z=cos(PI*x)*sin(PI*y)
	public static double f1(double x, double y) {
		double z = cos(x * PI) * sin(y * PI);
		return z;
	}
 
	// grid version of the function
	public static double[][] f1(double[] x, double[] y) {
		double[][] z = new double[y.length][x.length];
		for (int i = 0; i < x.length; i++)
			for (int j = 0; j < y.length; j++)
				z[j][i] = f1(x[i], y[j]);
		return z;
	}
 
	// another function definition: z=sin(PI*x)*cos(PI*y)
	public static double f2(double x, double y) {
		double z = sin(x * PI) * cos(y * PI);
		return z;
	}
 
	// grid version of the function
	public static double[][] f2(double[] x, double[] y) {
		double[][] z = new double[y.length][x.length];
		for (int i = 0; i < x.length; i++)
			for (int j = 0; j < y.length; j++)
				z[j][i] = f2(x[i], y[j]);
		return z;
	}
}