#include "RichEdit.h"

namespace Upp {

Image DiagramEditor::MakeIcon(DiagramItem& m, Size isz)
{
	struct IconMaker : ImageMaker {
		Size         isz;
		DiagramItem  m;
		bool         dark;
		String Key() const override {
			String key = StoreAsString(const_cast<DiagramItem&>(m));
			RawCat(key, isz);
			RawCat(key, dark);
			return key;
		}
		Image Make() const override {
			ImagePainter iw(isz);
			iw.Clear();
			m.Paint(iw, Diagram(), dark ? DiagramItem::DARK : 0);
			return iw;
		}
	};

	IconMaker mk;
	m.ink = SColorText();
	mk.dark = IsDarkTheme();
	mk.m = m;
	mk.isz = isz;
	return MakeImage(mk);
}

Image DiagramEditor::ShapeIcon(int i)
{
	Size isz = icon_sz;
	DiagramItem m;
	m.shape = i;
	if(m.IsLine()) {
		m.pos = Point(2, 2);
		m.size = isz - 4;
	}
	else {
		m.pos = Pointf(Point(isz)) / 2;
		m.size = m.pos - 2.0;
	}
	m.width = DPI(1);
	m.paper = Null;
	return MakeIcon(m, isz);
}

Image DiagramEditor::CapIcon(int start, int end)
{
	Size isz = icon_sz;
	DiagramItem m;
	m.pos = Point(findarg(start, DiagramItem::CAP_CIRCLEL, DiagramItem::CAP_DISCL) >= 0 ? DPI(6) : DPI(4), isz.cy / 2);
	m.size = Size(isz.cx - (findarg(end, DiagramItem::CAP_CIRCLEL, DiagramItem::CAP_DISCL) >= 0 ? DPI(10) :
	                        findarg(end, DiagramItem::CAP_CIRCLE, DiagramItem::CAP_DISC) >= 0 ? DPI(8) : DPI(6)), 0);
	m.shape = DiagramItem::SHAPE_LINE;
	m.width = DPI(1);
	m.cap[0] = start;
	m.cap[1] = end;
	return MakeIcon(m, isz);
}

Image DiagramEditor::DashIcon(int i)
{
	return MakeValue(
		[=] { return String((char *)&i, sizeof(i)) + String("D", (int)IsDarkTheme()); },
		[=](Value& v) {
			Size isz = icon_sz;
			ImagePainter p(isz);
			p.Clear();
		
			Vector<double> h = clone(DiagramItem::GetDash(i));
			for(double& x : h)
				x *= DPI(2);
			p.Move(DPI(2), isz.cy / 2 - (i & 1) * 0.5)
			 .RelLine(isz.cx - DPI(4), 0)
			 .Dash(h, 0)
			 .Stroke(DPI(2), SColorText());
			Image m = p;
			v = m;
			return m.GetLength() * sizeof(RGBA);
		}
	).To<Image>();
}

Image DiagramEditor::WidthIcon(int i)
{
	return MakeValue(
		[=] { return String((char *)&i, sizeof(i)) + String("D", (int)IsDarkTheme()); },
		[=](Value& v) {
			Size isz = icon_sz;
			ImagePainter p(isz);
			p.Clear();
			p.Move(DPI(2), isz.cy / 2.0 - (i & 1) * 0.5)
			 .RelLine(isz.cx - DPI(4), 0)
			 .Stroke(i, SColorText());
			Image m = p;
			v = m;
			return m.GetLength() * sizeof(RGBA);
		}
	).To<Image>();
}

void DiagramEditor::Dashes(ColumnPopUp& m)
{
	m.count = DiagramItem::DASH_COUNT;
	m.columns = 4;
	m.isz = icon_sz + Size(DPI(4), DPI(4));
	m.WhenPaintItem = [=](Draw& w, const Rect& r, int ii, bool sel) {
		PopPaint(w, r, DashIcon(ii), sel);
	};
}

void DiagramEditor::PopPaint(Draw& w, const Rect& r, const Image& m, bool sel)
{
	Point p = r.CenterPos(m.GetSize());
	if(sel)
		w.DrawImage(p.x, p.y, m, SColorHighlightText());
	else
		w.DrawImage(p.x, p.y, m);
}

void DiagramEditor::Shapes(ColumnPopUp& shape)
{
	shape.count = DiagramItem::SHAPE_COUNT;
	shape.columns = 5;
	shape.isz = icon_sz + Size(DPI(4), DPI(4));
	shape.WhenPaintItem = [=](Draw& w, const Rect& r, int ii, bool sel) {
		PopPaint(w, r, ii == DiagramItem::SHAPE_SVGPATH ? DiagramImg::FontSvg() :
		               ii == DiagramItem::SHAPE_IMAGE   ? CtrlImg::open()
		                                                : ShapeIcon(ii),
		         sel && ii != DiagramItem::SHAPE_IMAGE);
	};
}

void DiagramEditor::Caps(ColumnPopUp& m, bool left)
{
	m.count = DiagramItem::CAP_COUNT;
	m.columns = 3;
	m.isz = icon_sz + Size(DPI(4), DPI(4));
	m.WhenPaintItem = [=](Draw& w, const Rect& r, int ii, bool sel) {
		PopPaint(w, r, left ? CapIcon(ii, 0) : CapIcon(0, ii), sel);
	};
}

void DiagramEditor::Widths(ColumnPopUp& m)
{
	m.count = 15;
	m.columns = 5;
	m.isz = icon_sz + Size(DPI(4), DPI(4));
	m.WhenPaintItem = [=](Draw& w, const Rect& r, int ii, bool sel) {
		PopPaint(w, r, WidthIcon(ii), sel);
	};
}

void Upp::DiagramEditor::DropColumns::Paint(Draw& w, const Rect& r, const Value& q, Color ink, Color paper, dword style) const
{
	popup.WhenPaintItem(w, r, (int)q, false);
}

DiagramEditor::DropColumns::DropColumns()
{
	AddButton().Main().WhenPush << [=] {
		int c = popup.Execute(GetScreenRect(), this);
		if(IsNull(c))
			return;
		SetData(c);
		UpdateAction();
	};
	SetDisplay(*this);
}

}