// NeL - MMORPG Framework <http://dev.ryzom.com/projects/nel/>
// Copyright (C) 2010  Winch Gate Property Limited
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as
// published by the Free Software Foundation, either version 3 of the
// License, or (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.


// particle_dlg.cpp : implementation file
//

#include "std_afx.h"
#include "object_viewer.h"
#include "particle_dlg.h"
#include "editable_range.h"
#include "located_properties.h"
#include "particle_system_edit.h"
#include "skippable_message_box.h"
#include "main_frame.h"
//
// TODO : remove these include when the test system will be removed
#include "nel/3d/particle_system.h"
#include "nel/3d/ps_force.h"
#include "nel/3d/ps_emitter.h"
#include "nel/3d/ps_particle.h"
#include "nel/3d/ps_util.h"
#include "nel/3d/ps_zone.h"
#include "nel/3d/ps_color.h"
#include "nel/3d/ps_float.h"
#include "nel/3d/ps_int.h"
#include "nel/3d/ps_plane_basis_maker.h"
#include "nel/3d/particle_system_model.h"
#include "nel/3d/particle_system_shape.h"
#include "nel/3d/texture_file.h"
#include "nel/3d/texture_grouped.h"
#include "nel/3d/nelu.h"
#include "nel/3d/font_manager.h"
//
#include "nel/misc/file.h"
#include "start_stop_particle_system.h"
//
#include "save_options_dlg.h"
#include "create_file_dlg.h"

using namespace NL3D;

//**************************************************************************************************************************
CParticleDlg::CParticleDlg(class CObjectViewer* main, CWnd *pParent, CMainFrame* mainFrame, CAnimationDlg *animDLG)
	: CDialog(CParticleDlg::IDD, pParent),
	  MainFrame(mainFrame),
	  CurrentRightPane(NULL),
	  _ActiveNode(NULL),
	  _ObjView(main),
	  _EmptyBBox(true),
	  _AutoUpdateBBox(false),
	  _PW(NULL)

{
	//{{AFX_DATA_INIT(CParticleDlg)
		// NOTE: the ClassWizard will add member initialization here
	//}}AFX_DATA_INIT
	nlverify (FontManager = main->getFontManager());
	nlverify (FontGenerator = main->getFontGenerator());
	NL3D::CParticleSystem::setSerializeIdentifierFlag(true); // serialize identifiers for edition	
	ParticleTreeCtrl = new CParticleTreeCtrl(this);
	StartStopDlg = new CStartStopParticleSystem(this, animDLG);
	/** register us, so that our 'go' method will be called
	  * this gives us a chance to display a bbox when needed
	  */
	_ObjView->registerMainLoopCallBack(this);	
}


//**************************************************************************************************************************
BOOL CParticleDlg::Create( UINT nIDTemplate, CWnd* pParentWnd /*= NULL*/ )
{
	if (!CDialog::Create(nIDTemplate, pParentWnd)) return FALSE;	
	return TRUE;
}

//**************************************************************************************************************************
void CParticleDlg::moveElement(const NLMISC::CMatrix &mat)
{
	ParticleTreeCtrl->moveElement(mat);
}

//**************************************************************************************************************************
NLMISC::CMatrix CParticleDlg::getElementMatrix(void) const
{
	return ParticleTreeCtrl->getElementMatrix();
}

//**************************************************************************************************************************
CParticleDlg::~CParticleDlg()
{
	_ObjView->removeMainLoopCallBack(this);	
	delete ParticleTreeCtrl;
	delete CurrentRightPane;
	delete StartStopDlg;
	if (_PW) _PW->setModificationCallback(NULL);
	delete _PW;
}

//**************************************************************************************************************************
void CParticleDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CParticleDlg)
	//}}AFX_DATA_MAP
}


