#version 400

//the following uniform values are set by Visions of Chaos prior to shader execution
uniform vec2 resolution;
uniform vec3 palette[256];
uniform double xmin;
uniform double xmax;
uniform double ymin;
uniform double ymax;
uniform double bailout;
uniform int maxiters;
uniform int samplepixels;

double sqrsamplepixels=double(samplepixels*samplepixels);
double bailout_squared=double(bailout*bailout);
double magnitude,r1,r2,g1,g2,b1,b2,tweenval;
float realiters;
vec4 finalcol,col;
int superx,supery;
double stepx=(xmax-xmin)/resolution.x/double(samplepixels);
double stepy=(ymax-ymin)/resolution.y/double(samplepixels);
int index,colval,colval2;
dvec2 z,c;

void main(void)
{
	finalcol=vec4(0,0,0,0);
	for (supery=0;supery<samplepixels;supery++)
	{
		for (superx=0;superx<samplepixels;superx++)
		{
			c.x = xmin+gl_FragCoord.x/resolution.x*(xmax-xmin)+(stepx*double(superx));
			c.y = ymin+gl_FragCoord.y/resolution.y*(ymax-ymin)+(stepy*double(supery));
			int i;
			z = dvec2(0.0,0.0);
			for(i=0; i<maxiters; i++) 
			{
				double x;
				double y;

				if (mod(i,4)==0) {
					//Mandelbrot
					x = (z.x * z.x - z.y * z.y) + c.x;
					y = (z.y * z.x + z.x * z.y) + c.y;
				}

				if (mod(i,4)==1) {
					//Burning Ship
					x = z.x * z.x - z.y * z.y + c.x;
					y = 2.0 * abs(z.y * z.x) + c.y;				
				}

				if (mod(i,4)==2) {
					//Mandelbrot
					x = (z.x * z.x - z.y * z.y) + c.x;
					y = (z.y * z.x + z.x * z.y) + c.y;
				}

				if (mod(i,4)==3) {
					//Mandelbrot
					x = (z.x * z.x - z.y * z.y) + c.x;
					y = (z.y * z.x + z.x * z.y) + c.y;
				}

				magnitude=(x * x + y * y);
				if(magnitude>bailout_squared) break;
				z.x = x;
				z.y = y;


			}

			if (i==maxiters) {
				col=vec4(0.0,0.0,0.0,1.0);
			} else {
				//CPM smooth colors
				//note that double precision does not support log so it needs to be cast as float
				realiters=float(i+1-((log(log(sqrt(float(magnitude))))/log(2.0))));
				colval=int(mod(realiters,255));
				colval2=int(mod(colval+1,255));
				tweenval=realiters-int(realiters);
				r1=palette[colval].r;
				g1=palette[colval].g;
				b1=palette[colval].b;
				r2=palette[colval2].r;
				g2=palette[colval2].g;
				b2=palette[colval2].b;
				col=vec4(r1+((r2-r1)*tweenval),g1+((g2-g1)*tweenval),b1+((b2-b1)*tweenval),1.0);
			}

			finalcol+=col;
		}
	}
	gl_FragColor = vec4(finalcol/double(sqrsamplepixels));
}