Counter Strike : Global Offensive Source Code
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.
 
 
 
 
 
 

360 lines
8.8 KiB

//====== Copyright c 1996-2007, Valve Corporation, All rights reserved. =======//
//
// Purpose:
//
// $NoKeywords: $
//
//=============================================================================//
#include "stdafx.h"
#include "hammer.h"
#include "filesystem.h"
#include "mapclass.h"
#include "mapentity.h"
#include "history.h"
#include "dlglistmanage.h"
// memdbgon must be the last include file in a .cpp file!!!
#include <tier0/memdbgon.h>
BEGIN_MESSAGE_MAP( CDlgListManage, CDialog )
ON_WM_SIZE()
ON_WM_GETMINMAXINFO()
ON_BN_CLICKED(IDC_SCRIPT_LIST_ADD, &CDlgListManage::OnBnClickedScriptListAdd)
ON_BN_CLICKED(IDC_SCRIPT_LIST_REMOVE, &CDlgListManage::OnBnClickedScriptListRemove)
ON_BN_CLICKED(IDC_SCRIPT_LIST_EDIT, &CDlgListManage::OnBnClickedScriptListEdit)
END_MESSAGE_MAP()
CDlgListManage::CDlgListManage(CWnd *pParent, IDlgListManageBrowse *pBrowseImpl, const CMapObjectList *pObjectList)
: CDialog(CDlgListManage::IDD, pParent)
{
m_pBrowseImpl = pBrowseImpl;
m_rcDialog.SetRectEmpty();
m_pObjectList = pObjectList;
}
static bool StringLessFunc( CString const &a, CString const &b )
{
return (a < b);
}
void CDlgListManage::PopulateScriptList( void )
{
// Clear us out!
m_ScriptList.ResetContent();
// Populate the list
CUtlMap<CString, int> mapStrings( StringLessFunc ); // A map of strings to their counts
// Iterate through all map objects currently being edited
FOR_EACH_OBJ( *m_pObjectList, pos )
{
const CMapClass *pObject = m_pObjectList->Element( pos );
if ( (pObject != NULL) && (pObject->IsMapClass(MAPCLASS_TYPE(CMapEntity))) )
{
CMapEntity *pEntity = (CMapEntity *)pObject;
CString strScriptName = pEntity->GetKeyValue( "vscripts" );
// Only consider this if we have strings here!
if ( strScriptName.IsEmpty() )
continue;
int nStart = 0;
CString strToken = strScriptName.Tokenize( " ", nStart );
// Now, iterate through all the listed scripts in the entity and check their counts
while( strToken != "" )
{
// If we've removed this, filter it out
if ( m_vSubtractions.Find( strToken ) != m_vSubtractions.InvalidIndex() )
{
strToken = strScriptName.Tokenize( " ", nStart );
continue;
}
// Also, keep a map into the global scripts we're adding for later!
int nIndex = mapStrings.Find( strToken );
if ( nIndex != mapStrings.InvalidIndex() )
{
mapStrings[nIndex]++;
}
else
{
int nIndex = mapStrings.Insert( strToken );
mapStrings[nIndex] = 1;
}
// Continue to the next token
strToken = strScriptName.Tokenize( " ", nStart );
}
}
}
// Include new additions
FOR_EACH_VEC( m_vAdditions, itr )
{
int nIndex = mapStrings.Insert( m_vAdditions[itr] );
mapStrings[nIndex] = m_pObjectList->Count(); // Always added to all edited members
}
// Now go through and add things to the list with proper coloring
FOR_EACH_MAP( mapStrings, itr )
{
// If it's the same for every object, it's bold
if ( mapStrings[itr] == m_pObjectList->Count() )
{
if ( m_pObjectList->Count() == 1 )
{
// Normal (single instance)
m_ScriptList.AddItemText( mapStrings.Key( itr ), LISTFONT_NORMAL );
}
else
{
// Bold (multiple instances)
m_ScriptList.AddItemText( mapStrings.Key( itr ), LISTFONT_BOLD );
}
}
else
{
// Red (different)
m_ScriptList.AddItemText( mapStrings.Key( itr ), LISTFONT_DUPLICATE );
}
}
}
void CDlgListManage::UpdateScriptChanges( void )
{
// Nothing to do currently!
}
void CDlgListManage::SaveScriptChanges( void )
{
GetHistory()->MarkUndoPosition( NULL, "VScript update" );
// Iterate through all map objects currently being edited
FOR_EACH_OBJ( *m_pObjectList, pos )
{
const CMapClass *pObject = m_pObjectList->Element( pos );
if ( (pObject != NULL) && (pObject->IsMapClass(MAPCLASS_TYPE(CMapEntity))) )
{
CMapEntity *pEntity = (CMapEntity *)pObject;
CString strScriptName = pEntity->GetKeyValue( "vscripts" );
GetHistory()->Keep( pEntity );
// Add in all new additions
FOR_EACH_VEC( m_vAdditions, itr )
{
// Do we already have this in the list somehow?
if ( strScriptName.Find( m_vAdditions[itr], 0 ) != -1 )
continue;
// Append it
strScriptName += " " + m_vAdditions[itr];
}
// Subtract things we deleted
FOR_EACH_VEC( m_vSubtractions, itr )
{
// Find an occurence of this string in our base strings
int nIndex = strScriptName.Find( m_vSubtractions[itr], 0 );
if ( nIndex == -1 )
continue;
// Kill it!
int nEndIndex = strScriptName.Find( ' ', nIndex );
if ( nEndIndex == -1 )
{
nEndIndex = strlen( strScriptName ); // Really? You can't ask a CString its length?
}
strScriptName.Delete( nIndex, (nEndIndex-nIndex) );
}
// Done!
pEntity->SetKeyValue( "vscripts", strScriptName );
}
}
// Clear it
m_vAdditions.RemoveAll();
m_vSubtractions.RemoveAll();
}
void CDlgListManage::DoDataExchange( CDataExchange* pDX )
{
DDX_Control( pDX, IDC_SCRIPT_LIST, m_ScriptList );
CDialog::DoDataExchange(pDX);
if ( pDX->m_bSaveAndValidate )
{
UpdateScriptChanges();
}
else
{
PopulateScriptList();
}
}
BOOL CDlgListManage::OnInitDialog()
{
BOOL bResult = CDialog::OnInitDialog();
GetWindowRect( m_rcDialog );
int arrCtlIds[] = { IDOK, IDCANCEL, IDC_SCRIPT_LIST, IDC_BROWSE };
int riFlags[] = { ResizeInfo_t::RI_TOP_AND_LEFT, ResizeInfo_t::RI_TOP_AND_LEFT, ResizeInfo_t::RI_WIDTH_AND_HEIGHT, ResizeInfo_t::RI_TOP };
for ( int k = 0; k < ARRAYSIZE( arrCtlIds ); ++ k )
{
CWnd *pWnd = GetDlgItem( arrCtlIds[ k ] );
if ( !pWnd )
continue;
ResizeInfo_t ri;
ri.flags = riFlags[k];
pWnd->GetWindowRect( ri.rc );
this->ScreenToClient( ri.rc );
m_ctlInfo[ arrCtlIds[ k ] ] = ri;
}
if ( !m_pBrowseImpl )
{
if ( CWnd *pWnd = GetDlgItem( IDC_BROWSE ) )
pWnd->ShowWindow( SW_HIDE );
}
return bResult;
}
void CDlgListManage::OnSize(UINT nType, int cx, int cy)
{
CDialog::OnSize( nType, cx, cy );
CRect rcDialog;
GetWindowRect( rcDialog );
int dx = rcDialog.Width() - m_rcDialog.Width();
int dy = rcDialog.Height() - m_rcDialog.Height();
for ( POSITION pos = m_ctlInfo.GetStartPosition(); pos; )
{
int ctlId;
ResizeInfo_t ri;
m_ctlInfo.GetNextAssoc( pos, ctlId, ri );
CWnd *pWnd = GetDlgItem( ctlId );
if ( !pWnd )
continue;
CRect rcWnd = ri.rc;
if ( ri.flags & ResizeInfo_t::RI_HEIGHT )
rcWnd.bottom += dy;
if ( ri.flags & ResizeInfo_t::RI_WIDTH )
rcWnd.right += dx;
if ( ri.flags & ResizeInfo_t::RI_LEFT )
rcWnd.OffsetRect( dx, 0 );
if ( ri.flags & ResizeInfo_t::RI_TOP )
rcWnd.OffsetRect( 0, dy );
pWnd->MoveWindow( rcWnd );
}
}
void CDlgListManage::OnGetMinMaxInfo(MINMAXINFO* lpMMI)
{
CDialog::OnGetMinMaxInfo( lpMMI );
if ( !m_rcDialog.IsRectEmpty() )
{
lpMMI->ptMinTrackSize.x = m_rcDialog.Width();
lpMMI->ptMinTrackSize.y = m_rcDialog.Height();
}
}
void CDlgListManage::OnBnClickedScriptListAdd()
{
// Launch a browse window
CStringList lstBrowse;
if ( m_pBrowseImpl == NULL || m_pBrowseImpl->HandleBrowse( lstBrowse ) == false )
return;
// Nothing was added!
if ( lstBrowse.IsEmpty() )
return;
// Operate on all strings we got back from the browser
for ( POSITION pos = lstBrowse.GetHeadPosition(); pos; )
{
const CString &strTemp = lstBrowse.GetNext( pos );
// Update our change lists
m_vSubtractions.FindAndRemove( strTemp );
// Don't add twice!
if ( m_vAdditions.Find( strTemp ) == -1 )
{
m_vAdditions.AddToTail( strTemp );
}
}
UpdateData();
UpdateData( FALSE );
}
void CDlgListManage::OnBnClickedScriptListRemove()
{
// Remove the current selection
int nSelectedItem = m_ScriptList.GetCurSel();
if ( nSelectedItem != LB_ERR )
{
CString strOldText;
m_ScriptList.GetText( nSelectedItem, strOldText );
// Update our change lists
m_vAdditions.FindAndRemove( strOldText );
// Don't add twice!
if ( m_vSubtractions.Find( strOldText ) == -1 )
{
m_vSubtractions.AddToTail( strOldText );
}
// Kill the line
m_ScriptList.DeleteString( nSelectedItem );
// Move our current selection around based on where we are in the list
if ( nSelectedItem >= m_ScriptList.GetCount() )
{
nSelectedItem = m_ScriptList.GetCount() - 1;
}
m_ScriptList.SetCurSel( nSelectedItem ); // Don't lose selection (annoying for multi-delete)
}
// Now update to reflect the change
UpdateData();
}
void CDlgListManage::OnBnClickedScriptListEdit()
{
// Get what they've currently selected
int nSelectedItem = m_ScriptList.GetCurSel();
if ( nSelectedItem == LB_ERR )
return;
// Get the file they've selected
CString strSelectedText;
m_ScriptList.GetText( nSelectedItem, strSelectedText );
// Cobble together a complete filename
CString strFilename = "scripts\\vscripts\\" + strSelectedText;
char pFullPath[MAX_PATH];
if ( g_pFullFileSystem->GetLocalPath( strFilename, pFullPath, MAX_PATH ) )
{
ShellExecute( NULL, "open", pFullPath, NULL, NULL, SW_SHOWNORMAL );
}
}