Class ComplexCarpet

src/org/opensourcephysics/display2d/ComplexCarpet.java:10–105

package org.opensourcephysics.display2d;

public class ComplexCarpet extends ComplexInterpolatedPlot implements ICarpet {
	/**
	 * Constructor ComplexCarpet
	 * 
	 * @param griddata
	 */
	public ComplexCarpet(GridData griddata) {
		super(griddata);
		setShowGridLines(false);
	}

	@Override
	public void clearData() {
		if (griddata instanceof ArrayData) {
			double[][][] data = griddata.getData();
			for (int ix = 0, nx = griddata.getNx(); ix < nx; ix++) {
				for (int iy = 0, ny = griddata.getNy(); iy < ny; iy++) {
					data[0][ix][iy] = 0;
					data[1][ix][iy] = 0;
					data[2][ix][iy] = 0;
				}
			}
		} else { // point data
			double[][][] data = griddata.getData();
			for (int ix = 0, nx = griddata.getNx(); ix < nx; ix++) {
				for (int iy = 0, ny = griddata.getNy(); iy < ny; iy++) {
					data[nx][ny][2] = 0;
					data[nx][ny][3] = 0;
					data[nx][ny][4] = 0;
				}
			}
		}
		update();
	}

	@Override
	public void setTopRow(double[][] line) {
		if (image == null) {
			return;
		}
		if (pixelData.length != image.getWidth() * image.getHeight() * 4) {
			return;
		}
		if (griddata instanceof ArrayData) { // replace top row of array
			for (int c = 0; c < line.length; c++) {
				double[][] data = griddata.getData()[c];
				int len = data[0].length - 1;
				for (int ix = 0, nx = data.length; ix < nx; ix++) {
					System.arraycopy(data[ix], 0, data[ix], 1, len);
					data[ix][0] = line[c][ix];
				}
			}
		} else {
			double[][][] data = griddata.getData();
			for (int ix = 0, nx = data.length; ix < nx; ix++) {
				int len = line.length;
				for (int ny = data[0].length - 1, iy = ny; iy > 0; iy--) {
					System.arraycopy(data[ix][iy - 1], 2, data[ix][iy], 2, len);