ultimatepp/uppsrc/ScatterDraw/ScatterDraw.cpp
koldo 496505a0fb ScatterDraw: Changed text drawing to Draw
git-svn-id: svn://ultimatepp.org/upp/trunk@6080 f0d560ea-af0d-0410-9eb7-867de7ffcac7
2013-05-18 22:30:12 +00:00

1424 lines
35 KiB
C++

#include "ScatterDraw.h"
NAMESPACE_UPP
ScatterDraw& ScatterDraw::SetColor(const Color& _color)
{
graphColor = _color;
return *this;
}
ScatterDraw& ScatterDraw::SetTitle(const String& _title)
{
title = _title;
return *this;
}
const String& ScatterDraw::GetTitle()
{
return title;
}
ScatterDraw& ScatterDraw::SetTitleFont(const Font& fontTitle)
{
titleFont = fontTitle;
return *this;
}
ScatterDraw& ScatterDraw::SetTitleColor(const Color& colorTitle)
{
titleColor = colorTitle;
return *this;
}
void ScatterDraw::SetLabels(const String& _xLabel, const String& _yLabel, const String& _yLabel2)
{
xLabel = _xLabel;
yLabel = _yLabel;
yLabel2 = _yLabel2;
}
ScatterDraw& ScatterDraw::SetLabelX(const String& _xLabel)
{
xLabel = _xLabel;
return *this;
}
ScatterDraw& ScatterDraw::SetLabelY(const String& _yLabel)
{
yLabel = _yLabel;
return *this;
}
ScatterDraw& ScatterDraw::SetLabelY2(const String& _yLabel)
{
yLabel2 = _yLabel;
return *this;
}
ScatterDraw& ScatterDraw::SetLabelsFont(const Font& fontLabels)
{
labelsFont = fontLabels;
return *this;
}
ScatterDraw& ScatterDraw::SetLabelsColor(const Color& colorLabels)
{
labelsColor = colorLabels;
return *this;
}
ScatterDraw& ScatterDraw::SetPlotAreaMargin(int hLeft, int hRight, int vTop, int vBottom)
{
hPlotLeft = hLeft;
hPlotRight = hRight;
vPlotTop = vTop;
vPlotBottom = vBottom;
return *this;
}
ScatterDraw& ScatterDraw::SetPlotAreaLeftMargin(int margin) {
hPlotLeft = margin;
return *this;
}
ScatterDraw& ScatterDraw::SetPlotAreaRightMargin(int margin) {
hPlotRight = margin;
return *this;
}
ScatterDraw& ScatterDraw::SetPlotAreaTopMargin(int margin) {
vPlotTop = margin;
return *this;
}
ScatterDraw& ScatterDraw::SetPlotAreaBottomMargin(int margin) {
vPlotBottom = margin;
return *this;
}
ScatterDraw& ScatterDraw::SetPlotAreaColor(const Color& p_a_color)
{
plotAreaColor = p_a_color;
return *this;
}
ScatterDraw& ScatterDraw::SetAxisColor(const Color& axis_color)
{
axisColor = axis_color;
return *this;
}
ScatterDraw& ScatterDraw::SetAxisWidth(int axis_width)
{
axisWidth = axis_width;
return *this;
}
ScatterDraw& ScatterDraw::SetGridColor(const Color& grid_color)
{
gridColor = grid_color;
return *this;
}
ScatterDraw& ScatterDraw::SetGridWidth(int grid_width)
{
gridWidth = grid_width;
return *this;
}
ScatterDraw& ScatterDraw::ShowHGrid(bool show)
{
drawHGrid = show;
return *this;
}
ScatterDraw& ScatterDraw::ShowVGrid(bool show)
{
drawVGrid = show;
return *this;
}
ScatterDraw &ScatterDraw::SetDrawXReticle(bool set)
{
drawXReticle = set;
return *this;
}
ScatterDraw &ScatterDraw::SetDrawYReticle(bool set)
{
drawYReticle = set;
return *this;
}
ScatterDraw &ScatterDraw::SetDrawY2Reticle(bool set)
{
drawY2Reticle = set;
return *this;
}
void ScatterDraw::DrawLegend(Draw& w, const Size &size, int scale) const
{
int nmr = fround((size.cx - 2*(hPlotLeft + hPlotRight))/legendWidth); // Max number of labels per row
if (nmr <= 0)
return;
int nLab = series.GetCount(); //number of labels
int nc; //number of complete rows
int lcr; //number of labels on complete row
int rl; //number of remaining labels on incomplete row
if(nmr > nLab) {
nc = 0; lcr = 0; rl = nLab;
} else if (nmr == nLab) {
nc = 1; lcr = nLab; rl = 0;
} else {
nc = nLab/nmr; lcr = nmr; rl = nLab%nmr;
}
for(int j = 0; j <= nc; j++) {
int start = nLab - (j+1)*lcr;
int end = nLab - j*lcr;
if (j == nc) {
start = 0;
end = rl;
}
for(int i = start; i < end; i++) {
Vector <Point> vp;
vp << Point(scale*(i-start)*legendWidth, scale*(4-12*(j+1))) <<
Point(scale*(i-start)*legendWidth + scale*23, scale*(4-12*(j+1)));
if (series[i].opacity > 0 && series[i].seriesPlot)
DrawPolylineOpa(w, vp, scale, 1, scale*series[i].thickness, series[i].color, series[i].dash);
Point p(scale*((i-start)*legendWidth+7),scale*(4-12*(j+1))/*+scale*Thick.At(i)/12*/);
if (series[i].markWidth >= 1 && series[i].markPlot)
series[i].markPlot->Paint(w, scale, p, series[i].markWidth, series[i].markColor);
Font scaledFont = GetStdFont();
scaledFont.Height(scale*GetStdFont().GetHeight());
DrawText(w, scale*(i-start)*legendWidth+scale*25, scale*(-2-12*(j+1)), 0,
series[i].legend, scaledFont, series[i].color);
}
}
}
void ScatterDraw::AdjustMinUnitX()
{
if (xMajorUnit > 0) {
if (xMinUnit < 0)
xMinUnit += (fabs(ceil(xMinUnit/xMajorUnit)) + 1)*xMajorUnit;
else if (xMinUnit > xMajorUnit)
xMinUnit -= (fabs(floor(xMinUnit/xMajorUnit)))*xMajorUnit;
}
}
void ScatterDraw::AdjustMinUnitY()
{
if (yMajorUnit > 0) {
if (yMinUnit < 0)
yMinUnit += (fabs(ceil(yMinUnit/yMajorUnit)) + 1)*yMajorUnit;
else if (yMinUnit > yMajorUnit)
yMinUnit -= (fabs(floor(yMinUnit/yMajorUnit)))*yMajorUnit;
}
}
void ScatterDraw::AdjustMinUnitY2()
{
if (yMajorUnit2 > 0) {
if (yMinUnit2 < 0)
yMinUnit2 += (fabs(ceil(yMinUnit2/yMajorUnit2)) + 1)*yMajorUnit2;
else if (yMinUnit2 > yMajorUnit2)
yMinUnit2 -= (fabs(floor(yMinUnit2/yMajorUnit2)))*yMajorUnit2;
}
}
ScatterDraw &ScatterDraw::SetRange(double rx, double ry, double ry2)
{
ASSERT(!(rx <= 0 || ry <= 0 || ry2 <= 0));
xRange = rx;
yRange = ry;
yRange2 = ry2;
xMajorUnit = xRange/10;
AdjustMinUnitX();
yMajorUnit = yRange/10;
AdjustMinUnitY();
yMajorUnit2 = yRange2/10;
AdjustMinUnitY2();
WhenSetRange();
return *this;
}
ScatterDraw &ScatterDraw::SetMajorUnits(double ux, double uy)
{
if (!IsNull(ux)) {
xMajorUnit = ux;
AdjustMinUnitX();
}
if (!IsNull(uy)) {
yMajorUnit = uy;
yMajorUnit2 = yMajorUnit*yRange2/yRange;
AdjustMinUnitY();
AdjustMinUnitY2();
}
return *this;
}
ScatterDraw &ScatterDraw::SetMajorUnitsNum(int nx, int ny)
{
if (!IsNull(nx)) {
xMajorUnit = xRange/nx;
AdjustMinUnitX();
}
if (!IsNull(ny)) {
yMajorUnit = yRange/ny;
yMajorUnit2 = yMajorUnit*yRange2/yRange;
AdjustMinUnitY();
AdjustMinUnitY2();
}
return *this;
}
ScatterDraw &ScatterDraw::SetMinUnits(double ux, double uy)
{
xMinUnit = ux;
yMinUnit = uy;
yMinUnit2 = yRange2*yMinUnit/yRange;
AdjustMinUnitX();
AdjustMinUnitY();
AdjustMinUnitY2();
return *this;
}
ScatterDraw &ScatterDraw::SetXYMin(double xmin, double ymin, double ymin2)
{
xMin = xmin;
yMin = ymin;
yMin2 = ymin2;
WhenSetXYMin();
return *this;
}
void ScatterDraw::FitToData(bool vertical) {
double minx, maxx, miny, miny2, maxy, maxy2;
minx = miny = miny2 = -DOUBLE_NULL;
maxx = maxy = maxy2 = DOUBLE_NULL;
for (int j = 0; j < series.GetCount(); j++) {
if (series[j].opacity == 0)
continue;
minx = min(minx, series[j].PointsData()->MinX());
maxx = max(maxx, series[j].PointsData()->MaxX());
}
if (vertical) {
for (int j = 0; j < series.GetCount(); j++) {
if (series[j].opacity == 0)
continue;
for (int64 i = 0; i < series[j].PointsData()->GetCount(); i++) {
if (series[j].primaryY) {
if (series[j].PointsData()->y(i) < miny)
miny = series[j].PointsData()->y(i);
if (series[j].PointsData()->y(i) > maxy)
maxy = series[j].PointsData()->y(i);
} else {
if (series[j].PointsData()->y(i) < miny2)
miny2 = series[j].PointsData()->y(i);
if (series[j].PointsData()->y(i) > maxy2)
maxy2 = series[j].PointsData()->y(i);
}
}
}
}
if (minx != -DOUBLE_NULL) {
double deltaX = xMin - minx;
xMin -= deltaX;
xMinUnit += deltaX;
AdjustMinUnitX();
if (maxx == minx) {
if (maxx == 0) {
xRange = 2;
xMin = -1;
} else
xRange = 2*maxx;
} else
xRange = maxx - minx;
}
if (vertical) {
if (miny != -DOUBLE_NULL) {
double deltaY = yMin - miny;
yMin -= deltaY;
yMinUnit += deltaY;
AdjustMinUnitY();
if (maxy == miny)
yRange = 2*maxy;
else
yRange = maxy - miny;
}
if (miny2 != -DOUBLE_NULL) {
double deltaY2 = yMin2 - miny2;
yMin2 -= deltaY2;
yMinUnit2 += deltaY2;
AdjustMinUnitY2();
if (maxy2 == miny2)
yRange2 = 2*maxy2;
else
yRange2 = maxy2 - miny2;
}
}
WhenSetRange();
WhenSetXYMin();
Refresh();
}
ScatterDraw &ScatterDraw::Graduation_FormatX(Formats fi)
{
switch (fi) {
case EXP: cbModifFormatX = THISBACK(ExpFormat); break;
case MON: cbModifFormatX = THISBACK(MonFormat); break;
case DY: cbModifFormatX = THISBACK(DyFormat); break;
default:break;
}
return *this;
}
ScatterDraw &ScatterDraw::Graduation_FormatY(Formats fi)
{
switch (fi) {
case EXP: cbModifFormatY = THISBACK(ExpFormat); break;
case MON: cbModifFormatY = THISBACK(MonFormat); break;
case DY: cbModifFormatY = THISBACK(DyFormat); break;
default:break;
}
return *this;
}
ScatterDraw &ScatterDraw::Graduation_FormatY2(Formats fi)
{
switch (fi) {
case EXP: cbModifFormatY2 = THISBACK(ExpFormat); break;
case MON: cbModifFormatY2 = THISBACK(MonFormat); break;
case DY: cbModifFormatY2 = THISBACK(DyFormat); break;
default:break;
}
return *this;
}
Color ScatterDraw::GetNewColor(int index)
{
switch(index) {
case 0: return LtBlue();
case 1: return LtRed();
case 2: return LtGreen();
case 3: return Black();
case 4: return LtGray();
case 5: return Brown();
case 6: return Blue();
case 7: return Red();
case 8: return Green();
case 9: return Gray();
case 10: return LtBlue();
case 11: return LtRed();
case 12: return LtGreen();
case 13: return Black();
case 14: return LtGray();
case 15: return Brown();
case 16: return Blue();
case 17: return Red();
case 18: return Green();
case 19: return Gray();
}
return Color(Random(), Random(), Random());
}
String ScatterDraw::GetNewDash(int index)
{
switch(index) {
case 0: return LINE_SOLID;
case 1: return LINE_DOTTED;
case 2: return LINE_DASHED;
case 3: return LINE_DASH_DOT;
case 4: return LINE_SOLID;
case 5: return LINE_DOTTED;
case 6: return LINE_DASHED;
case 7: return LINE_DASH_DOT;
case 8: return LINE_SOLID;
case 9: return LINE_DOTTED;
case 10: return LINE_DASHED;
case 11: return LINE_DASH_DOT;
}
dword r = Random();
if (r < 8000)
r += 2000;
String ret = FormatInt(r).Right(4);
String space = " ";
return ret.Mid(0, 1) + space + ret.Mid(1, 1) + space + ret.Mid(2, 1) + space + ret.Mid(3, 1);
}
MarkPlot *ScatterDraw::GetNewMarkPlot(int index)
{
switch(index) {
case 0: return new CircleMarkPlot();
case 1: return new SquareMarkPlot();
case 2: return new TriangleMarkPlot();
case 3: return new CrossMarkPlot();
case 4: return new XMarkPlot();
case 5: return new RhombMarkPlot();
}
return new CircleMarkPlot();
}
Color GetOpaqueColor(const Color &color, const Color &background, double opacity)
{
if (opacity == 1)
return color;
if (opacity == 0)
return background;
return Color(int(opacity*(color.GetR() - background.GetR()) + background.GetR()),
int(opacity*(color.GetG() - background.GetG()) + background.GetG()),
int(opacity*(color.GetB() - background.GetB()) + background.GetB()));
}
ScatterDraw::ScatterBasicSeries::ScatterBasicSeries()
{
color = Null;
thickness = 3;
legend = "";
opacity = 1;
primaryY = true;
sequential = false;
dash = LINE_SOLID;
seriesPlot = new LineSeriesPlot();
markPlot = new CircleMarkPlot();
markWidth = 8;
markColor = Null;
fillColor = Null;
}
void ScatterDraw::ScatterBasicSeries::Init(int index)
{
color = GetNewColor(index);
markColor = Color(max(color.GetR()-30, 0), max(color.GetG()-30, 0), max(color.GetB()-30, 0));
dash = GetNewDash(int(index/6));
markPlot = GetNewMarkPlot(index);
}
ScatterDraw &ScatterDraw::AddSeries(DataSource &data)
{
ScatterSeries &s = series.Add();
s.Init(series.GetCount()-1);
s.SetDataSource(&data, false);
Refresh();
return *this;
}
ScatterDraw &ScatterDraw::_AddSeries(DataSource *data)
{
ScatterSeries &s = series.Add();
s.Init(series.GetCount()-1);
s.SetDataSource(data);
Refresh();
return *this;
}
void ScatterDraw::InsertSeries(int index, double *yData, int numData, double x0, double deltaX)
{
InsertSeries<CArray>(index, yData, numData, x0, deltaX);
}
void ScatterDraw::InsertSeries(int index, double *xData, double *yData, int numData)
{
InsertSeries<CArray>(index, xData, yData, numData);
}
void ScatterDraw::InsertSeries(int index, Vector<double> &xData, Vector<double> &yData)
{
InsertSeries<VectorDouble>(index, xData, yData);
}
void ScatterDraw::InsertSeries(int index, Array<double> &xData, Array<double> &yData)
{
InsertSeries<ArrayDouble>(index, xData, yData);
}
void ScatterDraw::InsertSeries(int index, Vector<Pointf> &points)
{
InsertSeries<VectorPointf>(index, points);
}
void ScatterDraw::InsertSeries(int index, Array<Pointf> &points)
{
InsertSeries<ArrayPointf>(index, points);
}
void ScatterDraw::InsertSeries(int index, double (*function)(double))
{
InsertSeries<FuncSource>(index, function);
}
void ScatterDraw::InsertSeries(int index, Pointf (*function)(double), int np, double from, double to)
{
InsertSeries<FuncSourcePara>(index, function, np, from, to);
}
void ScatterDraw::InsertSeries(int index, PlotExplicFunc &function)
{
InsertSeries<PlotExplicFuncSource>(index, function);
}
void ScatterDraw::InsertSeries(int index, PlotParamFunc function, int np, double from, double to)
{
InsertSeries<PlotParamFuncSource>(index, function, np, from, to);
}
void ScatterDraw::_InsertSeries(int index, DataSource *data)
{
ASSERT(IsValid(index));
ScatterSeries &s = series.Insert(index);
s.Init(index);
s.SetDataSource(data);
Refresh();
}
int64 ScatterDraw::GetCount(int index)
{
ASSERT(IsValid(index));
if (series[index].PointsData()->IsParam())
return Null;
else if (IsNull(series[index].PointsData()->GetCount()))
return Null;
return series[index].PointsData()->GetCount();
}
void ScatterDraw::GetValues(int index, int64 idata, double &x, double &y)
{
ASSERT(IsValid(index) && !IsNull(GetCount(index)));
ASSERT(idata >= 0 && idata < series[index].PointsData()->GetCount());
x = series[index].PointsData()->x(idata);
y = series[index].PointsData()->y(idata);
}
double ScatterDraw::GetValueX(int index, int64 idata)
{
ASSERT(IsValid(index) && !IsNull(GetCount(index)));
ASSERT(idata >= 0 && idata < series[index].PointsData()->GetCount());
return series[index].PointsData()->x(idata);
}
double ScatterDraw::GetValueY(int index, int64 idata)
{
ASSERT(IsValid(index) && !IsNull(GetCount(index)));
ASSERT(idata >= 0 && idata < series[index].PointsData()->GetCount());
return series[index].PointsData()->y(idata);
}
ScatterDraw &ScatterDraw::PlotStyle(SeriesPlot *data)
{
int index = series.GetCount() - 1;
series[index].seriesPlot = data;
return *this;
}
ScatterDraw &ScatterDraw::MarkStyle(MarkPlot *data)
{
int index = series.GetCount() - 1;
return MarkStyle(index, data);
}
ScatterDraw &ScatterDraw::MarkStyle(int index, MarkPlot *data)
{
ASSERT(IsValid(index));
series[index].markPlot = data;
return *this;
}
ScatterDraw &ScatterDraw::MarkStyle(int index, const String name)
{
ASSERT(IsValid(index));
int typeidx = MarkPlot::TypeIndex(name);
if (typeidx >= 0)
series[index].markPlot = MarkPlot::Create(typeidx);
else
series[index].markPlot = 0;
return *this;
}
const String ScatterDraw::GetMarkStyleName(int index)
{
ASSERT(IsValid(index));
if (series[index].markPlot)
return MarkPlot::TypeName(series[index].markPlot->GetType());
else
return t_("No mark");
}
ScatterDraw &ScatterDraw::Stroke(double thickness, Color color)
{
int index = series.GetCount() - 1;
if (IsNull(color))
color = GetNewColor(index);
series[index].color = color;
series[index].thickness = thickness;
//series[index].dash = GetNewDash(index);
Refresh();
return *this;
}
ScatterDraw &ScatterDraw::Fill(Color color)
{
int index = series.GetCount() - 1;
if (IsNull(color)) {
color = GetNewColor(index);
color = Color(min(color.GetR()+30, 255), min(color.GetG()+30, 255), min(color.GetB()+30, 255));
}
series[index].fillColor = color;
Refresh();
return *this;
}
ScatterDraw &ScatterDraw::MarkColor(Color color)
{
int index = series.GetCount() - 1;
if (IsNull(color)) {
color = GetNewColor(index);
color = Color(max(color.GetR()-30, 0), max(color.GetG()-30, 0), max(color.GetB()-30, 0));
}
series[index].markColor = color;
Refresh();
return *this;
}
ScatterDraw &ScatterDraw::MarkWidth(double markWidth)
{
int index = series.GetCount() - 1;
series[index].markWidth = markWidth;
Refresh();
return *this;
}
ScatterDraw &ScatterDraw::Dash(const char *dash)
{
int index = series.GetCount() - 1;
return Dash(index, dash);
}
ScatterDraw &ScatterDraw::Dash(int index, const char *dash)
{
ASSERT(IsValid(index));
series[index].dash = dash;
Refresh();
return *this;
}
const String ScatterDraw::GetDash(int index)
{
ASSERT(IsValid(index));
return series[index].dash;
}
ScatterDraw &ScatterDraw::Legend(const String legend)
{
int index = series.GetCount() - 1;
return Legend(index, legend);
}
ScatterDraw& ScatterDraw::Legend(int index, const String legend)
{
ASSERT(IsValid(index));
series[index].legend = legend;
return *this;
}
const String& ScatterDraw::GetLegend(int index)
{
ASSERT(IsValid(index));
return series[index].legend;
}
void ScatterDraw::SetDataColor(int index, const Color& color)
{
ASSERT(IsValid(index));
series[index].color = color;
Refresh();
}
Color ScatterDraw::GetDataColor(int index) const
{
ASSERT(IsValid(index));
return series[index].color;
}
void ScatterDraw::SetDataThickness(int index, double thickness)
{
ASSERT(IsValid(index));
series[index].thickness = thickness;
Refresh();
}
double ScatterDraw::GetDataThickness(int index) const
{
ASSERT(IsValid(index));
return series[index].thickness;
}
void ScatterDraw::SetFillColor(int index, const Color& color)
{
ASSERT(IsValid(index));
series[index].fillColor = color;
Refresh();
}
Color ScatterDraw::GetFillColor(int index) const
{
ASSERT(IsValid(index));
return series[index].fillColor;
}
ScatterDraw &ScatterDraw::SetMarkWidth(int index, double markWidth)
{
ASSERT(IsValid(index));
series[index].markWidth = markWidth;
Refresh();
return *this;
}
double ScatterDraw::GetMarkWidth(int index)
{
ASSERT(IsValid(index));
return series[index].markWidth;
}
void ScatterDraw::SetMarkColor(int index, const Color& color)
{
ASSERT(IsValid(index));
series[index].markColor = color;
Refresh();
}
Color ScatterDraw::GetMarkColor(int index) const
{
ASSERT(IsValid(index));
return series[index].markColor;
}
void ScatterDraw::NoMark(int index)
{
ASSERT(IsValid(index));
series[index].markWidth = 0;
}
bool ScatterDraw::IsShowMark(int index) const throw (Exc)
{
ASSERT(IsValid(index));
return series[index].markWidth > 0;
}
void ScatterDraw::SetDataPrimaryY(int index, bool primary)
{
ASSERT(IsValid(index));
series[index].primaryY = primary;
Refresh();
}
ScatterDraw &ScatterDraw::SetDataPrimaryY(bool primary)
{
SetDataPrimaryY(series.GetCount()-1, primary);
return *this;
}
bool ScatterDraw::IsDataPrimaryY(int index) const throw (Exc)
{
ASSERT(IsValid(index));
return series[index].primaryY;
}
void ScatterDraw::SetSequentialX(int index, bool sequential)
{
ASSERT(IsValid(index));
series[index].sequential = sequential;
Refresh();
}
ScatterDraw &ScatterDraw::SetSequentialX(bool sequential)
{
SetSequentialX(series.GetCount()-1, sequential);
return *this;
}
ScatterDraw &ScatterDraw::SetSequentialXAll(bool sequential)
{
for (int i = 0; i < series.GetCount(); ++i)
SetSequentialX(i, sequential);
sequentialXAll = sequential;
return *this;
}
void ScatterDraw::Show(int index, bool show)
{
ASSERT(IsValid(index));
series[index].opacity = show ? 1 : 0;
Refresh();
}
bool ScatterDraw::IsVisible(int index)
{
ASSERT(IsValid(index));
return series[index].opacity > 0;
}
ScatterDraw &ScatterDraw::ShowAll(bool show)
{
for (int i = 0; i < series.GetCount(); ++i)
series[i].opacity = 1;
return *this;
}
ScatterDraw& ScatterDraw::Id(int id)
{
return Id(series.GetCount()-1, id);
}
ScatterDraw& ScatterDraw::Id(int index, int id)
{
ASSERT(IsValid(index));
series[index].id = id;
return *this;
}
int ScatterDraw::GetId(int index)
{
ASSERT(IsValid(index));
return series[index].id;
}
void ScatterDraw::RemoveSeries(int index)
{
ASSERT(IsValid(index));
series.Remove(index);
Refresh();
}
void ScatterDraw::RemoveAllSeries()
{
series.Clear();
Refresh();
}
Drawing ScatterDraw::GetDrawing(bool ctrl)
{
DrawingDraw ddw(3*GetSize());
SetDrawing(ddw, GetSize(), 3);
if (ctrl)
PlotTexts(ddw, GetSize(), 3);
return ddw;
}
Image ScatterDraw::GetImage(const Size &size, int scale, bool ctrl)
{
#ifndef flagGUI
ASSERT(mode != MD_DRAW);
#endif
ImageBuffer ib(scale*size);
BufferPainter bp(ib, mode);
SetDrawing(bp, size, scale);
if (ctrl)
PlotTexts(bp, size, scale);
return ib;
}
Image ScatterDraw::GetImage(int scale)
{
return GetImage(GetSize(), scale, false);
}
double ScatterDraw::GetXByPoint(double x)
{
return (x - hPlotLeft)*GetXRange()/(GetSize().cx - (hPlotLeft + hPlotRight) - 1) + GetXMin();
}
double ScatterDraw::GetYByPoint(double y)
{
return (GetSize().cy - vPlotTop - y - 1)*GetYRange()/(GetSize().cy - (vPlotTop + vPlotBottom) - GetTitleFont().GetHeight() - 1) + GetYMin();
}
double ScatterDraw::GetY2ByPoint(double y)
{
return (GetSize().cy - vPlotTop - y - 1)*GetY2Range()/(GetSize().cy - (vPlotTop + vPlotBottom) - GetTitleFont().GetHeight() - 1) + GetYMin2();
}
double ScatterDraw::GetXPointByValue(double x)
{
return (x - GetXMin())/GetXRange()*(GetSize().cx - (hPlotLeft + hPlotRight) - 1) + hPlotLeft;
}
double ScatterDraw::GetYPointByValue(double y)
{
return (GetSize().cy - vPlotTop - 1) - (y - GetYMin())/GetYRange()*(GetSize().cy - (vPlotTop + vPlotBottom) - GetTitleFont().GetHeight() - 1);
}
void ScatterDraw::Zoom(double scale, bool mouseX, bool mouseY)
{
bool canX = (minXZoom > 0 && xRange*scale > minXZoom) || (minXZoom < 0) && (maxXZoom > 0 && xRange*scale < maxXZoom) || (maxXZoom < 0);
bool canY = (minYZoom > 0 && yRange*scale > minYZoom) || (minYZoom < 0) && (maxYZoom > 0 && yRange*scale < maxYZoom) || (maxYZoom < 0);
double plotW = scale*(GetSize().cx - (hPlotLeft + hPlotRight));
double plotH = scale*(GetSize().cy - (vPlotTop + vPlotBottom)) - titleHeight;
double dw = plotW*xMajorUnit/double(xRange);
double dh = plotH*yMajorUnit/double(yRange);
bool can = canX && canY && (min<double>(dw, dh) > 20 || scale < 1);
if (mouseX && can) {
if (zoomStyleX == TO_CENTER) {
double oldXMin = xMin;
xMin += xRange*(1-scale)/2.;
xMinUnit = oldXMin + xMinUnit - xMin;
AdjustMinUnitX();
}
xRange *= scale;
if (!IsNull(maxMajorUnitsX)) {
if (xRange < 2*xMajorUnit)
xMajorUnit /= 10;
else if (xRange/xMajorUnit > maxMajorUnitsX)
xMajorUnit *= 10;
AdjustMinUnitX();
}
lastxRange = xRange;
}
if (mouseY && can) {
if (zoomStyleY == TO_CENTER) {
double oldYMin = yMin;
yMin += yRange*(1-scale)/2.;
yMinUnit = oldYMin + yMinUnit - yMin;
AdjustMinUnitY();
double oldYMin2 = yMin2;
yMin2 += yRange2*(1-scale)/2.;
yMinUnit2 = oldYMin2 + yMinUnit2 - yMin2;
AdjustMinUnitY2();
}
yRange *= scale;
yRange2 *= scale;
if (!IsNull(maxMajorUnitsY)) {
if (yRange < 2*yMajorUnit) {
yMajorUnit /= 5;
yMajorUnit2 /= 5;
} else if (yRange/yMajorUnit > maxMajorUnitsY) {
yMajorUnit *= 5;
yMajorUnit2 *= 5;
}
AdjustMinUnitY();
}
lastyRange = yRange;
}
if ((mouseX || mouseY) && can) {
WhenSetRange();
Refresh();
WhenZoomScroll();
}
}
void ScatterDraw::Scroll(double factorX, double factorY)
{
if (factorX != 0) {
double deltaX = factorX*xRange;
xMin -= deltaX;
xMinUnit += deltaX;
AdjustMinUnitX();
}
if (factorY != 0) {
double deltaY = -factorY*yRange;
yMin -= deltaY;
yMinUnit += deltaY;
AdjustMinUnitY();
if (drawY2Reticle) {
double deltaY2 = -factorY*yRange2;
yMin2 -= deltaY2;
yMinUnit2 += deltaY2;
AdjustMinUnitY2();
}
}
if (factorX != 0 || factorY != 0) {
Refresh();
WhenZoomScroll();
}
}
Pointf PointAtLen(const Pointf &p0, const Pointf &p1, double len)
{
Pointf ret;
if (p1.y == p0.y) {
ret.x = p0.x + ((p0.x < p1.x) ? len : -len);
ret.y = p0.y;
} else if (p1.x == p0.x) {
ret.x = p0.x;
ret.y = p0.y + ((p0.y < p1.y) ? len : -len);
} else {
double k = sqr((p1.x-p0.x)/(p1.y-p0.y));
double deltax = len/sqrt(1/k + 1);
double deltay = len/sqrt(k + 1);
ret.x = p0.x + ((p0.x < p1.x) ? deltax : -deltax);
ret.y = p0.y + ((p0.y < p1.y) ? deltay : -deltay);
}
return ret;
}
inline bool Even(int val) {return !(val%2);}
Vector <double> GetLineDash(String dash)
{
Vector<double> d;
CParser p(dash);
try {
while(!p.IsEof())
if(!p.Char(':')) {
double data = p.ReadDouble();
if (data > 0)
d.Add(p.ReadDouble());
}
}
catch(CParser::Error) {}
if(d.GetCount() & 1) {
Vector<double> dash1;
dash1.Append(d);
dash1.Append(d);
return dash1;
}
return d;
}
Vector <double> &GetDashedArray(String dash)
{
static VectorMap <String, Vector <double> > pats;
int pos = pats.Find(dash);
if (pos < 0) {
pats.Add(dash, GetLineDash(dash));
pos = pats.GetCount()-1;
}
return pats.GetValues()[pos];
}
void ScatterDraw::ParseTextMultiline(const String &text, Font fnt, Upp::Array <String> &texts, Upp::Array <Size> &sizes) {
Size ret(0, 0);
int npos = 0;
for (int pos = 0; npos != -1; pos = npos+1) {
npos = text.Find('\n', pos);
String &t = texts.Add();
if (npos != -1)
t = text.Mid(pos, npos-pos);
else
t = text.Mid(pos);
Size &s = sizes.Add();
s.cx = GetTextSize(t, fnt).cx;
s.cy = GetTextSize(t, fnt).cy;
}
}
Size GetTextSizeMultiline(Array <Size> &sizes) {
Size ret(0, 0);
for (int i = 0; i < sizes.GetCount(); ++i) {
if (sizes[i].cx > ret.cx)
ret.cx = sizes[i].cx;
ret.cy += sizes[i].cy;
}
return ret;
}
ScatterDraw::ScatterDraw()
{
mode = MD_ANTIALIASED;
size = Size(800, 600);
titleColor = SColorText();
graphColor = White();
titleFont = Roman(20);
labelsFont = GetStdFont();
labelsColor = SColorText();
plotAreaColor = SColorLtFace();
axisColor = SColorText();
axisWidth = 6;
hPlotLeft = hPlotRight = vPlotTop = vPlotBottom = 30;
xRange = yRange = yRange2 = 100.0;
xMin = yMin = yMin2 = xMinUnit = yMinUnit = yMinUnit2 = 0;
gridColor = SColorDkShadow();
gridWidth = 1;
drawXReticle = true; drawYReticle = true;
drawY2Reticle = false;
drawVGrid = drawHGrid = showLegend = true;
legendWidth = 80;
minXZoom = maxXZoom = minYZoom = maxYZoom = -1;
fastViewX = false;
sequentialXAll = false;
zoomStyleX = zoomStyleY = TO_CENTER;
maxMajorUnitsX = maxMajorUnitsY = 10;
SetMajorUnitsNum(5, 10);
Color(graphColor);
lastxRange = xRange;
lastyRange = yRange;
}
void DrawLine(Draw &w, double x0, double y0, double x1, double y1, double width, const Color &color)
{
w.DrawLine(fround(x0), fround(y0), fround(x1), fround(y1), fround(width), color);
}
void DrawLine(Painter &w, double x0, double y0, double x1, double y1, double width, const Color &color)
{
w.Move(x0, y0).Line(x0, y0).Line(x1, y1).Stroke(width, color);
}
void DrawVArrow(Draw &w, double x0, double y0, double x1, double y1, double width, double aWidth, double aHeight, const Color &color)
{
DrawLine(w, x0, y0, x1, y1, width, color);
if (fabs(y0 - y1) > 2*aHeight) {
if (x0 > x1)
Swap(x0, x1);
if (y0 > y1)
Swap(y0, y1);
Vector <Point> arrowU, arrowL;
arrowU << Point(fround(x0), fround(y0)) << Point(fround(x0-aWidth), fround(y0+aHeight)) << Point(fround(x0+aWidth), fround(y0+aHeight));
w.DrawPolygon(arrowU, SColorHighlight());
arrowL << Point(fround(x0), fround(y1)) << Point(fround(x0-aWidth), fround(y1-aHeight)) << Point(fround(x0+aWidth), fround(y1-aHeight));
w.DrawPolygon(arrowL, SColorHighlight());
}
}
void DrawVArrow(Painter &w, double x0, double y0, double x1, double y1, double width, double aWidth, double aHeight, const Color &color)
{
DrawLine(w, x0, y0, x1, y1, width, color);
if (fabs(y0 - y1) > 2*aHeight) {
if (x0 > x1)
Swap(x0, x1);
if (y0 > y1)
Swap(y0, y1);
w.Move(x0, y0).Line(x0-aWidth, y0+aHeight).Line(x0+aWidth, y0+aHeight).Fill(SColorHighlight());
w.Move(x0, y1).Line(x0-aWidth, y1-aHeight).Line(x0+aWidth, y1-aHeight).Fill(SColorHighlight());
}
}
void DrawHArrow(Draw &w, double x0, double y0, double x1, double y1, double width, double aWidth, double aHeight, const Color &color)
{
DrawLine(w, x0, y0, x1, y1, width, color);
if (fabs(x0 - x1) > 2*aHeight) {
if (x0 > x1)
Swap(x0, x1);
if (y0 > y1)
Swap(y0, y1);
Vector <Point> arrowL, arrowR;
arrowL << Point(fround(x0), fround(y0)) << Point(fround(x0+aHeight), fround(y0+aWidth)) << Point(fround(x0+aHeight), fround(y0-aWidth));
w.DrawPolygon(arrowL, SColorHighlight());
arrowR << Point(fround(x1), fround(y0)) << Point(fround(x1-aHeight), fround(y0+aWidth)) << Point(fround(x1-aHeight), fround(y0-aWidth));
w.DrawPolygon(arrowR, SColorHighlight());
}
}
void DrawHArrow(Painter &w, double x0, double y0, double x1, double y1, double width, double aWidth, double aHeight, const Color &color)
{
DrawLine(w, x0, y0, x1, y1, width, color);
if (fabs(x0 - x1) > 2*aHeight) {
if (x0 > x1)
Swap(x0, x1);
if (y0 > y1)
Swap(y0, y1);
w.Move(x0, y0).Line(x0+aHeight, y0+aWidth).Line(x0+aHeight, y0-aWidth).Fill(SColorHighlight());
w.Move(x1, y0).Line(x1-aHeight, y0+aWidth).Line(x1-aHeight, y0-aWidth).Fill(SColorHighlight());
}
}
void DrawText(Draw &w, double x, double y, int angle, const String &text, Font font, Color color)
{
w.DrawText(fround(x), fround(y), angle, text, font, color);
}
void DrawText(Painter &w, double x, double y, int angle, const String &text, Font font, Color color)
{
w.Begin();
w.Translate(x, y).Rotate(-angle*M_PI/1800.);
w.Text(0, 0, text, font).Fill(color);
w.End();
}
void Clip(Draw &w, double x, double y, double cx, double cy)
{
w.Clip(fround(x), fround(y), fround(cx), fround(cy));
}
void Clip(Painter &w, double x, double y, double cx, double cy)
{
w.Rectangle(x, y, cx, cy).Clip();
}
void ClipEnd(Draw &w)
{
w.End();
}
void ClipEnd(Painter &w)
{
;
}
void DrawLineOpa(Draw& w, int x0, int y0, int x1, int y1, int scale, double opacity,
double thick, const Color &_color, String dash, const Color &background)
{
Vector<Point> p;
p << Point(x0, y0) << Point(x1, y1);
DrawPolylineOpa(w, p, scale, opacity, thick, _color, dash, background);
}
void DashScaled(Painter& w, const String dash, double scale)
{
if (!dash.IsEmpty()) {
Vector<double> d;
double start = 0;
CParser p(dash);
try {
while(!p.IsEof())
if(p.Char(':'))
start = p.ReadDouble();
else
d.Add(scale*p.ReadDouble());
}
catch(CParser::Error) {}
w.Dash(d, scale*start);
}
}
void DrawLineOpa(Painter& w, int x0, int y0, int x1, int y1, int scale,
double opacity, double thick, const Color &color, String dash,
const Color &background)
{
w.Move(Pointf(x0, y0));
w.Line(Pointf(x1, y1));
DashScaled(w, dash, scale);
w.Opacity(opacity); // Before Stroke()
w.Stroke(thick*scale, color);
}
void FillRectangleOpa(Draw &w, double x0, double y0, double x1, double y1, int scale,
double opacity, const Color &background, const Color &color)
{
if (IsNull(color))
return;
Color opacolor = GetOpaqueColor(color, background, opacity);
if (x0 > x1)
Swap(x0, x1);
if (y0 > y1)
Swap(y0, y1);
w.DrawRect(int(x0), int(y0), abs(int(x1 - x0)), abs(int(y1 - y0)), opacolor);
}
void FillRectangleOpa(Painter &w, double x0, double y0, double x1, double y1, int scale,
double opacity, const Color &background, const Color &color)
{
if (IsNull(color))
return;
w.Rectangle(x0, y0, x1 - x0, y1 - y0).Opacity(opacity).Fill(color);
}
void DrawPolylineOpa(Draw& w, const Vector<Point> &p, int scale, double opacity,
double thick, const Color &_color, String dash, const Color &background)
{
ASSERT(!p.IsEmpty());
Color color = GetOpaqueColor(_color, background, opacity) ;
if (dash == LINE_SOLID)
w.DrawPolyline(p, fround(thick), color);
else {
Vector <double> &pat = GetDashedArray(dash);
if (pat.IsEmpty())
return;
int iPat = 0;
double len = pat[0]*scale; // Pixels per bar
Pointf begin, end;
begin = p[0];
for (int i = 1; i < p.GetCount();) {
double d = Distance(begin, p[i]);
if (d >= len)
end = PointAtLen(begin, p[i], len);
else {
end = p[i];
len -= d;
++i;
}
if (Even(iPat))
w.DrawLine(begin, end, fround(thick), color);
if (d >= len) {
iPat++;
if (iPat == pat.GetCount())
iPat = 0;
len = pat[iPat]*scale;
}
begin = end;
}
}
}
void DrawPolylineOpa(Painter& w, const Vector<Point> &p, int scale, double opacity,
double thick, const Color &color, String dash, const Color &background)
{
ASSERT(!p.IsEmpty());
w.Move(p[0]);
for (int i = 1; i < p.GetCount(); ++i)
w.Line(p[i]);
DashScaled(w, dash, scale);
w.Opacity(opacity); // Before Stroke()
w.Stroke(thick*scale, color);
}
void FillPolylineOpa(Draw& w, const Vector<Point> &p, int scale, double opacity,
const Color &background, const Color &fillColor)
{
ASSERT(!p.IsEmpty());
Color opacolor = GetOpaqueColor(fillColor, background, opacity) ;
w.DrawPolygon(p, opacolor);
}
void FillPolylineOpa(Painter& w, const Vector<Point> &p, int scale, double opacity,
const Color &background, const Color &fillColor)
{
ASSERT(!p.IsEmpty());
w.Move(p[0]);
for (int i = 1; i < p.GetCount(); ++i)
w.Line(p[i]);
w.Opacity(opacity);
w.Fill(fillColor); // Before Stroke()
}
void debug_h() {
; // Do nothing. Just to set a breakpoint in templated functions
}
INITBLOCK{
SeriesPlot::Register<LineSeriesPlot>("Line");
SeriesPlot::Register<StaggeredSeriesPlot>("Staggered");
SeriesPlot::Register<BarSeriesPlot>("Bar");
MarkPlot::Register<CircleMarkPlot>("Circle");
MarkPlot::Register<SquareMarkPlot>("Square");
MarkPlot::Register<TriangleMarkPlot>("Triangle");
MarkPlot::Register<CrossMarkPlot>("Cross");
MarkPlot::Register<XMarkPlot>("X");
MarkPlot::Register<RhombMarkPlot>("Rhomb");
}
END_UPP_NAMESPACE