#include "CtrlLib/CtrlLib.h"

using namespace Upp;

struct App : TopWindow {
	TreeCtrl   tree;

	Option options[10000];

	typedef App CLASSNAME;

	void DropInsert(int parent, int ii, PasteClip& d)
	{
		tree.AdjustAction(parent, d);
		if(AcceptInternal<TreeCtrl>(d, "mytreedrag")) {
			tree.InsertDrop(parent, ii, d);
			tree.SetFocus();
			return;
		}
		if(AcceptText(d)) {
			tree.SetCursor(tree.Insert(parent, ii, Image(), GetString(d)));
			tree.SetFocus();
			return;
		}
	}

	void Drag()
	{
		if(tree.DoDragAndDrop(InternalClip(tree, "mytreedrag"),
		                       tree.GetDragSample()) == DND_MOVE)
			tree.RemoveSelection();
	}

	App() {
		Add(tree.SizePos());
		Vector<int> parent, parent2;
		parent.Add(0);
		tree.SetRoot(Image(), "The Tree");
		for(int i = 1; i < 10000; i++) 
		{
			TreeCtrl::Node node(CtrlImg::File(),options[i],20);
			node.Set(FormatIntRoman(i, true));
			parent.Add(tree.Add(parent[rand() % parent.GetCount()],node));
		
//			parent.Add(tree.Add(parent[rand() % parent.GetCount()], Image(),
//			            FormatIntRoman(i, true)));
			if((rand() & 3) == 0)
				tree.Open(parent.Top());
		}
		tree.Open(0);
		tree.WhenDropInsert = THISBACK(DropInsert);
		tree.WhenDrag = THISBACK(Drag);
		tree.MultiSelect();
		Sizeable();
	}
};

GUI_APP_MAIN
{
	App().Run();
}

Size   TreeCtrl::Item::GetValueSize() const
{
	return display ? display->GetStdSize(value) : StdDisplay().GetStdSize(value);
}

Size   TreeCtrl::Item::GetCtrlSize() const
{
	return IsNull(size) ? (ctrl ? ctrl->GetStdSize() : Size(0, 0)) : size;
}

Size   TreeCtrl::Item::GetSize() const
{
	Size sz = GetValueSize();
	Size csz = GetCtrlSize();
	sz += Size(2 * margin, 2 * margin);
	Size isz = image.GetSize();
	sz.cx += isz.cx;
	sz.cy = max(sz.cy, isz.cy);
	sz.cx += csz.cx;
	sz.cy = max(sz.cy, csz.cy);
	return sz;
}

			if(m.ctrl)
				x += m.GetCtrlSize().cx;