#include <creaImageIOWxViewer.h>
#include <creaImageIOSystem.h>
#include <fstream>
#include <vtkCamera.h>
#include <vtkRenderer.h>
#include <vtkImageData.h>

#include <creaMessageManager.h>
#include <stdio.h>
#include <time.h>

using namespace crea;

namespace creaImageIO
{

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

  //=====================================================================
  class WxViewerPlayer: public wxThread
  {
  public:
    WxViewerPlayer(WxViewer* v) :
      mWxViewer(v)
    {}
    
    void* Entry();
    //    void SetImagesToPlay(std::vector<vtkImageData*> im);
    //    void ShowImage(vtkImageData* v);
    //	  void StartIterator();
    void  OnExit();
	
  private:
    //std::vector<vtkImageData*> mImagesToPlay;
    //vtkImageViewer2* mViewer;
    /// Associated wxvtk interactor
    //crea::creawxVTKRenderWindowInteractor  *mInteractor;
    //std::vector<vtkImageData*>::iterator i;
    WxViewer* mWxViewer;
    
    //    int mx1,mx2,my1,my2,mz1,mz2;
    //    double mspx,mspy,mspz;
  };
  
  //=====================================================================



  

  //=====================================================================
  // CTor
  WxViewer::WxViewer(wxWindow *parent, 
		     wxWindowID id,
		     wxString title,
		     const wxPoint& pos,
		     const wxSize& size)
    :   wxPanel( parent, 
		 id, 
		 pos,
		 size)
  {
    wxMutexLocker lock(mMutex);
    GimmickDebugMessage(1,"WxViewer::WxViewer"
			<<std::endl);

    mNeedRefresh = false;
    mLastImageShown = NULL;

    wxBoxSizer *topsizer = new wxBoxSizer(wxVERTICAL);

	
    // previewer
	
    mInteractor = new crea::creawxVTKRenderWindowInteractor(this,-1);
    mInteractor->UseCaptureMouseOn();	
 
    mViewer	= vtkImageViewer2::New();
    mViewer->SetupInteractor ( mInteractor );
    
    mCurrent = 0;

    mPlayer = new WxViewerPlayer(this); //images, mViewer, mInteractor, this);
	
	mPlayer->Create();
	mPlayer->Run();
	
    topsizer-> Add( mInteractor ,1,wxGROW  ,0);
    SetSizer( topsizer );     
    Layout(); 
  }
  //=====================================================================

  //=====================================================================
  /// Destructor
  WxViewer::~WxViewer()
  {
    wxMutexLocker lock(mMutex);
    GimmickDebugMessage(1,"WxViewer::~WxViewer"
			<<std::endl);
    SetMovieSize(0);
    // TO DO : desallocate cleanly
    //    delete mPlayer;
    //    delete mInteractor;
  }
  //=====================================================================


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

  void WxViewer::SetImage(int i, vtkImageData* im)
  {
    wxMutexLocker lock(mMutex);
	if(images.size()>0)
	{
		GimmickDebugMessage(5,"WxViewer::SetImage "<<i+1<<"/"<<images.size()
			<<std::endl);
		if (i<images.size())
		{
		//	if (images[i]!=0) images[i]->UnRegister(NULL);
		images[i] = im;
		//	if (im!=0) im->Register(NULL);
		}
		
	}
    
  }

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


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

  bool WxViewer::ImagesEmpty()
  {
    wxMutexLocker lock(mMutex);
    return images.empty();
  }
  //================================================================

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

  void WxViewer::SetMovieSize(unsigned int si)
  {
    wxMutexLocker lock(mMutex);
    GimmickDebugMessage(5,"WxViewer::SetMovieSize("<<(int)si<<")"
			<<std::endl);
    for (unsigned int i=0;i<images.size();++i)
      {
	if (images[i]!=0) 
	  {
	    //	    images[i]->UnRegister(NULL);
	  }
      }
    images.clear();
    for (unsigned int i=0;i<si;++i) images.push_back(0);
    mCurrent = 0;
  }
  //================================================================

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

  void WxViewer::ShowNextImage()
  {
    wxMutexLocker lock(mMutex);
	
    /*
    GimmickMessage(1,"WxViewer::ShowNextImage() "
		   <<mCurrent+1<<"/"
		   <<images.size()<<std::endl);
    */
    
    if (mCurrent<images.size()) 
      {
	ShowImage(images[mCurrent]);
	if ( images[mCurrent] != mLastImageShown ) mNeedRefresh = true;
	mLastImageShown = images[mCurrent];
      }
    mCurrent++;
    if (mCurrent >= images.size()) mCurrent = 0;
  }
  //================================================================