BEGIN_MESSAGE_MAP(CParticleDlg, CDialog)
	//{{AFX_MSG_MAP(CParticleDlg)
	ON_WM_DESTROY()
	ON_WM_SIZE()
	ON_WM_SHOWWINDOW()
	ON_WM_CHAR()
	ON_COMMAND(IDM_CREATE_NEW_PS_WORKSPACE, OnCreateNewPsWorkspace)
	ON_COMMAND(IDM_LOAD_PS_WORKSPACE, OnLoadPSWorkspace)
	ON_COMMAND(IDM_SAVE_ALL_PS_WORKSPACE, OnSaveAllPsWorkspace)
	ON_COMMAND(IDM_SAVE_PS_WORKSPACE, OnSavePsWorkspace)
	ON_COMMAND(IDM_VIEW_PS_FILENAME, OnViewPsFilename)	
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

//**************************************************************************************************************************
void CParticleDlg::OnDestroy() 
{
	checkModifiedWorkSpace();	
	if (CurrentRightPane)
	{
		CurrentRightPane->DestroyWindow();
		delete CurrentRightPane;
		CurrentRightPane = NULL;
	}
	setRegisterWindowState (this, REGKEY_OBJ_PARTICLE_DLG);
	CDialog::OnDestroy();		
}

//**************************************************************************************************************************
BOOL CParticleDlg::OnInitDialog() 
{
	CDialog::OnInitDialog();
	CRect r;
	GetWindowRect(&r);	
	ParticleTreeCtrl->Create(WS_VISIBLE | WS_TABSTOP | WS_CHILD | WS_BORDER
							   | TVS_HASBUTTONS | TVS_LINESATROOT | TVS_HASLINES | TVS_SHOWSELALWAYS | TVS_EDITLABELS 
							   | TVS_DISABLEDRAGDROP , r, this, 0x1005);	
	ParticleTreeCtrl->init();
	ParticleTreeCtrl->ShowWindow(SW_SHOW);
	StartStopDlg->Create(IDD_PARTICLE_SYSTEM_START_STOP, this);	
	// create menu bar that allow to create / load a particle workspace
	CMenu menu;
	menu.LoadMenu(MAKEINTRESOURCE(IDR_PARTICLE_DLG_MENU));
	this->SetMenu(&menu);
	menu.Detach();		
	updateMenu();
	//
	_StatusBar.Create(this);
	UINT indicators = ID_PS_EDITOR_STATUS;
	_StatusBar.SetIndicators(&indicators, 1);		
	_StatusBar.SetPaneInfo(0, ID_PS_EDITOR_STATUS, SBPS_NORMAL, computeStatusBarWidth());
	RepositionBars(AFX_IDW_CONTROLBAR_FIRST, AFX_IDW_CONTROLBAR_LAST, ID_PS_EDITOR_STATUS);
	return TRUE;  // return TRUE unless you set the focus to a control
	              // EXCEPTION: OCX Property Pages should return FALSE
}

//**************************************************************************************************************************
void CParticleDlg::setStatusBarText(CString &str)
{
	_StatusBar.SetPaneText(0, str, TRUE);
}

//**************************************************************************************************************************
void CParticleDlg::OnSize(UINT nType, int cx, int cy) 
{	
	AFX_MANAGE_STATE(AfxGetStaticModuleState());
	bool blocked = false;
	if (ParticleTreeCtrl->m_hWnd && this->m_hWnd)
	{	
		CRect r = getTreeRect(cx, cy);			
		ParticleTreeCtrl->MoveWindow(&r);
		if (CurrentRightPane)
		{								
			CurrentRightPane->MoveWindow(r.right + 10, r.top, r.right + CurrRightPaneWidth + 10, r.top + CurrRightPaneHeight);
		}			
		CDialog::OnSize(nType, cx, cy);	
		if (IsWindow(_StatusBar))
		{					
			_StatusBar.SetPaneInfo(0, ID_PS_EDITOR_STATUS, SBPS_NORMAL, computeStatusBarWidth());
			RepositionBars(AFX_IDW_CONTROLBAR_FIRST, AFX_IDW_CONTROLBAR_LAST, ID_PS_EDITOR_STATUS);
		}
	}
}

