#include <creaImageIOWxTreeView.h>
#include <creaImageIOSystem.h>
#include <wx/splitter.h>

namespace creaImageIO
{
  //=====================================================================
  // CTor
  WxTreeView::WxTreeView(TreeHandler* handler,
			 wxWindow* parent,
			 const wxWindowID id)
    : wxPanel(parent,id),
      TreeView(handler)
  {
    GimmickDebugMessage(1,"WxTreeView::WxTreeView"
			<<std::endl);

    
    // Split part below toolbar into notebook for views and panel
    // for preview, messages...
    // TO DO : Splitter
    //    mSplitter = new wxSplitterWindow( this , -1);

    // Global sizer
    wxBoxSizer	*sizer = new wxBoxSizer(wxHORIZONTAL);
    
    int ctrl_style = wxLC_REPORT | wxLC_VRULES;
    int col_style = wxLIST_FORMAT_LEFT;

    // Creating the ListCtrl for the levels > 0 (not for Root level)
    for (int i = 1;
	 i < handler->GetTree().GetNumberOfLevels();
	 ++i)
      {

	GimmickDebugMessage(5,"Creating view for level "<<i
			    <<std::endl);
	LevelType level;

	// If the first level : parent = this
	wxWindow* sparent = this;
	// else parent = last splitter
	if (i>1) sparent = mLevelList[i-2].wxSplitter;

	level.wxSplitter = new wxSplitterWindow( sparent , -1);
	//	    level.wxSplitter->SetMinimumPaneSize(100);
	
	wxListCtrl* ctrl = new wxListCtrl(level.wxSplitter,
					  i,
					  wxDefaultPosition, 
					  wxDefaultSize,
					  ctrl_style);
	level.wxCtrl = ctrl;
	level.wxSplitter->Initialize(ctrl);

	// Create the columns : one for each attribute of the level
	int col = 0;
			//ctrl->InsertItem(0, "1");
	
	tree::LevelDescriptor::AttributeDescriptorListType::const_iterator a;
	for (a  = handler->GetTree().GetAttributeDescriptorList(i).begin();
	     a != handler->GetTree().GetAttributeDescriptorList(i).end();
	     ++a)
	  {
		  if(col==0)
		  {
			ctrl->InsertColumn(col, _("Children"), col_style);
			//ctrl->InsertItem(0, "1");
			//level.key.push_back(_("Children"));
			//level.key.push_back(handler->GetTree().GetChildrenList().size());
			col++;
		  }
		  
		GimmickDebugMessage(5,"Creating column "<<col<<" : "
				<<a->GetName()
				<<std::endl);
		ctrl->InsertColumn(col, 
				crea::std2wx(a->GetName()),
				col_style);
		level.key.push_back(a->GetKey());
		//	    ctrl->SetColumnWidth(col, wxLIST_AUTOSIZE );
		col++;
		
		
	  }
	  
	mLevelList.push_back(level);
      }
    

    /// Initialize the first level splitter
 
      sizer->Add( mLevelList[0].wxSplitter ,1, wxGROW  ,0);

    UpdateLevel(1);

    SetSizer( sizer );     
    SetAutoLayout(true);
    Layout();

  }
  //=====================================================================

  //=====================================================================
  /// Destructor
  WxTreeView::~WxTreeView()
  {
    GimmickDebugMessage(1,"WxTreeView::~WxTreeView"
			<<std::endl);
  }
  //=====================================================================
  
  
  //=====================================================================
  struct ItemData
  {
    tree::Node* node;
  };
 
  //=====================================================================
   std::vector<tree::Node*> WxTreeView::GetSelected(int level)
  {
    int l = level - 1;
    // the selection of upper level
    std::vector<tree::Node*> sel;
	
    if (level == 1) 
      {
	sel.push_back(GetTreeHandler()->GetTree().GetTree());
      }
    else 
      {
	int n = GetCtrl(l-1)->GetItemCount();
	for (int i = 0; i < n; i++)
	  {
	    if ( GetCtrl(l-1)->GetItemState(i,wxLIST_STATE_SELECTED))
	      {
		long adr = GetCtrl(l-1)->GetItemData(i);
		tree::Node* n = ((ItemData*)adr)->node;
		sel.push_back(n);
	      }
	  }	
      }
	
	  return sel;
  }

  //=====================================================================
  
  ///Removes selected nodes on given level
  void WxTreeView::RemoveSelected( int level )
  {
	  std::vector<tree::Node*> sel=GetSelected(level+1);
	  bool erase=false;
	  if (wxMessageBox(_T("Delete file(s) ?"),
			 _T("Remove Files"),
			 wxYES_NO,this ) == wxYES)
	  {
	    erase = true;
	  }
	  if(erase)
	  {
		std::vector<tree::Node*>::iterator i;
		for (i=sel.begin(); i!=sel.end(); ++i)
		{
			GimmickDebugMessage(2,
					"deleting '"
					<<(*i)->GetLabel()
					<<"'"<<level
					<<std::endl);
				GetTreeHandler()->Remove(*i);
		}

		UpdateLevel(level);
	  }
	  
  }
    

