Modified source engine (2017) developed by valve and leaked in 2020. Not for commercial purporses
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

420 lines
12 KiB

//========= 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 );
}