//**************************************************************************************************************************
CRect CParticleDlg::getTreeRect(int cx, int cy) const
{
	const uint ox = 0, oy = 10;

	if (CurrentRightPane)
	{		
		CRect res(ox, oy, cx - CurrRightPaneWidth - 10, cy - 20); 
		return res;
	}
	else
	{
		CRect res(ox, oy, cx - 10, cy - 20);
		return res;
	}
}

//**************************************************************************************************************************
void CParticleDlg::setRightPane(CWnd *pane)
{
	AFX_MANAGE_STATE(AfxGetStaticModuleState());
	if (CurrentRightPane)
	{
		CurrentRightPane->DestroyWindow();
	}
	delete CurrentRightPane;
	CurrentRightPane = pane;
	RECT r;	
	if (pane)
	{
		
		pane->ShowWindow(SW_SHOW);
	
	
		CurrentRightPane->GetClientRect(&r);

		CurrRightPaneWidth = r.right;
		CurrRightPaneHeight = r.bottom;
	
	}

	GetClientRect(&r);
	this->SendMessage(WM_SIZE, SIZE_RESTORED, r.right + (r.bottom << 16));	
	GetWindowRect(&r);
	this->MoveWindow(&r);
	if (CurrentRightPane)
	{
		CurrentRightPane->Invalidate();
	}
	this->Invalidate();
	ParticleTreeCtrl->Invalidate();
}

//**************************************************************************************************************************
LRESULT CParticleDlg::WindowProc(UINT message, WPARAM wParam, LPARAM lParam) 
{

	if (message == WM_GETMINMAXINFO)
	{
		sint cx = 150, cy = 150;
		if (CurrentRightPane)
		{
			RECT r;
			CurrentRightPane->GetClientRect(&r);
			cx += CurrRightPaneWidth;
			if (cy < (CurrRightPaneHeight + 20) ) cy = CurrRightPaneHeight + 20;
		}


		MINMAXINFO *inf = 	(MINMAXINFO *) lParam;
		inf->ptMinTrackSize.x = cx;
		inf->ptMinTrackSize.y = cy;

	}
	
	return CDialog::WindowProc(message, wParam, lParam);
	
}

//**************************************************************************************************************************
void CParticleDlg::OnShowWindow(BOOL bShow, UINT nStatus) 
{
	CDialog::OnShowWindow(bShow, nStatus);
	StartStopDlg->ShowWindow(bShow);	
		
}

//**************************************************************************************************************************
void CParticleDlg::goPostRender()
{
	NL3D::CParticleSystem *currPS = _ActiveNode ? _ActiveNode->getPSPointer() : NULL;
	if (!currPS) return;
	if (StartStopDlg->isBBoxDisplayEnabled() && currPS)
	{
		NL3D::CNELU::Driver->setupModelMatrix(currPS->getSysMat());		
		if (_AutoUpdateBBox)
		{
			NLMISC::CAABBox currBBox;
			currPS->forceComputeBBox(currBBox);
			if (_EmptyBBox)
			{
				_EmptyBBox = false;
				_CurrBBox = currBBox;
			}
			else
			{			
				NL3D::CPSUtil::displayBBox(NL3D::CNELU::Driver, _CurrBBox, CRGBA::Blue);			
				_CurrBBox = NLMISC::CAABBox::computeAABBoxUnion(currBBox, _CurrBBox);
			}
			currPS->setPrecomputedBBox(_CurrBBox);
		}	
		else
		{
			currPS->getLastComputedBBox(_CurrBBox);
		}
		NL3D::CPSUtil::displayBBox(NL3D::CNELU::Driver, _CurrBBox, currPS->getAutoComputeBBox() ? CRGBA::White : CRGBA::Red);
	}
	// copy user matrix into current fx
	nlassert(_ObjView);
	_ActiveNode->getPSModel()->setUserMatrix(_ObjView->getFXUserMatrix());
}

