#include "vtkDefOrgViewerWithKWState.h"
namespace mial{
vtkDefOrgViewerWithKWState::~vtkDefOrgViewerWithKWState(){
	std::vector<vtkKWTopLevel *>::iterator it;
	for (it=m_OpenedWindows.begin();it!=m_OpenedWindows.end();it++){
		(*it)->Withdraw();
	}
	delete theDefOrg;
	theDefOrg=NULL;
}

vtkDefOrgViewerWithKWState::SOViewerPointer vtkDefOrgViewerWithKWState::GetCurrentSOViewer(){
	itkScenePointer primaryItkScenePtr = (this->theDefOrg->m_OutputItkSpatialObjects[currSOViewerIndex]).theItkScene;
	return (this->m_OutputSceneSOVMap[primaryItkScenePtr]).theSoViewerPointer;
}

vtkKWRenderWidget* vtkDefOrgViewerWithKWState::GetCurrentSOViewerRenderWidget(){
	itkScenePointer primaryItkScenePtr = (this->theDefOrg->m_OutputItkSpatialObjects[currSOViewerIndex]).theItkScene;
	return (this->m_OutputSceneSOVMap[primaryItkScenePtr]).theRenderWidget;
}

vtkDefOrgViewerWithKWState::SOViewerPointer vtkDefOrgViewerWithKWState::GetSOViewer(itkScenePointer itkScenePointerKey){
	return (this->m_OutputSceneSOVMap[itkScenePointerKey]).theSoViewerPointer;
}

void vtkDefOrgViewerWithKWState::SetSOViewer(itkScenePointer itkScenePointerKey, vtkDefOrgViewerWithKWState::SOViewerPointer sovPointer){
	if (m_OutputSceneSOVMap.find(itkScenePointerKey)==m_OutputSceneSOVMap.end()) itkScenePointerKey->Register();
	(this->m_OutputSceneSOVMap[itkScenePointerKey]).theSoViewerPointer = sovPointer;
	
}

vtkKWRenderWidget* vtkDefOrgViewerWithKWState::GetSOViewerRenderWidget(itkScenePointer itkScenePointerKey){
	return (this->m_OutputSceneSOVMap[itkScenePointerKey]).theRenderWidget;
}

void vtkDefOrgViewerWithKWState::SetSOViewerRenderWidget(itkScenePointer itkScenePointerKey, vtkKWRenderWidget* renderWidget){
	if (m_OutputSceneSOVMap.find(itkScenePointerKey)==m_OutputSceneSOVMap.end()) itkScenePointerKey->Register();
	(this->m_OutputSceneSOVMap[itkScenePointerKey]).theRenderWidget = renderWidget;
	renderWidget->Register(NULL);
}

/*
void vtkDefOrgViewerWithKWState::SetPrimarySOViewer(vtkDefOrgViewerWithKWState::SOViewerPointer newPrimarySOViewer){
	itkScenePointer primaryItkScenePtr = (this->theDefOrg->m_OutputItkSpatialObjects[currSOViewerIndex]).theItkScene;
	this->m_OutputSceneSOVMap[primaryItkScenePtr] = newPrimarySOViewer;
}*/

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

vtkImageViewer2* vtkDefOrgViewerWithKWState::GetCurrentImageViewer(){
	vtkImageImport* primaryImagePtr = (this->theDefOrg->m_OutputImages[currImageViewerIndex]).theImageVolume;
	return (this->m_OutputImageViewerMap[primaryImagePtr]).theImageViewerPointer;
}
vtkKWRenderWidget* vtkDefOrgViewerWithKWState::GetCurrentImageViewerRenderWidget(){
	vtkImageImport* primaryImagePtr = (this->theDefOrg->m_OutputImages[currImageViewerIndex]).theImageVolume;
	return (this->m_OutputImageViewerMap[primaryImagePtr]).theRenderWidget;
}
vtkImageViewer2* vtkDefOrgViewerWithKWState::GetImageViewer(vtkImageImport* vtkImageImportPointerKey){
	return (this->m_OutputImageViewerMap[vtkImageImportPointerKey]).theImageViewerPointer;
}
vtkKWRenderWidget* vtkDefOrgViewerWithKWState::GetImageViewerRenderWidget(vtkImageImport* vtkImageImportPointerKey){
	return (this->m_OutputImageViewerMap[vtkImageImportPointerKey]).theRenderWidget;
}

void vtkDefOrgViewerWithKWState::SetImageViewerRenderWidget(vtkImageImport* vtkImageImportPointerKey, vtkKWRenderWidget* renderWidget){
	(this->m_OutputImageViewerMap[vtkImageImportPointerKey]).theRenderWidget = renderWidget;
	renderWidget->Register(NULL);
}

void vtkDefOrgViewerWithKWState::SetImageViewer(vtkImageImport* vtkImageImportPointerKey, vtkImageViewer2* newImageViewerPointer){
	(this->m_OutputImageViewerMap[vtkImageImportPointerKey]).theImageViewerPointer = newImageViewerPointer;
}
}