#include "morphmesh.h"

track::track()
{
}

/*
void track::addKey(float time,float weight)
{
    keys.push_back(key(time,weight));
}
*/

float track::evaluate(float time)
{
    //interpolate key track
    if (time<=keys.begin()->first) return keys.begin()->second;
    if (time>=keys.rbegin()->first) return keys.rbegin()->second;
    map< float,float >::iterator key2=keys.upper_bound( time );
    map< float,float >::iterator key1=key2--;
    float interval=key2->first-key1->first;
    return (((1.0-((key2->first-time)/interval))*key2->second)+((1.0-((time-key1->first)/interval))*key1->second));
}

sequence::sequence(string _name,float _length,float _fadeinTime,float _fadeoutTime)
{
    name=_name;
    length=_length;
    startTime=0;
    stopTime=0;
    fadeinTime=_fadeinTime;
    fadeoutTime=_fadeoutTime;
    active=false;
}

//-----------------------------------------------------
// stopTime & startTime are absolute
//
// all other time markers are relative to the sequence
//
//
void sequence::reset(){
	active=false;
	fadeinTime=-1;
	fadeoutTime=-1;
	startTime=0;
	stopTime=0;
}
void sequence::start(){
	reset();
	active=true;
	startTime=ofGetElapsedTimef();
}
void sequence::startAt(float time){
	reset();
	active=true;
	startTime=ofGetElapsedTimef()+time;
}
void sequence::stopAt(float time){
	reset();
	active=true;
	stopTime=ofGetElapsedTimef()+time;
}
void sequence::fadeout(float time){
	if(active==true) {
		reset();
		active=true;
		startTime=ofGetElapsedTimef();
		fadeoutTime=0;
		stopTime=ofGetElapsedTimef()+time;
	}
}
void sequence::fadein(float time){
	reset();
	active=true;
	fadeinTime=time;
	startTime=ofGetElapsedTimef();
}
void sequence::stop(){
	reset();
}

vector<morphWeight> sequence::evaluate(float time){
	float now=ofGetElapsedTimef();
	float seqtime=time-startTime; 
	float looptime=fmod(seqtime,length); 
	float fadeWeight=(seqtime<fadeinTime?seqtime/fadeinTime:seqtime>fadeoutTime&&fadeoutTime>-1?1.0-(seqtime/(stopTime-startTime)):1.0);
	vector<morphWeight> weights;
	map<string,track>::iterator i;
	for( i = tracks.begin(); i != tracks.end(); i++ ) {
		weights.push_back(morphWeight(i->first,(i->second.evaluate(looptime))*fadeWeight));
	}
	return weights;
}

morphmesh::morphmesh()
{
	loaded=false;

    /*
	//testing track function
	track testrack;
	testrack.keys.insert( pair<float,float>(0.0,2.0) );
	testrack.keys.insert( pair<float,float>(1.0,4.0) );
	testrack.keys.insert( pair<float,float>(0.5,10.0) );
	for (float f=-0.4;f<1.4;f+=0.1) {
        printf("%f : %f\n",f,testrack.evaluate(f));
	}
    */
}

morphmesh::morphmesh(string filename)
{
	morphmesh();
	loadMesh(filename);
}


int morphmesh::getNumTargets(){
	return morphs.size();
}
int morphmesh::getNumSequences(){
	return sequences.size();
}
void morphmesh::draw() {
	draw(0);
}
void morphmesh::draw(int target){
	map<string,vector<ofVec3f> >::iterator it=morphs.begin();
    for ( int i=0;i<target;i++ ) it++;
	draw(it->first);
}
void morphmesh::draw(string target){
	clearVertices();
	addVertices(morphs[target]);
	ofMesh::draw();
}
void morphmesh::draw(const vector<morphWeight>& weights){
	clearVertices();
	//normalise weights
	float totalWeights=0;
	for (int i=0;i<weights.size();i++) totalWeights+=weights[i].weight;
	float weightFactor=1.0/totalWeights;
	float bx,by,bz;
	for (int j=0;j<morphs.begin()->second.size();j++) {
		bx=by=bz=0;
		for (int i=0;i<weights.size();i++) {
			bx+=morphs[weights[i].name][j].x*weights[i].weight;
			by+=morphs[weights[i].name][j].y*weights[i].weight;
			bz+=morphs[weights[i].name][j].z*weights[i].weight;
		}
		addVertex(ofVec3f(bx*weightFactor,by*weightFactor,bz*weightFactor));
	}
	ofMesh::draw();
}

void morphmesh::drawAnimated(){
	float time=ofGetElapsedTimef();
	vector<morphWeight> weights;
	map<string,sequence>::iterator i;
	for( i = sequences.begin(); i != sequences.end(); i++ ) {
		if (i->second.active) {
			vector<morphWeight> newWeights=i->second.evaluate(time);
			weights.insert(weights.end(),newWeights.begin(),newWeights.end());
			if (i->second.stopTime>0&&time>i->second.stopTime) i->second.active=false;
		}
	}
	draw(weights);
}


