#include "bbmaracasvisuShowNPoints.h"
#include "bbcreaMaracasVisuPackage.h"

#include "vtkProperty.h"
#include "vtkSphereSource.h"
#include "vtkPolyDataMapper.h"
#include "vtkRenderWindow.h"
#include "vtkTextActor3D.h"

namespace bbcreaMaracasVisu
{


//----------------------------------------------------------------------
	WidgetShowNPoints::WidgetShowNPoints(wxWindow *parent,  bbcreaMaracasVisu::ShowNPoints *box)
    : wxPanel( parent, -1, wxDefaultPosition, wxDefaultSize, wxTAB_TRAVERSAL)
  {
    mbbShowNPoints				= box;
    this->renderer				= NULL;
	wxPanel	*panel				= this;
    wxSizer *sizer				= NULL;


	// Widget interface
	textCtrl						= new wxTextCtrl(panel, -1);
	wxButton *btnAddPoint			= new wxButton( panel, -1, _T("Add Point"));
	wxButton *btnRenamePoint		= new wxButton( panel, -1, _T("Rename point"));
	wxButton *btnEraseLastPoint		= new wxButton( panel, -1, _T("Erase Last point"));
	wxButton *btnErasePoint			= new wxButton( panel, -1, _T("Erase point"));
	wxButton *btnDeleteAllPoints	= new wxButton( panel, -1, _T("Delete all points"));
	wxButton *btnSavePoints			= new wxButton( panel, -1, _T("Save points"));
	wxButton *btnLoadPoints			= new wxButton( panel, -1, _T("Load points"));
	txtNrPoints						= new wxStaticText(panel,-1, _T(" "));

	wxFlexGridSizer *sizer1 = new wxFlexGridSizer(1);
//    sizer1->Add(new wxStaticText(panel,-1,_T("  ")));

	  Connect(btnAddPoint->GetId()			, wxEVT_COMMAND_BUTTON_CLICKED  , (wxObjectEventFunction) &WidgetShowNPoints::OnAddPoint);
	  Connect(btnRenamePoint->GetId()		, wxEVT_COMMAND_BUTTON_CLICKED  , (wxObjectEventFunction) &WidgetShowNPoints::OnRenamePoint);
	  Connect(btnEraseLastPoint->GetId()	, wxEVT_COMMAND_BUTTON_CLICKED  , (wxObjectEventFunction) &WidgetShowNPoints::OnEraseLastPoint);
	  Connect(btnErasePoint->GetId()		, wxEVT_COMMAND_BUTTON_CLICKED  , (wxObjectEventFunction) &WidgetShowNPoints::OnErasePoint);
	  Connect(btnDeleteAllPoints->GetId()	, wxEVT_COMMAND_BUTTON_CLICKED  , (wxObjectEventFunction) &WidgetShowNPoints::OnDeleteAllPoints);
	  Connect(btnSavePoints->GetId()		, wxEVT_COMMAND_BUTTON_CLICKED  , (wxObjectEventFunction) &WidgetShowNPoints::OnSavePoints);
	  Connect(btnLoadPoints->GetId()		, wxEVT_COMMAND_BUTTON_CLICKED  , (wxObjectEventFunction) &WidgetShowNPoints::OnLoadPoints);

	  sizer1->Add(textCtrl);
	  sizer1->Add(btnAddPoint);
	  sizer1->Add(btnRenamePoint);
	  sizer1->Add(btnErasePoint);
	  sizer1->Add(btnEraseLastPoint);
	  sizer1->Add(btnDeleteAllPoints);
	  sizer1->Add(txtNrPoints);
	  sizer1->Add(btnSavePoints);
	  sizer1->Add(btnLoadPoints);

	  sizer = sizer1;
	  panel	->	SetSizer(sizer);
	  panel	->	SetAutoLayout(true);
	  panel	->	Layout();

}


//------------------------------------------------------------------------
WidgetShowNPoints::~WidgetShowNPoints()
{
}

void WidgetShowNPoints::SetRadio(double radio)
{
	mradio=radio;
}

//------------------------------------------------------------------------
std::vector<int> WidgetShowNPoints::GetLstPointsX()
{
	return lstPointsX;
}

//------------------------------------------------------------------------
std::vector<int> WidgetShowNPoints::GetLstPointsY()
{
	return lstPointsY;
}

//------------------------------------------------------------------------
std::vector<int> WidgetShowNPoints::GetLstPointsZ()
{
	return lstPointsZ;
}

//------------------------------------------------------------------------
	std::vector<std::string> WidgetShowNPoints::GetLstLabels()
{
	return lstLabels;
}




//------------------------------------------------------------------------
void WidgetShowNPoints::SetPoint(std::vector<int> ppoint)
{
	mpoint = ppoint;
}

//------------------------------------------------------------------------
void WidgetShowNPoints::SetColour(std::vector<double> colour)
{
	this->mcolour = colour;
}

//------------------------------------------------------------------------
void WidgetShowNPoints::SetOpacity(double opacity)
{
	this->mopacity=opacity;
}

//------------------------------------------------------------------------
void WidgetShowNPoints::SetImage(vtkImageData *image)
{
	this->mimage=image;
}

//------------------------------------------------------------------------
void  WidgetShowNPoints::SetRenderer(vtkRenderer *renderer)
{
	this->renderer	= renderer;
}


std::string WidgetShowNPoints::CleanSpaces(std::string ss)
{
	int i;
	while( (i=ss.find(32))>=0 )
	{
		ss.replace(i,1,"_");
	}
	return ss;
}

//------------------------------------------------------------------------
void WidgetShowNPoints::AddPoint(int x, int y, int z, std::string label)
{
	lstPointsX.push_back( x );
	lstPointsY.push_back( y );
	lstPointsZ.push_back( z );

	std::string strLabel = CleanSpaces(  label );

	lstLabels.push_back( strLabel );

	// Sphere
	vtkSphereSource *vtksphere		= vtkSphereSource::New();
	vtksphere->SetThetaResolution (20);
	vtksphere->SetPhiResolution (20);
	vtksphere->SetRadius( mradio  );
	vtkPolyDataMapper *sphereMapper	= vtkPolyDataMapper::New();
	sphereMapper->SetInput( vtksphere->GetOutput() );
	vtkActor *sphereActor	= vtkActor::New();
	sphereActor->SetMapper(sphereMapper);
	sphereActor->SetOrigin(0, 0, 0);
	double spc[3];
	if(mimage==NULL){
		wxMessageDialog dialog(this, _T("Image Not Set"),_T("Image Not Set"),wxICON_ERROR);
		dialog.ShowModal();
		return;
	}
	mimage->GetSpacing(spc);
	sphereActor->SetPosition( spc[0]*x , spc[1]*y , spc[2]*z );
	sphereActor->GetProperty()->SetColor( mcolour[0] , mcolour[1] , mcolour[2] );
	sphereActor->GetProperty()->SetOpacity( mopacity );

	lstActorsSphere.push_back(sphereActor);
	if(renderer==NULL){
		wxMessageDialog dialog(this, _T("Renderer Not Set"),_T("Renderer Not Set"),wxICON_ERROR);
		dialog.ShowModal();		
		return;
	}
	renderer->AddActor( sphereActor );

	// Actor
	vtkTextActor3D *textActor = vtkTextActor3D::New();
	textActor->SetPosition(  mradio+spc[0]*x , spc[1]*y , spc[2]*z );
	textActor->SetInput( strLabel.c_str()  );
	renderer->AddActor( textActor );
	lstActorsText.push_back(textActor);

	SetOutputBox();

}

//------------------------------------------------------------------------
void WidgetShowNPoints::OnAddPoint (wxCommandEvent& event)
{
	if (this->renderer!=NULL){
		if (mpoint.size()==3){
			AddPoint(mpoint[0],mpoint[1],mpoint[2], (const char*) ( textCtrl->GetValue().mb_str() ) );
		} else {//mpoint.size
			printf("creaMaracasVisu::ShowNPoints (not match point) \n");
		}
	} // renderer
}


