ultimatepp/uppsrc/ScatterDraw/DataSource.h
koldo d78cd9fc58 ScatterDraw: Added new plots
git-svn-id: svn://ultimatepp.org/upp/trunk@8645 f0d560ea-af0d-0410-9eb7-867de7ffcac7
2015-07-08 05:50:55 +00:00

321 lines
11 KiB
C++

#ifndef _ScatterDraw_DataSource_h_
#define _ScatterDraw_DataSource_h_
NAMESPACE_UPP
class DataSource {
public:
typedef double (DataSource::*Getdatafun)(int64 id);
DataSource() : isParam(false), isExplicit(false), key(111111) {}
virtual ~DataSource() {key = 0;}
virtual double y(int64 id) {NEVER(); return Null;}
virtual double x(int64 id) {NEVER(); return Null;}
virtual double znx(int n, int64 id) {NEVER(); return Null;}
virtual double zny(int n, int64 id) {NEVER(); return Null;}
virtual double znFixed(int n, int64 id) {NEVER(); return Null;}
virtual double y(double t) {NEVER(); return Null;}
virtual double x(double t) {NEVER(); return Null;}
virtual double f(double x) {NEVER(); return Null;}
virtual double f(Vector<double> zn) {NEVER(); return Null;}
virtual int64 GetCount() {NEVER(); return Null;}
virtual int GetznxCount() {return 0;}
virtual int GetznyCount() {return 0;}
virtual int GetznFixedCount() {return 0;}
bool IsParam() {return isParam;}
bool IsExplicit() {return isExplicit;}
bool IsDeleted() {return key != 111111;}
virtual double MinY() {return Min(&DataSource::y);}
virtual double MinX() {return Min(&DataSource::x);}
virtual double MaxY() {return Max(&DataSource::y);}
virtual double MaxX() {return Max(&DataSource::x);}
virtual double AvgY() {return Avg(&DataSource::y);}
virtual double StdDevY(double avg = Null) {return StdDev(&DataSource::y, avg);}
virtual double VarianceY(double avg = Null) {return Variance(&DataSource::y, avg);}
double Min(Getdatafun getdata);
double Max(Getdatafun getdata);
double Avg(Getdatafun getdata);
double StdDev(Getdatafun getdata, double avg = Null);
double Variance(Getdatafun getdata, double avg = Null);
double SinEstim_Amplitude(double avg = Null);
bool SinEstim_FreqPhase(double &frequency, double &phase, double avg = Null);
protected:
bool isParam, isExplicit;
private:
int key;
};
class CArray : public DataSource {
private:
double *xData, *yData, *zData;
int64 numData;
double x0, deltaX;
public:
CArray(double *yData, int numData, double x0, double deltaX) : yData(yData), numData(numData), x0(x0), deltaX(deltaX) {xData = NULL;}
CArray(double *yData, double *xData, int numData) : yData(yData), xData(xData), numData(numData) {zData = NULL; x0 = deltaX = 0;}
CArray(double *yData, double *xData, double *zData, int numData) : yData(yData), xData(xData), zData(zData), numData(numData) {x0 = deltaX = 0;}
virtual inline double y(int64 id) {return yData[ptrdiff_t(id)];}
virtual inline double x(int64 id) {return xData ? xData[ptrdiff_t(id)] : id*deltaX + x0;}
virtual double znFixed(int n, int64 id);
virtual int GetznFixedCount() {return 1;}
virtual inline int64 GetCount() {return numData;}
};
template <class Y>
class VectorY : public DataSource {
private:
Vector<Y> *yData;
double x0, deltaX;
public:
VectorY(Vector<Y> &yData, double x0, double deltaX) : yData(&yData), x0(x0), deltaX(deltaX) {}
virtual inline double y(int64 id) {return (*yData)[int(id)];}
virtual inline double x(int64 id) {return id*deltaX + x0;}
virtual inline int64 GetCount() {return yData->GetCount();}
virtual double MinX() {return x0;}
virtual double MaxX() {return x0 + (yData->GetCount() - 1)*deltaX;}
virtual double AvgX() {return x0 + ((yData->GetCount() - 1)*deltaX)/2.;}
};
template <class Y>
class ArrayY : public DataSource {
private:
Upp::Array<Y> *yData;
double x0, deltaX;
public:
ArrayY(Upp::Array<Y> &yData, double x0, double deltaX) : yData(&yData), x0(x0), deltaX(deltaX) {}
virtual inline double y(int64 id) {return (*yData)[ptrdiff_t(id)];}
virtual inline double x(int64 id) {return id*deltaX + x0;}
virtual inline int64 GetCount() {return yData->GetCount();}
virtual double MinX() {return x0;}
virtual double MaxX() {return x0 + yData->GetCount()*deltaX;}
virtual double AvgX() {return (x0 + yData->GetCount()*deltaX)/2.;}
};
template <class Y>
class VectorVectorY : public DataSource {
private:
Vector<Vector<Y> > *data;
bool useRows;
int idx, idy;
Vector<int> idsx, idsy, idsFixed;
int beginData;
int64 numData;
public:
VectorVectorY() : data(0), useRows(true), beginData(0), numData(Null), idx(0), idy(1) {}
VectorVectorY(Vector<Vector<Y> > &data, int idy, int idx,
Vector<int> &idsy, Vector<int> &idsx, Vector<int> &idsFixed, bool useRows = true, int beginData = 0, int numData = Null) :
data(&data), useRows(useRows), beginData(beginData), numData(numData) {
Init(data, idy, idx, idsy, idsx, idsFixed, useRows, beginData, numData);
}
void Init(Vector<Vector<Y> > &_data, int _idy, int _idx, Vector<int> &_idsy, Vector<int> &_idsx, Vector<int> &_idsFixed,
bool _useRows = true, int _beginData = 0, int _numData = Null) {
data = &_data;
useRows = _useRows;
idx = _idx;
idy = _idy;
idsx = clone(_idsx);
idsy = clone(_idsy);
idsFixed = clone(_idsFixed);
beginData = _beginData;
numData = _numData;
if (IsNull(_numData)) {
if (!useRows) {
if (data->IsEmpty())
numData = 0;
else
numData = (*data)[0].GetCount() - beginData;
} else
numData = data->GetCount() - beginData;
}
}
void Init(Vector<Vector<Y> > &_data, int idy, int idx, bool _useRows = true, int _beginData = 0, int _numData = Null) {
static Vector<int> idsVoid;
Init(_data, idy, idx, idsVoid, idsVoid, idsVoid, _useRows, _beginData, _numData);
}
virtual inline double y(int64 id) {return useRows ? (*data)[beginData + int(id)][idy] : (*data)[idy][beginData + int(id)];}
virtual inline double x(int64 id) {return useRows ? (*data)[beginData + int(id)][idx] : (*data)[idx][beginData + int(id)];}
//virtual inline double xn(int n, int64 id) {return useRows ? (*data)[beginData + int(id)][ids[n]] : (*data)[ids[n]][beginData + int(id)];}
virtual inline int64 GetCount() {return numData;};
virtual double znx(int n, int64 id) {return useRows ? (*data)[beginData + int(id)][idsx[n]] : (*data)[idsx[n]][beginData + int(id)];}
virtual double zny(int n, int64 id) {return useRows ? (*data)[beginData + int(id)][idsy[n]] : (*data)[idsy[n]][beginData + int(id)];}
virtual double znFixed(int n, int64 id) {return useRows ? (*data)[beginData + int(id)][idsFixed[n]] : (*data)[idsFixed[n]][beginData + int(id)];}
virtual int GetznxCount() {return idsx.GetCount();}
virtual int GetznyCount() {return idsy.GetCount();}
virtual int GetznFixedCount() {return idsFixed.GetCount();}
};
class VectorDouble : public DataSource {
private:
const Vector<double> *xData, *yData;
public:
VectorDouble(const Vector<double> &yData, Vector<double> &xData) : xData(&xData), yData(&yData) {}
virtual inline double y(int64 id) {return (*yData)[int(id)];}
virtual inline double x(int64 id) {return (*xData)[int(id)];}
virtual inline int64 GetCount() {return min(xData->GetCount(), yData->GetCount());}
};
class ArrayDouble : public DataSource {
private:
const Upp::Array<double> *xData, *yData;
public:
ArrayDouble(const Upp::Array<double> &yData, Upp::Array<double> &xData) : xData(&xData), yData(&yData) {}
virtual inline double y(int64 id) {return (*yData)[int(id)];}
virtual inline double x(int64 id) {return (*xData)[int(id)];}
virtual inline int64 GetCount() {return min(xData->GetCount(), yData->GetCount());}
};
class VectorPointf : public DataSource {
private:
const Vector<Pointf> *data;
public:
VectorPointf(const Vector<Pointf> &data) : data(&data) {}
VectorPointf(Vector<Pointf> *data) : data(data) {}
virtual inline double y(int64 id) {return (*data)[int(id)].y;}
virtual inline double x(int64 id) {return (*data)[int(id)].x;}
virtual inline int64 GetCount() {return data->GetCount();}
};
class ArrayPointf : public DataSource {
private:
Upp::Array<Pointf> *data;
public:
ArrayPointf(Upp::Array<Pointf> &data) : data(&data) {}
virtual inline double y(int64 id) {return (*data)[int(id)].y;}
virtual inline double x(int64 id) {return (*data)[int(id)].x;}
virtual inline int64 GetCount() {return data->GetCount();}
};
template <class X, class Y>
class VectorMapXY : public DataSource {
private:
VectorMap<X, Y> *data;
public:
VectorMapXY(VectorMap<X, Y> &data) : data(&data) {}
virtual inline double y(int64 id) {return (*data)[int(id)];}
virtual inline double x(int64 id) {return (*data).GetKey(int(id));}
virtual inline int64 GetCount() {return data->GetCount();}
};
template <class X, class Y>
class ArrayMapXY : public DataSource {
private:
ArrayMap<X, Y> *data;
public:
ArrayMapXY(ArrayMap<X, Y> &data) : data(&data) {}
virtual inline double y(int64 id) {return (*data)[int(id)];}
virtual inline double x(int64 id) {return (*data).GetKey(int(id));}
virtual inline int64 GetCount() {return data->GetCount();}
};
class FuncSource : public DataSource {
private:
double (*function)(double);
public:
FuncSource(double (*function)(double)) : function(function) {isExplicit = true;}
virtual inline double f(double x) {return function(x);}
};
class FuncSourceV : public DataSource {
private:
void (*function)(double&, double);
public:
FuncSourceV(void (*function)(double&, double)) : function(function) {isExplicit = true;}
virtual inline double f(double x) {double y; function(y, x); return y;}
};
class FuncSourcePara : public DataSource {
private:
Pointf (*function)(double);
Pointf lastPointf;
double lastT;
int numPoints;
double minT, maxT;
public:
FuncSourcePara(Pointf (*function)(double), int np, double from, double to) :
function(function), numPoints(np), minT(from), maxT(to) {
isParam = true;
lastT = Null;
}
virtual inline double y(double t) {
if (IsNull(lastT) || t != lastT) {
lastPointf = function(minT + t*(maxT-minT)/numPoints);
lastT = t;
}
return lastPointf.y;
}
virtual inline double x(double t) {
if (IsNull(lastT) || t != lastT) {
lastPointf = function(minT + t*(maxT-minT)/numPoints);
lastT = t;
}
return lastPointf.x;
}
virtual inline int64 GetCount() {return numPoints;}
};
typedef Callback2<double&, double> PlotExplicFunc;
typedef Callback2<Pointf&, double> PlotParamFunc;
class PlotExplicFuncSource : public DataSource {
private:
PlotExplicFunc function;
public:
PlotExplicFuncSource(PlotExplicFunc &function) : function(function) {isExplicit = true;}
virtual inline double f(double t) {double y; function(y, t); return y;}
};
class PlotParamFuncSource : public DataSource {
private:
PlotParamFunc function;
Pointf lastPointf;
double lastT;
int numPoints;
double minT, maxT;
public:
PlotParamFuncSource(PlotParamFunc function, int np, double from, double to) :
function(function), numPoints(np), minT(from), maxT(to) {
isParam = true;
lastT = Null;
}
inline double y(double t) {
if (IsNull(lastT) || t != lastT) {
function(lastPointf, minT + t*(maxT-minT)/numPoints);
lastT = t;
}
return lastPointf.y;
}
inline double x(double t) {
if (IsNull(lastT) || t != lastT) {
function(lastPointf, minT + t*(maxT-minT)/numPoints);
lastT = t;
}
return lastPointf.x;
}
virtual inline int64 GetCount() {return numPoints;}
};
END_UPP_NAMESPACE
#endif