//prog13.cpp
//(c) 1997, Oguz Yetkin
 12/2/1997
//yetkin@cs.wisc.edu
//(the code for the graphics library is at
//http://www.cs.wisc.edu/~yetkin/code
//
//Physics 505, Assignment 13
//
//general purpose iterated function systems generator!!!
//
//the program comes with its own 2 built in fractals.  You can
//make your own, though.  With no arguments, the program will
//go into demo mode.  If you want to use your own fractals,
//cut and paste the following into a file called demo.ifs
//and then type
//prog13 demo.ifs
//
//cut here
/*
-3 3 -0 10
0       0       0       0.16    0       0
0.85    0.04    -0.04   0.85    0       1.6
0.2     -0.23   0.23    0.2     0       1.6
-0.15   0.28    0.26    0.24    0       0.4
*/
//the first line is xmin xmax ymin ymax
//(note xmin must be -xmax...an unfortunate bug I haven't gotten around
//to fixing yet).
//
//the next 4 lines each contain 6 values and specify f1,f2,f3, and f4
//needed to generate the fractal.  You can change these values...the given
//values generate Barnsley's Fern

#include<iostream.h>
#include<stdio.h>
#include<stdlib.h>	//for int random(int)
#include<conio.h>
#include<time.h>
#include<fstream.h>


#include "graph.h"


#define MAX_ITERS 50000
const int MAX_FERN=40000;
const int DISCARD=200;


//to store ifs values
const int POLYNOMIAL_SIZE=6;		//a,b,c,d,e,f

const int NUM_SIERPINSKI_FUNCS=3;
const int NUM_FERN_FUNCS=4;
const int MAX_FUNCS=4;

const int MAX=100;

//constant POLYNOMIAL_SIZE must exist
void plotxy(float* p, float &x, float &y);

float frand();	//returns a random number between 0 and 1

