/*
 * JAME 6.2.1
 * http://jame.sourceforge.net
 *
 * Copyright 2001, 2016 Andrea Medeghini
 *
 * This file is part of JAME.
 *
 * JAME is an application for creating fractals and other graphics artifacts.
 *
 * JAME is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * JAME is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with JAME.  If not, see <http://www.gnu.org/licenses/>.
 *
 */
package net.sf.jame.mandelbrot.extensions.processingFormula;

import net.sf.jame.core.math.Complex;
import net.sf.jame.mandelbrot.processingFormula.extension.ProcessingFormulaExtensionRuntime;
import net.sf.jame.mandelbrot.renderer.RenderedPoint;

/**
 * @author Andrea Medeghini
 */
public class CurvatureFormulaRuntime extends ProcessingFormulaExtensionRuntime {
	private final Complex z0 = new Complex();
	private final Complex z1 = new Complex();
	private final Complex z2 = new Complex();
	private final Complex t1 = new Complex();
	private final Complex t2 = new Complex();
	private final Complex t3 = new Complex();
	private double sumMod;
	private double sumArg;
	private double avgMod;
	private double avgArg;
	private boolean flag;
	private int k;
	private double m;
	private double c;
	private double r;

	/**
	 * @see net.sf.jame.mandelbrot.processingFormula.extension.ProcessingFormulaExtensionRuntime#prepareForProcessing()
	 */
	@Override
	public void prepareForProcessing() {
		k = 0;
		z0.r = 0;
		z0.i = 0;
		z1.r = 0;
		z1.i = 0;
		z2.r = 0;
		z2.i = 0;
		sumMod = 0;
		sumArg = 0;
		avgMod = 0;
		avgArg = 0;
		flag = false;
	}

	/**
	 * @see net.sf.jame.mandelbrot.processingFormula.extension.ProcessingFormulaExtensionRuntime#processPoint(net.sf.jame.mandelbrot.renderer.RenderedPoint)
	 */
	@Override
	public void processPoint(final RenderedPoint cp) {
		if (!flag) {
			k += 1;
			z0.r = z1.r;
			z0.i = z1.i;
			z1.r = z2.r;
			z1.i = z2.i;
			z2.r = cp.zr;
			z2.i = cp.zi;
			if (k > 2) {
				Complex.sub(t1, z2, z1);
				Complex.sub(t2, z1, z0);
				Complex.div(t3, t1, t2);
				m = Complex.mod(t3);
				c = Math.abs(Complex.arg(t3));
				sumMod += m;
				sumArg += c;
				r = Complex.mod(t2);
				if (r < 0.000000001) {
					flag = true;
				}
			}
		}
	}

	/**
	 * @see net.sf.jame.mandelbrot.processingFormula.extension.ProcessingFormulaExtensionRuntime#renderPoint(net.sf.jame.mandelbrot.renderer.RenderedPoint)
	 */
	@Override
	public void renderPoint(final RenderedPoint cp) {
		if (k > 2) {
			avgMod = sumMod / (k - 2);
			avgArg = sumArg / (k - 2);
		}
		cp.tr = avgMod * Math.cos(avgArg);
		cp.ti = avgMod * Math.sin(avgArg);
	}
}