//**************************************************************************************************************************
void CParticleDlg::OnChar(UINT nChar, UINT nRepCnt, UINT nFlags) 
{
	if (nChar == (UINT) 'p' || nChar == (UINT) 'P' || nChar == (UINT) ' ')
	{
		// simulate a start / stop on the system
		StartStopDlg->toggle();
	}
	
	CDialog::OnChar(nChar, nRepCnt, nFlags);
}

//**************************************************************************************************************************
const NLMISC::CMatrix &CParticleDlg::getPSMatrix() const
{
	if (!getActivePSM()) return NLMISC::CMatrix::Identity;		
	return getActivePSM()->getMatrix();
}

//**************************************************************************************************************************
const NLMISC::CMatrix &CParticleDlg::getPSWorldMatrix() const
{
	if (!getActivePSM()) return NLMISC::CMatrix::Identity;	
	return getActivePSM()->getWorldMatrix();
}

//**************************************************************************************************************************
void CParticleDlg::setPSMatrix(const NLMISC::CMatrix &mat)
{
	if (!getActivePSM()) return;		
	getActivePSM()->setMatrix(mat);
}

//**************************************************************************************************************************
void CParticleDlg::setPSWorldMatrix(const NLMISC::CMatrix &mat)
{
	if (!getActivePSM()) return;	
	CMatrix invParentMat =  getActivePSM()->getMatrix() * getActivePSM()->getWorldMatrix().inverted();
	invParentMat.normalize(CMatrix::XYZ);
	CMatrix newMat = invParentMat * mat;
	newMat.normalize(CMatrix::XYZ);
	getActivePSM()->setMatrix(newMat);
}


//**************************************************************************************************************************
void CParticleDlg::stickPSToSkeleton(CParticleWorkspace::CNode *node,
									 NL3D::CSkeletonModel *skel,
									 uint bone,
									 const std::string &parentSkelName,
									 const std::string &parentBoneName)
{
	if (!node) return;	
	node->stickPSToSkeleton(skel, bone, parentSkelName, parentBoneName);
	if (skel)
	{
		if (_ObjView->getMainFrame()->MouseMoveType == CMainFrame::MoveFX)
		{
			_ObjView->getMainFrame()->OnEditMovecamera();
		}		
		_ObjView->getMainFrame()->ToolBar.Invalidate();		
	}
}

//**************************************************************************************************************************
void CParticleDlg::unstickPSFromSkeleton(CParticleWorkspace::CNode *node)
{
	if (!node) return;
	node->unstickPSFromSkeleton();	
	_ObjView->getMainFrame()->ToolBar.Invalidate();		
}

//**************************************************************************************************************************
bool CParticleDlg::savePS(HWND parent, CParticleWorkspace::CNode &psNode, bool askToContinueWhenError)
{
	return savePSAs(parent, psNode, psNode.getFullPath(), askToContinueWhenError);
}

//**************************************************************************************************************************
bool CParticleDlg::savePSAs(HWND parent, CParticleWorkspace::CNode &psNode ,const std::string &fullPath, bool askToContinueWhenError)
{
	nlassert(psNode.getPSPointer());
	if (psNode.getResetAutoCountFlag() && psNode.getPSPointer()->getAutoCountFlag())
	{		
		MessageBox(psNode.getFilename().c_str() + getStrRsc(IDS_AUTO_COUNT_ERROR), getStrRsc(IDS_WARNING), MB_ICONEXCLAMATION);
		return false;
	}
	StartStopDlg->stop();
	try
	{	
		psNode.savePSAs(fullPath);
		psNode.setModified(false);
		setStatusBarText(CString(fullPath.c_str()) + " " + getStrRsc(IDS_SAVED));
	}
	catch (NLMISC::Exception &e)
	{
		if (askToContinueWhenError)
		{		
			int result = ::MessageBox(parent, CString(e.what()) + getStrRsc(IDS_CONTINUE_SAVING) , getStrRsc(IDS_ERROR), MB_ICONEXCLAMATION);
			return result == MB_OK;
		}
		else
		{
			::MessageBox(parent, e.what(), getStrRsc(IDS_ERROR), MB_ICONEXCLAMATION);
			return false;
		}
	}
	return true;
}

