#include <creaImageIOWxGimmickView.h>
#include <creaImageIOWxTreeView.h>
#include <creaImageIOSystem.h>
using namespace crea;
// Icons
#include "icons/accept.xpm"
#include "icons/add.xpm"
#include "icons/folder-down.xpm"
#include "icons/page-down.xpm"
#include "icons/remove.xpm"
#include "icons/database-add.xpm"
#include "icons/help.xpm"

#include <wx/imaglist.h>
#include<boost/filesystem/operations.hpp>

namespace creaImageIO
{
   
  //======================================================================
  // The ids of the different tools
  enum
    {
      TOOL_ADDFILES_ID = 1,
      TOOL_ADDDIR_ID = 2,
      TOOL_REMOVE_ID = 3,
      TOOL_ADDDATABASE_ID = 4,
      TOOL_HELP_ID = 5
    };
  //======================================================================

  //================================================================
  // 
  const int icon_number = 7;
  // Icon ids
  typedef enum
    {
      Icon_accept,
      Icon_add,
      Icon_folder_down,
      Icon_page_down,
      Icon_remove,
      Icon_database_add,
      Icon_help
    }
    icon_id;
  //================================================================

  //================================================================
  /*
  const icon_id Icon[5] = { Icon_Database,  
			    Icon_Patient,
			    Icon_Study,
			    Icon_Series,
			    Icon_Image };
  */
  //================================================================


  //======================================================================
  // CTor
  WxGimmickView::WxGimmickView(Gimmick* gimmick,
			       wxWindow *parent, 
			       const wxWindowID id,
			       const wxPoint& pos, const wxSize& size,
			       int min_dim,
				   int max_dim,
			       int number_of_threads)
    : wxPanel(parent,id,pos,size),
      GimmickView(gimmick, number_of_threads),
      mProgressDialog(0),
      mConstructed(false)
  {
    GimmickDebugMessage(1,"WxGimmickView::WxGimmickView"
			<<std::endl);
    // Sets the current directory to the home dir
    mCurrentDirectory =  std2wx(gimmick->GetHomeDirectory());

     // Connect the AddProgress callback
    gimmick->ConnectAddProgressObserver
      ( boost::bind( &WxGimmickView::OnAddProgress , this, _1 ) );

    // Create the list of icons (mIcon)
    CreateIconList();

    // Global sizer
    wxBoxSizer	*sizer = new wxBoxSizer(wxVERTICAL);

    // Create the tool bar
    CreateToolBar(); 
    sizer->Add( mToolBar ,0, wxGROW  ,0);

    // Split part below toolbar into notebook for views and panel
    // for preview, messages...
    mSplitter = new wxSplitterWindow( this , -1);
 
   
    // Notebook
    mNotebook = new wxNotebook(mSplitter,
			       -1,wxDefaultPosition, wxDefaultSize, 0);

    //Gimmick
    mGimmick=gimmick;

      
    mSelectionMaxDimension= max_dim;
    mSelectionMinDimension= min_dim;
    
    // Create the views
    CreateTreeViews();

    // Bottom panel 
    mBottomPanel = new wxPanel(mSplitter,-1);
    
	  wxBoxSizer	*bottom_sizer = new wxBoxSizer(wxVERTICAL); //HORIZONTAL);
    
    
    // Previewer
    mViewer = new WxViewer(mBottomPanel, wxID_ANY, wxT("Gimmick! Viewer"),wxDefaultPosition, wxDefaultSize );
	pointers.push_back(new ImagePointerHolder(GetDefaultImage()));
	mViewer->SetImageVector(pointers);
	mViewer->StartPlayer();


    bottom_sizer->Add(mViewer,1,wxGROW,1);
    //    mViewer->Show();

	  mText = new wxStaticText(mBottomPanel, wxID_ANY, wxT("Welcome to Gimmick!"));
	  bottom_sizer->Add(mText,0,wxGROW,0);

	  
	  
    mBottomPanel->SetSizer(bottom_sizer);

    // Splitting
    int hsize = size.GetHeight();

    int top_minsize = 450;
    int bottom_minsize = 50;

    mSplitter->SetMinimumPaneSize( bottom_minsize );
    mSplitter->SplitHorizontally( mNotebook, mBottomPanel, 
				  top_minsize);

    sizer->Add( mSplitter,1,wxGROW  ,0);


    SetSizer( sizer );     
    SetAutoLayout(true);
    Layout();
    
    mConstructed = true;
  }
  //======================================================================