  //=====================================================================

 
  //=====================================================================
  /// 
  void WxTreeView::UpdateLevel( int level )
  {
    GimmickDebugMessage(1,
			GetTreeHandler()->GetTree().GetLabel()
			<<" view : updating level "<<level
			<<std::endl);
    
    wxBusyCursor busy;
    RecursiveUpdateLevel(level);
    int i;
    for (i=0; i<level-1; i++)
      {
	if (!GetSplitter(i)->IsSplit()) 
	  GetSplitter(i)->SplitVertically(  GetCtrl(i), GetSplitter(i+1),
					    100 );
      }
    if (GetSplitter(i)->IsSplit()) GetSplitter(i)->Unsplit();    
    
  }
  //=====================================================================
  
  /// 
  void WxTreeView::RecursiveUpdateLevel( int level )
  {
	  GimmickDebugMessage(2,
			GetTreeHandler()->GetTree().GetLabel()
			<<" view : updating level (recursive)"<<level
			<<std::endl);


    std::vector<tree::Node*> sel=GetSelected(level);

	  int l = level - 1;
 
    // to speed up inserting we hide the control temporarily
    GetCtrl(l)->Hide();
    GetCtrl(l)->DeleteAllItems();
    
    std::vector<tree::Node*>::iterator i;
    for (i=sel.begin(); i!=sel.end(); ++i)
      {
	GimmickDebugMessage(2,
			    "adding children of '"
			    <<(*i)->GetLabel()
			    <<"'"<<level
			    <<std::endl);
	
	//Adds columns
	GetTreeHandler()->LoadChildren(*i,1);
	tree::Node::ChildrenListType::reverse_iterator j;
	for (j = (*i)->GetChildrenList().rbegin(); 
	     j!= (*i)->GetChildrenList().rend(); 
	     ++j)
	  {
	    wxListItem item;
	    item.SetMask(wxLIST_MASK_STATE | 
			 wxLIST_MASK_TEXT |
			 //			 wxLIST_MASK_IMAGE |
			 wxLIST_MASK_DATA |
			 //			 wxLIST_MASK_WIDTH |
			 wxLIST_MASK_FORMAT
			 );

	    ItemData* data = new ItemData;
	    data->node = *j;
	    item.SetData(data);
	    
	  
	    long id = GetCtrl(l)->InsertItem(item);

		std::ostringstream oss;
		
		int	n= GetTreeHandler()->GetNumberOfChildren(*j);
		
		oss << n;
		std::string s(oss.str());
		
		GetCtrl(l)->SetItem(id,0, crea::std2wx(s));
		GetCtrl(l)->SetColumnWidth(0, wxLIST_AUTOSIZE );

	  for (int k=1; k<GetCtrl(l)->GetColumnCount(); k++)
	      {
		std::string val = (*j)->GetAttribute(mLevelList[l].key[k-1]);
		if (val.size()==0) val = "?";
		GetCtrl(l)->SetItem(id,k, crea::std2wx(val));
		GetCtrl(l)->SetColumnWidth(k, wxLIST_AUTOSIZE );
		
	      }
	    
	  }
      }
    
    GetCtrl(l)->Show();

    
    if (level<mLevelList.size()) UpdateLevel(level+1);
 }
  //=====================================================================


  //================================================================
  void WxTreeView::OnSelected(wxListEvent& event)
  { 
    GimmickDebugMessage(1,
			GetTreeHandler()->GetTree().GetLabel()
			<<" view : item selected "
			<<std::endl);

    wxListItem info;
    info.m_itemId = event.m_itemIndex;
    

    // retrieve the level
    wxObject* obj = event.GetEventObject();   
    unsigned int level = 0;
    for (level = 0; level<mLevelList.size(); ++level)
      {
	if ( GetCtrl(level) == obj ) break;
      }
    GimmickDebugMessage(1,
			" Level "<<level+1
			<<std::endl);
    if (level<mLevelList.size()-1) UpdateLevel( level + 2 ); 

  }
  //================================================================



