#include <creaImageIOWxTreeView.h>
#include <creaImageIOSystem.h>
#include <wx/splitter.h>
int wxCALLBACK MyCompareFunction(long item1, long item2, long WXUNUSED(sortData))
{
    // inverse the order
    if (item1 < item2)
        return -1;
    if (item1 > item2)
        return 1;

    return 0;
}
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)
		  {
		    wxListItem it;
		    it.SetTextColour(*wxRED);
		    it.SetText(_T("#C"));
		    it.SetColumn(col);
		    //		    ctrl->InsertColumn(col, _("Children"), col_style);
		    ctrl->InsertColumn(col,it);
		    //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());
	    
	    item.SetText( crea::std2wx(s));
	    //	    item.SetTextColour(*wxRED);
	    //	    item.SetBackgroundColour(*wxBLACK); 
	    item.SetColumn(0);

		//Sets the last level as selecte....How to make it select only the first time?
		//if (level==mLevelList.size()) item.SetState(wxLIST_STATE_SELECTED);

	    GetCtrl(l)->SetItem(item);
		
	    //	    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 = "?";
		item.SetText( crea::std2wx(val));
		//		item.SetTextColour(*wxBLACK);
		//		item.SetBackgroundColour(*wxWHITE); 
		item.SetColumn(k);
		GetCtrl(l)->SetItem(item);

		//		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::OnSelectedChanged(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 ); 
	if (level==mLevelList.size()-1) ValidateSelectedImages ();

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

//================================================================
  void WxTreeView::OnColClick(wxListEvent& event)
  { 
	  /*
	//Obtain the column name and the level that needs to be organized
	int colNum=event.m_col;
    GimmickMessage(1,
			" Column " <<event.m_col
			<<std::endl);

	
    wxObject* ctrl = event.GetEventObject(); 
	unsigned int level = 0;
    for (level = 0; level<mLevelList.size(); ++level)
      {
	if ( GetCtrl(level) == ctrl ) break;
      }

	  wxBusyCursor busy;
    std::vector<tree::Node*> sel=GetSelected(level+1);

	tree::Node* nodes[20];

	  int l = level - 1;
 
    // to speed up inserting we hide the control temporarily
    GetCtrl(level)->Hide();
    GetCtrl(level)->DeleteAllItems();
    
    std::vector<tree::Node*>::iterator i;
	int num=0;
    for (i=sel.begin(); i!=sel.end(); ++i)
      {
	
		//Adds columns
		GetTreeHandler()->LoadChildren(*i,1);
		tree::Node::ChildrenListType::reverse_iterator j;
		
		for (j = (*i)->GetChildrenList().rbegin(); 
			j!= (*i)->GetChildrenList().rend(); 
			++j)
			{
				nodes[num]=*j;
				num++;
			}
      }
	
	  //Sorting elements
	  int k, m; 
	  tree::Node* index;
	  
	
	for (k=1; k<num; ++k)
	{
		index = nodes[k];
		m = k;
		GimmickMessage(1,
			"Comparison Key 1: "
			    <<(nodes[m-1])->GetAttribute(mLevelList[level].key[colNum-1])
				<<"Comparison Key 2: "<<(index)->GetAttribute(mLevelList[level].key[colNum-1])
			    <<std::endl);
		while ((m > 0) && ((nodes[m-1])->GetAttribute(mLevelList[level].key[colNum-1]) > 
						   (index)->GetAttribute(mLevelList[level].key[colNum-1])))
		{
		nodes[m] = nodes[m-1];
		m = m - 1;
		}
		nodes[m] = index;
	}

	for (m=0; m<num; ++m)
	{
		tree::Node* node=nodes[m];
	    wxListItem item;
	    item.SetMask(wxLIST_MASK_STATE | 
			 wxLIST_MASK_TEXT |
			 //			 wxLIST_MASK_IMAGE |
			 wxLIST_MASK_DATA |
			 //			 wxLIST_MASK_WIDTH |
			 wxLIST_MASK_FORMAT
			 );

	    long id = GetCtrl(level)->InsertItem(item);
		int n;
		if ((node)->GetChildrenLoaded()) 
		{
			n=(node)->GetChildrenList().size();
		}
		else
		{
			n= GetTreeHandler()->GetNumberOfChildren(node);
		}
		std::ostringstream oss;
		
		oss << n;
		std::string s(oss.str());

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

    
    GetCtrl(level)->Show();
*/
  }
  //================================================================

  void WxTreeView::ValidateSelectedImages()
  {
		//Send an event telling wether the selection is valid or not
		wxCommandEvent event( 0, GetId() );
		event.SetEventObject( this );
		std::vector<tree::Node*> sel=GetSelected((mLevelList.size()+1));
		event.SetClientData(&sel);
		GetEventHandler()->ProcessEvent( event );
  }
  //================================================================
  void WxTreeView::GetSelectedAsString(std::vector<std::string>&s)
  {
	  int level=mLevelList.size();
	std::vector<tree::Node*> sel=GetSelected(level+1);
	std::vector<tree::Node*>::iterator i;
	
    for (i=sel.begin(); i!=sel.end(); ++i)
      {
		  std::string filename=(*i)->GetAttribute("FullFileName");
		  s.push_back(filename);
	  }
  }
  //================================================================
  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::OnSelectedChanged)
  
    EVT_LIST_ITEM_DESELECTED(-1, WxTreeView::OnSelectedChanged)
	/*
    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_RIGHT_CLICK(LIST_CTRL, MyListCtrl::OnColRightClick)
	*/
    EVT_LIST_COL_CLICK(-1, WxTreeView::OnColClick)
	/*
    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