	//------------------------------------------------------------------------
	void WidgetShowNPoints::SetOutputBox()
	{
		renderer->GetRenderWindow()->Render();
		wxString strTmp;
		strTmp.Printf(_T("Nbr of points: %d"), (int)(lstPointsX.size()) );
		txtNrPoints->SetLabel(  strTmp );

		//--BBTK
		mbbShowNPoints->bbSetOutputlstPointsX( lstPointsX );
		mbbShowNPoints->bbSetOutputlstPointsY( lstPointsY );
		mbbShowNPoints->bbSetOutputlstPointsZ( lstPointsZ );
		mbbShowNPoints->bbSetOutputlstLabels( lstLabels );
		mbbShowNPoints->bbSignalOutputModification(std::string("lstPointsX"));
		mbbShowNPoints->bbSignalOutputModification(std::string("lstPointsY"));
		mbbShowNPoints->bbSignalOutputModification(std::string("lstPointsZ"));
		mbbShowNPoints->bbSignalOutputModification(std::string("lstLabels"));
	}

	//------------------------------------------------------------------------
	void WidgetShowNPoints::OnSavePoints(wxCommandEvent& event)
	{

		wxFileDialog* FD = new wxFileDialog( 0,
											_T("Save points .."),
											_T(""),
											_T(""),
											_T("(*.xls)|*.xls"),
											 wxSAVE | wxOVERWRITE_PROMPT,
											wxDefaultPosition);
		//EED

		int result_FD = FD->ShowModal();

		// This line is need it by windows //EED
		FD->SetReturnCode( result_FD );

		if (FD->GetReturnCode()==wxID_OK)
		{
			FILE *ff;
			std::string filename= (const char*) ( FD->GetPath().mb_str() );
			ff = fopen( filename.c_str() , "w+" );
			if (ff!=NULL)
			{
				int i , size = (int) (lstActorsSphere.size());
				fprintf(ff,"NumberOfPoints %d \n",size);
				fprintf(ff," X\tY\tZ\tvalue\tLabel\n");
				int x, y, z;
				double value;
				for (i=0; i<size; i++)
				{
					x=lstPointsX[i];
					y=lstPointsY[i];
					z=lstPointsZ[i];
					value= mimage->GetScalarComponentAsDouble(x,y,z,0);
					fprintf(ff,"%d\t %d\t %d\t %f\t %s\n", x , y , z , value  , lstLabels[i].c_str());
				} // for
				fclose(ff);
			} else {   // else ff
				printf("WidgetShowNPoints::OnSavePoints  ...Error... creating file");
			} //ff
		} 	// dialog box

	}

