1HEADmaster

1 files changed, 420 insertions, 0 deletions

diff --git a/utils/elementviewer/elementviewerpanel.cpp b/utils/elementviewer/elementviewerpanel.cpp
new file mode 100644
index 0000000..8866060
--- /dev/null
+++ b/utils/elementviewer/elementviewerpanel.cpp
@@ -0,0 +1,420 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+//
+// Purpose: 
+//
+// $NoKeywords: $
+//=============================================================================
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <malloc.h>
+#include "tier0/dbg.h"
+#include "vgui_controls/Panel.h"
+#include "elementviewer.h"
+#include "vgui_controls/MenuBar.h"
+#include "vgui/ISurface.h"
+#include "vgui/IInput.h"
+#include "vgui_controls/Menu.h"
+#include "KeyValues.h"
+#include "tier0/icommandline.h"
+#include "datamodel/dmelement.h"
+#include "datamodel/idatamodel.h"
+#include "vgui_controls/FileOpenDialog.h"
+#include "filesystem.h"
+#include "vgui/IVGui.h"
+#include "movieobjects/movieobjects.h"
+//#include "view.h"
+#include "dme_controls/INotifyUI.h"
+#include "dme_controls/ElementPropertiesTree.h"
+#include "dme_controls/filelistmanager.h"
+#include "dme_controls/dmecontrols.h"
+
+using namespace vgui;
+
+typedef vgui::DHANDLE< CElementPropertiesTree > viewList_t;
+typedef CUtlRBTree< CDmElement *, int > ElementDict_t;
+ 
+struct ViewerDoc_t
+{
+	ViewerDoc_t() : m_fileid( DMFILEID_INVALID ), m_bDirty( false ) {}
+
+	DmFileId_t	m_fileid;
+	bool		m_bDirty;
+};
+
+
+//-----------------------------------------------------------------------------
+// main editor panel
+//-----------------------------------------------------------------------------
+class CElementViewerPanel : public vgui::Panel, public IDmNotify, public CBaseElementPropertiesChoices
+{
+	DECLARE_CLASS_SIMPLE( CElementViewerPanel, vgui::Panel );
+
+public:
+
+	CElementViewerPanel();
+	~CElementViewerPanel();
+
+	virtual void NotifyDataChanged( const char *pReason, int nNotifySource, int nNotifyFlags );
+
+	// Resize this panel to match its parent
+	virtual void PerformLayout();
+
+	virtual void OnCommand( const char *cmd );
+
+	virtual void OnThink();
+
+	void	OnOpen();
+	void	OnSaveAs();
+	void	OnSave();
+	void	OnNew();
+
+	int NumDocs()
+	{
+		return m_Docs.Count();
+	}
+
+	CDmElement *GetRoot( int docNum )
+	{
+		return GetElement< CDmElement >( g_pDataModel->GetFileRoot( m_Docs[ docNum ].m_fileid ) );
+	}
+
+protected:
+	MESSAGE_FUNC_PARAMS( OnFileSelected, "FileSelected", kv );
+
+private:
+	void CreateNewView( CDmElement *pRoot, const char *title );
+
+	vgui::MenuBar *m_pMenuBar;
+
+	// FIXME: Is there a better way?
+	// A panel that represents all area under the menu bar
+	vgui::Panel *m_pClientArea;
+	CFileManagerFrame *m_pFileManager;
+	CUtlVector< viewList_t >	m_Views;
+	CUtlVector< ViewerDoc_t >	m_Docs;
+};
+
+
+vgui::Panel *CreateElementViewerPanel()
+{
+	// add our main window
+	vgui::Panel *pElementViewer = new CElementViewerPanel;
+	//g_pVGuiSurface->CreatePopup(pElementViewer->GetVPanel(), false );
+	pElementViewer->SetParent( g_pVGuiSurface->GetEmbeddedPanel() );
+	return pElementViewer;
+}
+
+//CElementView *CreateView( vgui::Panel *parent, CDmElement *pRoot, const char *title );
+
+CElementViewerPanel::CElementViewerPanel() : vgui::Panel( NULL, "ElementViewer" )
+{
+	SetElementPropertiesChoices( this );
+	m_pMenuBar = new vgui::MenuBar( this, "Main Menu Bar" );
+	m_pMenuBar->SetSize( 10, 28 );
+
+	// Next create a menu
+	Menu *pMenu = new Menu(NULL, "File Menu");
+	pMenu->AddMenuItem("&New", new KeyValues ( "Command", "command", "OnNew"), this);
+	pMenu->AddMenuItem("&Open", new KeyValues ("Command", "command", "OnOpen"), this);
+	pMenu->AddMenuItem("&Save", new KeyValues ("Command", "command", "OnSave"), this);
+	pMenu->AddMenuItem("Save &As", new KeyValues ("Command", "command", "OnSaveAs"), this);
+	pMenu->AddMenuItem("E&xit", new KeyValues ("Command", "command", "OnExit"), this);
+
+	m_pMenuBar->AddMenu( "&File", pMenu );
+
+	m_pClientArea = new vgui::Panel( this, "ElementViewer Client Area" );
+
+	m_pFileManager = new CFileManagerFrame( m_pClientArea );
+
+	SetKeyBoardInputEnabled( true );
+
+	// load a file from the commandline
+	const char *fileName = NULL;
+	CommandLine()->CheckParm("-loadDmx", &fileName );
+
+	if ( fileName )
+	{
+		// trim off any quotes (paths with spaces need to be quoted on the commandline)
+		char buf[ MAX_PATH ];
+		V_StrSubst( fileName, "\"", "", buf, sizeof( buf ) );
+
+		KeyValues *pKeyValues = new KeyValues( "OnFileSelected", "fullpath", buf );
+		OnFileSelected( pKeyValues );
+		pKeyValues->deleteThis();
+	}
+
+	g_pDataModel->InstallNotificationCallback( this );
+}
+
+void RemoveFileId( DmFileId_t fileid )
+{
+	Assert( fileid != DMFILEID_INVALID );
+	if ( fileid != DMFILEID_INVALID )
+	{
+		g_pDataModel->RemoveFileId( fileid );
+	}
+}
+
+void ReportElementStats()
+{
+	int nCurrentElements = g_pDataModel->GetAllocatedElementCount();
+	int nTotalElements = g_pDataModel->GetElementsAllocatedSoFar();
+	int nMaxElements = g_pDataModel->GetMaxNumberOfElements();
+//	int nCurrentAttributes = g_pDataModel->GetAllocatedAttributeCount();
+	Msg( "element count: current = %d max = %d total = %d\n", nCurrentElements, nMaxElements, nTotalElements );
+
+	int nElementsInFiles = 0;
+	int nFiles = g_pDataModel->NumFileIds();
+	for ( int fi = 0; fi < nFiles; ++fi )
+	{
+		DmFileId_t fileid = g_pDataModel->GetFileId( fi );
+		int nElements = g_pDataModel->NumElementsInFile( fileid );
+		nElementsInFiles += nElements;
+		const char *pFileName = g_pDataModel->GetFileName( fileid );
+		Msg( "elements in file \"%s\" = %d\n", pFileName, nElements );
+	}
+	Msg( "elements not in any file = %d\n", nCurrentElements - nElementsInFiles );
+}
+
+CElementViewerPanel::~CElementViewerPanel()
+{
+	int nDocs = m_Docs.Count();
+	for ( int i = 0; i < nDocs; ++i )
+	{
+		RemoveFileId( m_Docs[ i ].m_fileid );
+	}
+	m_Docs.RemoveAll();
+
+	ReportElementStats();
+	g_pDataModel->RemoveNotificationCallback( this );
+	SetElementPropertiesChoices( NULL );
+}
+
+void CElementViewerPanel::OnThink()
+{
+	if ( vgui::input()->IsKeyDown( KEY_ESCAPE ) )
+	{
+		vgui::ivgui()->Stop();
+	}
+	else
+	{
+		BaseClass::OnThink();
+	}
+}
+
+void CElementViewerPanel::CreateNewView( CDmElement *pRoot, const char *title )
+{
+	vgui::DHANDLE< CElementPropertiesTree > f;
+	// f = CreateView( m_pClientArea, pRoot, title );
+	f = new CElementPropertiesTree( m_pClientArea, this, pRoot );
+	f->Init();
+	f->SetPos( 10, 30 );
+	f->SetSize( 600, 500 );
+	f->SetVisible( true );
+	m_Views.AddToTail( f );
+
+	m_pFileManager->Refresh();
+}
+
+void CElementViewerPanel::NotifyDataChanged( const char *pReason, int nNotifySource, int nNotifyFlags )
+{
+//	if ( flags & INotifyUI::NOTIFY_REFRESH_PROPERTIES_VALUES ) // FIXME - do we need new flags for file association changes?
+	{
+		m_pFileManager->Refresh();
+
+		int nViews = m_Views.Count();
+		for ( int i = 0; i < nViews; ++i )
+		{
+			if ( m_Views[ i ] )
+			{
+				m_Views[ i ]->Refresh();
+			}
+		}
+	}
+}
+
+void CElementViewerPanel::OnCommand( const char *cmd )
+{
+	if ( !V_stricmp( cmd, "OnOpen" ) )
+	{
+		OnOpen();
+	}
+	else if ( !V_stricmp( cmd, "OnSave" ) )
+	{
+		OnSave();
+	}
+	else if ( !V_stricmp( cmd, "OnSaveAs" ) )
+	{
+		OnSaveAs();
+	}
+	else if ( !V_stricmp( cmd, "OnNew" ) )
+	{
+		OnNew();
+	}
+	else if ( !V_stricmp( cmd, "OnExit" ) )
+	{
+		// Throw up a "save" dialog?
+		vgui::ivgui()->Stop();
+	}
+	else
+	{
+		BaseClass::OnCommand( cmd );
+	}
+}
+
+void CElementViewerPanel::OnSaveAs()
+{
+	if ( m_Docs.Count() < 1 )
+		return;
+
+	DmFileId_t fileid = m_Docs[ 0 ].m_fileid;
+	// Save As file
+	KeyValues *pContextKeyValues = new KeyValues( "OnSaveAs" );
+	FileOpenDialog *pFileOpenDialog = new FileOpenDialog( this, "Save .dmx File As", false, pContextKeyValues );
+
+	const char *pFileFormat = g_pDataModel->GetFileFormat( fileid );
+	const char *pDescription = ( pFileFormat && *pFileFormat ) ? g_pDataModel->GetFormatDescription( pFileFormat ) : NULL;
+
+	if ( pDescription && *pDescription )
+	{
+		char description[ 256 ];
+		V_snprintf( description, sizeof( description ), "%s (*.dmx)", g_pDataModel->GetFormatDescription( pFileFormat ) );
+		pFileOpenDialog->AddFilter( "*.dmx", description, true, pFileFormat );
+	}
+	else
+	{
+		pFileOpenDialog->AddFilter( "*.dmx", "DMX File (*.dmx)", true, "dmx" );
+	}
+
+	pFileOpenDialog->AddActionSignalTarget( this );
+	pFileOpenDialog->DoModal( false );
+}
+
+void CElementViewerPanel::OnSave()
+{
+	int docCount = m_Docs.Count();
+	if ( docCount > 0 )
+	{
+		DmFileId_t fileid = m_Docs[ docCount - 1 ].m_fileid;
+		const char *pFormat = g_pDataModel->GetFileFormat( fileid );
+		const char *pEncoding = g_pDataModel->GetDefaultEncoding( pFormat );
+		const char *pFileName = g_pDataModel->GetFileName( fileid );
+		CDmElement *pRoot = GetElement< CDmElement >( g_pDataModel->GetFileRoot( fileid ) );
+		g_pDataModel->SaveToFile( pFileName, NULL, pEncoding, pFormat, pRoot );
+		// TODO - figure out what file type this was
+	}
+}
+
+struct DataModelFilenameArray
+{
+	int Count() const
+	{
+		return g_pDataModel->NumFileIds();
+	}
+	const char *operator[]( int i ) const
+	{
+		return g_pDataModel->GetFileName( g_pDataModel->GetFileId( i ) );
+	}
+};
+
+void CElementViewerPanel::OnNew()
+{
+	char filename[ MAX_PATH ];
+	V_GenerateUniqueName( filename, sizeof( filename ), "unnamed", DataModelFilenameArray() );
+
+	ViewerDoc_t doc;
+	doc.m_fileid = g_pDataModel->FindOrCreateFileId( filename );
+	CDmElement *pRoot = CreateElement< CDmElement >( "root", doc.m_fileid );
+	g_pDataModel->SetFileRoot( doc.m_fileid, pRoot->GetHandle() );
+
+	m_Docs.AddToTail( doc );
+
+	CreateNewView( pRoot, filename );
+}
+
+void CElementViewerPanel::OnOpen()
+{
+	// Open file
+	FileOpenDialog *pFileOpenDialog = new FileOpenDialog( this, "Choose .dmx file", true );
+	pFileOpenDialog->AddFilter( "*.*", "All Files (*.*)", false );
+	pFileOpenDialog->AddFilter( "*.dmx", "DmElement Files (*.dmx)", true );
+	for ( int i = 0; i < g_pDataModel->GetFormatCount(); ++i )
+	{
+		const char *pFormatName = g_pDataModel->GetFormatName(i);
+		const char *pDesc = g_pDataModel->GetFormatDescription(pFormatName);
+		const char *pExt = g_pDataModel->GetFormatExtension(pFormatName);
+
+		char pExtBuf[512];
+		char pDescBuf[512];
+		Q_snprintf( pExtBuf, sizeof(pExtBuf), "*.%s", pExt );
+		Q_snprintf( pDescBuf, sizeof(pDescBuf), "%s (*.%s)", pDesc, pExt );
+
+		pFileOpenDialog->AddFilter( pExtBuf, pDescBuf, false );
+	}
+	pFileOpenDialog->AddActionSignalTarget( this );
+	pFileOpenDialog->DoModal( false );
+}
+
+void CElementViewerPanel::OnFileSelected( KeyValues *pKeyValues )
+{
+	const char *pFullPath = pKeyValues->GetString( "fullpath" );
+	if ( !pFullPath || !pFullPath[ 0 ] )
+		return;
+
+	if ( pKeyValues->FindKey( "OnSaveAs" ) )
+	{
+		const char *pFormat = pKeyValues->GetString( "filterinfo" );
+		Assert( pFormat );
+		if ( !pFormat )
+			return;
+
+		// TODO - figure out which panel is on top, and save the file associated with it
+
+		int docCount = m_Docs.Count();
+		if ( docCount == 1 )
+		{
+			g_pDataModel->SetFileName( m_Docs[ 0 ].m_fileid, pFullPath );
+			g_pDataModel->SaveToFile( pFullPath, NULL, g_pDataModel->GetDefaultEncoding( pFormat ), pFormat, GetElement< CDmElement >( g_pDataModel->GetFileRoot( m_Docs[ 0 ].m_fileid ) ) );
+		}
+		return;
+	}
+
+//	char relativepath[ 512 ];
+//	g_pFileSystem->FullPathToRelativePath( fullpath, relativepath, sizeof( relativepath ) );
+
+	g_pDataModel->OnlyCreateUntypedElements( true );
+	g_pDataModel->SetDefaultElementFactory( NULL );
+
+	// Open the path as a KV and parse stuff from it...
+	CDmElement *pRoot = NULL;
+	DmFileId_t fileid = g_pDataModel->RestoreFromFile( pFullPath, NULL, NULL, &pRoot, CR_DELETE_NEW );
+	if ( pRoot )
+	{
+		ViewerDoc_t doc;
+		doc.m_fileid = fileid;
+		m_Docs.AddToTail( doc );
+
+		CreateNewView( pRoot, pFullPath );
+	}
+}
+
+//-----------------------------------------------------------------------------
+// The editor panel should always fill the space...
+//-----------------------------------------------------------------------------
+void CElementViewerPanel::PerformLayout()
+{
+	// Make the editor panel fill the space
+	int iWidth, iHeight;
+
+	vgui::VPANEL parent = GetParent() ? GetParent()->GetVPanel() : vgui::surface()->GetEmbeddedPanel(); 
+	vgui::ipanel()->GetSize( parent, iWidth, iHeight );
+	SetSize( iWidth, iHeight );
+	m_pMenuBar->SetSize( iWidth, 28 );
+
+	// Make the client area also fill the space not used by the menu bar
+	int iTemp, iMenuHeight;
+	m_pMenuBar->GetSize( iTemp, iMenuHeight );
+	m_pClientArea->SetPos( 0, iMenuHeight );
+	m_pClientArea->SetSize( iWidth, iHeight - iMenuHeight );
+}