1HEADmaster

1 files changed, 490 insertions, 0 deletions

diff --git a/utils/modelcheckin/modelcheckindlg.cpp b/utils/modelcheckin/modelcheckindlg.cpp
new file mode 100644
index 0000000..4987509
--- /dev/null
+++ b/utils/modelcheckin/modelcheckindlg.cpp
@@ -0,0 +1,490 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+//
+// Purpose: 
+//
+// $NoKeywords: $
+//
+//=============================================================================//
+// ModelCheckInDlg.cpp : implementation file
+//
+
+#include "stdafx.h"
+#include "ModelCheckIn.h"
+#include "ModelCheckInDlg.h"
+#include <direct.h>
+#include <sys/stat.h>
+
+#ifdef _DEBUG
+#define new DEBUG_NEW
+#undef THIS_FILE
+static char THIS_FILE[] = __FILE__;
+#endif
+
+/////////////////////////////////////////////////////////////////////////////
+// CModelCheckInDlg dialog
+
+CModelCheckInDlg::CModelCheckInDlg(CWnd* pParent /*=NULL*/)
+	: CDialog(CModelCheckInDlg::IDD, pParent)
+{
+	//{{AFX_DATA_INIT(CModelCheckInDlg)
+	m_HL2GameDirectory = _T("");
+	m_TF2GameDirectory = _T("");
+	m_UserName = _T("");
+	m_HL2Radio = -1;
+	m_TF2Radio = -1;
+	//}}AFX_DATA_INIT
+	// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
+	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
+}
+
+void CModelCheckInDlg::DoDataExchange(CDataExchange* pDX)
+{
+	CDialog::DoDataExchange(pDX);
+	//{{AFX_DATA_MAP(CModelCheckInDlg)
+	DDX_Text(pDX, IDC_EDIT_HL2_GAME_DIRECTORY, m_HL2GameDirectory);
+	DDX_Text(pDX, IDC_EDIT_TF2_GAME_DIRECTORY, m_TF2GameDirectory);
+	DDX_Text(pDX, IDC_EDIT_USERNAME, m_UserName);
+	//}}AFX_DATA_MAP
+}
+
+BEGIN_MESSAGE_MAP(CModelCheckInDlg, CDialog)
+	//{{AFX_MSG_MAP(CModelCheckInDlg)
+	ON_WM_PAINT()
+	ON_WM_QUERYDRAGICON()
+	//}}AFX_MSG_MAP
+
+	ON_COMMAND(ID_FILE_CHECK_OUT,  OnFileCheckOut)
+	ON_COMMAND(ID_FILE_CHECK_IN,   OnFileCheckIn)
+	ON_COMMAND(ID_FILE_EXIT,   OnFileExit)
+
+END_MESSAGE_MAP()
+
+/////////////////////////////////////////////////////////////////////////////
+// CModelCheckInDlg message handlers
+
+BOOL CModelCheckInDlg::OnInitDialog()
+{
+	CDialog::OnInitDialog();
+
+	// Restore state
+	RestoreStateFromRegistry();
+	UpdateData( FALSE );
+
+	// Set the icon for this dialog.  The framework does this automatically
+	//  when the application's main window is not a dialog
+	SetIcon(m_hIcon, TRUE);			// Set big icon
+	SetIcon(m_hIcon, FALSE);		// Set small icon
+	
+	// TODO: Add extra initialization here
+	
+	return TRUE;  // return TRUE  unless you set the focus to a control
+}
+
+// If you add a minimize button to your dialog, you will need the code below
+//  to draw the icon.  For MFC applications using the document/view model,
+//  this is automatically done for you by the framework.
+
+void CModelCheckInDlg::OnPaint() 
+{
+	if (IsIconic())
+	{
+		CPaintDC dc(this); // device context for painting
+
+		SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
+
+		// Center icon in client rectangle
+		int cxIcon = GetSystemMetrics(SM_CXICON);
+		int cyIcon = GetSystemMetrics(SM_CYICON);
+		CRect rect;
+		GetClientRect(&rect);
+		int x = (rect.Width() - cxIcon + 1) / 2;
+		int y = (rect.Height() - cyIcon + 1) / 2;
+
+		// Draw the icon
+		dc.DrawIcon(x, y, m_hIcon);
+	}
+	else
+	{
+		CDialog::OnPaint();
+	}
+}
+
+// The system calls this to obtain the cursor to display while the user drags
+//  the minimized window.
+HCURSOR CModelCheckInDlg::OnQueryDragIcon()
+{
+	return (HCURSOR) m_hIcon;
+}
+
+
+//-----------------------------------------------------------------------------
+// Resets our last used directory
+//-----------------------------------------------------------------------------
+
+void CModelCheckInDlg::ResetDirectoryEntry( CString &fullPath, char *pRegEntry )
+{
+	// reset the last directory...
+	CString temp = fullPath;
+	int i = temp.ReverseFind( '\\' );
+	int j = temp.ReverseFind( '/' );
+	if (i < j) i = j;
+	temp.SetAt(i, 0);
+
+	AfxGetApp()->WriteProfileString( MDL_CHECKOUT_REG_CLASS, pRegEntry, temp );
+}
+
+
+//-----------------------------------------------------------------------------
+// Stores state
+//-----------------------------------------------------------------------------
+
+void CModelCheckInDlg::StoreStateIntoRegistry( )
+{
+	AfxGetApp()->WriteProfileString( MDL_CHECKOUT_REG_CLASS, 
+		MDL_CHECKOUT_REG_USER, m_UserName );
+	AfxGetApp()->WriteProfileString( MDL_CHECKOUT_REG_CLASS, 
+		MDL_CHECKOUT_REG_HL2_PATH, m_HL2GameDirectory );
+	AfxGetApp()->WriteProfileString( MDL_CHECKOUT_REG_CLASS, 
+		MDL_CHECKOUT_REG_TF2_PATH, m_TF2GameDirectory );
+}
+
+
+//-----------------------------------------------------------------------------
+// Restores state
+//-----------------------------------------------------------------------------
+
+void CModelCheckInDlg::RestoreStateFromRegistry( )
+{
+	m_UserName = AfxGetApp()->GetProfileString( MDL_CHECKOUT_REG_CLASS, 
+		MDL_CHECKOUT_REG_USER, "" );
+	m_HL2GameDirectory = AfxGetApp()->GetProfileString( MDL_CHECKOUT_REG_CLASS, 
+		MDL_CHECKOUT_REG_HL2_PATH, "u:/hl2/hl2" );
+	m_TF2GameDirectory = AfxGetApp()->GetProfileString( MDL_CHECKOUT_REG_CLASS, 
+		MDL_CHECKOUT_REG_TF2_PATH, "u:/hl2/tf2" );
+}
+
+
+//-----------------------------------------------------------------------------
+// Extensions of all files related to the model file
+//-----------------------------------------------------------------------------
+
+static char* s_ppExtensions[] = 
+{
+	".dx7_2bone.vtx",
+	".dx80.vtx",
+	".phy",
+	".mdl",
+	""
+};
+
+
+static char* s_ppProjectDir[] = 
+{
+	"$/HL2/release/dev/hl2/",
+	"$/TF2/release/dev/tf2/"
+};
+
+//-----------------------------------------------------------------------------
+// Checks it out/ checks it in baby
+//-----------------------------------------------------------------------------
+
+void CModelCheckInDlg::PerformCheckoutCommand( ProjectType_t project, 
+	char const* pRelativeDir, char const* pDestPath, char const* pFileName )
+{
+	char error[1024];
+	char buf[1024];
+	int len = 0;
+
+	int currExtension;
+	for ( currExtension = 0; s_ppExtensions[currExtension][0]; ++currExtension )
+	{
+		// Check for the existence of the file in source safe...
+		sprintf( buf, "ss filetype %s%s/%s%s -O- -y%s\n",
+			s_ppProjectDir[project], pRelativeDir, pFileName, s_ppExtensions[currExtension],
+			(char const*)m_UserName );
+		int retVal = system( buf );
+		if (retVal > 0)
+			continue;
+
+		// It's there, try to check it out
+		sprintf( buf, "ss checkout %s%s/%s%s -GL%s -GWA -O- -y%s\n",
+			s_ppProjectDir[project], pRelativeDir, pFileName, s_ppExtensions[currExtension],
+			pDestPath, (char const*)m_UserName );
+		retVal = system( buf );
+		if (retVal > 0)
+		{
+			len += sprintf( &error[len], "*** SourceSafe error attempting to check out \"%s%s\"\n", 
+				pFileName, s_ppExtensions[currExtension] );
+		}
+	}
+
+	if (len > 0)
+	{
+		MessageBox( error, "Error!" );					
+	}
+}
+
+//-----------------------------------------------------------------------------
+// Checks it out/ checks it in baby
+//-----------------------------------------------------------------------------
+
+void CModelCheckInDlg::PerformCheckinCommand( ProjectType_t project, 
+	char const* pRelativeDir, char const* pDestPath, char const* pFileName )
+{
+	char buf[1024];
+	char error[1024];
+	int len = 0;
+
+	int currExtension;
+	for ( currExtension = 0; s_ppExtensions[currExtension][0]; ++currExtension )
+	{
+		// Check for the existence of the file on disk. If it's not there, don't bother
+		sprintf( buf, "%s/%s%s", pDestPath, pFileName, s_ppExtensions[currExtension] );
+		struct _stat statbuf;
+		int result = _stat( buf, &statbuf );
+		if (result != 0)
+			continue;
+
+		// Check for the existence of the file in source safe...
+		sprintf( buf, "ss filetype %s%s/%s%s -O- -y%s\n",
+			s_ppProjectDir[project], pRelativeDir, pFileName, s_ppExtensions[currExtension],
+			(char const*)m_UserName );
+		int retVal = system( buf );
+		if (retVal > 0)
+		{
+			// Try to add the file to source safe...
+			sprintf( buf, "ss add %s%s/%s%s -GL%s -O- -y%s\n",
+				s_ppProjectDir[project], pRelativeDir, pFileName, s_ppExtensions[currExtension],
+				pDestPath, (char const*)m_UserName );
+			int retVal = system( buf );
+			if (retVal > 0)
+			{
+				len += sprintf( &error[len], "SourceSafe error attempting to add \"%s%s\"\n", 
+					pFileName, s_ppExtensions[currExtension] );
+			}
+		}
+		else
+		{
+			// It's there, just check it in
+			sprintf( buf, "ss checkin %s%s/%s%s -GL%s -O- -y%s\n",
+				s_ppProjectDir[project], pRelativeDir, pFileName, s_ppExtensions[currExtension],
+				pDestPath, (char const*)m_UserName );
+			retVal = system( buf );
+			if (retVal > 0)
+			{
+				len += sprintf( &error[len], "SourceSafe error attempting to check in \"%s%s\"\n", 
+					pFileName, s_ppExtensions[currExtension] );
+			}
+		}
+	}
+
+	if (len > 0)
+	{
+		MessageBox( error, "Error!" );					
+	}
+}
+
+//-----------------------------------------------------------------------------
+// Fixes up a filename
+//-----------------------------------------------------------------------------
+
+static int FixupFileName( char const* pInFile, char* pOutBuf )
+{
+	if (!pInFile)
+	{
+		*pOutBuf = '\0';
+		return 0;
+	}
+
+	int len = strlen(pInFile); 
+	for (int i = 0; i <= len; ++i)
+	{
+		pOutBuf[i] = tolower( pInFile[i] );
+		if (pOutBuf[i] == '\\')
+			pOutBuf[i] = '/';
+	}
+
+	// Make sure no trailing '/'
+	if (pOutBuf[len - 1] == '/')
+		pOutBuf[--len] = '\0';
+
+	return len;
+}
+
+
+//-----------------------------------------------------------------------------
+// Gets the relative file name
+//-----------------------------------------------------------------------------
+
+CModelCheckInDlg::ProjectType_t CModelCheckInDlg::ComputeRelativeFileName( char const* pInFile, char* pRelativeFile )
+{
+	// Depending on which project is selected, strip out the game directory
+	char tempIn[MAX_PATH];
+	char tempPath[MAX_PATH];
+
+	FixupFileName( pInFile, tempIn );
+
+	int len;
+	if (!m_HL2GameDirectory.IsEmpty())
+	{
+		len = FixupFileName( m_HL2GameDirectory, tempPath );
+		if (!strncmp( tempIn, m_HL2GameDirectory, len ))
+		{
+			strcpy( pRelativeFile, &tempIn[len+1] );
+			return PROJECT_HL2;
+		}
+	}
+
+	if (!m_TF2GameDirectory.IsEmpty())
+	{
+		len = FixupFileName( m_TF2GameDirectory, tempPath );
+		if (!strncmp( tempIn, m_TF2GameDirectory, len ))
+		{
+			strcpy( pRelativeFile, &tempIn[len+1] );
+			return PROJECT_TF2;
+		}
+	}
+
+	return PROJECT_ERROR;
+}
+
+
+//-----------------------------------------------------------------------------
+// Have we typed in anything for our user name?
+//-----------------------------------------------------------------------------
+
+bool CModelCheckInDlg::CheckInfo() 
+{
+	if (m_UserName.IsEmpty())
+	{
+		MessageBox( "Please enter your user name.\n", "Error!" );
+		return false;
+	}
+
+	if (m_HL2GameDirectory.IsEmpty() && m_TF2GameDirectory.IsEmpty())
+	{
+		MessageBox( "Please enter the game directories for HL2 and/or TF2.\n", "Error!" );
+		return false;
+	}
+
+	return true;
+}
+
+
+//-----------------------------------------------------------------------------
+// Resets our last used directory
+//-----------------------------------------------------------------------------
+
+CModelCheckInDlg::ProjectType_t CModelCheckInDlg::GetFileNames( char const* pTitle, 
+					char*& pRelativePath, char*& pFileName, char*& pDestPath ) 
+{
+	UpdateData( TRUE );
+
+	StoreStateIntoRegistry();
+
+	if (!CheckInfo())
+		return PROJECT_ERROR;
+	
+	CFileDialog dlg( TRUE, ".mdl", "*.mdl", OFN_HIDEREADONLY, 
+		"Model Files (*.mdl)|*.mdl||" );
+	
+	// Set up initial paths
+	char pCwd[MAX_PATH];
+	_getcwd( pCwd, MAX_PATH );
+															   
+	// Setup the title and initial directory
+	CString temp = AfxGetApp()->GetProfileString( MDL_CHECKOUT_REG_CLASS, 
+		MDL_CHECKOUT_REG_LAST_PATH, pCwd );
+	dlg.m_ofn.lpstrInitialDir = temp;
+	dlg.m_ofn.lpstrTitle = pTitle;
+	
+	// Grab the data from the dialog...
+	if (dlg.DoModal() != IDOK) 
+		return PROJECT_ERROR;
+	
+	// Get the relative file name
+	static char relativeFile[MAX_PATH];
+	ProjectType_t projectType = ComputeRelativeFileName( dlg.GetPathName(), relativeFile );
+	if (projectType == PROJECT_ERROR)
+	{
+		char buf[MAX_PATH];
+		sprintf( buf, "%s\nwas not found under either game directory!\n", dlg.GetPathName() );
+		MessageBox( buf, "Error!" );
+		return PROJECT_ERROR;
+	}
+	
+	// reset the last directory...
+	ResetDirectoryEntry( dlg.GetPathName(), MDL_CHECKOUT_REG_LAST_PATH );
+	
+	// Split into relative file + path...
+	char* pSlash = strrchr( relativeFile, '/' );
+	if (pSlash)
+	{
+		*pSlash = 0;
+		pRelativePath = relativeFile;
+		pFileName = pSlash + 1;
+	}
+	else
+	{
+		pFileName = relativeFile;
+		pRelativePath = "";
+	}
+
+	// Remove extension
+	char* pExt = strchr( pFileName, '.' );
+	if (pExt)
+		*pExt = 0;
+
+	static char destPath[MAX_PATH];
+	FixupFileName( dlg.GetPathName(), destPath );
+	pSlash = strrchr( destPath, '/' );
+	if (pSlash)
+		*pSlash = 0;
+	else
+		destPath[0] = 0;
+	pDestPath = destPath;
+
+	return projectType;
+}
+
+
+
+//-----------------------------------------------------------------------------
+// Does checkout and check-in
+//-----------------------------------------------------------------------------
+
+void CModelCheckInDlg::OnFileCheckOut() 
+{
+	char* pRelativePath;
+	char* pFileName;
+	char* pDestPath;
+	ProjectType_t projectType;
+	projectType = GetFileNames( "Select MDL file to check out...", pRelativePath,
+								pFileName, pDestPath );
+
+	if (projectType != PROJECT_ERROR)
+	{
+		PerformCheckoutCommand( projectType, pRelativePath, pDestPath, pFileName ); 
+	}
+}
+
+
+void CModelCheckInDlg::OnFileCheckIn()
+{
+	char* pRelativePath;
+	char* pFileName;
+	char* pDestPath;
+	ProjectType_t projectType;
+	projectType = GetFileNames( "Select MDL file to check in...", pRelativePath,
+								pFileName, pDestPath );
+
+	if (projectType != PROJECT_ERROR)
+	{
+		PerformCheckinCommand( projectType, pRelativePath, pDestPath, pFileName ); 
+	}
+}
+
+void CModelCheckInDlg::OnFileExit()
+{
+	PostQuitMessage(0);
+}