	//------------------------------------------------------------------------
	void WidgetShowNPoints::OnLoadPoints(wxCommandEvent& event)
	{
		wxFileDialog* FD = new wxFileDialog( 0,
											_T("Load points .."),
											_T(""),
											_T(""),
											_T("(*.xls)|*.xls"),
											wxOPEN | wxFILE_MUST_EXIST,
											wxDefaultPosition);
		//EED

		int result_FD = FD->ShowModal();

		// This line is need it by windows //EED
		FD->SetReturnCode( result_FD );

		if (FD->GetReturnCode()==wxID_OK)
		{
			FILE *ff;
			std::string filename= (const char*) ( FD->GetPath().mb_str() );
			ff = fopen( filename.c_str() , "r+" );
			if (ff!=NULL)
			{
				int i,size;
				char chartmp[256];
				fscanf(ff," %s %d",chartmp,&size);
				fscanf(ff," %s %s %s %s %s",chartmp, chartmp,chartmp,chartmp,chartmp );

				float value;
				int x,y,z;
				for (i=0; i<size; i++)
				{
					fscanf(ff," %d %d %d %f %s",&x,&y,&z,&value,chartmp );  // x,y,z,value,label
					AddPoint(x,y,z, chartmp );
				}
				fclose(ff);
			} else {   // else ff
				printf("WidgetShowNPoints::OnLoadPoints  ...Error... reading file");
			} //ff
		} 	// dialog box


	}

	int WidgetShowNPoints::GetNearestPoint()
	{
		int id=-1;
		int i, size=(int)(lstActorsSphere.size());
		double spc[3];

		if(mimage ==NULL){			
			wxMessageDialog* diag = new wxMessageDialog(this, _T("Image not set"), _T("Image Not Set"), wxICON_ERROR);
			diag->ShowModal();
			return -1;
		}else{

			mimage->GetSpacing(spc);

			for ( i=0  ; i<size; i++ )
			{
				double rx =  spc[0]*(mpoint[0] - lstPointsX [i]);
				double ry =  spc[1]*(mpoint[1] - lstPointsY [i]);
				double rz =  spc[2]*(mpoint[2] - lstPointsZ [i]);
				if ( rx*rx + ry*ry + rz*rz <= mradio*mradio)
				{
					id=i;
				}	// if
			} // for

			return id;
		}
	}

