#pragma warning(disable: 4786)
#include <stdlib.h>
#include <fstream>
#include <iostream>
#include <vector>
#include <string>
#include <algorithm>
#include <math.h>
#include "Drawline.h"
#include "Render.h"
#include "OBJParser.h"

using namespace std;

Render::Render()
{
                
}

Render::~Render()
{
                 
}
void Render::input(char *fileName) {
	char buf[1024];
	char* stopString;
	char* faceColor=NULL;
	
	string line;
	string separators = " ";
	vector<string> tokenList;
     	ifstream fin(fileName); // Open the OBJ file
        //Read line 1, X and Y resolution of the out.ppm image    
	    fin.getline(buf,1024); 
		tokenList.clear();
		line = buf;
		getTokens(line, separators, tokenList);
		convertFromString(Xres,tokenList[0]);		
		convertFromString(Yres,tokenList[1]);		
		//Read line 2,<3D mesh object filename <.obj>
	    fin.getline(buf,1024); 
		tokenList.clear();
		objfile = buf;
        //Read line 3, perspective transform    
	    fin.getline(buf,1024);
		tokenList.clear();
		line = buf;
		getTokens(line, separators, tokenList);
		for(int i=0;i<6;i++)
    		convertFromString(ps[i],tokenList[i]);		
        //Read line 4, camera specifications   
	    fin.getline(buf,1024);
		tokenList.clear();
		line = buf;
		getTokens(line, separators, tokenList);
		
        convertFromString(rx,tokenList[0]);
		convertFromString(ry,tokenList[1]);	        
		convertFromString(rz,tokenList[2]);	        		
		convertFromString(ux,tokenList[3]);
		convertFromString(uy,tokenList[4]);	        
		convertFromString(uz,tokenList[5]);		
		convertFromString(vx,tokenList[6]);
		convertFromString(vy,tokenList[7]);	        
		convertFromString(vz,tokenList[8]);
		convertFromString(nx,tokenList[9]);
		convertFromString(ny,tokenList[10]);	        
		convertFromString(nz,tokenList[11]);	
	
        //Read line 5, Scaling Transform
	    fin.getline(buf,1024);
		tokenList.clear();
		line = buf;
		getTokens(line, separators, tokenList);
        convertFromString(sx,tokenList[0]);
		convertFromString(sy,tokenList[1]);	        
		convertFromString(sz,tokenList[2]);        	
        //Read line 6, Translations vector
	    fin.getline(buf,1024);
		tokenList.clear();
		line = buf;
		getTokens(line, separators, tokenList);
		convertFromString(tx,tokenList[0]);
		convertFromString(ty,tokenList[1]);	        
		convertFromString(tz,tokenList[2]);                
         //Read line 7, Rotation angles
	    fin.getline(buf,1024);
		tokenList.clear();
		line = buf;
		getTokens(line, separators, tokenList);
		convertFromString(thx,tokenList[0]);
		convertFromString(thy,tokenList[1]);	        
		convertFromString(thz,tokenList[2]);     

        //showInput();//Debug               
}

void Render::getTokens(string& stmt, string& separators, vector<string>& token_list)
{
   int n = stmt.length();
   int start, stop;

   start = stmt.find_first_not_of(separators);
   while ((start >= 0) && (start < n)) {
      stop = stmt.find_first_of(separators, start);
      if ((stop < 0) || (stop > n)) stop = n;
      token_list.push_back(stmt.substr(start, stop - start));
      start = stmt.find_first_not_of(separators, stop+1);
   }
}
//Display the input file
void Render::showInput()
{
cout<<Xres<<" "<<Yres<<" "<<endl;
cout<<objfile<<" "<<endl;
for(int i=0;i<6;i++)
        cout<<ps[i]<<" ";
cout<<endl                ;
cout<<"r "<<rx<<" "<<ry<<" "<<rz<<" ";
cout<<ux<<" "<<uy<<" "<<uz<<" "<<vx<<" "<<vy<<" ";
cout<<vz<<" "<<nx<<" "<<ny<<" "<<nz<<" "<<endl;
cout<<"s "<<sx<<" "<<sy<<" "<<sz<<" "<<endl;
cout<<"t "<<tx<<" "<<ty<<" "<<tz<<" "<<endl;
cout<<"th "<<thx<<" "<<thy<<" "<<thz<<" "<<endl;             
}