  //======================================================================
  /// Destructor
  WxGimmickView::~WxGimmickView()
  {
	// stop the viewer before application exit.
	mViewer->StopPlayer();
    GimmickDebugMessage(1,"WxGimmickView::~WxGimmickView"
			<<std::endl);
  }
  //======================================================================
  
  //======================================================================
  /// Creates the tool bar
  void WxGimmickView::CreateToolBar()
  {
    long style = wxTB_HORIZONTAL | wxNO_BORDER | wxTB_TEXT;
    mToolBar = new wxToolBar(this,-1,wxDefaultPosition,wxDefaultSize,
			     style);

    mToolAddFile = mToolBar->AddTool( TOOL_ADDFILES_ID, 
				      _T("Add file(s)"),
				      mIcon->GetBitmap(Icon_page_down),
				      _T("Add one or more file to database")
				      );
    mToolAddDir = mToolBar->AddTool( TOOL_ADDDIR_ID, 
				      _T("Add folder"),
				      mIcon->GetBitmap(Icon_folder_down),
				      _T("Add the content of a folder to database")
				      );
    mToolRemove = mToolBar->AddTool( TOOL_REMOVE_ID, 
				      _T("Remove"),
				      mIcon->GetBitmap(Icon_remove),
				      _T("Remove selected items")
				      );
    mToolAddDatabase = mToolBar->AddTool( TOOL_ADDDATABASE_ID, 
				      _T("Open database"),
				      mIcon->GetBitmap(Icon_database_add),
				      _T("Open a local or distant database")
				      );
    mToolHelp = mToolBar->AddTool( TOOL_HELP_ID, 
				      _T("Help"),
				      mIcon->GetBitmap(Icon_help),
				      _T("Open help window")
				      );
    //const wxBitmap& bitmap1, const wxString& shortHelpString = "", wxItemKind kind = wxITEM_NORMAL)

    mToolBar->Realize();
  }
  //======================================================================

 
  //======================================================================
  /// Create the tree view for TreeHandler provided
  void WxGimmickView::CreateTreeView( TreeHandler* h, TimestampDatabaseHandler* tdh)
  {
    std::string name(h->GetTree().GetAttribute("Name"));
    GimmickMessage(2,"Creating the tree view for '"<<
		   name<<"'"<<std::endl);
    // Create the WxTreeView
    WxTreeView* view = new WxTreeView(h,tdh,this,mNotebook,-1);

    // TO DO : TEST THAT A VIEW WITH SAME NAME IS NOT
    // ALREADY IN THE MAP
    GetTreeViewMap()[name] = view;

    // Add Notebook page
    mNotebook->AddPage( view, crea::std2wx(name) );

  }

  //======================================================================
  void WxGimmickView::GetSelectedImages(std::vector<vtkImageData*>& s, int dim)
  {
	std::vector<std::string> files;
	GetTreeViewMap()["Local database"]->GetSelectedAsString(files);
	ReadImagesNotThreaded(s,files,dim);
  }
  //======================================================================

  //======================================================================
  void WxGimmickView::GetSelectedFiles(std::vector<std::string>& s)
  {
	GetTreeViewMap()["Local database"]->GetSelectedAsString(s);
  }
  //======================================================================

  //======================================================================
  void WxGimmickView::GetImages(int dim, 
				const std::vector<std::string>& files, 
				std::vector<vtkImageData*>& s)
  {
	ReadImagesNotThreaded(s,files,dim);
  }
  //======================================================================