	//------------------------------------------------------------------------
	void WidgetShowNPoints::OnRenamePoint(wxCommandEvent& event)
	{
		int id=GetNearestPoint();
		if (id>=0)
		{
			std::string strLabel = CleanSpaces(  (const char*) ( textCtrl->GetValue().mb_str() ) );
			lstLabels[id] = CleanSpaces( (const char*) (textCtrl->GetValue().mb_str()) ) ;
			lstActorsText[id]->SetInput(  lstLabels[id].c_str() );
			SetOutputBox();
		}
	}

	//------------------------------------------------------------------------
	void WidgetShowNPoints::ErasePoint(int id)
	{
		if (this->renderer!=NULL){
			if (id>=0){
				renderer->RemoveActor( lstActorsSphere[id] );
				renderer->RemoveActor( lstActorsText[id] );
				 lstActorsSphere[id]->Delete();
				 lstActorsText[id]->Delete();
				lstActorsSphere.erase( lstActorsSphere.begin()+id );
				lstActorsText.erase( lstActorsText.begin()+id );

				lstPointsX.erase( lstPointsX.begin()+id );
				lstPointsY.erase( lstPointsY.begin()+id );
				lstPointsZ.erase( lstPointsZ.begin()+id );
				lstLabels.erase( lstLabels.begin()+id );

				SetOutputBox();

			} // if id
		} // if renderer
	}

	//------------------------------------------------------------------------
	void WidgetShowNPoints::OnErasePoint(wxCommandEvent& event)
	{
		ErasePoint( GetNearestPoint() );
	}

	//------------------------------------------------------------------------
	void WidgetShowNPoints::OnEraseLastPoint(wxCommandEvent& event)
	{
		ErasePoint(lstActorsSphere.size()-1);
	}


//------------------------------------------------------------------------
void WidgetShowNPoints::OnDeleteAllPoints(wxCommandEvent& event)
{
	int id,size=lstActorsSphere.size();
	for (id=size-1;id>=0;id--)
	{
		ErasePoint(id);
	}

}



BBTK_ADD_BLACK_BOX_TO_PACKAGE(creaMaracasVisu,ShowNPoints)
BBTK_BLACK_BOX_IMPLEMENTATION(ShowNPoints,bbtk::WxBlackBox);

void ShowNPoints::Process()
{
	if (mwxwidget!=NULL){
		mwxwidget->SetRenderer( bbGetInputRenderer() );
		mwxwidget->SetPoint( bbGetInputIn() );
		mwxwidget->SetImage( bbGetInputImage() );
		mwxwidget->SetColour( bbGetInputColour() );
		mwxwidget->SetOpacity( bbGetInputOpacity() );
		mwxwidget->SetRadio( bbGetInputRadio() );

		bbSetOutputlstPointsX( mwxwidget->GetLstPointsX() );
		bbSetOutputlstPointsY( mwxwidget->GetLstPointsY() );
		bbSetOutputlstPointsZ( mwxwidget->GetLstPointsZ() );
		bbSetOutputlstLabels( mwxwidget->GetLstLabels() );
	} // mwxwidget
}


void ShowNPoints::CreateWidget(wxWindow* parent)
{
	mwxwidget = new WidgetShowNPoints( parent ,  this);
	mwxwidget->SetPoint( bbGetInputIn() );

	if (bbGetInputImage()==NULL)
	{
		printf("Missing Image  (ShowNPoints) \n");
	}

   bbSetOutputWidget( mwxwidget );
}

void ShowNPoints::bbUserSetDefaultValues()
{
	mwxwidget = NULL;

	bbSetInputRadio(1);
	bbSetInputOpacity(1);

	std::vector<double> colour;
	colour.push_back(1.0);
	colour.push_back(0.0);
	colour.push_back(0.0);
	bbSetInputColour(colour);

	bbSetInputImage(NULL);
	bbSetInputRenderer(NULL);
}

	//-----------------------------------------------------------------
	void ShowNPoints::bbUserInitializeProcessing()
	{
	}

	//-----------------------------------------------------------------
	void ShowNPoints::bbUserFinalizeProcessing()
	{
	}

	//-----------------------------------------------------------------

}
// EO namespace bbcreaMaracasVisu