//**************************************************************************************************************************
bool CParticleDlg::loadPS(HWND parent, CParticleWorkspace::CNode &psNode, TLoadPSBehav behav)
{
	bool loadingOK = true;	
	try
	{	
		if (!psNode.loadPS())
		{
			loadingOK = false;
			switch(behav)
			{
				case Silent: return false; // no op
				case ReportError:					
					::MessageBox(parent, (LPCTSTR) (CString(psNode.getFilename().c_str()) + " : " + getStrRsc(IDS_COULDNT_INSTANCIATE_PS)), getStrRsc(IDS_ERROR), MB_OK|MB_ICONEXCLAMATION);
					return true;
				break;
				case ReportErrorSkippable:
				{
					CSkippableMessageBox mb(getStrRsc(IDS_ERROR), CString(psNode.getFilename().c_str()) + " : " + getStrRsc(IDS_COULDNT_INSTANCIATE_PS), CWnd::FromHandle(parent));
					mb.DoModal();
					return !mb.getBypassFlag();
				}
				break;
				default:
					nlassert(0);
				break;
			}			
		}
		else
		{
			setStatusBarText(CString(psNode.getFullPath().c_str()) + " " + getStrRsc(IDS_LOADED));
		}
	}
	catch (NLMISC::Exception &e)
	{
		switch(behav)
		{
			case Silent: return false; // no op
			case ReportError:	
				::MessageBox(parent, e.what(), getStrRsc(IDS_ERROR), MB_OK|MB_ICONEXCLAMATION);				
				return true;
			break;
			case ReportErrorSkippable:
			{
				CSkippableMessageBox mb(getStrRsc(IDS_ERROR), CString(e.what()), CWnd::FromHandle(parent));
				mb.DoModal();
				return !mb.getBypassFlag();
			}
			break;
			default:
				nlassert(0);
			break;
		}		
	}
	if (psNode.getPSPointer()->hasLoop())
	{		
		localizedMessageBox(parent, IDS_FX_HAS_LOOP, IDS_WARNING, MB_OK|MB_ICONEXCLAMATION);
	}	
	return behav != Silent;
}


//**************************************************************************************************************************
void CParticleDlg::checkModifiedWorkSpace()
{
	if (_PW)
	{
		// see if current tree has been changed				
		if (_PW->isModified())
		{			
			int result = localizedMessageBox(*this, IDS_PS_WORKSPACE_MODIFIED, IDS_PARTICLE_EDITOR, MB_YESNO|MB_ICONQUESTION);
			if (result == IDYES)
			{
				saveWorkspaceStructure();
			}
		}
		if (_PW->isContentModified())
		{			
			saveWorkspaceContent(true);			
		}
	}
}

//**************************************************************************************************************************
void CParticleDlg::closeWorkspace()
{
	setActiveNode(NULL);
	ParticleTreeCtrl->setActiveNode(NULL);
	ParticleTreeCtrl->reset();
	delete _PW;
	_PW = NULL;
}

