#include "CtrlLib/CtrlLib.h"

using namespace Upp;

const char *MyDnDName = "MyReferenceDragAndDropExample";

struct MyApp : TopWindow {
	TreeCtrl   tree;

	typedef MyApp CLASSNAME;

	void DropInsert(int parent, int ii, PasteClip& d)
	{
		tree.AdjustAction(parent, d);
		if(d.Accept(MyDnDName)) {
			tree.SetCursor(tree.Insert(parent, ii, Image(), ~d));
			tree.SetFocus();
		}
	}

	void Drag()
	{
		if(!tree.IsCursor())
			return;
		int id = tree.GetCursor();
		String text = tree.Get();
		Size isz = GetTextSize(text.ToWString(), StdFont());
		ImageDraw iw(isz);
		iw.DrawRect(isz, White);
		iw.DrawText(0, 0, text);
		
		VectorMap<String, ClipData> clip;
		clip.Add(MyDnDName, text);

		if(DoDragAndDrop(clip, iw) == DND_MOVE)
			tree.Remove(id);
	}

	MyApp() {
		Add(tree.SizePos());
		Vector<int> parent, parent2;
		parent.Add(0);
		tree.SetRoot(Image(), "The Tree");
		for(int i = 1; i < 10000; i++) {
			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);
		Sizeable();
	}
};

GUI_APP_MAIN
{
	MyApp().Run();
}

void LayerTree::DropInsert(int parent, int ii, PasteClip& d)
{
	AdjustAction(parent, d);
	if(AcceptInternal<LayerTree>(d, "mytreedrag")) {
		const TreeCtrl &src = GetInternal<LayerTree>(d); 
		Vector<int> sel = src.GetSel(); 
		SaveStateToLayers(); // we need to rebuild the tree
		for(int i=0;i<sel.GetCount();i++)
		{
			Drop(parent,sel[i],ii);
		}			
		m_ptr->SetTree();
		SetFocus();
		return;
	}
	else if(d.Accept("externalInstance")) 
	{
		m_ptr->SetStatus("dropped");
		
		StringStream ss(d.Get());
		
		Layer* pLayer = Layer::Load(ss);

		if(pLayer==NULL)
			return;

		m_ptr->AddLayer(pLayer);
	
	}

//	if(AcceptText(d)) 
//	{
//		SetCursor(Insert(parent, ii, Image(), GetString(d)));
//		SetFocus();
//		return;
//	}
}

void LayerTree::Drag()
{
	if(m_ptr->AreWeBusy())
		return;

	if(DoDragAndDrop(InternalClip(*this, "mytreedrag"),
		                       this->GetDragSample()) == DND_MOVE)
	{
		RemoveSelection();		
	}
}

void LayerTree::DragLeave()
{
	int id;
	
	id = GetCursor();
	if(id<=0) // if id==0 then its the root node which is not a layer
		return;
		
	// copy to clipboard
	TreeCtrl::Node node = GetNode(id);	
	LayerOption* ptr = (LayerOption*)~node.ctrl;
	Layer* pLayer = ptr->GetLayer();
	
	StringStream ss;

	ss.SetStoring();
	
	ss.Put("layer");
	int type = int(pLayer->GetType());
	ss.Put32(type);
	
	pLayer->Serialize(ss);

	String sLayer(ss);

	String text = pLayer->GetName();
	
	Size isz = GetTextSize(text.ToWString(), StdFont());
	
	ImageDraw iw(isz);
	
	iw.DrawRect(isz, White);
	
	iw.DrawText(0, 0, text);
		
	VectorMap<String, ClipData> clip;
	
	clip.Add("externalInstance", sLayer);
		
	if(DoDragAndDrop(clip, iw) == DND_MOVE)
	{

	}

	TreeCtrl::DragLeave();	

}

void LayerTree::DropInsert(int parent, int ii, PasteClip& d)
{
	AdjustAction(parent, d);
	if(!IsDragAndDropSource() && d.Accept("externalInstance"))
	{
		m_ptr->SetStatus("dropped");
		
		StringStream ss(d.Get());
		
		Layer* pLayer = Layer::Load(ss);

		if(pLayer==NULL)
			return;

		if(parent==0)
		{
			m_ptr->AddLayer(pLayer);
			return;
		}

		TreeCtrl::Node node = GetNode(parent);	
		LayerOption* ptr = (LayerOption*)~node.ctrl;
		Layer* pParent = ptr->GetLayer();

		if(pParent)
		{
			pParent->AddChild(pLayer);	
			m_ptr->SetTree();
			m_ptr->RefreshNow();
		}
		else
		{
			m_ptr->AddLayer(pLayer);
		}
	}
	else if(AcceptInternal<LayerTree>(d, "mytreedrag") ) 
	{
		const TreeCtrl &src = GetInternal<LayerTree>(d); 
		Vector<int> sel = src.GetSel(); 
		SaveStateToLayers(); // we need to rebuild the tree
		for(int i=0;i<sel.GetCount();i++)
		{
			Drop(parent,sel[i],ii);
		}			
		m_ptr->SetTree();
		SetFocus();
		return;
	}
	
}

void LayerTree::Drag()
{
	if(m_ptr->AreWeBusy())
		return;

	if(DoDragAndDrop(InternalClip(*this, "mytreedrag"),
		                       this->GetDragSample()) == DND_MOVE)
	{
		RemoveSelection();		
	}	
}


void LayerTree::DragLeave()
{


	int id;
	
	id = GetCursor();
	if(id<=0) // if id==0 then its the root node which is not a layer
		return;
		
	// copy to clipboard
	TreeCtrl::Node node = GetNode(id);	
	LayerOption* ptr = (LayerOption*)~node.ctrl;
	Layer* pLayer = ptr->GetLayer();
	
	StringStream ss;

	ss.SetStoring();
	
	ss.Put("layer");
	int type = int(pLayer->GetType());
	ss.Put32(type);
	
	pLayer->Serialize(ss);

	String sLayer(ss);

	String text = pLayer->GetName();
	
	Size isz = GetTextSize(text.ToWString(), StdFont());
	
	ImageDraw iw(isz);
	
	iw.DrawRect(isz, White);
	
	iw.DrawText(0, 0, text);
		
	VectorMap<String, ClipData> clip;
	
	clip.Add("externalInstance", sLayer);
		
	if(DoDragAndDrop(clip, iw) == DND_MOVE)
	{

	}

	TreeCtrl::DragLeave();	

}