string Render::getObjfileName()
{
       return objfile; 
}      
int Render::getXres()
{
    return Xres;
}
int Render::getYres()
{
    return Yres;
}   
//rotation
Vertex Render::rotationRx(Vertex v)
{
    Vertex nv;
    nv.x = v.x;
    nv.y = v.y*cos(thx)-v.z*sin(thx);
    nv.z = v.y*sin(thx)+v.z*cos(thx);
    return nv; 
}  
Vertex Render::rotationRy(Vertex v)  
{
    Vertex nv;
    nv.x = v.x*cos(thy)+v.z*sin(thy);
    nv.y = v.y;
    nv.z = -v.x*sin(thy)+v.z*cos(thy);
    return nv;        
}
Vertex Render::rotationRz(Vertex v)  
{
    Vertex nv;
    nv.x = v.x*cos(thz)-v.y*sin(thz);
    nv.y = v.x*sin(thz)+v.y*cos(thz);
    nv.z = v.z;
    return nv;
             
}
//Transformation
Vertex Render::trans(Vertex v)
{
    Vertex nv;
    nv.x = v.x+tx;
    nv.y = v.y+ty;
    nv.z = v.z+tz;
    return nv;       
} 
//Scalling
Vertex Render::scale(Vertex v)
{
    Vertex nv;
    nv.x = v.x*sx;
    nv.y = v.y*sy;
    nv.z = v.z*sz;
    return nv;       
}     
//To camera view      
Vertex Render::camera(Vertex v) 
{
    Vertex nv;
    nv.x = v.x*ux+v.y*uy+v.z*uz-rx;    
    nv.y = v.x*vx+v.y*vy+v.z*vz-ry;    
    nv.z = v.x*nx+v.y*ny+v.z*nz-rz;  
    return nv;         
}
//To NDC
Vertex Render::perspective(Vertex v) 
{
    Vertex nv;
    float nrl = (-2*ps[2]/(ps[3]-ps[0]));
    float rl = ((ps[3]+ps[0])/(ps[3]-ps[0]));
    float w =  (-1*v.z);  
    double ntbd = (ps[4]-ps[1]);
    double ntbu = (-2*ps[2]);
    double ntb = ntbu/ntbd;
    double tpu =  ps[4]+ps[1];
    double tpd =  ps[4]-ps[1]; 
    double tb = tpu/tpd;

    nv.x = (v.x)/w;
    nv.y = (v.y)/w;
    nv.z = v.z;  
    return nv;         
}
//To view window
void Render::viewWindow(Vertex v,int *x,int *y)
{
       *x = (int)((v.x+1)*0.5*(getXres()-1));
       *y = (int)((v.y+1)*0.5*(getYres()-1)); 
}       
Vertex Render::rtscp(Vertex v) 
{    
    v = rotationRx(v); 
    v = rotationRy(v); 
    v = rotationRz(v); 
    v = trans(v);
    v = scale(v);  
    v = camera(v);
    v = perspective(v);       
    return v; 
} 
void Render::renderImage(OBJParser &parser,Drawline &drawline)                
{
    int num = parser.getNumOfFaces();
    //cout<<"Num="<<num<<endl;
	for(int x=1; x <= num; x++) {
		Face f = parser.getFace(x);
		//cout << "Face " << f.faceNum << ": "<<endl;		    
		for(int y=0; y < f.faceVertices.size(); y++) {
            //Get First Point               
			Face::FaceVertex fVertex = f.faceVertices[y];
			Vertex v1 = parser.getVertex(fVertex.v); 			
			//Get Second Point,if it's last point then set to 1st point
			if(y==f.faceVertices.size()-1)
               fVertex = f.faceVertices[0];                                                            
            else   
			   fVertex = f.faceVertices[y+1];
            Vertex v2 = parser.getVertex(fVertex.v);              	
			
			v1 = rtscp(v1);			
			v2 = rtscp(v2);
			
			int p_x,p_y,p_x2,p_y2;			
			viewWindow(v1,&p_x,&p_y);
			viewWindow(v2,&p_x2,&p_y2);	  

            drawline.drawline(p_x,p_y,p_x2,p_y2);                       			   			
		}
		//cout<<endl;
	}		
    
}
void Render::makeAnimation(OBJParser &parser,Drawline &drawline)
{
         char fileName[10];
    for(int i=0;i<30;i++){
       
       renderImage(parser,drawline);	
       thy=thy+0.1;
       thz=thz+0.1;
       sprintf(fileName,"mov_%3d.ppm",i);
       drawline.output(fileName);
       drawline.initGraph();
       cout<<"output image "<<fileName<<endl;
}
    for(int i=30;i<90;i++){       
       
       renderImage(parser,drawline);	
       thx=thx+0.1;       
       sx=sx+0.01;       
       sy=sy+0.01; 
       sz=sz+0.01;               
       sprintf(fileName,"mov_%3d.ppm",i);
       drawline.output(fileName);
       drawline.initGraph();
       cout<<"output image "<<fileName<<endl;       
}
    for(int i=90;i<150;i++){
       
       renderImage(parser,drawline);	
       thx=thx-0.1;       
       sx=sx-0.01;       
       sy=sy-0.01; 
       sz=sz-0.01;               
       sprintf(fileName,"mov_%3d.ppm",i);
       drawline.output(fileName);
       drawline.initGraph();
       cout<<"output image "<<fileName<<endl;   
}
    for(int i=150;i<180;i++){
       renderImage(parser,drawline);	
       thy=thy-0.1;
       thz=thz-0.1;
       sprintf(fileName,"mov_%3d.ppm",i);
       drawline.output(fileName);
       drawline.initGraph();
       cout<<"output image "<<fileName<<endl;
}         
}
int main(int argc, char* argv[]) 
{
    Render render;
   	OBJParser parser;  
    Drawline drawline;
    render.input(argv[1]); // read an input file
                   
	parser.parse(render.getObjfileName().c_str()); // parse an OBJ file    
    drawline.setRES(render.getXres(),render.getYres()); //Set graph size
    //render.makeAnimation(parser,drawline);
    render.renderImage(parser,drawline);
    //If there given output file name
    if(argc>=3)    
        drawline.output(argv[2]);
    else
        drawline.output();    
    
    //system("Pause");   	
	return 0;		
}