//**************************************************************************************************************************
void CParticleDlg::OnCreateNewPsWorkspace() 
{		
	checkModifiedWorkSpace();
	// ask name of the new workspace to create
	CCreateFileDlg cf(getStrRsc(IDS_CHOOSE_WORKSPACE_NAME), "", "pws");
	INT_PTR result = cf.DoModal();
	if (result = IDOK)
	{			
		if (cf.touchFile())
		{				
			CParticleWorkspace *newPW = new CParticleWorkspace;
			newPW->setModificationCallback(ParticleTreeCtrl);
			newPW->init(_ObjView, cf.getFullPath(), _ObjView->getFontManager(), _ObjView->getFontGenerator());
			// save empty workspace
			try
			{		
				newPW->save();
			}
			catch(NLMISC::EStream &e)
			{
				MessageBox(e.what(), getStrRsc(IDS_ERROR), MB_ICONEXCLAMATION);
			}
			closeWorkspace();			
			_PW = newPW;
			ParticleTreeCtrl->buildTreeFromWorkSpace(*_PW);
		}
	}

}

//**************************************************************************************************************************
void CParticleDlg::OnLoadPSWorkspace() 
{
	checkModifiedWorkSpace();
	static const char BASED_CODE szFilter[] = "particle workspaces(*.pws)|*.pws||";
	CFileDialog fd( TRUE, ".pws", "*.pws", 0, szFilter);
	INT_PTR result = fd.DoModal();
	if (result != IDOK) return;
	loadWorkspace((LPCTSTR) fd.GetPathName());
}

//**************************************************************************************************************************
void CParticleDlg::loadWorkspace(const std::string &fullPath)
{
	// Add to the path
	std::auto_ptr<CParticleWorkspace> newPW(new CParticleWorkspace);
	newPW->init(_ObjView, fullPath, _ObjView->getFontManager(), _ObjView->getFontGenerator());
	newPW->setModificationCallback(ParticleTreeCtrl);
	// save empty workspace
	try
	{		
		newPW->load();
		setStatusBarText(CString(newPW->getFilename().c_str()) + " " + getStrRsc(IDS_LOADED));
	}
	catch(NLMISC::EStream &e)
	{
		MessageBox(e.what(), getStrRsc(IDS_ERROR), MB_ICONEXCLAMATION);
		setStatusBarText(CString(e.what()));
		return;
	}	
	// try to load each ps
	CParticleWorkspace::CNode *firstLoadedNode = NULL;
	bool displayErrorMsg = true;
	for(uint k = 0; k < newPW->getNumNode(); ++k)
	{
		
		displayErrorMsg = loadPS(*this, *newPW->getNode(k), displayErrorMsg ? ReportErrorSkippable : Silent);		
		if (newPW->getNode(k)->isLoaded() && !firstLoadedNode)
		{
			firstLoadedNode = newPW->getNode(k);
		}			
	}	
	closeWorkspace();
	_PW = newPW.release();
	ParticleTreeCtrl->buildTreeFromWorkSpace(*_PW);	
	setActiveNode(firstLoadedNode);
	ParticleTreeCtrl->setActiveNode(firstLoadedNode);
	ParticleTreeCtrl->expandRoot();
	setStatusBarText(getStrRsc(IDS_READY));
}

//**************************************************************************************************************************
void CParticleDlg::OnSaveAllPsWorkspace() 
{
	saveWorkspaceStructure();
	saveWorkspaceContent(false);
}

//**************************************************************************************************************************
void CParticleDlg::OnSavePsWorkspace() 
{
	saveWorkspaceStructure();
	saveWorkspaceContent(true);
}

//**************************************************************************************************************************
void CParticleDlg::saveWorkspaceStructure()
{
	nlassert(_PW);
	try
	{
		_PW->save();		
		setStatusBarText(CString(_PW->getFilename().c_str()) + " " + getStrRsc(IDS_SAVED));
	}
	catch(NLMISC::EStream &e)
	{
		localizedMessageBox(*this, e.what(), IDS_ERROR, MB_ICONEXCLAMATION);
		setStatusBarText(CString(e.what()));
	}
}


