paulstretch_cpp

FreeEdit.cpp (5733B)

---

 /*
   Copyright (C) 2009 Nasca Octavian Paul
   Author: Nasca Octavian Paul
 
   This program is free software; you can redistribute it and/or modify
   it under the terms of version 2 of the GNU General Public License 
   as published by the Free Software Foundation.
 
   This program 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 (version 2) for more details.
 
   You should have received a copy of the GNU General Public License (version 2)
   along with this program; if not, write to the Free Software Foundation,
   Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
 */
 
 #include <stdio.h>
 #include <math.h>
 #include "FreeEdit.h"
 
 FreeEdit::FreeEdit(){			
 			enabled=false;
 			smooth=0.0;
 			interp_mode=LINEAR;
 			npos=FREE_EDIT_MAX_POINTS;
 			pos=new FreeEditPos[npos];
 			for (int i=0;i<npos;i++){
 				pos[i].x=pos[i].y=0;
 				pos[i].enabled=false;
 			};
 			pos[0].x=0;
 			pos[0].y=0.5;
 			pos[0].enabled=true;
 			pos[1].x=1;
 			pos[1].y=0.5;
 			pos[1].enabled=true;
 			
 			curve.data=NULL;
 			curve.size=0;
 };
 
 void FreeEdit::deep_copy_from(const FreeEdit &other){
 	enabled=other.enabled;
 	smooth=other.smooth;
 	interp_mode=other.interp_mode;
 	npos=other.npos;
 	pos=new FreeEditPos[npos];
 	for (int i=0;i<npos;i++){
 		pos[i].x=other.pos[i].x;
 		pos[i].y=other.pos[i].y;
 		pos[i].enabled=other.pos[i].enabled;		
 	};
 	curve.size=other.curve.size;
 	if (other.curve.data&&other.curve.size){
 		curve.data=new REALTYPE[curve.size];
 		for (int i=0;i<curve.size;i++) curve.data[i]=other.curve.data[i];
 	}else curve.data=NULL;
 	extreme_x=other.extreme_x;
 	extreme_y=other.extreme_y;
 };
 FreeEdit::FreeEdit (const FreeEdit &other){
 	deep_copy_from(other);
 };
 const FreeEdit &FreeEdit::operator=(const FreeEdit &other){
 	if (this == &other) return *this;
 	deep_copy_from(other);
 	return *this;
 };
 void FreeEdit::get_curve(int datasize,REALTYPE *data,bool real_values){
 	int npos_used=0;
 	for (int i=0;i<npos;i++) if (is_enabled(i)) npos_used++;
 	if (!npos_used){
 		for (int i=0;i<datasize;i++) data[i]=(real_values?extreme_y.get_min():0.0); 
 		return;
 	};
 
 	//get the enabled points
 	REALTYPE posx[npos_used],posy[npos_used];
 	int k=0;
 	for (int i=0;i<npos;i++){
 		if (is_enabled(i)){
 			posx[k]=get_posx(i);
 			posy[k]=get_posy(i);
 			k++;
 		};
 	};
 
 	//sort the points
 	for (int j=0;j<npos_used-1;j++){
 		for (int i=j+1;i<npos_used;i++){
 			if (posx[i]<posx[j]){
 				swap(posx[i],posx[j]);
 				swap(posy[i],posy[j]);
 			};
 		};
 	};
 
 	//generate the curve
 	int p1=0,p2=1;
 	for (int i=0;i<datasize;i++){
 		REALTYPE x=(REALTYPE)i/(REALTYPE)datasize;
 		while ((x>posx[p2])&&(p2<npos_used)){
 			p1=p2;
 			p2++;
 		};
 		REALTYPE px1=posx[p1];
 		REALTYPE px2=posx[p2];
 		REALTYPE diffx=px2-px1;
 		REALTYPE x0=0;		
 		if (diffx>1e-5) x0=(x-px1)/diffx;
 		if (interp_mode==COSINE) x0=(1.0-cos(x0*M_PI))*0.5;
 		REALTYPE y=y=posy[p1]*(1.0-x0)+posy[p2]*x0;
 		data[i]=y;
 	};
 
 
 	//smooth the curve
 	if (smooth>0.01){
 		const int max_times=4;
 		REALTYPE a=exp(log(0.25)/(smooth*smooth*datasize*0.25));
 		if ((a<=0.0)||(a>=1.0)) return;
 		a=pow(a,max_times);
 		for (k=0;k<max_times;k++){
 			for (int i=1;i<datasize;i++) data[i]=data[i]*(1.0-a)+data[i-1]*a;
 			for (int i=datasize-2;i>=0;i--) data[i]=data[i]*(1.0-a)+data[i+1]*a;
 		};
 	};
 
 	if (real_values){
 		for (int i=0;i<datasize;i++) data[i]=extreme_y.coord_to_real_value(data[i]);
 		if (extreme_y.get_scale()==FE_DB){
 			for (int i=0;i<datasize;i++) data[i]=dB2rap(data[i]);
 		};
 	};	
 
 };
 
 void FreeEdit::update_curve(int size){
 	if (curve.data) delete []curve.data;
 	if (size<2) size=2;
 	curve.size=size;
 	curve.data=new REALTYPE[size];
 
 	get_curve(curve.size,curve.data,true);
 
 //	for(int i=0;i<size;i++) printf("_%d %g\n",i,curve.data[i]);
 };
 
 REALTYPE FreeEdit::get_value(REALTYPE x){
 	if (!curve.data) {
 		return 0.0;// update_curve();
 	};
 	if (extreme_x.get_scale()==FE_LOG){
 		if (x<=0.0) x=1e-9;
 	};
 	
 //	printf("%g\n",curve.data[1]);
 	
 	x=extreme_x.real_value_to_coord(x);
 	if (x<0) x=0.0;
 	else if (x>1.0) x=1.0;
 	REALTYPE rx=x*curve.size;
 	REALTYPE rxh=floor(rx);
 	int k=(int)rxh;
 	REALTYPE rxl=rx-rxh;
 	if (k<0) k=0;
 	if (k>(curve.size-1)) k=curve.size-1;
 	int k1=k+1; if (k1>(curve.size-1)) k1=curve.size-1;
 	return curve.data[k]*(1.0-rxl)+curve.data[k1]*rxl;
 };
 
 void FreeEdit::add2XML(XMLwrapper *xml){
 	xml->addparbool("enabled",enabled);
 	xml->addparreal("smooth",smooth);
 	xml->addpar("interp_mode",interp_mode);
 	xml->beginbranch("POINTS");
 		for (int i=0;i<FREE_EDIT_MAX_POINTS;i++){
 			if (!pos[i].enabled) continue;
 			xml->beginbranch("POINT",i);
 				xml->addparbool("enabled",pos[i].enabled);
 				xml->addparreal("x",pos[i].x);
 				xml->addparreal("y",pos[i].y);
 	
 			xml->endbranch();
 		};
 	xml->endbranch();
 
 	xml->beginbranch("EXTREME_X");
 		extreme_x.add2XML(xml);
 	xml->endbranch();
 
 	xml->beginbranch("EXTREME_Y");
 		extreme_y.add2XML(xml);
 	xml->endbranch();
 };
 
 
 void FreeEdit::getfromXML(XMLwrapper *xml){
 	enabled=xml->getparbool("enabled",enabled);
 	smooth=xml->getparreal("smooth",smooth);
 	interp_mode=(INTERP_MODE)xml->getpar("interp_mode",interp_mode,0,1);
 	
 	
 	
 	if (xml->enterbranch("POINTS")){
 		for (int i=0;i<FREE_EDIT_MAX_POINTS;i++){
 			if (xml->enterbranch("POINT",i)){
 				pos[i].enabled=xml->getparbool("enabled",pos[i].enabled);
 				pos[i].x=xml->getparreal("x",pos[i].x);
 				pos[i].y=xml->getparreal("y",pos[i].y);
 
 				xml->exitbranch();
 			};
 		};
 		xml->exitbranch();
 	};
 
 	if (xml->enterbranch("EXTREME_X")){
 		extreme_x.getfromXML(xml);
 		xml->exitbranch();
 	};
 
 	if (xml->enterbranch("EXTREME_Y")){
 		extreme_y.getfromXML(xml);
 		xml->exitbranch();
 	};
 
 	update_curve();
 };