  //=====================================================================
  void WxViewer::ShowImage(vtkImageData* im)
  {
    GimmickDebugMessage(5,"WxViewer::ShowImage"
			<<std::endl);
    if (im==0) return;

    mViewer->SetInput(im);
	
    mViewer->SetSlice( 0 );
    
    int x1,x2,y1,y2,z1,z2;
    double spx,spy,spz;
    im->Update();
    
    im->GetSpacing(spx,spy,spz);
    im->GetExtent (x1,x2,y1,y2,z1,z2);
    /*
    std::cout << "-----------------------------"<<std::endl;
    std::cout << x1 << "-"<<x2<<std::endl; 
    std::cout << y1 << "-"<<y2<<std::endl; 
    std::cout << z1 << "-"<<z2<<std::endl; 
    std::cout << spx << "-"<<spy<<"-"<<spz<<std::endl; 
    */
    
    if ((x1!=mx1) ||
	(x2!=mx2) ||
	(y1!=my1) ||
	(y2!=my2) ||
	(z1!=mz1) ||
	(z2!=mz2) ||
	(spx!=mspx) ||
	(spy!=mspy) ||
	(spz!=mspz) 
	)
      {
	mx1 = x1;
	mx2 = x2;
	my1 = y1;
	my2 = y2;
	mz1 = z1;
	mz2 = z2;
	mspx = spx;
	mspy = spy;
	mspz = spz;
	
	double *range = im->GetScalarRange();
	 mViewer->SetColorWindow(range[1] - range[0]);
	 mViewer->SetColorLevel(0.5 * (range[1] + range[0]));

	 mViewer->GetRenderer()->ResetCamera();
	double bounds[6];


	 mViewer->GetRenderer()->ComputeVisiblePropBounds(bounds);

	 /*
	std::cout <<"bounds : "<<bounds[0]<<","
<<bounds[1]<<","
<<bounds[2]<<","
<<bounds[3]<<","
<<bounds[4]<<","
		  <<bounds[5]<<std::endl;
	 */
	  
	 mViewer->GetRenderer()->ResetCameraClippingRange(bounds);
	/*
	vtkCamera *camera = mViewer->GetRenderer()->GetActiveCamera();
	
	camera->SetViewUp ( spx*0, -spy*1, spz*0);
	camera->SetPosition( spx*(x1+x2)/2, spy*(y1+y2)/2, spz*10000000); 
	camera->SetFocalPoint	( spx*(x1+x2)/2	, spy*(y1+y2)/2	, spz*0); 
	
	camera->ComputeViewPlaneNormal();
	camera->SetParallelScale(  spx*(x2-x1)/2.0 );
	
	camera->Roll ( 180 );
	*/
	  
      }
    
    //  mInteractor->Refresh();
    //mInteractor->Render();
    //    mViewer->Render();
  
   //::wxWakeUpIdle();
  } 
  //================================================================
  
  //==================================================
  void WxViewer::OnInternalIdle()
  {
    //    mInteractor->Refresh();
	if(images.size()>0)
	{
//	 mInteractor->Render();
	}
    //mViewer->Render();      
  }

   //================================================================
  bool WxViewer::RefreshIfNecessary()
  {
    if (mNeedRefresh)
      {
	GimmickMessage(1,"WxViewer : Refreshing"<<std::endl);

	mInteractor->Render();
	mNeedRefresh = false;
	return true;
      }
    return false;
  }
  //==================================================
  void WxViewer::StopPlayer()
  {
    mPlayer->Delete();  
  }
  
 

  //  BEGIN_EVENT_TABLE(WxGimmickFrame, wxDialog)
  //    END_EVENT_TABLE()
  //================================================================

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

  void*  WxViewerPlayer::Entry()
  {
	  
    GimmickMessage(1,"WxViewerPlayer::Entry()"<<std::endl);
			
    while(true)
      { 
		if(!TestDestroy())
	    { 
			
			mWxViewer->ShowNextImage();
			//			mWxViewer->Refresh();
			::wxWakeUpIdle();
			clock_t endwait;
			endwait = clock () + 0.2 * CLOCKS_PER_SEC ;
			while (clock() < endwait) {}
			
		}
		else
		{
			break;
		}
	/*
	for(i=mImagesToPlay.begin();i!=mImagesToPlay.end();++i)
	  {
	    if(i!=mImagesToPlay.end())
	      {   

		GimmickMessage(1,"ThreadedMovie next image"<<std::endl);
		
		ShowImage(*i);
		mParent->Refresh();
		endwait = clock () + 0.2 * CLOCKS_PER_SEC ;
		while (clock() < endwait) {}
	      
	    
	  }
	*/
      }
    return 0;
  }

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

  //=====================================================================
  void WxViewerPlayer::OnExit()
  {
    GimmickMessage(1,"WxViewerPlayer::OnExit() "<<std::endl);
  }

 
} // EO namespace creaImageIO