class EditableTree : public TreeCtrl {
public:
	typedef EditableTree CLASSNAME;
		
	EditableTree() : selId_(0) {
		NoRoot().MultiSelect(true).HighlightCtrl(false);
		WhenSel = THISBACK(OnSelect);				
		WhenOpen = THISBACK1(ChangeNodeImg, true);
	}
	
	int AddNode(int parent) {
		EditString &edit = edit_.Add();
   		edit.SetData("NODE" + AsString(GetChildCount(0) + 1));
    	edit.AutoSize();
		edit.SetFrame(NullFrame());
	
		Node node;
		node.SetCtrl(edit);        
       	return Add(parent, node);       
	}
	
	void OnAddNode() {
		int id = AddNode(selId_);	        
		Open(selId_);
        SelectNode(id);
	}
	
	void SelectNode(int id) {
		ClearSelection();
		SelectOne(id);
		selId_ = GetSel()[0];
		ASSERT(selId_ == id);//failed
	}

	void OnSelect() {
		if (GetSelectCount() <= 0 || selId_ == GetSel()[0])
			return;
		
		selId_ = GetSel()[0];
	}
	
	void ChangeNodeImg(int selId, bool open) {
		if (!selId) 
			return;
	
		Node node = GetNode(selId);
		//...some image assignment
		
		SetNode(selId, node);//if comment => no ASSERT(selId_ == id); failed
	}
	
private:
	int selId_;
    Array<EditString> edit_;
};
 
GUI_APP_MAIN
{	
	TopWindow wnd;
	Button btn;
	wnd.Add(btn.SetLabel("Add").LeftPos(10).TopPos(10));
	
	EditableTree tree;
	wnd.Add(tree.VSizePos(40).HSizePos());

	tree.AddNode(0);
	tree.AddNode(0);
	btn <<= callback(&tree, &EditableTree::OnAddNode);
    wnd.Run();
}