mirror of
https://github.com/nillerusr/source-engine.git
synced 2024-12-22 14:16:50 +00:00
1743 lines
54 KiB
C++
1743 lines
54 KiB
C++
|
//========= Copyright Valve Corporation, All rights reserved. ============//
|
||
|
//
|
||
|
// Purpose:
|
||
|
//
|
||
|
//=============================================================================
|
||
|
|
||
|
#include "matsys_controls/mdlpicker.h"
|
||
|
#include "tier1/KeyValues.h"
|
||
|
#include "tier1/utldict.h"
|
||
|
#include "filesystem.h"
|
||
|
#include "studio.h"
|
||
|
#include "matsys_controls/matsyscontrols.h"
|
||
|
#include "matsys_controls/mdlpanel.h"
|
||
|
#include "vgui_controls/Splitter.h"
|
||
|
#include "vgui_controls/ComboBox.h"
|
||
|
#include "vgui_controls/Button.h"
|
||
|
#include "vgui_controls/PropertySheet.h"
|
||
|
#include "vgui_controls/MessageBox.h"
|
||
|
#include "vgui_controls/PropertyPage.h"
|
||
|
#include "vgui_controls/CheckButton.h"
|
||
|
#include "vgui_controls/DirectorySelectDialog.h"
|
||
|
#include "vgui/IVGui.h"
|
||
|
#include "vgui/IInput.h"
|
||
|
#include "vgui/ISurface.h"
|
||
|
#include "vgui/Cursor.h"
|
||
|
#include "matsys_controls/assetpicker.h"
|
||
|
#include "matsys_controls/colorpickerpanel.h"
|
||
|
#include "dmxloader/dmxloader.h"
|
||
|
#include "tier1/utlbuffer.h"
|
||
|
#include "bitmap/tgawriter.h"
|
||
|
#include "tier3/tier3.h"
|
||
|
#include "istudiorender.h"
|
||
|
#include "../vgui2/src/VPanel.h"
|
||
|
#include "tier2/p4helpers.h"
|
||
|
#include "ivtex.h"
|
||
|
#include "bitmap/tgaloader.h"
|
||
|
|
||
|
// memdbgon must be the last include file in a .cpp file!!!
|
||
|
#include "tier0/memdbgon.h"
|
||
|
|
||
|
|
||
|
using namespace vgui;
|
||
|
|
||
|
static bool SaveTgaAndAddToP4( unsigned char *pImage, ImageFormat imageFormat, int Width, int Height, const char *szDestFilename )
|
||
|
{
|
||
|
|
||
|
// allocate a buffer to write the tga into
|
||
|
int iMaxTGASize = 1024 + (Width * Height * 4);
|
||
|
void *pTGA = malloc( iMaxTGASize );
|
||
|
CUtlBuffer buffer( pTGA, iMaxTGASize );
|
||
|
|
||
|
if( !TGAWriter::WriteToBuffer( pImage, buffer, Width, Height, imageFormat, IMAGE_FORMAT_BGRA8888 ) )
|
||
|
{
|
||
|
Error( "Couldn't write bitmap data snapshot.\n" );
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
CP4AutoEditAddFile autop4( szDestFilename );
|
||
|
|
||
|
// async write to disk (this will take ownership of the memory)
|
||
|
char szDirName[ _MAX_PATH ];
|
||
|
strcpy( szDirName, szDestFilename );
|
||
|
V_StripFilename( szDirName );
|
||
|
g_pFullFileSystem->CreateDirHierarchy( szDirName, "" );
|
||
|
g_pFullFileSystem->AsyncWrite( szDestFilename, buffer.Base(), buffer.TellPut(), true );
|
||
|
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
//
|
||
|
// MDL Picker
|
||
|
//
|
||
|
//-----------------------------------------------------------------------------
|
||
|
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
// Sort by MDL name
|
||
|
//-----------------------------------------------------------------------------
|
||
|
static int __cdecl MDLBrowserSortFunc( vgui::ListPanel *pPanel, const ListPanelItem &item1, const ListPanelItem &item2 )
|
||
|
{
|
||
|
const char *string1 = item1.kv->GetString("mdl");
|
||
|
const char *string2 = item2.kv->GetString("mdl");
|
||
|
return stricmp( string1, string2 );
|
||
|
}
|
||
|
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
// Purpose: Constructor
|
||
|
//-----------------------------------------------------------------------------
|
||
|
CMDLPicker::CMDLPicker( vgui::Panel *pParent, int nFlags ) :
|
||
|
BaseClass( pParent, "MDL Files", "mdl", "models", "mdlName" )
|
||
|
{
|
||
|
for( int i = 0; i < MAX_SELECTED_MODELS; i++ )
|
||
|
{
|
||
|
m_hSelectedMDL[ i ] = MDLHANDLE_INVALID;
|
||
|
}
|
||
|
|
||
|
m_nFlags = nFlags; // remember what we show and what not
|
||
|
|
||
|
m_pRenderPage = NULL;
|
||
|
m_pSequencesPage = NULL;
|
||
|
m_pActivitiesPage = NULL;
|
||
|
m_pSkinsPage = NULL;
|
||
|
m_pInfoPage = NULL;
|
||
|
m_pScreenCapsPage = NULL;
|
||
|
|
||
|
m_pSequencesList = NULL;
|
||
|
m_pActivitiesList = NULL;
|
||
|
|
||
|
m_hDirectorySelectDialog = NULL;
|
||
|
|
||
|
// Horizontal splitter for mdls
|
||
|
m_pFileBrowserSplitter = new Splitter( this, "FileBrowserSplitter", SPLITTER_MODE_VERTICAL, 1 );
|
||
|
|
||
|
float flFractions[] = { 0.33f, 0.67f };
|
||
|
|
||
|
m_pFileBrowserSplitter->RespaceSplitters( flFractions );
|
||
|
|
||
|
vgui::Panel *pSplitterLeftSide = m_pFileBrowserSplitter->GetChild( 0 );
|
||
|
vgui::Panel *pSplitterRightSide = m_pFileBrowserSplitter->GetChild( 1 );
|
||
|
|
||
|
// Standard browser controls
|
||
|
pSplitterLeftSide->RequestFocus();
|
||
|
CreateStandardControls( pSplitterLeftSide, false );
|
||
|
|
||
|
// property sheet - revisions, changes, etc.
|
||
|
m_pPreviewSplitter = new Splitter( pSplitterRightSide, "PreviewSplitter", SPLITTER_MODE_HORIZONTAL, 1 );
|
||
|
|
||
|
vgui::Panel *pSplitterTopSide = m_pPreviewSplitter->GetChild( 0 );
|
||
|
vgui::Panel *pSplitterBottomSide = m_pPreviewSplitter->GetChild( 1 );
|
||
|
|
||
|
// MDL preview
|
||
|
m_pMDLPreview = new CMDLPanel( pSplitterTopSide, "MDLPreview" );
|
||
|
SetSkipChildDuringPainting( m_pMDLPreview );
|
||
|
|
||
|
m_pViewsSheet = new vgui::PropertySheet( pSplitterBottomSide, "ViewsSheet" );
|
||
|
m_pViewsSheet->AddActionSignalTarget( this );
|
||
|
|
||
|
// now add wanted features
|
||
|
if ( nFlags & PAGE_RENDER )
|
||
|
{
|
||
|
m_pRenderPage = new vgui::PropertyPage( m_pViewsSheet, "RenderPage" );
|
||
|
|
||
|
m_pRenderPage->AddActionSignalTarget( this );
|
||
|
|
||
|
m_pRenderPage->LoadControlSettingsAndUserConfig( "resource/mdlpickerrender.res" );
|
||
|
|
||
|
RefreshRenderSettings();
|
||
|
|
||
|
// ground
|
||
|
Button *pSelectProbe = (Button*)m_pRenderPage->FindChildByName( "ChooseLightProbe" );
|
||
|
pSelectProbe->AddActionSignalTarget( this );
|
||
|
}
|
||
|
|
||
|
if ( nFlags & PAGE_SEQUENCES )
|
||
|
{
|
||
|
m_pSequencesPage = new vgui::PropertyPage( m_pViewsSheet, "SequencesPage" );
|
||
|
|
||
|
m_pSequencesList = new vgui::ListPanel( m_pSequencesPage, "SequencesList" );
|
||
|
m_pSequencesList->AddColumnHeader( 0, "sequence", "sequence", 52, 0 );
|
||
|
m_pSequencesList->AddActionSignalTarget( this );
|
||
|
m_pSequencesList->SetSelectIndividualCells( true );
|
||
|
m_pSequencesList->SetEmptyListText("No .MDL file currently selected.");
|
||
|
m_pSequencesList->SetDragEnabled( true );
|
||
|
m_pSequencesList->SetAutoResize( Panel::PIN_TOPLEFT, Panel::AUTORESIZE_DOWNANDRIGHT, 6, 6, -6, -6 );
|
||
|
}
|
||
|
|
||
|
if ( nFlags & PAGE_ACTIVITIES )
|
||
|
{
|
||
|
m_pActivitiesPage = new vgui::PropertyPage( m_pViewsSheet, "ActivitiesPage" );
|
||
|
|
||
|
m_pActivitiesList = new vgui::ListPanel( m_pActivitiesPage, "ActivitiesList" );
|
||
|
m_pActivitiesList->AddColumnHeader( 0, "activity", "activity", 52, 0 );
|
||
|
m_pActivitiesList->AddActionSignalTarget( this );
|
||
|
m_pActivitiesList->SetSelectIndividualCells( true );
|
||
|
m_pActivitiesList->SetEmptyListText( "No .MDL file currently selected." );
|
||
|
m_pActivitiesList->SetDragEnabled( true );
|
||
|
m_pActivitiesList->SetAutoResize( Panel::PIN_TOPLEFT, Panel::AUTORESIZE_DOWNANDRIGHT, 6, 6, -6, -6 );
|
||
|
}
|
||
|
|
||
|
if ( nFlags & PAGE_SKINS )
|
||
|
{
|
||
|
m_pSkinsPage = new vgui::PropertyPage( m_pViewsSheet, "SkinsPage" );
|
||
|
|
||
|
m_pSkinsList = new vgui::ListPanel( m_pSkinsPage, "SkinsList" );
|
||
|
m_pSkinsList->AddColumnHeader( 0, "skin", "skin", 52, 0 );
|
||
|
m_pSkinsList->AddActionSignalTarget( this );
|
||
|
m_pSkinsList->SetSelectIndividualCells( true );
|
||
|
m_pSkinsList->SetEmptyListText( "No .MDL file currently selected." );
|
||
|
m_pSkinsList->SetDragEnabled( true );
|
||
|
m_pSkinsList->SetAutoResize( Panel::PIN_TOPLEFT, Panel::AUTORESIZE_DOWNANDRIGHT, 6, 6, -6, -6 );
|
||
|
}
|
||
|
|
||
|
if ( nFlags & PAGE_INFO )
|
||
|
{
|
||
|
m_pInfoPage = new vgui::PropertyPage( m_pViewsSheet, "InfoPage" );
|
||
|
|
||
|
m_pInfoPage->AddActionSignalTarget( this );
|
||
|
|
||
|
m_pInfoPage->LoadControlSettingsAndUserConfig( "resource/mdlpickerinfo.res" );
|
||
|
|
||
|
CheckButton * pTempCheck = (CheckButton *)m_pInfoPage->FindChildByName( "PhysicsObject" );
|
||
|
pTempCheck->SetDisabledFgColor1( pTempCheck->GetFgColor());
|
||
|
pTempCheck->SetDisabledFgColor2( pTempCheck->GetFgColor());
|
||
|
pTempCheck = (CheckButton *)m_pInfoPage->FindChildByName( "StaticObject" );
|
||
|
pTempCheck->SetDisabledFgColor1( pTempCheck->GetFgColor());
|
||
|
pTempCheck->SetDisabledFgColor2( pTempCheck->GetFgColor());
|
||
|
pTempCheck = (CheckButton *)m_pInfoPage->FindChildByName( "DynamicObject" );
|
||
|
pTempCheck->SetDisabledFgColor1( pTempCheck->GetFgColor());
|
||
|
pTempCheck->SetDisabledFgColor2( pTempCheck->GetFgColor());
|
||
|
|
||
|
m_pPropDataList = new vgui::ListPanel( m_pInfoPage, "PropData" );
|
||
|
m_pPropDataList->AddColumnHeader( 0, "key", "key", 250, ListPanel::COLUMN_FIXEDSIZE );
|
||
|
m_pPropDataList->AddColumnHeader( 1, "value", "value", 52, 0 );
|
||
|
m_pPropDataList->AddActionSignalTarget( this );
|
||
|
m_pPropDataList->SetSelectIndividualCells( false );
|
||
|
m_pPropDataList->SetEmptyListText( "No prop_data available." );
|
||
|
m_pPropDataList->SetDragEnabled( true );
|
||
|
m_pPropDataList->SetAutoResize( Panel::PIN_TOPLEFT, Panel::AUTORESIZE_DOWNANDRIGHT, 6, 72, -6, -6 );
|
||
|
|
||
|
RefreshRenderSettings();
|
||
|
}
|
||
|
|
||
|
if ( nFlags & PAGE_SCREEN_CAPS )
|
||
|
{
|
||
|
m_pScreenCapsPage = new vgui::PropertyPage( m_pViewsSheet, "ScreenCapsPage" );
|
||
|
|
||
|
// not sure why we have to do this for the color picker
|
||
|
CColorPickerButton *m_pBackgroundColor;
|
||
|
m_pBackgroundColor = new CColorPickerButton( m_pScreenCapsPage, "BackgroundColor", this );
|
||
|
|
||
|
m_pScreenCapsPage->LoadControlSettingsAndUserConfig( "resource/mdlpickerscreencaps.res" );
|
||
|
|
||
|
TextEntry *pTempValue;
|
||
|
pTempValue = ( TextEntry * )m_pScreenCapsPage->FindChildByName( "WidthText" );
|
||
|
pTempValue->SetText( "256" );
|
||
|
pTempValue = ( TextEntry * )m_pScreenCapsPage->FindChildByName( "HeightText" );
|
||
|
pTempValue->SetText( "256" );
|
||
|
|
||
|
// not sure why this doesn't work
|
||
|
// m_pBackgroundColor = ( CColorPickerButton * )m_pScreenCapsPage->FindChildByName( "BackgroundColor" );
|
||
|
m_pBackgroundColor->SetColor( 255, 0, 0, 255 );
|
||
|
|
||
|
Label *m_pOutputDirectory;
|
||
|
m_pOutputDirectory = ( Label * )m_pScreenCapsPage->FindChildByName( "OutputDirectory" );
|
||
|
m_pOutputDirectory->SetText( "c:\\" );
|
||
|
|
||
|
Button *pSelectProbe = ( Button * )m_pScreenCapsPage->FindChildByName( "Capture" );
|
||
|
pSelectProbe->AddActionSignalTarget( this );
|
||
|
pSelectProbe = ( Button * )m_pScreenCapsPage->FindChildByName( "OutputDirectorySelect" );
|
||
|
pSelectProbe->AddActionSignalTarget( this );
|
||
|
pSelectProbe = ( Button * )m_pScreenCapsPage->FindChildByName( "SaveCaps" );
|
||
|
pSelectProbe->AddActionSignalTarget( this );
|
||
|
pSelectProbe = ( Button * )m_pScreenCapsPage->FindChildByName( "RestoreCaps" );
|
||
|
pSelectProbe->AddActionSignalTarget( this );
|
||
|
pSelectProbe = ( Button * )m_pScreenCapsPage->FindChildByName( "GenerateBackpackIcons" );
|
||
|
pSelectProbe->AddActionSignalTarget( this );
|
||
|
}
|
||
|
|
||
|
// Load layout settings; has to happen before pinning occurs in code
|
||
|
LoadControlSettingsAndUserConfig( "resource/mdlpicker.res" );
|
||
|
|
||
|
// Pages must be added after control settings are set up
|
||
|
if ( m_pRenderPage )
|
||
|
{
|
||
|
m_pViewsSheet->AddPage( m_pRenderPage, "Render" );
|
||
|
}
|
||
|
if ( m_pSequencesPage )
|
||
|
{
|
||
|
m_pViewsSheet->AddPage( m_pSequencesPage, "Sequences" );
|
||
|
}
|
||
|
if ( m_pActivitiesPage )
|
||
|
{
|
||
|
m_pViewsSheet->AddPage( m_pActivitiesPage, "Activities" );
|
||
|
}
|
||
|
if ( m_pSkinsPage )
|
||
|
{
|
||
|
m_pViewsSheet->AddPage( m_pSkinsPage, "Skins" );
|
||
|
}
|
||
|
if ( m_pInfoPage )
|
||
|
{
|
||
|
m_pViewsSheet->AddPage( m_pInfoPage, "Info" );
|
||
|
}
|
||
|
if ( m_pScreenCapsPage )
|
||
|
{
|
||
|
m_pViewsSheet->AddPage( m_pScreenCapsPage, "Screen Caps" );
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void CMDLPicker::RefreshRenderSettings()
|
||
|
{
|
||
|
vgui::CheckButton *pToggle;
|
||
|
|
||
|
if ( !m_pRenderPage )
|
||
|
return;
|
||
|
|
||
|
// ground
|
||
|
pToggle = (vgui::CheckButton*)m_pRenderPage->FindChildByName("NoGround");
|
||
|
pToggle->AddActionSignalTarget( this );
|
||
|
m_pMDLPreview->SetGroundGrid( !pToggle->IsSelected() );
|
||
|
|
||
|
// collision
|
||
|
pToggle = (vgui::CheckButton*)m_pRenderPage->FindChildByName("Collision");
|
||
|
pToggle->AddActionSignalTarget( this );
|
||
|
m_pMDLPreview->SetCollsionModel( pToggle->IsSelected() );
|
||
|
|
||
|
// wireframe
|
||
|
pToggle = (vgui::CheckButton*)m_pRenderPage->FindChildByName("Wireframe");
|
||
|
pToggle->AddActionSignalTarget( this );
|
||
|
m_pMDLPreview->SetWireFrame( pToggle->IsSelected() );
|
||
|
|
||
|
// lockview
|
||
|
pToggle = (vgui::CheckButton*)m_pRenderPage->FindChildByName("LockView");
|
||
|
pToggle->AddActionSignalTarget( this );
|
||
|
m_pMDLPreview->SetLockView( pToggle->IsSelected() );
|
||
|
|
||
|
// look at camera
|
||
|
pToggle = (vgui::CheckButton*)m_pRenderPage->FindChildByName("LookAtCamera");
|
||
|
pToggle->AddActionSignalTarget( this );
|
||
|
m_pMDLPreview->SetLookAtCamera( pToggle->IsSelected() );
|
||
|
|
||
|
// thumbnail safe zone
|
||
|
pToggle = (vgui::CheckButton*)m_pRenderPage->FindChildByName("ThumbnailSafeZone");
|
||
|
pToggle->AddActionSignalTarget( this );
|
||
|
m_pMDLPreview->SetThumbnailSafeZone( pToggle->IsSelected() );
|
||
|
}
|
||
|
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
// Purpose: Destructor
|
||
|
//-----------------------------------------------------------------------------
|
||
|
CMDLPicker::~CMDLPicker()
|
||
|
{
|
||
|
}
|
||
|
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
// Performs layout
|
||
|
//-----------------------------------------------------------------------------
|
||
|
void CMDLPicker::PerformLayout()
|
||
|
{
|
||
|
// NOTE: This call should cause auto-resize to occur
|
||
|
// which should fix up the width of the panels
|
||
|
BaseClass::PerformLayout();
|
||
|
|
||
|
int w, h;
|
||
|
GetSize( w, h );
|
||
|
|
||
|
// Layout the mdl splitter
|
||
|
m_pFileBrowserSplitter->SetBounds( 0, 0, w, h );
|
||
|
}
|
||
|
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
// Buttons on various pages
|
||
|
//-----------------------------------------------------------------------------
|
||
|
void CMDLPicker::OnAssetSelected( KeyValues *pParams )
|
||
|
{
|
||
|
const char *pAsset = pParams->GetString( "asset" );
|
||
|
|
||
|
char pProbeBuf[MAX_PATH];
|
||
|
Q_snprintf( pProbeBuf, sizeof(pProbeBuf), "materials/lightprobes/%s", pAsset );
|
||
|
|
||
|
BeginDMXContext();
|
||
|
CDmxElement *pLightProbe = NULL;
|
||
|
bool bOk = UnserializeDMX( pProbeBuf, "GAME", true, &pLightProbe );
|
||
|
if ( !pLightProbe || !bOk )
|
||
|
{
|
||
|
char pBuf[1024];
|
||
|
Q_snprintf( pBuf, sizeof(pBuf), "Error loading lightprobe file '%s'!\n", pProbeBuf );
|
||
|
vgui::MessageBox *pMessageBox = new vgui::MessageBox( "Error Loading File!\n", pBuf, GetParent() );
|
||
|
pMessageBox->DoModal( );
|
||
|
|
||
|
EndDMXContext( true );
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
m_pMDLPreview->SetLightProbe( pLightProbe );
|
||
|
EndDMXContext( true );
|
||
|
}
|
||
|
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
// Buttons on various pages
|
||
|
//-----------------------------------------------------------------------------
|
||
|
void CMDLPicker::OnCommand( const char *pCommand )
|
||
|
{
|
||
|
if ( !Q_stricmp( pCommand, "ChooseLightProbe" ) )
|
||
|
{
|
||
|
CAssetPickerFrame *pPicker = new CAssetPickerFrame( this, "Select Light Probe (.prb) File",
|
||
|
"Light Probe", "prb", "materials/lightprobes", "lightprobe" );
|
||
|
pPicker->DoModal();
|
||
|
return;
|
||
|
}
|
||
|
else if ( !Q_stricmp( pCommand, "OutputDirectorySelect" ) )
|
||
|
{
|
||
|
if ( !m_hDirectorySelectDialog.Get() )
|
||
|
{
|
||
|
m_hDirectorySelectDialog = new DirectorySelectDialog( this, "Choose Screen Caps output folder" );
|
||
|
}
|
||
|
|
||
|
Label *m_pOutputDirectory;
|
||
|
char temp[ MAX_PATH ];
|
||
|
m_pOutputDirectory = ( Label * )m_pScreenCapsPage->FindChildByName( "OutputDirectory" );
|
||
|
m_pOutputDirectory->GetText( temp, sizeof( temp ) );
|
||
|
|
||
|
m_hDirectorySelectDialog->MakeReadyForUse();
|
||
|
m_hDirectorySelectDialog->SetStartDirectory( temp );
|
||
|
m_hDirectorySelectDialog->DoModal();
|
||
|
return;
|
||
|
}
|
||
|
else if ( !Q_stricmp( pCommand, "Capture" ) )
|
||
|
{
|
||
|
CaptureScreenCaps();
|
||
|
return;
|
||
|
}
|
||
|
else if ( !Q_stricmp( pCommand, "GenerateBackpackIcons" ) )
|
||
|
{
|
||
|
// shut off the ground grid
|
||
|
vgui::CheckButton *pGroundToggle = (vgui::CheckButton *)m_pRenderPage->FindChildByName( "NoGround" );
|
||
|
bool bOriginalGridState = pGroundToggle->IsSelected();
|
||
|
|
||
|
vgui::CheckButton *pSafeZoneToggle = (vgui::CheckButton *)m_pRenderPage->FindChildByName( "ThumbnailSafeZone" );
|
||
|
bool bOriginalSafeZoneState = pSafeZoneToggle->IsSelected();
|
||
|
|
||
|
m_pMDLPreview->SetGroundGrid( false );
|
||
|
m_pMDLPreview->SetThumbnailSafeZone( false );
|
||
|
|
||
|
// make the icons
|
||
|
GenerateBackpackIcons();
|
||
|
|
||
|
// return the ground grid to its original state
|
||
|
m_pMDLPreview->SetGroundGrid( !bOriginalGridState );
|
||
|
m_pMDLPreview->SetThumbnailSafeZone( bOriginalSafeZoneState );
|
||
|
|
||
|
return;
|
||
|
}
|
||
|
else if ( !Q_stricmp( pCommand, "SaveCaps" ) )
|
||
|
{
|
||
|
SaveCaps( NULL );
|
||
|
return;
|
||
|
}
|
||
|
else if ( !Q_stricmp( pCommand, "RestoreCaps" ) )
|
||
|
{
|
||
|
if ( input()->IsKeyDown( KEY_RCONTROL ) || input()->IsKeyDown( KEY_LCONTROL ) )
|
||
|
{
|
||
|
int nCount = m_AssetList.Count();
|
||
|
for ( int i = 0; i < nCount; ++i )
|
||
|
{
|
||
|
if ( m_pAssetBrowser->IsItemVisible( m_AssetList[ i ].m_nItemId ) &&
|
||
|
m_pAssetBrowser->IsItemSelected( m_AssetList[ i ].m_nItemId ) )
|
||
|
{
|
||
|
KeyValues *pItemKeyValues = m_pAssetBrowser->GetItem( m_AssetList[ i ].m_nItemId );
|
||
|
const char *pSelectedAsset = pItemKeyValues->GetString( "asset" );
|
||
|
|
||
|
char szBathPath[ _MAX_PATH ];
|
||
|
Label *m_pOutputDirectory;
|
||
|
|
||
|
m_pOutputDirectory = ( Label * )m_pScreenCapsPage->FindChildByName( "OutputDirectory" );
|
||
|
m_pOutputDirectory->GetText( szBathPath, sizeof( szBathPath ) );
|
||
|
|
||
|
char szPathedFileName[ _MAX_PATH ];
|
||
|
sprintf( szPathedFileName, "%s%s", szBathPath, pSelectedAsset );
|
||
|
|
||
|
Label *m_pResults = ( Label * )m_pScreenCapsPage->FindChildByName( "CaptureResults" );
|
||
|
if ( RestoreCaps( szPathedFileName ) )
|
||
|
{
|
||
|
m_pResults->SetText( "Prefs Restored" );
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
m_pResults->SetText( "Prefs NOT FOUND" );
|
||
|
}
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
RestoreCaps( NULL );
|
||
|
}
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
BaseClass::OnCommand( pCommand );
|
||
|
}
|
||
|
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
// Handles the directory selection for screen caps
|
||
|
//-----------------------------------------------------------------------------
|
||
|
void CMDLPicker::OnDirectorySelected( char const *dir )
|
||
|
{
|
||
|
if ( m_hDirectorySelectDialog != 0 )
|
||
|
{
|
||
|
m_hDirectorySelectDialog->MarkForDeletion();
|
||
|
}
|
||
|
|
||
|
Label *m_pOutputDirectory;
|
||
|
m_pOutputDirectory = ( Label * )m_pScreenCapsPage->FindChildByName( "OutputDirectory" );
|
||
|
m_pOutputDirectory->SetText( dir );
|
||
|
}
|
||
|
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
// Screen captures the specific model and writes out a .tga. Assumes the MDLPreview
|
||
|
// panel has been properly adjusted to 0,0 in screen space and that width / height
|
||
|
// have been set.
|
||
|
//-----------------------------------------------------------------------------
|
||
|
const char *CMDLPicker::CaptureModel( int nModIndex, const char *AssetName, const char *OutputPath, int Width, int Height, Color BackgroundColor, bool bSelectedOnly )
|
||
|
{
|
||
|
char pBuf[ MAX_PATH ];
|
||
|
Q_snprintf( pBuf, sizeof( pBuf ), "%s\\%s\\%s", GetModPath( nModIndex ), m_pAssetSubDir, AssetName );
|
||
|
Q_FixSlashes( pBuf );
|
||
|
|
||
|
if ( !bSelectedOnly )
|
||
|
{
|
||
|
SelectMDL( pBuf, false, -1 );
|
||
|
}
|
||
|
|
||
|
CMatRenderContextPtr pRenderContext( materials );
|
||
|
|
||
|
g_pMaterialSystem->BeginFrame( 0 );
|
||
|
g_pStudioRender->BeginFrame();
|
||
|
|
||
|
// pRenderContext->ClearColor4ub( 0, 0, 0, 0 );
|
||
|
// pRenderContext->ClearBuffers( true, true );
|
||
|
|
||
|
Color NewPanelColor;
|
||
|
|
||
|
NewPanelColor.SetColor( 0, 0, 0, 0 );
|
||
|
m_pMDLPreview->SetBackgroundColor( NewPanelColor );
|
||
|
|
||
|
g_pVGuiSurface->PaintTraverseEx( m_pMDLPreview->GetVPanel(), false );
|
||
|
|
||
|
g_pStudioRender->EndFrame();
|
||
|
g_pMaterialSystem->EndFrame( );
|
||
|
|
||
|
// get the data from the backbuffer and save to disk
|
||
|
// bitmap bits
|
||
|
unsigned char *pImageBlack = ( unsigned char * )malloc( Width * 4 * Height );
|
||
|
|
||
|
// Get Bits from the material system
|
||
|
pRenderContext->ReadPixels( 0, 0, Width, Height, pImageBlack, IMAGE_FORMAT_BGRA8888 );
|
||
|
|
||
|
|
||
|
g_pMaterialSystem->BeginFrame( 0 );
|
||
|
g_pStudioRender->BeginFrame();
|
||
|
|
||
|
// pRenderContext->ClearColor4ub( 255, 255, 255, 0 );
|
||
|
// pRenderContext->ClearBuffers( true, true );
|
||
|
|
||
|
NewPanelColor.SetColor( 255, 255, 255, 0 );
|
||
|
m_pMDLPreview->SetBackgroundColor( NewPanelColor );
|
||
|
|
||
|
g_pVGuiSurface->PaintTraverseEx( m_pMDLPreview->GetVPanel(), false );
|
||
|
|
||
|
g_pStudioRender->EndFrame();
|
||
|
g_pMaterialSystem->EndFrame( );
|
||
|
|
||
|
// get the data from the backbuffer and save to disk
|
||
|
// bitmap bits
|
||
|
unsigned char *pImageWhite = ( unsigned char * )malloc( Width * 4 * Height );
|
||
|
|
||
|
// Get Bits from the material system
|
||
|
pRenderContext->ReadPixels( 0, 0, Width, Height, pImageWhite, IMAGE_FORMAT_BGRA8888 );
|
||
|
|
||
|
unsigned char *pBlackPos = pImageBlack;
|
||
|
unsigned char *pWhitePos = pImageWhite;
|
||
|
for( int y = 0; y < Height; y++ )
|
||
|
{
|
||
|
for( int x = 0; x < Width; x++, pBlackPos += 4, pWhitePos += 4 )
|
||
|
{
|
||
|
if ( ( *( pBlackPos + 0 ) ) != ( *( pWhitePos + 0 ) ) || // blue
|
||
|
( *( pBlackPos + 1 ) ) != ( *( pWhitePos + 1 ) ) || // green
|
||
|
( *( pBlackPos + 2 ) ) != ( *( pWhitePos + 2 ) ) ) // red
|
||
|
{
|
||
|
unsigned char nBlueDiff = ( *( pBlackPos + 0 ) );
|
||
|
unsigned char nGreenDiff = ( *( pBlackPos + 1 ) );
|
||
|
unsigned char nRedDiff = ( *( pBlackPos + 2 ) );
|
||
|
|
||
|
unsigned char nMax = nBlueDiff;
|
||
|
if ( nGreenDiff > nMax )
|
||
|
{
|
||
|
nMax = nGreenDiff;
|
||
|
}
|
||
|
if ( nRedDiff > nMax )
|
||
|
{
|
||
|
nMax = nRedDiff;
|
||
|
}
|
||
|
|
||
|
*( pBlackPos + 3 ) = nMax;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
*( pBlackPos + 3 ) = 0xff;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static char szPathedFileName[ _MAX_PATH ];
|
||
|
sprintf( szPathedFileName, "%s%s", OutputPath, AssetName );
|
||
|
V_SetExtension( szPathedFileName, ".tga", sizeof( szPathedFileName ) );
|
||
|
|
||
|
bool bResult = SaveTgaAndAddToP4( pImageBlack, IMAGE_FORMAT_BGRA8888, Width, Height, szPathedFileName );
|
||
|
|
||
|
free( pImageBlack );
|
||
|
free( pImageWhite );
|
||
|
|
||
|
if ( !bResult) return NULL;
|
||
|
|
||
|
if ( bSelectedOnly )
|
||
|
{
|
||
|
SaveCaps( szPathedFileName );
|
||
|
}
|
||
|
|
||
|
return szPathedFileName;
|
||
|
}
|
||
|
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
// Will go through the asset browser and capture each visible item.
|
||
|
//-----------------------------------------------------------------------------
|
||
|
void CMDLPicker::CaptureScreenCaps( void )
|
||
|
{
|
||
|
char temp[ 256 ];
|
||
|
TextEntry *pTempValue;
|
||
|
int width;
|
||
|
int height;
|
||
|
char szBathPath[ _MAX_PATH ];
|
||
|
Label *m_pOutputDirectory;
|
||
|
|
||
|
m_pOutputDirectory = ( Label * )m_pScreenCapsPage->FindChildByName( "OutputDirectory" );
|
||
|
m_pOutputDirectory->GetText( szBathPath, sizeof( szBathPath ) );
|
||
|
|
||
|
pTempValue = ( TextEntry * )m_pScreenCapsPage->FindChildByName( "WidthText" );
|
||
|
pTempValue->GetText( temp, sizeof( temp ) );
|
||
|
width = atoi( temp );
|
||
|
pTempValue = ( TextEntry * )m_pScreenCapsPage->FindChildByName( "HeightText" );
|
||
|
pTempValue->GetText( temp, sizeof( temp ) );
|
||
|
height = atoi( temp );
|
||
|
|
||
|
int PanelX, PanelY, PanelWidth, PanelHeight;
|
||
|
Color PanelColor;
|
||
|
|
||
|
Panel *pParent = m_pMDLPreview->GetParent();
|
||
|
m_pMDLPreview->GetPos( PanelX, PanelY );
|
||
|
m_pMDLPreview->GetSize( PanelWidth, PanelHeight );
|
||
|
PanelColor = m_pMDLPreview->GetBackgroundColor();
|
||
|
|
||
|
m_pMDLPreview->SetParent( ( vgui::Panel * )NULL );
|
||
|
m_pMDLPreview->SetPos( 0, 0 );
|
||
|
m_pMDLPreview->SetSize( width, height );
|
||
|
|
||
|
CColorPickerButton *m_pBackgroundColor;
|
||
|
m_pBackgroundColor = ( CColorPickerButton * )m_pScreenCapsPage->FindChildByName( "BackgroundColor" );
|
||
|
|
||
|
Color NewPanelColor = m_pBackgroundColor->GetColor();
|
||
|
NewPanelColor[3] = 0;
|
||
|
m_pMDLPreview->SetBackgroundColor( NewPanelColor );
|
||
|
((VPanel *)m_pMDLPreview->GetVPanel())->Solve();
|
||
|
|
||
|
bool bSelectedOnly = false;
|
||
|
if ( input()->IsKeyDown( KEY_RCONTROL ) || input()->IsKeyDown( KEY_LCONTROL ) )
|
||
|
{
|
||
|
bSelectedOnly = true;
|
||
|
}
|
||
|
|
||
|
int nCount = m_AssetList.Count();
|
||
|
int nNumItems = 0;
|
||
|
for ( int i = 0; i < nCount; ++i )
|
||
|
{
|
||
|
if ( m_pAssetBrowser->IsItemVisible( m_AssetList[ i ].m_nItemId ) &&
|
||
|
( !bSelectedOnly || m_pAssetBrowser->IsItemSelected( m_AssetList[ i ].m_nItemId ) ) )
|
||
|
{
|
||
|
KeyValues *pItemKeyValues = m_pAssetBrowser->GetItem( m_AssetList[ i ].m_nItemId );
|
||
|
const char *pSelectedAsset = pItemKeyValues->GetString( "asset" );
|
||
|
int nModIndex = pItemKeyValues->GetInt( "modIndex" );
|
||
|
|
||
|
CaptureModel( nModIndex, pSelectedAsset, szBathPath, width, height, NewPanelColor, bSelectedOnly );
|
||
|
nNumItems++;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
m_pMDLPreview->SetParent( pParent );
|
||
|
m_pMDLPreview->SetPos( PanelX, PanelY );
|
||
|
m_pMDLPreview->SetSize( PanelWidth, PanelHeight );
|
||
|
m_pMDLPreview->SetBackgroundColor( PanelColor );
|
||
|
((VPanel *)m_pMDLPreview->GetVPanel())->Solve();
|
||
|
|
||
|
Label *m_pResults;
|
||
|
|
||
|
sprintf( temp, "Captured %d items", nNumItems );
|
||
|
m_pResults = ( Label * )m_pScreenCapsPage->FindChildByName( "CaptureResults" );
|
||
|
m_pResults->SetText( temp );
|
||
|
}
|
||
|
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
// Stub for XBox360 compiles
|
||
|
//-----------------------------------------------------------------------------
|
||
|
#if defined( _X360 )
|
||
|
const char *getenv( const char *varname )
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
#endif
|
||
|
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
// Writes two very simple .vmt file, one for the passed in asset,
|
||
|
// and the other for <asset>_large.
|
||
|
//-----------------------------------------------------------------------------
|
||
|
void CMDLPicker::WriteBackbackVMTFiles( const char *pAssetName )
|
||
|
{
|
||
|
const char *pVProject = getenv( "VPROJECT" );
|
||
|
if ( !pVProject )
|
||
|
return;
|
||
|
|
||
|
char pStrippedAssetName[ MAX_PATH ];
|
||
|
V_StripExtension( pAssetName, pStrippedAssetName, sizeof( pStrippedAssetName ) );
|
||
|
V_strcat_safe( pStrippedAssetName, GetOutputFileSuffix().Get() );
|
||
|
|
||
|
char pVMTFilename[ MAX_PATH ];
|
||
|
Q_snprintf( pVMTFilename, sizeof( pVMTFilename ), "%s\\materials\\backpack\\%s.vmt", pVProject, pStrippedAssetName );
|
||
|
Q_FixSlashes( pVMTFilename );
|
||
|
|
||
|
char pBaseTextureName[ MAX_PATH ];
|
||
|
Q_snprintf( pBaseTextureName, sizeof( pBaseTextureName ), "backpack\\%s", pStrippedAssetName );
|
||
|
Q_FixSlashes( pBaseTextureName );
|
||
|
|
||
|
{
|
||
|
CP4AutoEditAddFile autop4( pVMTFilename );
|
||
|
FileHandle_t fileHandle = g_pFullFileSystem->Open( pVMTFilename, "w" );
|
||
|
if ( fileHandle )
|
||
|
{
|
||
|
|
||
|
g_pFullFileSystem->FPrintf( fileHandle, "\"UnlitGeneric\"\n" );
|
||
|
g_pFullFileSystem->FPrintf( fileHandle, "{\n" );
|
||
|
g_pFullFileSystem->FPrintf( fileHandle, " \"$baseTexture\" \"%s\"\n", pBaseTextureName );
|
||
|
g_pFullFileSystem->FPrintf( fileHandle, " $translucent 1\n" );
|
||
|
g_pFullFileSystem->FPrintf( fileHandle, " $vertexcolor 1\n" );
|
||
|
g_pFullFileSystem->FPrintf( fileHandle, "}\n" );
|
||
|
|
||
|
g_pFullFileSystem->Close( fileHandle );
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// now write the _large version
|
||
|
Q_snprintf( pVMTFilename, sizeof( pVMTFilename ), "%s\\materials\\backpack\\%s_large", pVProject, pStrippedAssetName );
|
||
|
V_SetExtension( pVMTFilename, ".vmt", sizeof( pVMTFilename ) );
|
||
|
Q_FixSlashes( pVMTFilename );
|
||
|
|
||
|
Q_snprintf( pBaseTextureName, sizeof( pBaseTextureName ), "backpack\\%s_large", pStrippedAssetName );
|
||
|
Q_FixSlashes( pBaseTextureName );
|
||
|
|
||
|
{
|
||
|
CP4AutoEditAddFile autop4( pVMTFilename );
|
||
|
FileHandle_t fileHandle = g_pFullFileSystem->Open( pVMTFilename, "w" );
|
||
|
if ( fileHandle )
|
||
|
{
|
||
|
g_pFullFileSystem->FPrintf( fileHandle, "\"UnlitGeneric\"\n" );
|
||
|
g_pFullFileSystem->FPrintf( fileHandle, "{\n" );
|
||
|
g_pFullFileSystem->FPrintf( fileHandle, " \"$baseTexture\" \"%s\"\n", pBaseTextureName );
|
||
|
g_pFullFileSystem->FPrintf( fileHandle, " $translucent 1\n" );
|
||
|
g_pFullFileSystem->FPrintf( fileHandle, "}\n" );
|
||
|
|
||
|
g_pFullFileSystem->Close( fileHandle );
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
void *VTexFilesystemFactory( const char *pName, int *pReturnCode )
|
||
|
{
|
||
|
return g_pFullFileSystem;
|
||
|
}
|
||
|
|
||
|
|
||
|
void* MdlPickerFSFactory( const char *pName, int *pReturnCode )
|
||
|
{
|
||
|
if ( IsX360() )
|
||
|
return NULL;
|
||
|
|
||
|
if ( Q_stricmp( pName, FILESYSTEM_INTERFACE_VERSION ) == 0 )
|
||
|
return g_pFullFileSystem;
|
||
|
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
// Creates the two required icons for the TF2 store/backpack system
|
||
|
//-----------------------------------------------------------------------------
|
||
|
void CMDLPicker::GenerateBackpackIcons( void )
|
||
|
{
|
||
|
if ( !g_pVTex )
|
||
|
return;
|
||
|
|
||
|
int width;
|
||
|
int height;
|
||
|
|
||
|
// find the index of the item we are currently viewing
|
||
|
int selectedItemIndex;
|
||
|
for ( selectedItemIndex = 0; selectedItemIndex < m_AssetList.Count(); ++selectedItemIndex )
|
||
|
{
|
||
|
if ( m_pAssetBrowser->IsItemVisible( m_AssetList[ selectedItemIndex ].m_nItemId ) &&
|
||
|
m_pAssetBrowser->IsItemSelected( m_AssetList[ selectedItemIndex ].m_nItemId ) )
|
||
|
{
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if ( selectedItemIndex >= m_AssetList.Count() )
|
||
|
return;
|
||
|
|
||
|
//
|
||
|
// Fetch and check environment variables
|
||
|
//
|
||
|
const char *pVContent = getenv( "VCONTENT" );
|
||
|
if ( !pVContent )
|
||
|
{
|
||
|
Error( "VCONTENT environment variable not set" );
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
const char *pVMod = getenv( "VMOD" );
|
||
|
if ( !pVMod )
|
||
|
{
|
||
|
Error( "VMOD environment variable not set" );
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
const char *pVProject = getenv( "VPROJECT" );
|
||
|
if ( !pVProject )
|
||
|
{
|
||
|
Error( "VPROJECT environment variable not set" );
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
// extract the filename of the model
|
||
|
KeyValues *pItemKeyValues = m_pAssetBrowser->GetItem( m_AssetList[ selectedItemIndex ].m_nItemId );
|
||
|
const char *pSelectedAsset = pItemKeyValues->GetString( "asset" );
|
||
|
|
||
|
// set the P4 changelist label to refer to this set of icons
|
||
|
char pChangelistLabel[ MAX_PATH ];
|
||
|
V_strcpy_safe( pChangelistLabel, pSelectedAsset );
|
||
|
V_FileBase( pChangelistLabel, pChangelistLabel, sizeof( pChangelistLabel ) );
|
||
|
V_strcat_safe( pChangelistLabel, " Auto Checkout", sizeof( pChangelistLabel ) );
|
||
|
g_p4factory->SetOpenFileChangeList( pChangelistLabel );
|
||
|
|
||
|
// generate .VMT files for normal and _large icons
|
||
|
WriteBackbackVMTFiles( pSelectedAsset );
|
||
|
|
||
|
// store original state of model preview panel
|
||
|
int PanelX, PanelY, PanelWidth, PanelHeight;
|
||
|
Color PanelColor;
|
||
|
|
||
|
Panel *pParent = m_pMDLPreview->GetParent();
|
||
|
m_pMDLPreview->GetPos( PanelX, PanelY );
|
||
|
m_pMDLPreview->GetSize( PanelWidth, PanelHeight );
|
||
|
PanelColor = m_pMDLPreview->GetBackgroundColor();
|
||
|
|
||
|
// slam preview panel to desired TGA size for large icon, and write it out
|
||
|
width = 512;
|
||
|
height = 512;
|
||
|
m_pMDLPreview->SetParent( ( vgui::Panel * )NULL );
|
||
|
m_pMDLPreview->SetPos( 0, 0 );
|
||
|
m_pMDLPreview->SetSize( width, height );
|
||
|
|
||
|
CColorPickerButton *m_pBackgroundColor;
|
||
|
m_pBackgroundColor = ( CColorPickerButton * )m_pScreenCapsPage->FindChildByName( "BackgroundColor" );
|
||
|
|
||
|
Color NewPanelColor = m_pBackgroundColor->GetColor();
|
||
|
NewPanelColor[3] = 0;
|
||
|
m_pMDLPreview->SetBackgroundColor( NewPanelColor );
|
||
|
((VPanel *)m_pMDLPreview->GetVPanel())->Solve();
|
||
|
|
||
|
char pLargeAssetName[ MAX_PATH ];
|
||
|
V_strcpy_safe( pLargeAssetName, pSelectedAsset );
|
||
|
V_StripExtension( pLargeAssetName, pLargeAssetName, ARRAYSIZE( pLargeAssetName ) );
|
||
|
|
||
|
CUtlString strExtention = GetOutputFileSuffix();
|
||
|
strExtention += "_large.mdl";
|
||
|
|
||
|
V_strcat_safe( pLargeAssetName, strExtention.String() );
|
||
|
|
||
|
char pOutputPath[ MAX_PATH ];
|
||
|
Q_snprintf( pOutputPath, sizeof( pOutputPath ), "%s\\%s\\materialsrc\\backpack\\", pVContent, pVMod );
|
||
|
Q_FixSlashes( pOutputPath );
|
||
|
|
||
|
int nModIndex = pItemKeyValues->GetInt( "modIndex" );
|
||
|
const char *pLargeTGAName = CaptureModel( nModIndex, pLargeAssetName, pOutputPath, width, height, NewPanelColor, true );
|
||
|
if ( !pLargeTGAName )
|
||
|
return;
|
||
|
|
||
|
// write corresponding .txt file with vtex options
|
||
|
char pVTexOptionsFileName[ MAX_PATH ];
|
||
|
V_strcpy_safe( pVTexOptionsFileName, pLargeTGAName );
|
||
|
V_SetExtension( pVTexOptionsFileName, ".txt", sizeof( pVTexOptionsFileName ) );
|
||
|
|
||
|
{
|
||
|
CP4AutoEditAddFile autop4( pVTexOptionsFileName );
|
||
|
FileHandle_t hVTexOptionsFile = g_pFullFileSystem->Open( pVTexOptionsFileName, "w" );
|
||
|
if ( hVTexOptionsFile )
|
||
|
{
|
||
|
g_pFullFileSystem->FPrintf( hVTexOptionsFile, "nomip 1\n" );
|
||
|
g_pFullFileSystem->FPrintf( hVTexOptionsFile, "nolod 1\n" );
|
||
|
g_pFullFileSystem->Close( hVTexOptionsFile );
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// !KLUDGE! Everybody I've talked to says that vtex is *supposed* to
|
||
|
// use VPROJECT. But it doesn't. I don't think I can safely change vtex
|
||
|
// without breaking tons of stuff. So just force the output directory.
|
||
|
// Determine the proper output directory based on VPROJECT
|
||
|
char pOutputPathGame[ MAX_PATH ];
|
||
|
Q_strncpy( pOutputPathGame, pVProject, sizeof( pOutputPathGame ) );
|
||
|
Q_StripTrailingSlash( pOutputPathGame );
|
||
|
Q_strncat( pOutputPathGame, "/materials/", sizeof( pOutputPathGame ) );
|
||
|
const char *pBackpack = Q_stristr( pLargeTGAName, "backpack" );
|
||
|
if ( pBackpack )
|
||
|
{
|
||
|
Q_strncat( pOutputPathGame, pBackpack, sizeof( pOutputPathGame ) );
|
||
|
Q_StripFilename( pOutputPathGame );
|
||
|
}
|
||
|
Q_FixSlashes( pOutputPathGame );
|
||
|
|
||
|
// run vtex on the TGA and .txt file to create .VTF and add it to our Perforce changelist
|
||
|
char *vTexArgv[64];
|
||
|
int vTexArgc = 0;
|
||
|
vTexArgv[ vTexArgc++ ] = "";
|
||
|
vTexArgv[ vTexArgc++ ] = "-quiet";
|
||
|
vTexArgv[ vTexArgc++ ] = "-UseStandardError";
|
||
|
vTexArgv[ vTexArgc++ ] = "-WarningsAsErrors";
|
||
|
vTexArgv[ vTexArgc++ ] = "-p4skip";
|
||
|
vTexArgv[ vTexArgc++ ] = "-outdir";
|
||
|
vTexArgv[ vTexArgc++ ] = pOutputPathGame;
|
||
|
vTexArgv[ vTexArgc++ ] = (char *)pLargeTGAName;
|
||
|
|
||
|
g_pVTex->VTex( MdlPickerFSFactory, pOutputPathGame, vTexArgc, vTexArgv );
|
||
|
|
||
|
// Generale small TGA name, by removing the "large" part
|
||
|
char pSmallTGAName[ MAX_PATH ];
|
||
|
strcpy( pSmallTGAName, pLargeTGAName );
|
||
|
char *_large = Q_stristr( pSmallTGAName, "_large");
|
||
|
Assert(_large);
|
||
|
strcpy(_large, _large+6);
|
||
|
|
||
|
// Load up the large icon
|
||
|
int nCheckWidth, nCheckHeight;
|
||
|
ImageFormat largeFmt;
|
||
|
float gamma;
|
||
|
CUtlBuffer largeTGAFileData;
|
||
|
CUtlMemory<unsigned char> largeTGAImageData;
|
||
|
|
||
|
if ( !g_pFullFileSystem->ReadFile( pLargeTGAName, NULL, largeTGAFileData )
|
||
|
|| !TGALoader::GetInfo( largeTGAFileData, &nCheckWidth, &nCheckHeight, &largeFmt, &gamma )
|
||
|
|| nCheckWidth != width || nCheckHeight != height )
|
||
|
{
|
||
|
Error( "Failed to reload image header %s", pLargeTGAName );
|
||
|
Assert( false );
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
largeTGAFileData.SeekGet( CUtlBuffer::SEEK_HEAD, 0 );
|
||
|
if ( !TGALoader::LoadRGBA8888( largeTGAFileData, largeTGAImageData, nCheckWidth, nCheckHeight )
|
||
|
|| nCheckWidth != width || nCheckHeight != height )
|
||
|
{
|
||
|
Error( "Failed to reload image data %s", pLargeTGAName );
|
||
|
Assert( false );
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Perform a downsample. This is better than just re-rendering at the smaller size,
|
||
|
// which essentially just point-samples the image.
|
||
|
//
|
||
|
CUtlMemory<unsigned char> smallTGAImageData;
|
||
|
const int kSmallSize = 128;
|
||
|
smallTGAImageData.EnsureCapacity(kSmallSize*kSmallSize*4);
|
||
|
ImageLoader::ResampleInfo_t resampleInfo;
|
||
|
resampleInfo.m_nSrcWidth = width;
|
||
|
resampleInfo.m_nSrcHeight = height;
|
||
|
resampleInfo.m_flSrcGamma = gamma;
|
||
|
resampleInfo.m_pSrc = (unsigned char *)largeTGAImageData.Base();
|
||
|
|
||
|
resampleInfo.m_nDestWidth = kSmallSize;
|
||
|
resampleInfo.m_nDestHeight = kSmallSize;
|
||
|
resampleInfo.m_flDestGamma = gamma;
|
||
|
resampleInfo.m_pDest = (unsigned char *)smallTGAImageData.Base();
|
||
|
|
||
|
resampleInfo.m_nFlags = ImageLoader::RESAMPLE_CLAMPS | ImageLoader::RESAMPLE_CLAMPT;
|
||
|
//resampleInfo.m_nFlags |= ImageLoader::RESAMPLE_NICE_FILTER; // Turn this off. It has some sort of edge enhancement or something. Causes edges to ring.
|
||
|
|
||
|
if ( !ImageLoader::ResampleRGBA8888( resampleInfo ) )
|
||
|
{
|
||
|
Error( "Failed to resample %s", pLargeTGAName );
|
||
|
Assert( false );
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
// Save it
|
||
|
if ( !SaveTgaAndAddToP4( resampleInfo.m_pDest, IMAGE_FORMAT_RGBA8888, resampleInfo.m_nDestWidth, resampleInfo.m_nDestHeight, pSmallTGAName ) )
|
||
|
{
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
// Save the .cfg file.
|
||
|
SaveCaps( pSmallTGAName );
|
||
|
|
||
|
// write corresponding .txt file with vtex options
|
||
|
V_strcpy_safe( pVTexOptionsFileName, pSmallTGAName );
|
||
|
V_SetExtension( pVTexOptionsFileName, ".txt", sizeof( pVTexOptionsFileName ) );
|
||
|
|
||
|
{
|
||
|
CP4AutoEditAddFile autop4( pVTexOptionsFileName );
|
||
|
FileHandle_t hVTexOptionsFile = g_pFullFileSystem->Open( pVTexOptionsFileName, "w" );
|
||
|
if ( hVTexOptionsFile )
|
||
|
{
|
||
|
g_pFullFileSystem->FPrintf( hVTexOptionsFile, "nomip 1\n" );
|
||
|
g_pFullFileSystem->FPrintf( hVTexOptionsFile, "nolod 1\n" );
|
||
|
g_pFullFileSystem->Close( hVTexOptionsFile );
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// run vtex on the TGA and .txt file to create .VTF and add it to our Perforce changelist
|
||
|
vTexArgc = 0;
|
||
|
vTexArgv[ vTexArgc++ ] = "";
|
||
|
vTexArgv[ vTexArgc++ ] = "-quiet";
|
||
|
vTexArgv[ vTexArgc++ ] = "-UseStandardError";
|
||
|
vTexArgv[ vTexArgc++ ] = "-WarningsAsErrors";
|
||
|
vTexArgv[ vTexArgc++ ] = "-p4skip";
|
||
|
vTexArgv[ vTexArgc++ ] = "-outdir";
|
||
|
vTexArgv[ vTexArgc++ ] = pOutputPathGame;
|
||
|
vTexArgv[ vTexArgc++ ] = (char *)pSmallTGAName;
|
||
|
g_pVTex->VTex( MdlPickerFSFactory, pOutputPathGame, vTexArgc, vTexArgv );
|
||
|
|
||
|
|
||
|
// restore the preview panel to its original state
|
||
|
m_pMDLPreview->SetParent( pParent );
|
||
|
m_pMDLPreview->SetPos( PanelX, PanelY );
|
||
|
m_pMDLPreview->SetSize( PanelWidth, PanelHeight );
|
||
|
m_pMDLPreview->SetBackgroundColor( PanelColor );
|
||
|
((VPanel *)m_pMDLPreview->GetVPanel())->Solve();
|
||
|
}
|
||
|
|
||
|
|
||
|
CUtlString CMDLPicker::GetOutputFileSuffix()
|
||
|
{
|
||
|
char temp[256];
|
||
|
TextEntry *pTempValue = ( TextEntry * )m_pScreenCapsPage->FindChildByName( "SuffixText" );
|
||
|
if ( pTempValue )
|
||
|
{
|
||
|
pTempValue->GetText( temp, sizeof( temp ) );
|
||
|
}
|
||
|
return temp;
|
||
|
}
|
||
|
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
// Saves the screen cap information and camera position
|
||
|
//-----------------------------------------------------------------------------
|
||
|
void CMDLPicker::SaveCaps( const char *szFileName )
|
||
|
{
|
||
|
char temp[ _MAX_PATH ];
|
||
|
|
||
|
KeyValues *CaptureData = new KeyValues( "ScreenCaps" );
|
||
|
|
||
|
Vector vecPos;
|
||
|
QAngle angDir;
|
||
|
m_pMDLPreview->GetCameraPositionAndAngles( vecPos, angDir );
|
||
|
sprintf( temp, "%g %g %g", vecPos.x, vecPos.y, vecPos.z );
|
||
|
CaptureData->SetString( "CameraPosition", temp );
|
||
|
sprintf( temp, "%g %g %g", angDir.x, angDir.y, angDir.z );
|
||
|
CaptureData->SetString( "CameraAngles", temp );
|
||
|
|
||
|
Vector vecOffset;
|
||
|
m_pMDLPreview->GetCameraOffset( vecOffset );
|
||
|
sprintf( temp, "%g %g %g", vecOffset.x, vecOffset.y, vecOffset.z );
|
||
|
CaptureData->SetString( "CameraOffset", temp );
|
||
|
|
||
|
CColorPickerButton *m_pBackgroundColor;
|
||
|
m_pBackgroundColor = ( CColorPickerButton * )m_pScreenCapsPage->FindChildByName( "BackgroundColor" );
|
||
|
Color color = m_pBackgroundColor->GetColor();
|
||
|
|
||
|
sprintf( temp, "%d %d %d %d", color.r(), color.g(), color.b(), color.a() );
|
||
|
CaptureData->SetString( "BackgroundColor", temp );
|
||
|
|
||
|
TextEntry *pTempValue;
|
||
|
pTempValue = ( TextEntry * )m_pScreenCapsPage->FindChildByName( "WidthText" );
|
||
|
pTempValue->GetText( temp, sizeof( temp ) );
|
||
|
CaptureData->SetString( "Width", temp );
|
||
|
|
||
|
pTempValue = ( TextEntry * )m_pScreenCapsPage->FindChildByName( "HeightText" );
|
||
|
pTempValue->GetText( temp, sizeof( temp ) );
|
||
|
CaptureData->SetString( "Height", temp );
|
||
|
|
||
|
vgui::CheckButton *pToggle;
|
||
|
pToggle = (vgui::CheckButton*)m_pRenderPage->FindChildByName( "NoGround" );
|
||
|
CaptureData->SetInt( "NoGround", pToggle->IsSelected() ? 1 : 0 );
|
||
|
|
||
|
pToggle = (vgui::CheckButton*)m_pRenderPage->FindChildByName( "Collision" );
|
||
|
CaptureData->SetInt( "Collision", pToggle->IsSelected() ? 1 : 0 );
|
||
|
|
||
|
pToggle = (vgui::CheckButton*)m_pRenderPage->FindChildByName( "Wireframe" );
|
||
|
CaptureData->SetInt( "Wifeframe", pToggle->IsSelected() ? 1 : 0 );
|
||
|
|
||
|
pToggle = (vgui::CheckButton*)m_pRenderPage->FindChildByName( "LockView" );
|
||
|
CaptureData->SetInt( "LockView", pToggle->IsSelected() ? 1 : 0 );
|
||
|
|
||
|
pToggle = (vgui::CheckButton*)m_pRenderPage->FindChildByName( "LookAtCamera" );
|
||
|
CaptureData->SetInt( "LookAtCamera", pToggle->IsSelected() ? 1 : 0 );
|
||
|
|
||
|
for( int i = 1; i < MAX_SELECTED_MODELS; i++ )
|
||
|
{
|
||
|
if ( m_hSelectedMDL[ i ] != MDLHANDLE_INVALID )
|
||
|
{
|
||
|
const char *MergedModelName = vgui::MDLCache()->GetModelName( m_hSelectedMDL[ i ] );
|
||
|
sprintf( temp, "Merged_%d", i );
|
||
|
CaptureData->SetString( temp, MergedModelName );
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if ( szFileName != NULL )
|
||
|
{
|
||
|
strcpy( temp, szFileName );
|
||
|
V_SetExtension( temp, ".cfg", sizeof( temp ) );
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
Label *m_pOutputDirectory = ( Label * )m_pScreenCapsPage->FindChildByName( "OutputDirectory" );
|
||
|
m_pOutputDirectory->GetText( temp, sizeof( temp ) );
|
||
|
|
||
|
strcat( temp, "ScreenCaps.cfg" );
|
||
|
}
|
||
|
|
||
|
CaptureData->SaveToFile( g_pFullFileSystem, temp );
|
||
|
CP4AutoAddFile autop4( temp );
|
||
|
}
|
||
|
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
// Restores the screen cap information and camera position
|
||
|
//-----------------------------------------------------------------------------
|
||
|
bool CMDLPicker::RestoreCaps( const char *szFileName )
|
||
|
{
|
||
|
char temp[ _MAX_PATH ];
|
||
|
|
||
|
if ( szFileName != NULL )
|
||
|
{
|
||
|
strcpy( temp, szFileName );
|
||
|
V_SetExtension( temp, ".cfg", sizeof( temp ) );
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
Label *m_pOutputDirectory = ( Label * )m_pScreenCapsPage->FindChildByName( "OutputDirectory" );
|
||
|
m_pOutputDirectory->GetText( temp, sizeof( temp ) );
|
||
|
strcat( temp, "ScreenCaps.cfg" );
|
||
|
}
|
||
|
|
||
|
KeyValues *CaptureData = new KeyValues( "ScreenCaps" );
|
||
|
|
||
|
if ( !CaptureData->LoadFromFile( g_pFullFileSystem, temp ) )
|
||
|
{
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
Vector vecPos;
|
||
|
QAngle angDir;
|
||
|
Vector vecOffset;
|
||
|
sscanf( CaptureData->GetString( "CameraPosition" ), "%g %g %g", &vecPos.x, &vecPos.y, &vecPos.z );
|
||
|
sscanf( CaptureData->GetString( "CameraAngles" ), "%g %g %g", &angDir.x, &angDir.y, &angDir.z );
|
||
|
sscanf( CaptureData->GetString( "CameraOffset" ), "%g %g %g", &vecOffset.x, &vecOffset.y, &vecOffset.z );
|
||
|
|
||
|
m_pMDLPreview->SetCameraOffset( vecOffset );
|
||
|
m_pMDLPreview->SetCameraPositionAndAngles( vecPos, angDir );
|
||
|
|
||
|
CColorPickerButton *m_pBackgroundColor;
|
||
|
int r, g, b, a;
|
||
|
m_pBackgroundColor = ( CColorPickerButton * )m_pScreenCapsPage->FindChildByName( "BackgroundColor" );
|
||
|
sscanf( CaptureData->GetString( "BackgroundColor" ), "%d %d %d %d", &r, &g, &b, &a );
|
||
|
m_pBackgroundColor->SetColor( r, g, b, a );
|
||
|
|
||
|
TextEntry *pTempValue;
|
||
|
pTempValue = ( TextEntry * )m_pScreenCapsPage->FindChildByName( "WidthText" );
|
||
|
pTempValue->SetText( CaptureData->GetString( "Width" ) );
|
||
|
|
||
|
pTempValue = ( TextEntry * )m_pScreenCapsPage->FindChildByName( "HeightText" );
|
||
|
pTempValue->SetText( CaptureData->GetString( "Height" ) );
|
||
|
|
||
|
|
||
|
vgui::CheckButton *pToggle;
|
||
|
pToggle = (vgui::CheckButton*)m_pRenderPage->FindChildByName( "NoGround" );
|
||
|
pToggle->SetSelected( ( CaptureData->GetInt( "NoGround" ) == 1 ) );
|
||
|
|
||
|
pToggle = (vgui::CheckButton*)m_pRenderPage->FindChildByName( "Collision" );
|
||
|
pToggle->SetSelected( ( CaptureData->GetInt( "Collision" ) == 1 ) );
|
||
|
|
||
|
pToggle = (vgui::CheckButton*)m_pRenderPage->FindChildByName( "Wireframe" );
|
||
|
pToggle->SetSelected( ( CaptureData->GetInt( "Wireframe" ) == 1 ) );
|
||
|
|
||
|
pToggle = (vgui::CheckButton*)m_pRenderPage->FindChildByName( "LockView" );
|
||
|
pToggle->SetSelected( ( CaptureData->GetInt( "LockView" ) == 1 ) );
|
||
|
|
||
|
pToggle = (vgui::CheckButton*)m_pRenderPage->FindChildByName( "LookAtCamera" );
|
||
|
pToggle->SetSelected( ( CaptureData->GetInt( "LookAtCamera" ) == 1 ) );
|
||
|
|
||
|
for( int i = 1; i < MAX_SELECTED_MODELS; i++ )
|
||
|
{
|
||
|
sprintf( temp, "Merged_%d", i );
|
||
|
const char *MergedModelName = CaptureData->GetString( temp, NULL );
|
||
|
if ( MergedModelName )
|
||
|
{
|
||
|
SelectMDL( MergedModelName, false, i );
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
// Purpose: rebuilds the list of activities
|
||
|
//-----------------------------------------------------------------------------
|
||
|
void CMDLPicker::RefreshActivitiesAndSequencesList()
|
||
|
{
|
||
|
m_pActivitiesList->RemoveAll();
|
||
|
m_pSequencesList->RemoveAll();
|
||
|
m_pMDLPreview->SetSequence( 0 );
|
||
|
|
||
|
if ( m_hSelectedMDL[ 0 ] == MDLHANDLE_INVALID )
|
||
|
{
|
||
|
m_pActivitiesList->SetEmptyListText("No .MDL file currently selected");
|
||
|
m_pSequencesList->SetEmptyListText("No .MDL file currently selected");
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
m_pActivitiesList->SetEmptyListText(".MDL file contains no activities");
|
||
|
m_pSequencesList->SetEmptyListText(".MDL file contains no sequences");
|
||
|
|
||
|
studiohdr_t *hdr = vgui::MDLCache()->GetStudioHdr( m_hSelectedMDL[ 0 ] );
|
||
|
|
||
|
CUtlDict<int, unsigned short> activityNames( true, 0, hdr->GetNumSeq() );
|
||
|
|
||
|
for (int j = 0; j < hdr->GetNumSeq(); j++)
|
||
|
{
|
||
|
if ( /*g_viewerSettings.showHidden ||*/ !(hdr->pSeqdesc(j).flags & STUDIO_HIDDEN))
|
||
|
{
|
||
|
const char *pActivityName = hdr->pSeqdesc(j).pszActivityName();
|
||
|
if ( pActivityName && pActivityName[0] )
|
||
|
{
|
||
|
// Multiple sequences can have the same activity name; only add unique activity names
|
||
|
if ( activityNames.Find( pActivityName ) == activityNames.InvalidIndex() )
|
||
|
{
|
||
|
KeyValues *pkv = new KeyValues("node", "activity", pActivityName );
|
||
|
int nItemID = m_pActivitiesList->AddItem( pkv, 0, false, false );
|
||
|
|
||
|
KeyValues *pDrag = new KeyValues( "drag", "text", pActivityName );
|
||
|
pDrag->SetString( "texttype", "activityName" );
|
||
|
pDrag->SetString( "mdl", vgui::MDLCache()->GetModelName( m_hSelectedMDL[ 0 ] ) );
|
||
|
m_pActivitiesList->SetItemDragData( nItemID, pDrag );
|
||
|
|
||
|
activityNames.Insert( pActivityName, j );
|
||
|
}
|
||
|
}
|
||
|
|
||
|
const char *pSequenceName = hdr->pSeqdesc(j).pszLabel();
|
||
|
if ( pSequenceName && pSequenceName[0] )
|
||
|
{
|
||
|
KeyValues *pkv = new KeyValues("node", "sequence", pSequenceName);
|
||
|
int nItemID = m_pSequencesList->AddItem( pkv, 0, false, false );
|
||
|
|
||
|
KeyValues *pDrag = new KeyValues( "drag", "text", pSequenceName );
|
||
|
pDrag->SetString( "texttype", "sequenceName" );
|
||
|
pDrag->SetString( "mdl", vgui::MDLCache()->GetModelName( m_hSelectedMDL[ 0 ] ) );
|
||
|
m_pSequencesList->SetItemDragData( nItemID, pDrag );
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
// A MDL was selected
|
||
|
//-----------------------------------------------------------------------------
|
||
|
void CMDLPicker::OnSelectedAssetPicked( const char *pMDLName )
|
||
|
{
|
||
|
char pRelativePath[MAX_PATH];
|
||
|
|
||
|
int nSelectSecondary = -1;
|
||
|
if ( input()->IsKeyDown( KEY_RCONTROL ) || input()->IsKeyDown( KEY_LCONTROL ) )
|
||
|
{
|
||
|
nSelectSecondary = 0;
|
||
|
}
|
||
|
else if ( input()->IsMouseDown(MOUSE_RIGHT) )
|
||
|
{
|
||
|
nSelectSecondary = 1;
|
||
|
}
|
||
|
|
||
|
if ( pMDLName )
|
||
|
{
|
||
|
Q_snprintf( pRelativePath, sizeof(pRelativePath), "models\\%s", pMDLName );
|
||
|
SelectMDL( pRelativePath, true, nSelectSecondary );
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
SelectMDL( NULL, true, nSelectSecondary );
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
// Allows external apps to select a MDL
|
||
|
//-----------------------------------------------------------------------------
|
||
|
void CMDLPicker::SelectMDL( const char *pRelativePath, bool bDoLookAt, int nSelectSecondary )
|
||
|
{
|
||
|
MDLHandle_t hSelectedMDL = pRelativePath ? vgui::MDLCache()->FindMDL( pRelativePath ) : MDLHANDLE_INVALID;
|
||
|
int index = ( nSelectSecondary > 0 ? nSelectSecondary : 0 );
|
||
|
|
||
|
// We didn't change models after all...
|
||
|
if ( hSelectedMDL == m_hSelectedMDL[ index ] )
|
||
|
{
|
||
|
// vgui::MDLCache()->FindMDL adds a reference by default we don't use, release it again
|
||
|
if ( hSelectedMDL != MDLHANDLE_INVALID )
|
||
|
{
|
||
|
vgui::MDLCache()->Release( hSelectedMDL );
|
||
|
}
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
m_hSelectedMDL[ index ] = hSelectedMDL;
|
||
|
|
||
|
if ( vgui::MDLCache()->IsErrorModel( m_hSelectedMDL[ index ] ) )
|
||
|
{
|
||
|
m_hSelectedMDL[ index ] = MDLHANDLE_INVALID;
|
||
|
}
|
||
|
if ( nSelectSecondary != -1 )
|
||
|
{
|
||
|
m_pMDLPreview->ClearMergeMDLs();
|
||
|
for( int i = 1; i < MAX_SELECTED_MODELS; i++ )
|
||
|
{
|
||
|
if ( i != index )
|
||
|
{
|
||
|
m_hSelectedMDL[ i ] = MDLHANDLE_INVALID;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if ( index > 0 )
|
||
|
{
|
||
|
m_pMDLPreview->SetMergeMDL( m_hSelectedMDL[ index ] );
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
m_pMDLPreview->SetMDL( m_hSelectedMDL[ index ] );
|
||
|
|
||
|
if ( bDoLookAt )
|
||
|
{
|
||
|
m_pMDLPreview->LookAtMDL();
|
||
|
}
|
||
|
|
||
|
if ( m_nFlags & ( PAGE_SKINS ) )
|
||
|
{
|
||
|
UpdateSkinsList();
|
||
|
}
|
||
|
|
||
|
if ( m_nFlags & ( PAGE_INFO ) )
|
||
|
{
|
||
|
UpdateInfoTab();
|
||
|
}
|
||
|
|
||
|
if ( m_nFlags & (PAGE_ACTIVITIES|PAGE_SEQUENCES) )
|
||
|
{
|
||
|
RefreshActivitiesAndSequencesList();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// vgui::MDLCache()->FindMDL adds a reference by default we don't use, release it again
|
||
|
if ( hSelectedMDL != MDLHANDLE_INVALID )
|
||
|
{
|
||
|
vgui::MDLCache()->Release( hSelectedMDL );
|
||
|
}
|
||
|
|
||
|
PostActionSignal( new KeyValues( "MDLPreviewChanged", "mdl", pRelativePath ? pRelativePath : "" ) );
|
||
|
}
|
||
|
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
// Purpose: updates revision view on a file being selected
|
||
|
//-----------------------------------------------------------------------------
|
||
|
void CMDLPicker::OnCheckButtonChecked(KeyValues *kv)
|
||
|
{
|
||
|
// RefreshMDLList();
|
||
|
BaseClass::OnCheckButtonChecked( kv );
|
||
|
RefreshRenderSettings();
|
||
|
}
|
||
|
|
||
|
|
||
|
void CMDLPicker::GetSelectedMDLName( char *pBuffer, int nMaxLen )
|
||
|
{
|
||
|
Assert( nMaxLen > 0 );
|
||
|
if ( GetSelectedAssetCount() > 0 )
|
||
|
{
|
||
|
Q_snprintf( pBuffer, nMaxLen, "models\\%s", GetSelectedAsset( ) );
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
pBuffer[0] = 0;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
// Gets the selected activity/sequence
|
||
|
//-----------------------------------------------------------------------------
|
||
|
int CMDLPicker::GetSelectedPage( )
|
||
|
{
|
||
|
if ( m_pSequencesPage && ( m_pViewsSheet->GetActivePage() == m_pSequencesPage ) )
|
||
|
return PAGE_SEQUENCES;
|
||
|
|
||
|
if ( m_pActivitiesPage && ( m_pViewsSheet->GetActivePage() == m_pActivitiesPage ) )
|
||
|
return PAGE_ACTIVITIES;
|
||
|
|
||
|
return PAGE_NONE;
|
||
|
}
|
||
|
|
||
|
const char *CMDLPicker::GetSelectedSequenceName()
|
||
|
{
|
||
|
if ( !m_pSequencesPage )
|
||
|
return NULL;
|
||
|
|
||
|
int nIndex = m_pSequencesList->GetSelectedItem( 0 );
|
||
|
if ( nIndex >= 0 )
|
||
|
{
|
||
|
KeyValues *pkv = m_pSequencesList->GetItem( nIndex );
|
||
|
return pkv->GetString( "sequence", NULL );
|
||
|
}
|
||
|
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
const char *CMDLPicker::GetSelectedActivityName()
|
||
|
{
|
||
|
if ( !m_pActivitiesPage )
|
||
|
return NULL;
|
||
|
|
||
|
int nIndex = m_pActivitiesList->GetSelectedItem( 0 );
|
||
|
if ( nIndex >= 0 )
|
||
|
{
|
||
|
KeyValues *pkv = m_pActivitiesList->GetItem( nIndex );
|
||
|
return pkv->GetString( "activity", NULL );
|
||
|
}
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
int CMDLPicker::GetSelectedSkin()
|
||
|
{
|
||
|
if ( !m_pSkinsPage )
|
||
|
return 0;
|
||
|
|
||
|
int nIndex = m_pSkinsList->GetSelectedItem( 0 );
|
||
|
if ( nIndex >= 0 )
|
||
|
{
|
||
|
return nIndex;
|
||
|
}
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
// Plays the selected activity
|
||
|
//-----------------------------------------------------------------------------
|
||
|
void CMDLPicker::SelectActivity( const char *pActivityName )
|
||
|
{
|
||
|
studiohdr_t *pstudiohdr = vgui::MDLCache()->GetStudioHdr( m_hSelectedMDL[ 0 ] );
|
||
|
for ( int i = 0; i < pstudiohdr->GetNumSeq(); i++ )
|
||
|
{
|
||
|
mstudioseqdesc_t &seqdesc = pstudiohdr->pSeqdesc( i );
|
||
|
if ( stricmp( seqdesc.pszActivityName(), pActivityName ) == 0 )
|
||
|
{
|
||
|
// FIXME: Add weighted sequence selection logic?
|
||
|
m_pMDLPreview->SetSequence( i );
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
PostActionSignal( new KeyValues( "SequenceSelectionChanged", "activity", pActivityName ) );
|
||
|
}
|
||
|
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
// Plays the selected sequence
|
||
|
//-----------------------------------------------------------------------------
|
||
|
void CMDLPicker::SelectSequence( const char *pSequenceName )
|
||
|
{
|
||
|
studiohdr_t *pstudiohdr = vgui::MDLCache()->GetStudioHdr( m_hSelectedMDL[ 0 ] );
|
||
|
for (int i = 0; i < pstudiohdr->GetNumSeq(); i++)
|
||
|
{
|
||
|
mstudioseqdesc_t &seqdesc = pstudiohdr->pSeqdesc( i );
|
||
|
if ( !Q_stricmp( seqdesc.pszLabel(), pSequenceName ) )
|
||
|
{
|
||
|
m_pMDLPreview->SetSequence( i );
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
PostActionSignal( new KeyValues( "SequenceSelectionChanged", "sequence", pSequenceName ) );
|
||
|
}
|
||
|
|
||
|
void CMDLPicker::SelectSkin( int nSkin )
|
||
|
{
|
||
|
m_pMDLPreview->SetSkin( nSkin );
|
||
|
PostActionSignal( new KeyValues( "SkinSelectionChanged", "skin", nSkin));
|
||
|
}
|
||
|
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
// Purpose: Updates preview when an item is selected
|
||
|
//-----------------------------------------------------------------------------
|
||
|
void CMDLPicker::OnItemSelected( KeyValues *kv )
|
||
|
{
|
||
|
Panel *pPanel = (Panel *)kv->GetPtr("panel", NULL);
|
||
|
if ( m_pSequencesList && (pPanel == m_pSequencesList ) )
|
||
|
{
|
||
|
const char *pSequenceName = GetSelectedSequenceName();
|
||
|
if ( pSequenceName )
|
||
|
{
|
||
|
SelectSequence( pSequenceName );
|
||
|
}
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
if ( m_pActivitiesList && ( pPanel == m_pActivitiesList ) )
|
||
|
{
|
||
|
const char *pActivityName = GetSelectedActivityName();
|
||
|
if ( pActivityName )
|
||
|
{
|
||
|
SelectActivity( pActivityName );
|
||
|
}
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
if ( m_pSkinsList && ( pPanel == m_pSkinsList ) )
|
||
|
{
|
||
|
int nSelectedSkin = GetSelectedSkin();
|
||
|
SelectSkin( nSelectedSkin );
|
||
|
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
BaseClass::OnItemSelected( kv );
|
||
|
}
|
||
|
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
// Purpose: Called when a page is shown
|
||
|
//-----------------------------------------------------------------------------
|
||
|
void CMDLPicker::OnPageChanged( )
|
||
|
{
|
||
|
if ( m_pSequencesPage && ( m_pViewsSheet->GetActivePage() == m_pSequencesPage ) )
|
||
|
{
|
||
|
m_pSequencesList->RequestFocus();
|
||
|
|
||
|
const char *pSequenceName = GetSelectedSequenceName();
|
||
|
|
||
|
if ( pSequenceName )
|
||
|
{
|
||
|
SelectSequence( pSequenceName );
|
||
|
}
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
if ( m_pActivitiesPage && ( m_pViewsSheet->GetActivePage() == m_pActivitiesPage ) )
|
||
|
{
|
||
|
m_pActivitiesList->RequestFocus();
|
||
|
|
||
|
const char *pActivityName = GetSelectedActivityName();
|
||
|
|
||
|
if ( pActivityName )
|
||
|
{
|
||
|
SelectActivity( pActivityName );
|
||
|
}
|
||
|
return;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
//
|
||
|
// Purpose: Modal picker frame
|
||
|
//
|
||
|
//-----------------------------------------------------------------------------
|
||
|
CMDLPickerFrame::CMDLPickerFrame( vgui::Panel *pParent, const char *pTitle, int nFlags ) :
|
||
|
BaseClass( pParent )
|
||
|
{
|
||
|
SetAssetPicker( new CMDLPicker( this, nFlags ) );
|
||
|
LoadControlSettingsAndUserConfig( "resource/mdlpickerframe.res" );
|
||
|
SetTitle( pTitle, false );
|
||
|
}
|
||
|
|
||
|
CMDLPickerFrame::~CMDLPickerFrame()
|
||
|
{
|
||
|
}
|
||
|
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
// Allows external apps to select a MDL
|
||
|
//-----------------------------------------------------------------------------
|
||
|
void CMDLPickerFrame::SelectMDL( const char *pRelativePath )
|
||
|
{
|
||
|
static_cast<CMDLPicker*>( GetAssetPicker() )->SelectMDL( pRelativePath );
|
||
|
}
|
||
|
|
||
|
int CMDLPicker::UpdateSkinsList()
|
||
|
{
|
||
|
int nNumSkins = 0;
|
||
|
|
||
|
if ( m_pSkinsList )
|
||
|
{
|
||
|
m_pSkinsList->RemoveAll();
|
||
|
|
||
|
studiohdr_t *hdr = vgui::MDLCache()->GetStudioHdr( m_hSelectedMDL[ 0 ] );
|
||
|
if ( hdr )
|
||
|
{
|
||
|
nNumSkins = hdr->numskinfamilies;
|
||
|
for ( int i = 0; i < nNumSkins; i++ )
|
||
|
{
|
||
|
char skinText[25] = "";
|
||
|
sprintf( skinText, "skin%i", i );
|
||
|
KeyValues *pkv = new KeyValues("node", "skin", skinText );
|
||
|
m_pSkinsList->AddItem( pkv, 0, false, false );
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return nNumSkins;
|
||
|
}
|
||
|
|
||
|
void CMDLPicker::UpdateInfoTab()
|
||
|
{
|
||
|
studiohdr_t *hdr = vgui::MDLCache()->GetStudioHdr( m_hSelectedMDL[ 0 ] );
|
||
|
if ( !hdr )
|
||
|
return;
|
||
|
|
||
|
int nMass = hdr->mass;
|
||
|
Panel *pTempPanel = m_pInfoPage->FindChildByName("MassValue");
|
||
|
char massBuff[10];
|
||
|
Q_snprintf( massBuff, 10, "%d", nMass );
|
||
|
((vgui::Label *)pTempPanel)->SetText( massBuff );
|
||
|
bool bIsStatic = hdr->flags & STUDIOHDR_FLAGS_STATIC_PROP;
|
||
|
bool bIsPhysics = false;
|
||
|
const char* buf = hdr->KeyValueText();
|
||
|
Label * pTempLabel = (Label *)m_pInfoPage->FindChildByName("StaticText");
|
||
|
pTempLabel->SetVisible( false );
|
||
|
if( buf )
|
||
|
{
|
||
|
buf = Q_strstr( buf, "prop_data" );
|
||
|
if ( buf )
|
||
|
{
|
||
|
int iPropDataCount = UpdatePropDataList( buf, bIsStatic );
|
||
|
if( iPropDataCount )
|
||
|
{
|
||
|
bIsPhysics = true;
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
m_pPropDataList->RemoveAll();
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
m_pPropDataList->RemoveAll();
|
||
|
}
|
||
|
|
||
|
CheckButton * pTempCheck = (CheckButton *)m_pInfoPage->FindChildByName("StaticObject");
|
||
|
pTempCheck->SetCheckButtonCheckable( true );
|
||
|
pTempCheck->SetSelected( bIsStatic );
|
||
|
pTempCheck->SetCheckButtonCheckable( false );
|
||
|
pTempCheck = (CheckButton *)m_pInfoPage->FindChildByName("PhysicsObject");
|
||
|
pTempCheck->SetCheckButtonCheckable( true );
|
||
|
pTempCheck->SetSelected( bIsPhysics );
|
||
|
pTempCheck->SetCheckButtonCheckable( false );
|
||
|
pTempCheck = (CheckButton *)m_pInfoPage->FindChildByName("DynamicObject");
|
||
|
pTempCheck->SetCheckButtonCheckable( true );
|
||
|
pTempCheck->SetSelected( !bIsPhysics );
|
||
|
pTempCheck->SetCheckButtonCheckable( false );
|
||
|
|
||
|
|
||
|
}
|
||
|
|
||
|
int CMDLPicker::UpdatePropDataList( const char* pszPropData, bool &bIsStatic )
|
||
|
{
|
||
|
int iCount = 0;
|
||
|
|
||
|
if ( m_pPropDataList )
|
||
|
{
|
||
|
m_pPropDataList->RemoveAll();
|
||
|
|
||
|
const char * endPropData = strchr( pszPropData, '}' );
|
||
|
char keyText[255] = "";
|
||
|
char valueText[255] = "";
|
||
|
const char *beginChunk = strchr( pszPropData, '\"' );
|
||
|
if ( !beginChunk )
|
||
|
{
|
||
|
return 0;
|
||
|
}
|
||
|
beginChunk++;
|
||
|
const char *endChunk = strchr( beginChunk, '\"' );
|
||
|
while( endChunk )
|
||
|
{
|
||
|
Q_memcpy( keyText, beginChunk, endChunk - beginChunk );
|
||
|
beginChunk = endChunk + 1;
|
||
|
beginChunk = strchr( beginChunk, '\"' ) + 1;
|
||
|
endChunk = strchr( beginChunk, '\"' );
|
||
|
Q_memcpy( valueText, beginChunk, endChunk - beginChunk );
|
||
|
if( !Q_strcmp( keyText, "allowstatic" ) && !Q_strcmp( valueText , "1" ) )
|
||
|
{
|
||
|
if ( !bIsStatic )
|
||
|
{
|
||
|
Label * pTempLabel = (Label *)m_pInfoPage->FindChildByName("StaticText");
|
||
|
pTempLabel->SetVisible( true );
|
||
|
}
|
||
|
bIsStatic &= true;
|
||
|
}
|
||
|
KeyValues *pkv = new KeyValues("node", "key", keyText, "value", valueText );
|
||
|
m_pPropDataList->AddItem( pkv, 0, false, false );
|
||
|
Q_memset( keyText, 0, 255 );
|
||
|
Q_memset( valueText, 0, 255 );
|
||
|
iCount++;
|
||
|
beginChunk = endChunk + 1;
|
||
|
beginChunk = strchr( beginChunk, '\"' );
|
||
|
if ( !beginChunk || beginChunk > endPropData )
|
||
|
{
|
||
|
return iCount;
|
||
|
}
|
||
|
beginChunk++;
|
||
|
endChunk = strchr( beginChunk, '\"' );
|
||
|
}
|
||
|
}
|
||
|
return iCount;
|
||
|
}
|