  BEGIN_EVENT_TABLE(WxTreeView, wxPanel)
  /*
    EVT_SIZE(MyFrame::OnSize)

    EVT_MENU(LIST_QUIT, MyFrame::OnQuit)
    EVT_MENU(LIST_ABOUT, MyFrame::OnAbout)
    EVT_MENU(LIST_LIST_VIEW, MyFrame::OnListView)
    EVT_MENU(LIST_REPORT_VIEW, MyFrame::OnReportView)
    EVT_MENU(LIST_ICON_VIEW, MyFrame::OnIconView)
    EVT_MENU(LIST_ICON_TEXT_VIEW, MyFrame::OnIconTextView)
    EVT_MENU(LIST_SMALL_ICON_VIEW, MyFrame::OnSmallIconView)
    EVT_MENU(LIST_SMALL_ICON_TEXT_VIEW, MyFrame::OnSmallIconTextView)
    EVT_MENU(LIST_VIRTUAL_VIEW, MyFrame::OnVirtualView)
    EVT_MENU(LIST_SMALL_VIRTUAL_VIEW, MyFrame::OnSmallVirtualView)

    EVT_MENU(LIST_FOCUS_LAST, MyFrame::OnFocusLast)
    EVT_MENU(LIST_TOGGLE_FIRST, MyFrame::OnToggleFirstSel)
    EVT_MENU(LIST_DESELECT_ALL, MyFrame::OnDeselectAll)
    EVT_MENU(LIST_SELECT_ALL, MyFrame::OnSelectAll)
    EVT_MENU(LIST_DELETE, MyFrame::OnDelete)
    EVT_MENU(LIST_ADD, MyFrame::OnAdd)
    EVT_MENU(LIST_EDIT, MyFrame::OnEdit)
    EVT_MENU(LIST_DELETE_ALL, MyFrame::OnDeleteAll)
    EVT_MENU(LIST_SORT, MyFrame::OnSort)
    EVT_MENU(LIST_SET_FG_COL, MyFrame::OnSetFgColour)
    EVT_MENU(LIST_SET_BG_COL, MyFrame::OnSetBgColour)
    EVT_MENU(LIST_TOGGLE_MULTI_SEL, MyFrame::OnToggleMultiSel)
    EVT_MENU(LIST_SHOW_COL_INFO, MyFrame::OnShowColInfo)
    EVT_MENU(LIST_SHOW_SEL_INFO, MyFrame::OnShowSelInfo)
    EVT_MENU(LIST_FREEZE, MyFrame::OnFreeze)
    EVT_MENU(LIST_THAW, MyFrame::OnThaw)
    EVT_MENU(LIST_TOGGLE_LINES, MyFrame::OnToggleLines)
    EVT_MENU(LIST_MAC_USE_GENERIC, MyFrame::OnToggleMacUseGeneric)

    EVT_UPDATE_UI(LIST_SHOW_COL_INFO, MyFrame::OnUpdateShowColInfo)
    EVT_UPDATE_UI(LIST_TOGGLE_MULTI_SEL, MyFrame::OnUpdateToggleMultiSel)
END_EVENT_TABLE()

BEGIN_EVENT_TABLE(MyListCtrl, wxListCtrl)
    EVT_LIST_BEGIN_DRAG(LIST_CTRL, MyListCtrl::OnBeginDrag)
    EVT_LIST_BEGIN_RDRAG(LIST_CTRL, MyListCtrl::OnBeginRDrag)
    EVT_LIST_BEGIN_LABEL_EDIT(LIST_CTRL, MyListCtrl::OnBeginLabelEdit)
    EVT_LIST_END_LABEL_EDIT(LIST_CTRL, MyListCtrl::OnEndLabelEdit)
    EVT_LIST_DELETE_ITEM(LIST_CTRL, MyListCtrl::OnDeleteItem)
    EVT_LIST_DELETE_ALL_ITEMS(LIST_CTRL, MyListCtrl::OnDeleteAllItems)
#if WXWIN_COMPATIBILITY_2_4
    EVT_LIST_GET_INFO(LIST_CTRL, MyListCtrl::OnGetInfo)
    EVT_LIST_SET_INFO(LIST_CTRL, MyListCtrl::OnSetInfo)
#endif
  */
    EVT_LIST_ITEM_SELECTED(-1, WxTreeView::OnSelected)
  /*
    EVT_LIST_ITEM_DESELECTED(LIST_CTRL, MyListCtrl::OnDeselected)
    EVT_LIST_KEY_DOWN(LIST_CTRL, MyListCtrl::OnListKeyDown)
    EVT_LIST_ITEM_ACTIVATED(LIST_CTRL, MyListCtrl::OnActivated)
    EVT_LIST_ITEM_FOCUSED(LIST_CTRL, MyListCtrl::OnFocused)

    EVT_LIST_COL_CLICK(LIST_CTRL, MyListCtrl::OnColClick)
    EVT_LIST_COL_RIGHT_CLICK(LIST_CTRL, MyListCtrl::OnColRightClick)
    EVT_LIST_COL_BEGIN_DRAG(LIST_CTRL, MyListCtrl::OnColBeginDrag)
    EVT_LIST_COL_DRAGGING(LIST_CTRL, MyListCtrl::OnColDragging)
    EVT_LIST_COL_END_DRAG(LIST_CTRL, MyListCtrl::OnColEndDrag)

    EVT_LIST_CACHE_HINT(LIST_CTRL, MyListCtrl::OnCacheHint)

#if USE_CONTEXT_MENU
    EVT_CONTEXT_MENU(MyListCtrl::OnContextMenu)
#endif
    EVT_CHAR(MyListCtrl::OnChar)

    EVT_RIGHT_DOWN(MyListCtrl::OnRightClick)
  */
END_EVENT_TABLE()
  
} // EO namespace creaImageIO