#include "Test.h"

Test::Test()
{
	CtrlLayout(*this, "Window title");

	int n;
	
	n = t.Add(0, "DUMMY1");
	t.Remove(t.GetChildIndex(0, n)+1);
	t.Add(0, "real1");

	n = t.Add(0, "DUMMY2");
	t.Remove(t.GetChild(0, 1));
	t.Add(0, "real2");

	t.Open(0);
}

GUI_APP_MAIN
{
	Test().Run();
}

void OptionTree::SetLabel(int id, const char *text)
{
	Node n = GetNode(id);
	Option *o = dynamic_cast<Option *>(~n.ctrl);
	if(o)
		o->SetLabel(text);
	SetNode(id, n);
}

TreeCtrl::Node::Node(const Image& img, Ctrl& ctrl, int cx, int cy)
{
	Init();
	SetCtrl(ctrl);
	image = img;
	size = Null;
	if(cx > 0)
		size.cx = cx;
	if(cy > 0)
		size.cy = cy;
}

int CopyIfSelected(TreeCtrl& dst, int did, const OptionTree& src, int id)
{
	TreeCtrl::Node x = src.GetNode(id);
	x.ctrl = NULL;
	int orig = did;
	if (src.Get(id))
		did = dst.Add(did, x);
	dst.Open(orig);
	for(int i = 0; i < src.GetChildCount(id); i++)
		CopyIfSelected(dst, did, src, src.GetChild(id, i));
	return did;
}

String OptionTree::GetLabel(int id) const
{
	Node n = GetNode(id);
	Option *o = dynamic_cast<Option *>(~n.ctrl);
	if(o)
		return o->GetLabel();
	
	return "";
}

// Add a callback to the Option (must be added not to break SetOption)
tree.GetNode(nodeid).ctrl->WhenAction << THISBACK(OptionClick);

// Callback function
// Fake left click as if the option wasn't there
void OptionClick()
{
	dword flags = GetMouseFlags();
	Point p = tree.GetMouseViewPos();
	tree.LeftDown(p, flags);
	tree.LeftUp(p, flags);
}

void TreeCtrl::ChildGotFocus()
{
	if (multiselect) return; // Don't clear multi-selection!
	for(int i = 0; i < line.GetCount(); i++) {
		Item& m = item[line[i].itemi];
		if(m.ctrl && m.ctrl->HasFocusDeep()) {
			SetCursorLine(i);
			return;
		}
	}
}

TreeCtrl::
	String GetLabel(int id) const                   { return option[id]->GetLabel(); }

TreeCtrl::Node::Node(const Image& img, Ctrl& ctrl, int cx, int cy)
{
	Init();
	SetCtrl(ctrl);
	image = img;
	size = Null;
	if(cx > 0)
		size.cx = cx;
	if(cy > 0)
		size.cy = cy;
}

TreeCtrl::Node::Node(const Image& img, Ctrl& ctrl, int cx, int cy)
{
	Init();
	SetCtrl(ctrl);
	image = img;
	size = ctrl.GetMinSize();
	if(cx > 0)
		size.cx = cx;
	if(cy > 0)
		size.cy = cy;
}