com.aparapi.examples.afmandelbrot.AfKernel Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of aparapi-examples Show documentation
A collection of examples for the aparapi framework.
The newest version!
/**
 * Copyright (c) 2016 - 2018 Syncleus, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.aparapi.examples.afmandelbrot;

import com.aparapi.Kernel;

/**
 * Aparapi Fractals
 *
 * the kernel executes the math with complex numbers. Coordinates refer to
 * complex plane. result is a vector of number of iterations, It is transformed
 * in a color in the GUI, not here
 *
 * @author marco.stefanetti at gmail.com
 * @version $Id: $Id
 * @since 2.0.1
 */
public class AfKernel extends Kernel {

	/** the result of all calculations, the iterations for each pixel */
	private int[][] result;

	/** max iterations for pixel */
	private int max_iterations;

	/** starting point, x lower-left */
	private double cx1;
	/** starting point, y lower-left */
	private double cy1;

	/** max width */
	private int wmax;
	/** max height */
	private int hmax;

	/** one pixel width */
	private double wx;
	/** one pixel height */
	private double hy;

	/**
	 * no values on the constructor, we will reuse the kernel after init
	 */
	public AfKernel() {
		super();
	}

	/**
	 * sets the parameters, send only few double to the device and a pointer to an
	 * array to retrieve iterations
	 *
	 * @param _cx1 a double.
	 * @param _cy1 a double.
	 * @param _cx2 a double.
	 * @param _cy2 a double.
	 * @param _W a int.
	 * @param _H a int.
	 * @param _max_iterations a int.
	 */
	public void init(double _cx1, double _cy1, double _cx2, double _cy2, int _W, int _H, int _max_iterations) {

		wmax = _W;
		hmax = _H;
		result = new int[_W][_H];
		max_iterations = _max_iterations;
		cx1 = _cx1;
		cy1 = _cy1;

		wx = (_cx2 - cx1) / (double) wmax;
		hy = (_cy2 - cy1) / (double) hmax;

	}

	/**
	 * {@inheritDoc}
	 *
	 * just executes the "simple" math on a pixel
	 */
	@Override
	public void run() {

		final int w = getGlobalId(0);
		final int h = getGlobalId(1);

		if ((w < wmax) && (h < hmax)) {

			/** from pixel to complex coordinates */
			final double cx = cx1 + w * wx;
			final double cy = cy1 + h * hy;

			double xn = cx;
			double yn = cy;

			double y2 = cy * cy;
			/** I don't save x2, x squared, It's slower **/

			int t = 0;

			/**
			 * the original code gave a "goto" error in some platform while(
			 * (++tGetter for the field result.
	 *
	 * @return an array of {@link int} objects.
	 */
	public int[][] getResult() {
		return result;
	}
	
	

}