#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