  //=================================================
  void WxGimmickView::CreateIconList()
  {
    // Size of the icons;
    int size = 24;

    wxIcon icons[20];
    // should correspond to Icon_xxx enum
    icons[Icon_accept] = wxIcon(accept_xpm);
    icons[Icon_add] = wxIcon(add_xpm);
    icons[Icon_folder_down] = wxIcon(folder_down_xpm);
    icons[Icon_page_down] = wxIcon(page_down_xpm);
    icons[Icon_remove] = wxIcon(remove_xpm);
    icons[Icon_database_add] = wxIcon(database_add_xpm);
    icons[Icon_help] = wxIcon(help_xpm);

    //   unsigned int NbIcons = 8;
    // Make an image list containing small icons
    mIcon = new wxImageList(size,size,true);
    
    // Make all icons the same size = size of the first one
    int sizeOrig = icons[0].GetWidth();
    for ( size_t i = 0; i < icon_number; i++ )
      {
	if ( size == sizeOrig )
	  {
	    mIcon->Add(icons[i]);
	  }
        else
	  {
            mIcon->Add(wxBitmap(wxBitmap(icons[i]).ConvertToImage().Rescale(size, size)));
	  }
      }
  }
  //=================================================


  //=================================================
  void WxGimmickView::OnAddFiles(wxCommandEvent& event)
  {
   long style = wxOPEN | wxFILE_MUST_EXIST | wxFD_MULTIPLE;
    std::string wc("*.*");
    wxFileDialog* FD = new wxFileDialog( 0, 
					 _T("Select file"),
					 _T(""),
					 _T(""),
					 crea::std2wx(wc),
					 style,
					 wxDefaultPosition);
    
    if (FD->ShowModal()==wxID_OK)
      {
	wxBusyCursor busy;

	wxArrayString files;
	FD->GetPaths(files);
	unsigned int i;
	std::vector<std::string> filenames;
	for (i=0;i<files.GetCount();++i)
	{
	  filenames.push_back(wx2std(files[i]));
	  GimmickMessage(2,"Adding File "<<files[i]<<"."<<std::endl);
	}

	mProgressDialog = 
	  new wxProgressDialog(_T("Adding file(s)"),
			       _T(""),
			       1000,
			       this,
			       wxPD_ELAPSED_TIME |
			       //			       wxPD_ESTIMATED_TIME | 
			       //			       wxPD_REMAINING_TIME |
			       wxPD_CAN_ABORT );

	// TO DO : select the current tree handler
	mGimmick->AddFiles("Local database",filenames);

	mProgressDialog->Pulse(_T("Updating view..."));

	UpdateTreeViewLevel("Local database",1);
	delete mProgressDialog;
	DisplayAddSummary();	
      }
	
  }
  //=================================================

  //=================================================
  void WxGimmickView::OnAddDir(wxCommandEvent& event)
  {
    long style = wxDD_DEFAULT_STYLE | wxDD_DIR_MUST_EXIST;
    wxDirDialog* FD = 
      new wxDirDialog( 0, 
		       _T("Select directory"),
		       mCurrentDirectory,
		       style);
    
    if (FD->ShowModal()==wxID_OK)
      {
	
		  std::string dirname = wx2std (FD->GetPath());
		  bool recurse =  isNeedRecursive(dirname);
		  if (recurse)
		 {
             recurse = wxMessageBox(_T("Recurse into sub-directories ?"),  _T("Scan directory"),	 wxYES_NO,this ) == wxYES ? true : false;
		 }
		
		wxBusyCursor busy;
		wxString title(_T("Adding directory"));
		if (recurse) 
		title = _T("Adding directory (recursive)");
		mProgressDialog = 
		new wxProgressDialog(_T("Adding directory"),
					_T(""),
					1000,
					this,
					wxPD_ELAPSED_TIME |
					//			       wxPD_ESTIMATED_TIME | 
					//			       wxPD_REMAINING_TIME |
					wxPD_CAN_ABORT );
		
		mCurrentDirectory = FD->GetPath();  
		
		// TO DO : select the current tree handler
		mGimmick->AddDir("Local database",dirname,recurse);
		
		mProgressDialog->Pulse(_T("Updating view..."));
		
		UpdateTreeViewLevel("Local database",1);
		delete mProgressDialog;
		DisplayAddSummary();
	  }
  }

