paulxstretch/Source/PS_Source/FreeEdit.cpp

249 lines
6.0 KiB
C++
Raw Normal View History

2017-11-13 15:06:08 +00:00
/*
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"
#include <vector>
#ifdef HAVE_PS_FREE_EDIT
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;
curve.allocsize = 0;
2017-11-13 15:06:08 +00:00
};
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){
if (curve.data) delete [] curve.data;
curve.data=new REALTYPE[curve.size];
curve.allocsize = curve.size;
2017-11-13 15:06:08 +00:00
for (int i=0;i<curve.size;i++) curve.data[i]=other.curve.data[i];
} else {
if (curve.data) delete [] curve.data;
curve.data=NULL;
}
2017-11-13 15:06:08 +00:00
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.0f);
return;
};
//get the enabled points
std::vector<REALTYPE> posx(npos_used);
std::vector<REALTYPE> 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.0f-cos(x0*(float)M_PI))*0.5f;
REALTYPE y=y=posy[p1]*(1.0f-x0)+posy[p2]*x0;
data[i]=y;
};
//smooth the curve
if (smooth>0.01){
const int max_times=4;
REALTYPE a=exp(log(0.25f)/(smooth*smooth*datasize*0.25f));
if ((a<=0.0f)||(a>=1.0f)) 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.0f-a)+data[i-1]*a;
for (int i=datasize-2;i>=0;i--) data[i]=data[i]*(1.0f-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]=(float)dB2rap(data[i]);
};
};
};
void FreeEdit::update_curve(int size){
if (size<2) size=2;
if (size > curve.allocsize || !curve.data) {
if (curve.data) delete []curve.data;
curve.data = new REALTYPE[size];
curve.allocsize = size;
}
curve.size = size;
2017-11-13 15:06:08 +00:00
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.0f) x=1e-9f;
};
// 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.0f-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();
};
*/
#endif