int main(int argc, char** argv){

	//the ifs function is stored in a dynamically allocated
	//2-D array of MAX_FUNCS rows and POLYNOMIAL_SIZE columns

	float** ifs1;		//for the sierpinski triangle
	float** ifs2;		//for the fern
	float** ifs3;		//general purpose
	int i=0;			//good old 'i'
	int current=0;		//current function
	float x,y,xnew,ynew;
	char ch;
	char filename[MAX];
	ifstream infile;
	int dummy;
     float xmin,xmax,ymin,ymax;	//to be read from file
	//ifstream infile;

	//allocate memory

	ifs1=new float*[NUM_SIERPINSKI_FUNCS];	//3
	for(i=0;i<NUM_SIERPINSKI_FUNCS;i++){
		ifs1[i]=new float[POLYNOMIAL_SIZE];
	}

	ifs2=new float*[NUM_FERN_FUNCS];		//4
	for(i=0;i<NUM_FERN_FUNCS;i++){
		ifs2[i]=new float[POLYNOMIAL_SIZE];
	}
	ifs3=new float*[MAX_FUNCS];			//4 for now
	for(i=0;i<MAX_FUNCS;i++){
		ifs3[i]=new float[POLYNOMIAL_SIZE];
	}


	if(argc>1){

		infile.open(argv[1]);
		if(!infile){

			cout<<"\nfile error!!!";
			exit(1);
		}
		infile>>xmin>>xmax>>ymin>>ymax;
		graphsetup(xmin,xmax,ymin,ymax);
			//F1

		//distort whatever is in ifs3 pretty badly
		//(these could be "weights" or just 1's and 0's
		//all 1's will result in random values between 1 and 0)

		//assume file is in correct format
		//xmin xmax ymin ymax
		//followed by 4 rows by 6 columns

		infile>>ifs3[0][0];
		infile>>ifs3[0][1];
		infile>>ifs3[0][2];
		infile>>ifs3[0][3];
		infile>>ifs3[0][4];
		infile>>ifs3[0][5];

		//F2
		infile>>ifs3[1][0];
		infile>>ifs3[1][1];
		infile>>ifs3[1][2];
		infile>>ifs3[1][3];
		infile>>ifs3[1][4];
		infile>>ifs3[1][5];
		//F3
		infile>>ifs3[2][0];
		infile>>ifs3[2][1];
		infile>>ifs3[2][2];
		infile>>ifs3[2][3];
		infile>>ifs3[2][4];
		infile>>ifs3[2][5];

		//F4
		infile>>ifs3[3][0];
		infile>>ifs3[3][1];
		infile>>ifs3[3][2];
		infile>>ifs3[3][3];
		infile>>ifs3[3][4];
		infile>>ifs3[3][5];

		for(i=0;i<DISCARD;i++){

			//pick function to use

			current=random(MAX_FUNCS);
			xnew=x*ifs3[current][0]+y*ifs3[current][1]+ifs3[current][4];
			ynew=x*ifs3[current][2]+y*ifs3[current][3]+ifs3[current][5];
			x=xnew;
			y=ynew;

			//plotxy(ifs1[current],x,y);
		}

		for(i=0;i<MAX_ITERS;i++){

			//pick function to use

			current=random(MAX_FUNCS);
			plotxy(ifs3[current],x,y);
			if(kbhit()){
				ch=getch();	//bail out if needed
				if(ch=='q'){
					cout<<"\naborted, press any key to continue"<<endl;
					break;

				}
			}
		}
	getch();
	graphclose();
	exit(0);		//terminate program
	}//end if


	//memory allocation done

	//if I were smart, I'd read these from a file :( --- DONE!!!

	//set up the sierpinski triangle (f1, f2, and f3)
	//F1
	ifs1[0][0]=0.5;
	ifs1[0][1]=0;
	ifs1[0][2]=0;
	ifs1[0][3]=0.5;
	ifs1[0][4]=0;
	ifs1[0][5]=0;

	//F2
	ifs1[1][0]=0.5;
	ifs1[1][1]=0;
	ifs1[1][2]=0;
	ifs1[1][3]=0.5;
	ifs1[1][4]=0.5;
	ifs1[1][5]=0;

	//F3
	ifs1[2][0]=0.5;
	ifs1[2][1]=0;
	ifs1[2][2]=0;
	ifs1[2][3]=0.5;
	ifs1[2][4]=0.25;
	ifs1[2][5]=0.5;

	//set up Barnsley's Fern

	//F1
	ifs2[0][0]=0;
	ifs2[0][1]=0;
	ifs2[0][2]=0;
	ifs2[0][3]=0.16;
	ifs2[0][4]=0;
	ifs2[0][5]=0;

	//F2
	ifs2[1][0]=0.85;
	ifs2[1][1]=0.04;
	ifs2[1][2]=-0.04;
	ifs2[1][3]=0.85;
	ifs2[1][4]=0;
	ifs2[1][5]=1.6;

	//F3
	ifs2[2][0]=0.2;
	ifs2[2][1]=-0.23;
	ifs2[2][2]=0.23;
	ifs2[2][3]=0.2;
	ifs2[2][4]=0;
	ifs2[2][5]=1.6;

	//F4
	ifs2[3][0]=-0.15;
	ifs2[3][1]=0.28;
	ifs2[3][2]=0.26;
	ifs2[3][3]=0.24;
	ifs2[3][4]=0;
	ifs2[3][5]=0.4;

	//set up playground!!!
	//F1
	ifs3[0][0]=0;
	ifs3[0][1]=0;
	ifs3[0][2]=0;
	ifs3[0][3]=0.16;
	ifs3[0][4]=0;
	ifs3[0][5]=0;

	//F2
	ifs3[1][0]=0.85;
	ifs3[1][1]=0.04;
	ifs3[1][2]=-0.04;
	ifs3[1][3]=0.85;
	ifs3[1][4]=0;
	ifs3[1][5]=1.6;

	//F3
	ifs3[2][0]=0.2;
	ifs3[2][1]=-0.23;
	ifs3[2][2]=0.23;
	ifs3[2][3]=0.2;
	ifs3[2][4]=0;
	ifs3[2][5]=1.6;

	//F4
	ifs3[3][0]=-0.15;
	ifs3[3][1]=0.28;
	ifs3[3][2]=0.26;
	ifs3[3][3]=0.24;
	ifs3[3][4]=0;
	ifs3[3][5]=0.4;


	//end initialization
	graphsetup(-1, 1, -1 ,1);


	//do the sierpinski thang
	x=-1.0;
	y= 0.30;


	for(i=0;i<DISCARD;i++){

		//pick function to use

		current=random(NUM_SIERPINSKI_FUNCS);
		xnew=x*ifs1[current][0]+y*ifs1[current][1]+ifs1[current][4];
		ynew=x*ifs1[current][2]+y*ifs1[current][3]+ifs1[current][5];
		x=xnew;
		y=ynew;

		//plotxy(ifs1[current],x,y);
	}
	
	for(i=0;i<MAX_ITERS;i++){

		//pick function to use

		current=random(NUM_SIERPINSKI_FUNCS);
		plotxy(ifs1[current],x,y);
		if(kbhit()){
			ch=getch();	//bail out if needed
			if(ch=='q'){
				cout<<"\naborted, press any key to continue"<<endl;
				break;

			}
		}
	}
	cout<<"sierpinski triangle\n";
	cout<<"use properly formatted ifs file at command line\n";
	cout<<"for manual input"<<endl;
	getch();



	//now draw that funky fern

	graphsetup(-5, 5, 0, 10);

	x=0;
	y=0;

	for(i=0;i<DISCARD;i++){

		//pick function to use

		current=random(NUM_FERN_FUNCS);
		xnew=x*ifs2[current][0]+y*ifs2[current][1]+ifs2[current][4];
		ynew=x*ifs2[current][2]+y*ifs2[current][3]+ifs2[current][5];
		x=xnew;
		y=ynew;
		//plotxy(ifs2[current],x,y);
	}


	dummy=20000;
	for(i=0;i<30000;i++){

	//pick function to use

		current=random(NUM_FERN_FUNCS);
		plotxy(ifs2[current],x,y);
		if(kbhit()){
			ch=getch();
			if(ch=='q'){
				cout<<"\naborted"<<endl;
				break;

			}
		}
	}
	cout<<"barnsley fern\n";
	cout<<"use properly formatted ifs file at command line\n";
	cout<<"for manual input"<<endl;
	cout<<"\nformat:\nxmin xmax ymin ymax\n[4x6 array specifying 4 functions]\n";

	getch();

    /*
	//random fractals
	for(int j=0;j<5;j++){
		graphsetup(-5,5,-5,10);
			//F1

		//distort whatever is in ifs3 pretty badly
		//(these could be "weights" or just 1's and 0's
		//all 1's will result in random values between 1 and 0)

		ifs3[0][0]=frand()*0;
		ifs3[0][1]=frand()*0;
		ifs3[0][2]=frand()*0;
		ifs3[0][3]=frand()*1.2;
		ifs3[0][4]=frand()*0;
		ifs3[0][5]=frand()*0;

		//F2
		ifs3[1][0]=frand()*0.85;
		ifs3[1][1]=frand()*0.04;
		ifs3[1][2]=frand();
		ifs3[1][3]=ifs3[1][0];
		ifs3[1][4]=frand();
		ifs3[1][5]=frand()*1.6;
						//F3
		ifs3[2][0]=frand();
		ifs3[2][1]=frand()*-0.23;
		ifs3[2][2]=ifs3[2][1];
		ifs3[2][3]=frand();
		ifs3[2][4]=frand()*0;
		ifs3[2][5]=ifs3[1][5];

		//F4
		ifs3[3][0]=frand()*-0.15;
		ifs3[3][1]=frand()*0.28*2;
		ifs3[3][2]=0.9*ifs3[3][1];
		ifs3[3][3]=frand();
		ifs3[3][4]=frand();
		ifs3[3][5]=frand()*0.8;

		for(i=0;i<DISCARD;i++){

			//pick function to use

			current=random(MAX_FUNCS);
			xnew=x*ifs3[current][0]+y*ifs3[current][1]+ifs3[current][4];
			ynew=x*ifs3[current][2]+y*ifs3[current][3]+ifs3[current][5];
			x=xnew;
			y=ynew;

			//plotxy(ifs1[current],x,y);
		}

		for(i=0;i<MAX_ITERS;i++){

			//pick function to use

			current=random(MAX_FUNCS);
			plotxy(ifs3[current],x,y);
			if(kbhit()){
				ch=getch();	//bail out if needed
				if(ch=='q'){
					cout<<"\naborted, press any key to continue"<<endl;
					break;

				}
			}
		}
	getch();


	}//end for
    */


	graphclose();
	return 0;

}//end main


//graphsetup() must have been called
void plotxy(float* p, float &x, float&y){

	float xnew,ynew;
	xnew=x*p[0]+y*p[1]+p[4];
	ynew=x*p[2]+y*p[3]+p[5];
	x=xnew;
	y=ynew;
	plot(x,y);


}


float frand(){


	float a,b,c;
	randomize();

	a=rand();
	randomize();
	b=a/RAND_MAX;
	return b;
	/*
	a=rand();
	b=a/13;
	c=b-int(b);
	return c;
	*/
}