    //=================================================
   // Test a directory to know if contains sub-directory to analyze
  bool WxGimmickView::isNeedRecursive(std::string i_name)
  {
      boost::filesystem::directory_iterator iter(i_name), end_iter;
	  bool bfindir = false;
		  for(; iter != end_iter; ++iter)
		  {
			  if(boost::filesystem::is_directory(*iter))
			  {
				  return true;
			  }
		  }
		  return false;
  }
  //=================================================

  //=================================================
  void WxGimmickView::OnSelectionChange(const std::vector<tree::Node*>& sel, bool isSelection, int selection, bool needProcess)
  {      
    GimmickDebugMessage(5,
			"WxGimmickView::OnSelectionChange"
			<<std::endl);
    wxBusyCursor busy;
	bool valid=true;
	
	if(sel.size()==0)
	{
		
		valid= ValidateSelected(NULL,
				mSelectionMinDimension,
				mSelectionMaxDimension );
	}
	else if(needProcess)
	{
		ResetExtent();
		std::vector<tree::Node*>::const_iterator i;
		for(i=sel.begin();i!=sel.end()&&valid;++i)
		{
			valid= ValidateSelected((*i),
				mSelectionMinDimension,
				mSelectionMaxDimension );
		}
	}
	else if(isSelection)
	{
		valid= ValidateSelected(sel.front(),
				mSelectionMinDimension,
				mSelectionMaxDimension );
	}
	else
	{
		ResetExtent();
		std::vector<tree::Node*>::const_iterator i;
		for(i=sel.begin();i!=sel.end()&&valid;++i)
		{
			valid= ValidateSelected((*i),
				mSelectionMinDimension,
				mSelectionMaxDimension );
		}
	}
	
    mText->SetLabel(crea::std2wx(GetMessage()));
    /*if(valid)
      {
	ReadImageThreaded(sel);
      }
    else
      {
		  ClearSelection();
      }*/
	ReadImageThreaded(sel);

    
   }

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

  //==================================================
  ///Reads Images (Threaded)
  void WxGimmickView::ReadImageThreaded(const std::vector<tree::Node*>& sel)
  {	
   GimmickDebugMessage(5,
		       "ReadImageThreaded"
		       <<std::endl);
   int maxprio = GetMaximalPriority();
   int prio = maxprio + 2000;

   if(sel.size()>0)
   {
   //First load the selected images
   mCurImageItemToShow = sel.front();
   pointers.clear();
   int index = 0;
   std::vector<tree::Node*>::const_iterator selected;
   for(selected=sel.begin();selected!=sel.end();++selected)
     {
       GimmickDebugMessage(5,
			   "Requesting image from selected "
			   <<(*selected)->GetAttribute("FullFileName")
			   <<std::endl);
	   ImagePointerHolder* ph=new ImagePointerHolder(GetDefaultImage());
	   pointers.push_back(ph);
       RequestReading(*selected,prio,index,ph);
       //       AddEntryToMap(*selected);
       prio--;
       index++;
     }
	mViewer->SetImageVector(pointers);
	//Going up
	prio = maxprio + 20;
	std::vector<tree::Node*> up;
	GetTreeViewMap()["Local database"]->GetNodes(up,true);
	std::vector<tree::Node*>::iterator iterUp;
	for(iterUp=up.begin();iterUp!=up.end();++iterUp)
	{
		GimmickDebugMessage(5,
				"Requesting image from neighbors up "
				<<(*iterUp)->GetAttribute("FullFileName")
				<<std::endl);
		ImagePointerHolder* ph=new ImagePointerHolder(GetDefaultImage());
		RequestReading(*iterUp,prio,-1,ph);
		//		AddEntryToMap(*iterUp);
		prio--;
		if (prio == maxprio) break;
	}

	//Going down
	prio = maxprio + 19;
	std::vector<tree::Node*> down;
	GetTreeViewMap()["Local database"]->GetNodes(down,false);
	std::vector<tree::Node*>::iterator iterDown;
	for(iterDown=down.begin();iterDown!=down.end();++iterDown)
	{
		GimmickDebugMessage(5,
				"Requesting image from neighbors down "
				<<(*iterDown)->GetAttribute("FullFileName")
				<<std::endl);
		ImagePointerHolder* ph=new ImagePointerHolder(GetDefaultImage());
		RequestReading(*iterDown,prio,-1,ph);
		//		AddEntryToMap(*iterDown);
		prio--;
		if (prio == maxprio) break;
	}
   }
   else
   {
	   pointers.clear();
	   ImagePointerHolder* ph=new ImagePointerHolder(GetDefaultImage());
	   pointers.push_back(ph);
	   mViewer->SetImageVector(pointers);
   }
  }

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

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

