1HEADmaster

1 files changed, 1078 insertions, 0 deletions

diff --git a/tools/pet/pettool.cpp b/tools/pet/pettool.cpp
new file mode 100644
index 0000000..2eb83e5
--- /dev/null
+++ b/tools/pet/pettool.cpp
@@ -0,0 +1,1078 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+//
+// Purpose: Core Movie Maker UI API
+//
+//=============================================================================
+
+#include "pettool.h"
+#include "vgui_controls/Menu.h"
+#include "tier1/KeyValues.h"
+#include "vgui/IInput.h"
+#include "vgui/KeyCode.h"
+#include "vgui_controls/FileOpenDialog.h"
+#include "filesystem.h"
+#include "vgui/ilocalize.h"
+#include "dme_controls/elementpropertiestree.h"
+#include "tier0/icommandline.h"
+#include "materialsystem/imaterialsystem.h"
+#include "VGuiMatSurface/IMatSystemSurface.h"
+#include "petdoc.h"
+#include "particlesystemdefinitionbrowser.h"
+#include "particlesystempropertiescontainer.h"
+#include "dme_controls/AttributeStringChoicePanel.h"
+#include "dme_controls/ParticleSystemPanel.h"
+#include "datamodel/dmelementfactoryhelper.h"
+#include "matsys_controls/picker.h"
+#include "tier2/fileutils.h"
+#include "tier3/tier3.h"
+#include "particles/particles.h"
+#include "dmserializers/idmserializers.h"
+#include "dme_controls/dmepanel.h"
+#include "vgui/ivgui.h"
+
+using namespace vgui;
+
+
+//-----------------------------------------------------------------------------
+// Methods needed by scenedatabase. They have to live here instead of toolutils
+// because this is a DLL but toolutils is only a static library
+//-----------------------------------------------------------------------------
+USING_DMEPANEL_FACTORY( CParticleSystemPreviewPanel, DmeParticleSystemDefinition );
+USING_DMEPANEL_FACTORY( CParticleSystemDmePanel, DmeParticleSystemDefinition );
+
+
+const char *GetVGuiControlsModuleName()
+{
+	return "PetTool";
+}
+
+//-----------------------------------------------------------------------------
+// Connect, disconnect
+//-----------------------------------------------------------------------------
+bool ConnectTools( CreateInterfaceFn factory )
+{
+	// Attach to the dmserializers instance of the particle system
+	return (materials != NULL) && (g_pMatSystemSurface != NULL) && (g_pMDLCache != NULL) && (studiorender != NULL) && (g_pMaterialSystemHardwareConfig != NULL);
+}
+
+void DisconnectTools( )
+{
+}
+
+
+//-----------------------------------------------------------------------------
+// Singleton
+//-----------------------------------------------------------------------------
+CPetTool	*g_pPetTool = NULL;
+
+void CreateTools()
+{
+	g_pPetTool = new CPetTool();
+}
+
+
+//-----------------------------------------------------------------------------
+// Constructor
+//-----------------------------------------------------------------------------
+CPetTool::CPetTool()
+{
+	m_pMenuBar = NULL;
+	m_pDoc = NULL;
+}
+
+
+//-----------------------------------------------------------------------------
+// Init, shutdown
+//-----------------------------------------------------------------------------
+bool CPetTool::Init( )
+{
+	m_hCurrentParticleSystem = NULL;
+	m_pDoc = NULL;
+	m_RecentFiles.LoadFromRegistry( GetRegistryName() );
+
+	// NOTE: This has to happen before BaseClass::Init
+	g_pVGuiLocalize->AddFile( "resource/toolpet_%language%.txt" );
+
+	if ( !BaseClass::Init( ) )
+		return false;
+
+	CreateInterfaceFn factory;
+	enginetools->GetClientFactory( factory );
+	IParticleSystemQuery *pQuery = (IParticleSystemQuery*)factory( PARTICLE_SYSTEM_QUERY_INTERFACE_VERSION, NULL );
+	g_pParticleSystemMgr->Init( pQuery );
+	// tell particle mgr to add the default simulation + rendering ops
+	g_pParticleSystemMgr->AddBuiltinSimulationOperators();
+	g_pParticleSystemMgr->AddBuiltinRenderingOperators();
+
+	// Create a directory for particles if it doesn't exist
+	char pStartingDir[ MAX_PATH ];
+	GetModSubdirectory( "particles", pStartingDir, sizeof(pStartingDir) );
+	g_pFullFileSystem->CreateDirHierarchy( pStartingDir );
+
+	return true;
+}
+
+void CPetTool::Shutdown()
+{
+	m_RecentFiles.SaveToRegistry( GetRegistryName() );
+
+	BaseClass::Shutdown();
+}
+
+
+//-----------------------------------------------------------------------------
+// returns the document
+//-----------------------------------------------------------------------------
+CPetDoc *CPetTool::GetDocument()
+{
+	return m_pDoc;
+}
+
+	
+//-----------------------------------------------------------------------------
+// Tool activation/deactivation
+//-----------------------------------------------------------------------------
+void CPetTool::OnToolActivate()
+{
+	BaseClass::OnToolActivate();
+}
+
+void CPetTool::OnToolDeactivate()
+{
+	BaseClass::OnToolDeactivate();
+}
+
+
+//-----------------------------------------------------------------------------
+// Used to hook DME VMF entities into the render lists
+//-----------------------------------------------------------------------------
+void CPetTool::Think( bool finalTick )
+{
+	BaseClass::Think( finalTick );
+
+	if ( IsActiveTool() )
+	{
+		// Force resolve calls to happen
+		// FIXME: Shouldn't this not have to happen here?
+		CUtlVector< IDmeOperator* > operators;
+		g_pDmElementFramework->SetOperators( operators );
+		g_pDmElementFramework->Operate( true );
+		g_pDmElementFramework->BeginEdit();
+	}
+}
+
+
+//-----------------------------------------------------------------------------
+// Derived classes can implement this to get a new scheme to be applied to this tool
+//-----------------------------------------------------------------------------
+vgui::HScheme CPetTool::GetToolScheme() 
+{ 
+	return vgui::scheme()->LoadSchemeFromFile( "Resource/BoxRocket.res", "BoxRocket" );
+}
+
+
+//-----------------------------------------------------------------------------
+//
+// The View menu
+//
+//-----------------------------------------------------------------------------
+class CPetViewMenuButton : public CToolMenuButton
+{
+	DECLARE_CLASS_SIMPLE( CPetViewMenuButton, CToolMenuButton );
+public:
+	CPetViewMenuButton( CPetTool *parent, const char *panelName, const char *text, vgui::Panel *pActionSignalTarget );
+	virtual void OnShowMenu(vgui::Menu *menu);
+
+private:
+	CPetTool *m_pTool;
+};
+
+CPetViewMenuButton::CPetViewMenuButton( CPetTool *parent, const char *panelName, const char *text, vgui::Panel *pActionSignalTarget )
+	: BaseClass( parent, panelName, text, pActionSignalTarget )
+{
+	m_pTool = parent;
+
+	AddCheckableMenuItem( "properties", "#PetProperties", new KeyValues( "OnToggleProperties" ), pActionSignalTarget );
+	AddCheckableMenuItem( "browser", "#PetParticleSystemBrowser", new KeyValues( "OnToggleParticleSystemBrowser" ), pActionSignalTarget );
+	AddCheckableMenuItem( "particlepreview", "#PetParticlePreview", new KeyValues( "OnToggleParticlePreview" ), pActionSignalTarget );
+
+	AddSeparator();
+
+	AddMenuItem( "defaultlayout", "#PetViewDefault", new KeyValues( "OnDefaultLayout" ), pActionSignalTarget );
+
+	SetMenu(m_pMenu);
+}
+
+void CPetViewMenuButton::OnShowMenu(vgui::Menu *menu)
+{
+	BaseClass::OnShowMenu( menu );
+
+	// Update the menu
+	int id;
+
+	CPetDoc *pDoc = m_pTool->GetDocument();
+	if ( pDoc )
+	{
+		id = m_Items.Find( "properties" );
+		m_pMenu->SetItemEnabled( id, true );
+
+		Panel *p;
+		p = m_pTool->GetProperties();
+		Assert( p );
+		m_pMenu->SetMenuItemChecked( id, ( p && p->GetParent() ) ? true : false );
+
+		id = m_Items.Find( "browser" );
+		m_pMenu->SetItemEnabled( id, true );
+		
+		p = m_pTool->GetParticleSystemDefinitionBrowser();
+		Assert( p );
+		m_pMenu->SetMenuItemChecked( id, ( p && p->GetParent() ) ? true : false );
+
+		id = m_Items.Find( "particlepreview" );
+		m_pMenu->SetItemEnabled( id, true );
+
+		p = m_pTool->GetParticlePreview();
+		Assert( p );
+		m_pMenu->SetMenuItemChecked( id, ( p && p->GetParent() ) ? true : false );
+	}
+	else
+	{
+		id = m_Items.Find( "properties" );
+		m_pMenu->SetItemEnabled( id, false );
+		id = m_Items.Find( "browser" );
+		m_pMenu->SetItemEnabled( id, false );
+		id = m_Items.Find( "particlepreview" );
+		m_pMenu->SetItemEnabled( id, false );
+	}
+}
+
+
+//-----------------------------------------------------------------------------
+//
+// The Tool menu
+//
+//-----------------------------------------------------------------------------
+class CPetToolMenuButton : public CToolMenuButton
+{
+	DECLARE_CLASS_SIMPLE( CPetToolMenuButton, CToolMenuButton );
+public:
+	CPetToolMenuButton( CPetTool *parent, const char *panelName, const char *text, vgui::Panel *pActionSignalTarget );
+	virtual void OnShowMenu(vgui::Menu *menu);
+
+private:
+	CPetTool *m_pTool;
+};
+
+CPetToolMenuButton::CPetToolMenuButton( CPetTool *parent, const char *panelName, const char *text, vgui::Panel *pActionSignalTarget )
+	: BaseClass( parent, panelName, text, pActionSignalTarget )
+{
+	m_pTool = parent;
+
+	SetMenu(m_pMenu);
+}
+
+void CPetToolMenuButton::OnShowMenu(vgui::Menu *menu)
+{
+	BaseClass::OnShowMenu( menu );
+}
+
+
+//-----------------------------------------------------------------------------
+// Initializes the menu bar
+//-----------------------------------------------------------------------------
+vgui::MenuBar *CPetTool::CreateMenuBar( CBaseToolSystem *pParent ) 
+{
+	m_pMenuBar = new CToolFileMenuBar( pParent, "Main Menu Bar" );
+
+	// Sets info in the menu bar
+	char title[ 64 ];
+	ComputeMenuBarTitle( title, sizeof( title ) );
+	m_pMenuBar->SetInfo( title );
+	m_pMenuBar->SetToolName( GetToolName() );
+
+	// Add menu buttons
+	CToolMenuButton *pFileButton = CreateToolFileMenuButton( m_pMenuBar, "File", "&File", GetActionTarget(), this );
+	CToolMenuButton *pEditButton = CreateToolEditMenuButton( this, "Edit", "&Edit", GetActionTarget() );
+	CPetToolMenuButton *pToolButton = new CPetToolMenuButton( this, "Pet", "&Pet", GetActionTarget() );
+	CPetViewMenuButton *pViewButton = new CPetViewMenuButton( this, "View", "&View", GetActionTarget() );
+	CToolMenuButton *pSwitchButton = CreateToolSwitchMenuButton( m_pMenuBar, "Switcher", "&Tools", GetActionTarget() );
+
+	pEditButton->AddMenuItem( "copy", "#BxEditCopy", new KeyValues( "OnCopy" ), GetActionTarget(), NULL, "edit_copy" );
+	pEditButton->AddMenuItem( "paste", "#BxEditPaste", new KeyValues( "OnPaste" ), GetActionTarget(), NULL, "edit_paste" );
+
+	pEditButton->MoveMenuItem( pEditButton->FindMenuItem( "paste" ), pEditButton->FindMenuItem( "editkeybindings" ) );
+	pEditButton->MoveMenuItem( pEditButton->FindMenuItem( "copy" ), pEditButton->FindMenuItem( "paste" ) );
+	pEditButton->AddSeparatorAfterItem( "paste" );
+
+	m_pMenuBar->AddButton( pFileButton );
+	m_pMenuBar->AddButton( pEditButton );
+	m_pMenuBar->AddButton( pToolButton );
+	m_pMenuBar->AddButton( pViewButton );
+	m_pMenuBar->AddButton( pSwitchButton );
+
+	return m_pMenuBar;
+}
+
+
+//-----------------------------------------------------------------------------
+// Updates the menu bar based on the current file
+//-----------------------------------------------------------------------------
+void CPetTool::UpdateMenuBar( )
+{
+	if ( !m_pDoc )
+	{
+		m_pMenuBar->SetFileName( "#PetNoFile" );
+		return;
+	}
+
+	const char *pFile = m_pDoc->GetFileName();
+	if ( !pFile[0] )
+	{
+		m_pMenuBar->SetFileName( "#PetNoFile" );
+		return;
+	}
+
+	if ( m_pDoc->IsDirty() )
+	{
+		char sz[ 512 ];
+		Q_snprintf( sz, sizeof( sz ), "* %s", pFile );
+		m_pMenuBar->SetFileName( sz );
+	}
+	else
+	{
+		m_pMenuBar->SetFileName( pFile );
+	}
+}
+
+
+//-----------------------------------------------------------------------------
+// Gets at tool windows
+//-----------------------------------------------------------------------------
+CParticleSystemPropertiesContainer *CPetTool::GetProperties()
+{
+	return m_hProperties.Get();
+}
+
+CParticleSystemDefinitionBrowser *CPetTool::GetParticleSystemDefinitionBrowser()
+{
+	return m_hParticleSystemDefinitionBrowser.Get();
+}
+
+CParticleSystemPreviewPanel *CPetTool::GetParticlePreview()
+{
+	return m_hParticlePreview.Get();
+}
+
+
+//-----------------------------------------------------------------------------
+// Copy/paste
+//-----------------------------------------------------------------------------
+void CPetTool::OnCopy()
+{
+	GetParticleSystemDefinitionBrowser()->CopyToClipboard();
+}
+
+void CPetTool::OnPaste()
+{
+	GetParticleSystemDefinitionBrowser()->PasteFromClipboard();
+}
+
+
+//-----------------------------------------------------------------------------
+// Sets/gets the current particle system
+//-----------------------------------------------------------------------------
+void CPetTool::SetCurrentParticleSystem( CDmeParticleSystemDefinition *pParticleSystem, bool bForceBrowserSelection )
+{
+	if ( !m_pDoc )
+		return;
+
+	if ( m_hCurrentParticleSystem.Get() == pParticleSystem )
+		return;
+
+	m_hCurrentParticleSystem = pParticleSystem;
+	if ( bForceBrowserSelection && m_hParticleSystemDefinitionBrowser.Get() )
+	{
+		m_hParticleSystemDefinitionBrowser->UpdateParticleSystemList();
+		m_hParticleSystemDefinitionBrowser->SelectParticleSystem( pParticleSystem );
+	}
+	if ( m_hParticlePreview.Get() )
+	{
+		m_hParticlePreview->SetParticleSystem( pParticleSystem );
+	}
+	if ( m_hProperties.Get() )
+	{
+		m_hProperties->SetParticleSystem( m_hCurrentParticleSystem );
+	}
+}
+
+CDmeParticleSystemDefinition* CPetTool::GetCurrentParticleSystem( void )
+{
+	return m_hCurrentParticleSystem;
+}
+
+
+//-----------------------------------------------------------------------------
+// Destroys all tool windows
+//-----------------------------------------------------------------------------
+void CPetTool::DestroyToolContainers()
+{
+	int c = ToolWindow::GetToolWindowCount();
+	for ( int i = c - 1; i >= 0 ; --i )
+	{
+		ToolWindow *kill = ToolWindow::GetToolWindow( i );
+		delete kill;
+	}
+}
+
+
+//-----------------------------------------------------------------------------
+// Sets up the default layout
+//-----------------------------------------------------------------------------
+void CPetTool::OnDefaultLayout()
+{
+	int y = m_pMenuBar->GetTall();
+
+	int usew, useh;
+	GetSize( usew, useh );
+
+	DestroyToolContainers();
+
+	Assert( ToolWindow::GetToolWindowCount() == 0 );
+
+	CParticleSystemPropertiesContainer *pProperties = GetProperties();
+	CParticleSystemDefinitionBrowser *pParticleSystemBrowser = GetParticleSystemDefinitionBrowser();
+	CParticleSystemPreviewPanel *pPreviewer = GetParticlePreview();
+
+	// Need three containers
+	ToolWindow *pPropertyWindow = m_ToolWindowFactory.InstanceToolWindow( GetClientArea(), false, pProperties, "#PetProperties", false );
+	ToolWindow *pBrowserWindow = m_ToolWindowFactory.InstanceToolWindow( GetClientArea(), false, pParticleSystemBrowser, "#PetParticleSystemBrowser", false );
+	ToolWindow *pPreviewWindow = m_ToolWindowFactory.InstanceToolWindow( GetClientArea(), false, pPreviewer, "#PetPreviewer", false );
+
+	int halfScreen = usew / 2;
+	int bottom = useh - y;
+	int sy = (bottom - y) / 2;
+	SetMiniViewportBounds( halfScreen, y, halfScreen, sy - y );
+	pPreviewWindow->SetBounds( halfScreen, sy, halfScreen, bottom - sy );
+	pBrowserWindow->SetBounds( 0, y, halfScreen, sy - y );
+	pPropertyWindow->SetBounds( 0, sy, halfScreen, bottom - sy );
+}
+
+void CPetTool::OnToggleProperties()
+{
+	if ( m_hProperties.Get() )
+	{ 
+		ToggleToolWindow( m_hProperties.Get(), "#PetProperties" );
+	}
+}
+
+void CPetTool::OnToggleParticleSystemBrowser()
+{
+	if ( m_hParticleSystemDefinitionBrowser.Get() )
+	{ 
+		ToggleToolWindow( m_hParticleSystemDefinitionBrowser.Get(), "#PetParticleSystemBrowser" );
+	}
+}
+
+void CPetTool::OnToggleParticlePreview()
+{
+	if ( m_hParticlePreview.Get() )
+	{ 
+		ToggleToolWindow( m_hParticlePreview.Get(), "#PetPreviewer" );
+	}
+}
+
+
+//-----------------------------------------------------------------------------
+// Creates
+//-----------------------------------------------------------------------------
+void CPetTool::CreateTools( CPetDoc *doc )
+{
+	if ( !m_hProperties.Get() )
+	{
+		m_hProperties = new CParticleSystemPropertiesContainer( m_pDoc, this );
+	}
+
+	if ( !m_hParticleSystemDefinitionBrowser.Get() )
+	{
+		m_hParticleSystemDefinitionBrowser = new CParticleSystemDefinitionBrowser( m_pDoc, this, "ParticleSystemDefinitionBrowser" );
+	}
+
+	if ( !m_hParticlePreview.Get() )
+	{
+		m_hParticlePreview = new CParticleSystemPreviewPanel( NULL, "Particle System Preview" );
+	}
+	RegisterToolWindow( m_hProperties );
+	RegisterToolWindow( m_hParticleSystemDefinitionBrowser );
+	RegisterToolWindow( m_hParticlePreview );
+}
+
+
+//-----------------------------------------------------------------------------
+// Initializes the tools
+//-----------------------------------------------------------------------------
+void CPetTool::InitTools()
+{
+	// FIXME: There are no tool windows here; how should this work?
+	// These panels are saved
+	windowposmgr->RegisterPanel( "properties", m_hProperties, false );
+	windowposmgr->RegisterPanel( "particlesystemdefinitionbrowser", m_hParticleSystemDefinitionBrowser, false );
+	windowposmgr->RegisterPanel( "previewpanel", m_hParticlePreview, false );
+
+	if ( !windowposmgr->LoadPositions( "cfg/pet.txt", this, &m_ToolWindowFactory, "Pet" ) )
+	{
+		OnDefaultLayout();
+	}
+}
+
+
+void CPetTool::DestroyTools()
+{
+	SetCurrentParticleSystem( NULL );
+
+	int c = ToolWindow::GetToolWindowCount();
+	for ( int i = c - 1; i >= 0 ; --i )
+	{
+		ToolWindow *kill = ToolWindow::GetToolWindow( i );
+		delete kill;
+	}
+
+	UnregisterAllToolWindows();
+
+	if ( m_hProperties.Get() )
+	{
+		windowposmgr->UnregisterPanel( m_hProperties.Get() );
+		delete m_hProperties.Get();
+		m_hProperties = NULL;
+	}
+
+	if ( m_hParticleSystemDefinitionBrowser.Get() )
+	{
+		windowposmgr->UnregisterPanel( m_hParticleSystemDefinitionBrowser.Get() );
+		delete m_hParticleSystemDefinitionBrowser.Get();
+		m_hParticleSystemDefinitionBrowser = NULL;
+	}
+
+	if ( m_hParticlePreview.Get() )
+	{
+		windowposmgr->UnregisterPanel( m_hParticlePreview.Get() );
+		delete m_hParticlePreview.Get();
+		m_hParticlePreview = NULL;
+	}
+}
+
+
+void CPetTool::ShowToolWindow( Panel *tool, char const *toolName, bool visible )
+{
+	Assert( tool );
+
+	if ( tool->GetParent() == NULL && visible )
+	{
+		m_ToolWindowFactory.InstanceToolWindow( this, false, tool, toolName, false );
+	}
+	else if ( !visible )
+	{
+		ToolWindow *tw = dynamic_cast< ToolWindow * >( tool->GetParent()->GetParent() );
+		Assert( tw );
+		tw->RemovePage( tool );
+	}
+}
+
+void CPetTool::ToggleToolWindow( Panel *tool, char const *toolName )
+{
+	Assert( tool );
+
+	if ( tool->GetParent() == NULL )
+	{
+		ShowToolWindow( tool, toolName, true );
+	}
+	else
+	{
+		ShowToolWindow( tool, toolName, false );
+	}
+}
+
+
+//-----------------------------------------------------------------------------
+// Creates the action menu
+//-----------------------------------------------------------------------------
+vgui::Menu *CPetTool::CreateActionMenu( vgui::Panel *pParent )
+{
+	vgui::Menu *pActionMenu = new Menu( pParent, "ActionMenu" );
+	pActionMenu->AddMenuItem( "#ToolHide", new KeyValues( "Command", "command", "HideActionMenu" ), GetActionTarget() );
+	return pActionMenu;
+}
+
+
+//-----------------------------------------------------------------------------
+// Inherited from IFileMenuCallbacks
+//-----------------------------------------------------------------------------
+int	CPetTool::GetFileMenuItemsEnabled( )
+{
+	int nFlags = FILE_ALL;
+	if ( m_RecentFiles.IsEmpty() )
+	{
+		nFlags &= ~(FILE_RECENT | FILE_CLEAR_RECENT);
+	}
+	return nFlags;
+}
+
+void CPetTool::AddRecentFilesToMenu( vgui::Menu *pMenu )
+{
+	m_RecentFiles.AddToMenu( pMenu, GetActionTarget(), "OnRecent" );
+}
+
+bool CPetTool::GetPerforceFileName( char *pFileName, int nMaxLen )
+{
+	if ( !m_pDoc )
+		return false;
+
+	Q_strncpy( pFileName, m_pDoc->GetFileName(), nMaxLen );
+	return pFileName[0] != 0;
+}
+
+
+//-----------------------------------------------------------------------------
+// Purpose: 
+// Input  :  - 
+//-----------------------------------------------------------------------------
+void CPetTool::OnExit()
+{
+	windowposmgr->SavePositions( "cfg/pet.txt", "Pet" );
+
+	enginetools->Command( "quit\n" );
+}
+
+//-----------------------------------------------------------------------------
+// Handle commands from the action menu and other menus
+//-----------------------------------------------------------------------------
+void CPetTool::OnCommand( const char *cmd )
+{
+	if ( !V_stricmp( cmd, "HideActionMenu" ) )
+	{
+		if ( GetActionMenu() )
+		{
+			GetActionMenu()->SetVisible( false );
+		}
+	}
+	else if ( const char *pSuffix = StringAfterPrefix( cmd, "OnRecent" ) )
+	{
+		int idx = Q_atoi( pSuffix );
+		OpenFileFromHistory( idx );
+	}
+	else if ( const char *pSuffix = StringAfterPrefix( cmd, "OnTool" ) )
+	{
+		int idx = Q_atoi( pSuffix );
+		enginetools->SwitchToTool( idx );
+	}
+	else if ( !V_stricmp( cmd, "OnUndo" ) )
+	{
+		OnUndo();
+	}
+	else if ( !V_stricmp( cmd, "OnRedo" ) )
+	{
+		OnRedo();
+	}
+	else if ( !V_stricmp( cmd, "OnDescribeUndo" ) )
+	{
+		OnDescribeUndo();
+	}
+	else
+	{
+		BaseClass::OnCommand( cmd );
+	}
+}
+
+
+//-----------------------------------------------------------------------------
+// Command handlers
+//-----------------------------------------------------------------------------
+void CPetTool::PerformNew()
+{
+	OnCloseNoSave();
+	NewDocument();
+}
+
+void CPetTool::OnNew()
+{
+	if ( m_pDoc && m_pDoc->IsDirty() )
+	{
+		SaveFile( m_pDoc->GetFileName(), PET_FILE_FORMAT, FOSM_SHOW_PERFORCE_DIALOGS | FOSM_SHOW_SAVE_QUERY,
+			new KeyValues( "OnNew" ) );
+		return;
+	}
+
+	PerformNew();
+}
+
+
+//-----------------------------------------------------------------------------
+// Called when the File->Open menu is selected
+//-----------------------------------------------------------------------------
+void CPetTool::OnOpen( )
+{
+	int nFlags = 0;
+	const char *pSaveFileName = NULL;
+	if ( m_pDoc && m_pDoc->IsDirty() )
+	{
+		nFlags = FOSM_SHOW_PERFORCE_DIALOGS | FOSM_SHOW_SAVE_QUERY;
+		pSaveFileName = m_pDoc->GetFileName();
+	}
+
+	OpenFile( PET_FILE_FORMAT, pSaveFileName, PET_FILE_FORMAT, nFlags );
+}
+
+bool CPetTool::OnReadFileFromDisk( const char *pFileName, const char *pFileFormat, KeyValues *pContextKeyValues )
+{
+	OnCloseNoSave();
+
+	if ( !LoadDocument( pFileName ) )
+		return false;
+
+	m_RecentFiles.Add( pFileName, pFileFormat );
+	m_RecentFiles.SaveToRegistry( GetRegistryName() );
+	UpdateMenuBar();
+	return true;
+}
+
+void CPetTool::Save()
+{
+	if ( m_pDoc )
+	{
+		SaveFile( m_pDoc->GetFileName(), PET_FILE_FORMAT, FOSM_SHOW_PERFORCE_DIALOGS );
+	}
+}
+
+void CPetTool::OnSaveAs()
+{
+	if ( m_pDoc )
+	{
+		SaveFile( NULL, PET_FILE_FORMAT, FOSM_SHOW_PERFORCE_DIALOGS );
+	}
+}
+
+void CPetTool::OnRestartLevel()
+{
+	enginetools->Command( "restart" );
+	enginetools->Execute();
+}
+
+void CPetTool::SaveAndTest()
+{
+	if ( m_pDoc && m_pDoc->IsDirty() )
+	{
+		SaveFile( m_pDoc->GetFileName(), PET_FILE_FORMAT, FOSM_SHOW_PERFORCE_DIALOGS, 
+			new KeyValues( "RestartLevel" ) );
+	}
+	else
+	{
+		OnRestartLevel();
+	}
+}
+
+bool CPetTool::OnWriteFileToDisk( const char *pFileName, const char *pFileFormat, KeyValues *pContextKeyValues )
+{
+	if ( !m_pDoc )
+		return true;
+
+	m_pDoc->SetFileName( pFileName );
+	m_pDoc->SaveToFile( );
+
+	m_RecentFiles.Add( pFileName, pFileFormat );
+	m_RecentFiles.SaveToRegistry( GetRegistryName() );
+	UpdateMenuBar();
+	return true;
+}
+
+void CPetTool::OnClose()
+{
+	if ( m_pDoc && m_pDoc->IsDirty() )
+	{
+		SaveFile( m_pDoc->GetFileName(), PET_FILE_FORMAT, FOSM_SHOW_PERFORCE_DIALOGS | FOSM_SHOW_SAVE_QUERY, 
+			new KeyValues( "OnClose" ) );
+		return;
+	}
+
+	OnCloseNoSave();
+}
+
+void CPetTool::OnCloseNoSave()
+{
+	DestroyTools();
+
+	if ( m_pDoc )
+	{
+		CAppNotifyScopeGuard sg( "CPetTool::OnCloseNoSave", NOTIFY_CHANGE_OTHER );
+
+		delete m_pDoc;
+		m_pDoc = NULL;
+
+		if ( m_hProperties )
+		{
+			m_hProperties->SetParticleSystem( NULL );
+		}
+	}
+
+	UpdateMenuBar( );
+}
+
+void CPetTool::OnMarkNotDirty()
+{
+	if ( m_pDoc )
+	{
+		m_pDoc->SetDirty( false );
+	}
+}
+
+
+//-----------------------------------------------------------------------------
+// Open a specific file
+//-----------------------------------------------------------------------------
+void CPetTool::OpenSpecificFile( const char *pFileName )
+{
+	int nFlags = 0;
+	const char *pSaveFileName = NULL;
+	if ( m_pDoc )
+	{
+		// File is already open
+		if ( !Q_stricmp( m_pDoc->GetFileName(), pFileName ) )
+			return;
+
+		if ( m_pDoc->IsDirty() )
+		{
+			nFlags = FOSM_SHOW_PERFORCE_DIALOGS | FOSM_SHOW_SAVE_QUERY;
+			pSaveFileName = m_pDoc->GetFileName();
+		}
+		else
+		{
+			OnCloseNoSave();
+		}
+	}
+
+	OpenFile( pFileName, PET_FILE_FORMAT, pSaveFileName, PET_FILE_FORMAT, nFlags );
+}
+
+
+//-----------------------------------------------------------------------------
+// Show the save document query dialog
+//-----------------------------------------------------------------------------
+void CPetTool::OpenFileFromHistory( int slot )
+{
+	const char *pFileName = m_RecentFiles.GetFile( slot );
+	if ( !pFileName )
+		return;
+	OpenSpecificFile( pFileName );
+}
+
+
+//-----------------------------------------------------------------------------
+// Derived classes can implement this to get notified when files are saved/loaded
+//-----------------------------------------------------------------------------
+void CPetTool::OnFileOperationCompleted( const char *pFileType, bool bWroteFile, vgui::FileOpenStateMachine::CompletionState_t state, KeyValues *pContextKeyValues )
+{
+	if ( bWroteFile )
+	{
+		OnMarkNotDirty();
+	}
+
+	if ( !pContextKeyValues )
+		return;
+
+	if ( state != FileOpenStateMachine::SUCCESSFUL )
+		return;
+
+	if ( !Q_stricmp( pContextKeyValues->GetName(), "OnNew" ) )
+	{
+		PerformNew();
+		return;
+	}
+
+	if ( !Q_stricmp( pContextKeyValues->GetName(), "OnClose" ) )
+	{
+		OnCloseNoSave();
+		return;
+	}
+
+	if ( !Q_stricmp( pContextKeyValues->GetName(), "OnQuit" ) )
+	{
+		OnCloseNoSave();
+		vgui::ivgui()->PostMessage( GetVPanel(), new KeyValues( "OnExit" ), 0 );
+		return;
+	}
+
+	if ( !Q_stricmp( pContextKeyValues->GetName(), "RestartLevel" ) )
+	{
+		OnRestartLevel();
+		return;
+	}
+}
+
+
+//-----------------------------------------------------------------------------
+// Show the File browser dialog
+//-----------------------------------------------------------------------------
+void CPetTool::SetupFileOpenDialog( vgui::FileOpenDialog *pDialog, bool bOpenFile, const char *pFileFormat, KeyValues *pContextKeyValues )
+{
+	char pStartingDir[ MAX_PATH ];
+
+	GetModSubdirectory( "particles", pStartingDir, sizeof(pStartingDir) );
+
+	// Open a bsp file to create a new commentary file
+	pDialog->SetTitle( "Choose Particle Configuration File", true );
+	pDialog->SetStartDirectoryContext( "pet_session", pStartingDir );
+	pDialog->AddFilter( "*.pcf", "Particle Configuration File (*.pcf)", true );
+}
+
+
+//-----------------------------------------------------------------------------
+// Can we quit?
+//-----------------------------------------------------------------------------
+bool CPetTool::CanQuit()
+{
+	if ( m_pDoc && m_pDoc->IsDirty() )
+	{
+		// Show Save changes Yes/No/Cancel and re-quit if hit yes/no
+		SaveFile( m_pDoc->GetFileName(), PET_FILE_FORMAT, FOSM_SHOW_PERFORCE_DIALOGS | FOSM_SHOW_SAVE_QUERY, 
+			new KeyValues( "OnQuit" ) );
+		return false;
+	}
+
+	return true;
+}
+
+
+//-----------------------------------------------------------------------------
+// Various command handlers related to the Edit menu
+//-----------------------------------------------------------------------------
+void CPetTool::OnUndo()
+{
+	CDisableUndoScopeGuard guard;
+	g_pDataModel->Undo();
+}
+
+void CPetTool::OnRedo()
+{
+	CDisableUndoScopeGuard guard;
+	g_pDataModel->Redo();
+}
+
+void CPetTool::OnDescribeUndo()
+{
+	CUtlVector< UndoInfo_t > list;
+	g_pDataModel->GetUndoInfo( list );
+
+	Msg( "%i operations in stack\n", list.Count() );
+
+	for ( int i = list.Count() - 1; i >= 0; --i )
+	{
+		UndoInfo_t& entry = list[ i ];
+		if ( entry.terminator )
+		{
+			Msg( "[ '%s' ] - %i operations\n", entry.undo, entry.numoperations );
+		}
+
+		Msg( "   +%s\n", entry.desc );
+	}
+}
+
+
+//-----------------------------------------------------------------------------
+// Background
+//-----------------------------------------------------------------------------
+const char *CPetTool::GetLogoTextureName()
+{
+	return NULL;
+}
+
+
+//-----------------------------------------------------------------------------
+// Inherited from IPetDocCallback
+//-----------------------------------------------------------------------------
+void CPetTool::OnDocChanged( const char *pReason, int nNotifySource, int nNotifyFlags )
+{
+	CDmeParticleSystemDefinition *pParticleSystem = GetCurrentParticleSystem();
+	if ( m_pDoc && GetParticlePreview() && pParticleSystem )
+	{
+		m_pDoc->UpdateParticleDefinition( pParticleSystem );
+	}
+
+	if ( nNotifyFlags & NOTIFY_CHANGE_TOPOLOGICAL )
+	{
+		if ( GetParticleSystemDefinitionBrowser() )
+		{
+			GetParticleSystemDefinitionBrowser()->UpdateParticleSystemList();
+		}
+	}
+
+	bool bRefreshProperties = ( nNotifySource != NOTIFY_SOURCE_PROPERTIES_TREE ) &&
+		( ( nNotifyFlags & ( NOTIFY_CHANGE_TOPOLOGICAL | NOTIFY_CHANGE_ATTRIBUTE_ARRAY_SIZE ) ) != 0 );
+	bool bRefreshPropertyValues = ( nNotifySource != NOTIFY_SOURCE_PROPERTIES_TREE ) &&
+		( nNotifyFlags & NOTIFY_CHANGE_ATTRIBUTE_VALUE ) != 0;
+	
+	if ( bRefreshProperties || bRefreshPropertyValues )
+	{
+		if ( m_hProperties.Get() )
+		{
+			m_hProperties->Refresh( !bRefreshProperties );
+		}
+	}
+
+	UpdateMenuBar();
+}
+
+
+//-----------------------------------------------------------------------------
+// Creates a new document
+//-----------------------------------------------------------------------------
+void CPetTool::NewDocument( )
+{
+	Assert( !m_pDoc );
+
+	DestroyTools();
+
+	m_pDoc = new CPetDoc( this );
+	m_pDoc->CreateNew( );
+
+	ShowMiniViewport( true );
+	CreateTools( m_pDoc );
+	UpdateMenuBar( );
+	InitTools();
+}
+
+
+//-----------------------------------------------------------------------------
+// Loads up a new document
+//-----------------------------------------------------------------------------
+bool CPetTool::LoadDocument( const char *pDocName )
+{
+	Assert( !m_pDoc );
+
+	DestroyTools();
+
+	m_pDoc = new CPetDoc( this );
+	if ( !m_pDoc->LoadFromFile( pDocName ) )
+	{
+		delete m_pDoc;
+		m_pDoc = NULL;
+		Warning( "Fatal error loading '%s'\n", pDocName );
+		return false;
+	}
+
+	ShowMiniViewport( true );
+
+	CreateTools( m_pDoc );
+	UpdateMenuBar( );
+	InitTools();
+
+	// Let the other tools know	we've loaded + therefore modified particle systems
+	CUtlBuffer buf;
+	g_pDataModel->Serialize( buf, "binary", PET_FILE_FORMAT, m_pDoc->GetRootObject()->GetHandle() );
+
+	KeyValues *pMessage = new KeyValues( "ParticleSystemUpdated" );
+	pMessage->SetPtr( "definitionBits", buf.Base() );
+	pMessage->SetInt( "definitionSize", buf.TellMaxPut() );
+	PostMessageToAllTools( pMessage );
+	pMessage->deleteThis();
+	return true;
+}
+
+