//**************************************************************************************************************************
void CParticleDlg::saveWorkspaceContent(bool askToSaveModifiedPS)
{
	StartStopDlg->stop();
	bool saveAll = !askToSaveModifiedPS;
	// check each component of the tree
	for(uint k = 0; k < _PW->getNumNode(); ++k)
	{
		if (_PW->getNode(k)->isModified())
		{				
			if (saveAll)
			{
				bool keepSaving = savePS(*this, *_PW->getNode(k), k != _PW->getNumNode());
				if (!keepSaving) break;
			}
			else
			{
				// ask if the user wants to save the ps, or save all ps
				CString mess;					
				mess = CString(_PW->getNode(k)->getFilename().c_str()) + getStrRsc(IDS_SAVE_MODIFIED_PS);
				CSaveOptionsDlg sop(getStrRsc(IDS_SAVE_FILE), mess, this);
				sop.DoModal();
				bool saveThisFile = false;
				bool stop = false;
				switch(sop.getChoice())
				{
					case CSaveOptionsDlg::Yes:		saveThisFile = true;  break;
					case CSaveOptionsDlg::No:		saveThisFile = false; break;
					case CSaveOptionsDlg::SaveAll:	saveAll =	   true;  break;
					case CSaveOptionsDlg::Stop:		stop =		   true;  break;
					default: nlassert(0);
				}
				if (stop) break;
				if (saveAll || saveThisFile)
				{
					bool keepSaving = savePS(*this, *_PW->getNode(k), k != _PW->getNumNode());
					if (!keepSaving) break;
				}
			}				
		}
	}
	setStatusBarText(getStrRsc(IDS_READY));
}

//**************************************************************************************************************************
void CParticleDlg::setActiveNode(CParticleWorkspace::CNode *node)
{
	if (node == _ActiveNode) return;	
	_ActiveNode = node;
	StartStopDlg->setActiveNode(node);		
	if(MainFrame->isMoveFX())
	{
		if (node)
		{	
			_ObjView->getMouseListener().setModelMatrix(node->getPSModel()->getMatrix());	
		}
		else
		{
			_ObjView->getMouseListener().setModelMatrix(NLMISC::CMatrix::Identity);
		}
	}
	else
	if(MainFrame->isMoveFXUserMatrix())
	{
		if (node)
		{
			_ObjView->getMouseListener().setModelMatrix(node->getPSModel()->getUserMatrix());
		}
		else
		{
			_ObjView->getMouseListener().setModelMatrix(NLMISC::CMatrix::Identity);
		}
	}
}

//**************************************************************************************************************************
NL3D::CParticleSystemModel *CParticleDlg::getModelFromPS(NL3D::CParticleSystem *ps) const
{
	if (!ps) return	NULL;
	if (!_PW) return NULL;
	CParticleWorkspace::CNode *node = _PW->getNodeFromPS(ps);
	if (!node) return NULL;
	return node->getPSModel();
}

//**************************************************************************************************************************
uint CParticleDlg::computeStatusBarWidth() const
{
	nlassert(ParticleTreeCtrl);
	CRect tcRect;
	ParticleTreeCtrl->GetClientRect(&tcRect);
	if (!CurrentRightPane) return (uint) std::max((sint) tcRect.Width() - 16, (sint) 0);
	return (uint) std::max((sint) tcRect.Width() - 4, (sint) 0);
}

//**************************************************************************************************************************
void CParticleDlg::OnViewPsFilename() 
{
	ParticleTreeCtrl->setViewFilenameFlag(!ParticleTreeCtrl->getViewFilenameFlag());
	updateMenu();
}

//**************************************************************************************************************************
void CParticleDlg::updateMenu()
{
	if (!ParticleTreeCtrl) return;
	CMenu *menu = GetMenu();
	if (!menu) return;
	// update the view menu
	menu->CheckMenuItem(IDM_VIEW_PS_FILENAME, MF_BYCOMMAND|(ParticleTreeCtrl->getViewFilenameFlag() ? MF_CHECKED : 0));
}