  //==================================================
   void  WxGimmickView::OnInternalIdle()
  {
   if (!mConstructed) return;
   static bool first_time = true;
   if (false)
   {
  	first_time = false;
     }
   //   GimmickMessage(1,"WxGimmickView : Refresh viewer"<<std::endl);
	//  mViewer->StartPlayer();
   if(mViewer)
   {
	   mViewer->RefreshIfNecessary();
   }
  }

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

  //==================================================
   void  WxGimmickView::ClearSelection()
  {
	pointers.clear();
	pointers.push_back(new ImagePointerHolder(GetDefaultImage()));
	mViewer->SetImageVector(pointers);
	mViewer->RefreshIfNecessary();
	ResetExtent();
  }
  
  //=================================================
 
  //=================================================
  void WxGimmickView::OnRemove(wxCommandEvent& event)
  {
  	//TODO Select current tree handler       
    wxBusyCursor busy;
    GetTreeViewMap()["Local database"]->RemoveSelected();

	ClearSelection();
  }
  //=================================================

  //=================================================
  /// AddProgress Gimmick callback
  void WxGimmickView::OnAddProgress( Gimmick::AddProgress& p)
  {

    char mess[200];
   
	sprintf(mess,"%i dirs : %i files :\n            %i handled - %i added",
	   p.GetNumberScannedDirs(),
	   p.GetNumberScannedFiles(),
	   p.GetNumberHandledFiles(),
	   p.GetNumberAddedFiles());
    //    std::cout << "OnAddProgress "<<mess<<std::endl;
    wxString s(wxString::From8BitData(mess));
    //  std::cout << "Pulse"<<std::endl;
    if (!mProgressDialog->Pulse(s)) 
      {
	p.SetStop();
      }
    //  std::cout << "OnAddProgress ok"<<std::endl;
  }
  //=================================================

  //=================================================
  void WxGimmickView::DisplayAddSummary()
  {
    const Gimmick::AddProgress& p = mGimmick->GetAddProgress();
    std::stringstream mess;
    mess << "Dirs \tscanned\t: " << p.GetNumberScannedDirs()  << "\n";
    mess << "Files\tscanned\t: " << p.GetNumberScannedFiles() << "\n";
    mess << "Files\thandled\t: " << p.GetNumberHandledFiles() << "\n\n";
    mess << "Files\tadded  \t: " << p.GetNumberAddedFiles()   << "\n\n";

    /*    char times[500];
    sprintf(times,"Time to parse dir \t\t: %ld ms \t%d°/o\nTime to read files info \t: %ld ms \t%d°/o\nTime to update structs \t: %ld ms \t%d°/o\nTime to update database \t: %ld ms \t%d°/o\nTotal time \t\t\t: %ld ms",
	    summary.parse_time,
	    (int)( summary.parse_time*100./summary.total_time),
	    summary.file_scan_time,
	    (int)(summary.file_scan_time*100./summary.total_time),
	    summary.update_structs_time,
	    (int)(summary.update_structs_time*100./summary.total_time),
	    summary.update_database_time,
	    (int)(summary.update_database_time*100./summary.total_time),
	    summary.total_time );
    
    mess << times;
    */
    wxMessageBox(std2wx(mess.str()),_T("Addition result"),wxOK,this);
  }

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

   //=================================================
  BEGIN_EVENT_TABLE(WxGimmickView, wxPanel)
    EVT_TOOL(TOOL_ADDFILES_ID, WxGimmickView::OnAddFiles)
    EVT_TOOL(TOOL_ADDDIR_ID, WxGimmickView::OnAddDir)
	EVT_TOOL(TOOL_REMOVE_ID, WxGimmickView::OnRemove)
  END_EVENT_TABLE()
  //=================================================

} // EO namespace creaImageIO