bool morphmesh::loadSeqs(string filename){
	loaded=false;
	ofxXmlSettings XML;
	if( !XML.loadFile(filename) ){
		printf("unable to load %s check data/ folder\n",filename.c_str());
	}else{
		if(XML.pushTag("Gauntletanim")) {
		    int numSeqs=XML.getNumTags("Sequence");
		    vector<string> seqnames;
		    for (int i=0;i<numSeqs;i++) seqnames.push_back(XML.getAttribute("Sequence","Name","none",i));
		    for (int i=0;i<numSeqs;i++) {
                float length=ofToFloat(XML.getAttribute("Sequence","Length","0",i));
                float fadeinTime=ofToFloat(XML.getAttribute("Sequence","Fadein","0",i));
                float fadeoutTime=ofToFloat(XML.getAttribute("Sequence","Fadeout","0",i));
                sequence seq=sequence(seqnames[i],length,fadeinTime,fadeoutTime);
			  
                XML.pushTag("Sequence",i);
                    int numTargs=XML.getNumTags("Target");
                    vector<string> targnames;
                    for (int j=0;j<numTargs;j++) targnames.push_back(XML.getAttribute("Target","Name","none",j));
                    for (int j=0;j<numTargs;j++) {
                        XML.pushTag("Target",j);

                            vector<float> times;
                            string keystring=XML.getAttribute("Keys","Times","none",0);
                            stringstream ks(keystring);
                            istream_iterator<string> kbegin(ks);
                            istream_iterator<string> kend;
                            vector<string> kstrings(kbegin, kend);

                            vector<float> values;
                            string valstring=XML.getAttribute("Keys","Values","none",0);
                            stringstream vs(valstring);
                            istream_iterator<string> vbegin(vs);
                            istream_iterator<string> vend;
                            vector<string> vstrings(vbegin, vend);

                            track tr;

                            for (int k=0;k<min(kstrings.size(),vstrings.size());k++) tr.keys[ofToFloat(kstrings[k])]=ofToFloat(vstrings[k]);

                            seq.tracks[targnames[j]]=tr;

                        XML.popTag();
                    }
		    sequences[seqnames[i]]=seq;
		    /*
		    printf("testing sequence: %s\n",seqnames[i].c_str());
		    for (int k=0;k<targnames.size();k++) {
			    printf("track: %s\n",targnames[k].c_str());
			    for (float l=-.2;l<=2.2;l+=0.1) {
				    printf("%f : %f\n",l,sequences[seqnames[i]].tracks[targnames[k]].evaluate(l));
			    }
		    }
		    */
		    loaded=true;
                XML.popTag();
            }
            XML.popTag();
		}
	}
	return loaded;
}



bool morphmesh::loadMesh(string filename){
	loaded=false;
	ofxXmlSettings XML;
	if( !XML.loadFile(filename) ){
		printf("unable to load %s check data/ folder\n",filename.c_str());
	}else{
		if(XML.pushTag("Oak3DModelDocument")) {
			if(XML.pushTag("MeshList")) {
				int numMeshes=XML.getNumTags("Mesh");
				vector<string> meshnames;
				for (int i=0;i<numMeshes;i++) meshnames.push_back(XML.getAttribute("Mesh","Name","none",i));
				for (int i=0;i<numMeshes;i++) {
					XML.pushTag("Mesh",i);
						if (XML.pushTag("AttributeList")) {
							vector<ofVec3f> verts;
							string vertstring=XML.getAttribute("Attribute","Data","none",0);
							stringstream ss(vertstring);
							istream_iterator<string> begin(ss);
							istream_iterator<string> end;
							vector<string> vstrings(begin, end);
							for (int j=0;j<vstrings.size();j+=3) {
								verts.push_back(ofVec3f(ofToFloat(vstrings[j]),ofToFloat(vstrings[j+1]),ofToFloat(vstrings[j+2])));
							}
							morphs[meshnames[i]]=verts;

							if (i==0) {
								addVertices(verts);

								vector<ofVec3f> norms;
								string normstring=XML.getAttribute("Attribute","Data","none",1);
								stringstream ns(normstring);
								istream_iterator<string> nbegin(ns);
								istream_iterator<string> nend;
								vector<string> nstrings(nbegin, nend);
								for (int j=0;j<nstrings.size();j+=3) {
									norms.push_back(ofVec3f(ofToFloat(nstrings[j]),ofToFloat(nstrings[j+1]),ofToFloat(nstrings[j+2])));
								}
								addNormals(norms);

								vector<ofVec2f> texcords;
								string texstring=XML.getAttribute("Attribute","Data","none",2);
								stringstream ts(texstring);
								istream_iterator<string> tbegin(ts);
								istream_iterator<string> tend;
								vector<string> tstrings(tbegin, tend);
								for (int j=0;j<tstrings.size();j+=2) {
									texcords.push_back(ofVec2f(ofToFloat(tstrings[j]),ofToFloat(tstrings[j+1])));
								}
								addTexCoords(texcords);

								XML.popTag();

								if (XML.pushTag("IndexList")) {
									vector<ofIndexType> faces;
									string facestring=XML.getAttribute("Index","Data","none",0);
									stringstream fs(facestring);
									istream_iterator<string> fbegin(fs);
									istream_iterator<string> fend;
									vector<string> fstrings(fbegin, fend);
									for (int j=0;j<fstrings.size();j+=3) {
										faces.push_back(ofIndexType(ofToInt(fstrings[j])));
										faces.push_back(ofIndexType(ofToInt(fstrings[j+1])));
										faces.push_back(ofIndexType(ofToInt(fstrings[j+2])));
									}
									addIndices(faces);
									loaded=true;
									XML.popTag();
								}
							}
							else XML.popTag();
						}
						XML.popTag();
					}
				}
			XML.popTag();
		}
		XML.popTag();
	}
	return loaded;
}

bool morphmesh::isLoaded()
{
	return loaded;
}