1HEADmaster

1 files changed, 450 insertions, 0 deletions

diff --git a/hammer/materialdlg.cpp b/hammer/materialdlg.cpp
new file mode 100644
index 0000000..c7bdefe
--- /dev/null
+++ b/hammer/materialdlg.cpp
@@ -0,0 +1,450 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+//
+// Purpose: 
+//
+//=============================================================================//
+
+#include <stdafx.h>
+#include "materialdlg.h"
+#include "FaceEditSheet.h"
+#include "mapdoc.h"
+
+// memdbgon must be the last include file in a .cpp file!!!
+#include <tier0/memdbgon.h>
+
+//=============================================================================
+//
+// Face Smoothing Dialog Functions
+//
+BEGIN_MESSAGE_MAP( CFaceSmoothingDlg, CDialog )
+	//{{AFX_MSG_MAP(CFaceSmoothingDlg)
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_1, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_2, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_3, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_4, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_5, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_6, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_7, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_8, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_9, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_10, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_11, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_12, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_13, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_14, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_15, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_16, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_17, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_18, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_19, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_20, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_21, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_22, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_23, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_24, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_25, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_26, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_27, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_28, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_29, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_30, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_31, OnButtonGroup )
+	ON_COMMAND_EX( ID_SMOOTHING_GROUP_32, OnButtonGroup )
+
+	ON_WM_CLOSE()
+	ON_WM_DESTROY()	
+	//}}AFX_MSG_MAP
+END_MESSAGE_MAP()
+
+//-----------------------------------------------------------------------------
+// Purpose: Constructor
+//-----------------------------------------------------------------------------
+CFaceSmoothingDlg::CFaceSmoothingDlg( CWnd *pParent ) : CDialog( CFaceSmoothingDlg::IDD, pParent )
+{
+	InitButtonIDs();
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: Deconstructor
+//-----------------------------------------------------------------------------
+CFaceSmoothingDlg::~CFaceSmoothingDlg()
+{
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: 
+//-----------------------------------------------------------------------------
+BOOL CFaceSmoothingDlg::OnInitDialog( void )
+{
+	static bool bInit = false;
+
+	CDialog::OnInitDialog();
+
+	if ( bInit )
+	{
+		SetWindowPos( &wndTop, m_DialogPosRect.left, m_DialogPosRect.top,
+			          m_DialogPosRect.Width(), m_DialogPosRect.Height(), SWP_NOZORDER );
+	}
+
+	// Update the controls.
+	UpdateControls();
+
+	// Initialized.
+	bInit = true;
+
+	return TRUE;
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: 
+//-----------------------------------------------------------------------------
+void CFaceSmoothingDlg::OnClose( void )
+{
+	// Get the parent and set the material tool as the current tool.
+	CFaceEditSheet *pSheet = static_cast<CFaceEditSheet*>( GetParent() );
+	if ( pSheet )
+	{
+		pSheet->m_MaterialPage.SetMaterialPageTool( CFaceEditMaterialPage::MATERIALPAGETOOL_MATERIAL );
+	}
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: 
+//-----------------------------------------------------------------------------
+void CFaceSmoothingDlg::OnDestroy( void )
+{
+	// Save the current window position.
+	GetWindowRect( &m_DialogPosRect );
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: 
+//-----------------------------------------------------------------------------
+void CFaceSmoothingDlg::InitButtonIDs( void )
+{
+	m_ButtonIDs[0]  = 0;						// Default - no button!
+	m_ButtonIDs[1]  = ID_SMOOTHING_GROUP_1;
+	m_ButtonIDs[2]  = ID_SMOOTHING_GROUP_2;
+	m_ButtonIDs[3]  = ID_SMOOTHING_GROUP_3;
+	m_ButtonIDs[4]  = ID_SMOOTHING_GROUP_4;
+	m_ButtonIDs[5]  = ID_SMOOTHING_GROUP_5;
+	m_ButtonIDs[6]  = ID_SMOOTHING_GROUP_6;
+	m_ButtonIDs[7]  = ID_SMOOTHING_GROUP_7;
+	m_ButtonIDs[8]  = ID_SMOOTHING_GROUP_8;
+	m_ButtonIDs[9]  = ID_SMOOTHING_GROUP_9;
+	m_ButtonIDs[10] = ID_SMOOTHING_GROUP_10;
+	m_ButtonIDs[11] = ID_SMOOTHING_GROUP_11;
+	m_ButtonIDs[12] = ID_SMOOTHING_GROUP_12;
+	m_ButtonIDs[13] = ID_SMOOTHING_GROUP_13;
+	m_ButtonIDs[14] = ID_SMOOTHING_GROUP_14;
+	m_ButtonIDs[15] = ID_SMOOTHING_GROUP_15;
+	m_ButtonIDs[16] = ID_SMOOTHING_GROUP_16;
+	m_ButtonIDs[17] = ID_SMOOTHING_GROUP_17;
+	m_ButtonIDs[18] = ID_SMOOTHING_GROUP_18;
+	m_ButtonIDs[19] = ID_SMOOTHING_GROUP_19;
+	m_ButtonIDs[20] = ID_SMOOTHING_GROUP_20;
+	m_ButtonIDs[21] = ID_SMOOTHING_GROUP_21;
+	m_ButtonIDs[22] = ID_SMOOTHING_GROUP_22;
+	m_ButtonIDs[23] = ID_SMOOTHING_GROUP_23;
+	m_ButtonIDs[24] = ID_SMOOTHING_GROUP_24;
+	m_ButtonIDs[25] = ID_SMOOTHING_GROUP_25;
+	m_ButtonIDs[26] = ID_SMOOTHING_GROUP_26;
+	m_ButtonIDs[27] = ID_SMOOTHING_GROUP_27;
+	m_ButtonIDs[28] = ID_SMOOTHING_GROUP_28;
+	m_ButtonIDs[29] = ID_SMOOTHING_GROUP_29;
+	m_ButtonIDs[30] = ID_SMOOTHING_GROUP_30;
+	m_ButtonIDs[31] = ID_SMOOTHING_GROUP_31;
+	m_ButtonIDs[32] = ID_SMOOTHING_GROUP_32;
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: 
+//-----------------------------------------------------------------------------
+BOOL CFaceSmoothingDlg::OnButtonGroup( UINT uCmd )
+{
+	int iGroup = GetSmoothingGroup( uCmd );
+	if ( iGroup == -1 )
+		return FALSE;
+
+	// Add or remove.
+	CButton *pGroupButton = ( CButton* )GetDlgItem( uCmd );
+	if ( !pGroupButton )
+		return FALSE;
+
+	bool bAdd = ( pGroupButton->GetCheck() != 0 );
+
+	// Get the selected faces and add them to the appropriate group.
+	CFaceEditSheet *pSheet = ( CFaceEditSheet* )GetParent();
+	if ( pSheet )
+	{
+		int nFaceCount = pSheet->GetFaceListCount();
+		for ( int iFace = 0; iFace < nFaceCount; ++iFace )
+		{
+			CMapFace *pFace = pSheet->GetFaceListDataFace( iFace );
+			if ( pFace )
+			{
+				if ( bAdd )
+				{
+					pFace->AddSmoothingGroup( iGroup );
+				}
+				else
+				{
+					pFace->RemoveSmoothingGroup( iGroup );
+				}
+			}
+		}
+	}
+
+	UpdateControls();
+
+	return TRUE;
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: 
+//-----------------------------------------------------------------------------
+UINT CFaceSmoothingDlg::GetSmoothingGroup( UINT uCmd )
+{
+	for ( int iGroup = 1; iGroup <= SMOOTHING_GROUP_MAX_COUNT; ++iGroup )
+	{
+		if ( m_ButtonIDs[iGroup] == uCmd )
+			return iGroup;
+	}
+
+	return 0xFFFFFFFF;
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: 
+//-----------------------------------------------------------------------------
+int CFaceSmoothingDlg::GetActiveSmoothingGroup( void )
+{
+	for ( int iButton = 1; iButton <= SMOOTHING_GROUP_MAX_COUNT; ++iButton )
+	{
+		CButton *pGroupButton = ( CButton* )GetDlgItem( m_ButtonIDs[iButton] );
+		if ( !pGroupButton )
+			continue;
+
+		if ( pGroupButton->GetCheck() != 0 )
+			return iButton;
+	}
+
+	return 0xFFFFFFFF;
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: 
+//-----------------------------------------------------------------------------
+void CFaceSmoothingDlg::UpdateControls( void )
+{
+	// Get the selected faces and add them to the appropriate group.
+	CFaceEditSheet *pSheet = ( CFaceEditSheet* )GetParent();
+	if ( !pSheet )
+		return;
+
+	// Check for a face count.
+	int nFaceCount = pSheet->GetFaceListCount();
+	if ( nFaceCount == 0 )
+		return;
+
+	// Keep a list of used groups.
+	int nGroupCounts[SMOOTHING_GROUP_MAX_COUNT+1];
+	for ( int iGroup = 0; iGroup <= SMOOTHING_GROUP_MAX_COUNT; ++iGroup )
+	{
+		nGroupCounts[iGroup] = 0;
+	}
+
+	for ( int iFace = 0; iFace < nFaceCount; ++iFace )
+	{
+		CMapFace *pFace = pSheet->GetFaceListDataFace( iFace );
+		if ( pFace )
+		{
+			for ( int iGroup = 1; iGroup <= SMOOTHING_GROUP_MAX_COUNT; ++iGroup )
+			{
+				if ( pFace->InSmoothingGroup( iGroup ) )
+				{
+					nGroupCounts[iGroup]++;
+				}
+			}
+		}
+	}
+
+	CheckGroupButtons( nGroupCounts, nFaceCount );
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: 
+//-----------------------------------------------------------------------------
+void CFaceSmoothingDlg::CheckGroupButtons( int *pGroupCounts, int nFaceCount )
+{
+	m_bColorOverride = true;
+
+	for ( int iButton = 1; iButton <= SMOOTHING_GROUP_MAX_COUNT; ++iButton )
+	{
+		CButton *pGroupButton = ( CButton* )GetDlgItem( m_ButtonIDs[iButton] );
+		if ( !pGroupButton )
+			continue;
+		
+		pGroupButton->SetCheck( FALSE );
+
+		if ( pGroupCounts[iButton] == 0 )
+			continue;
+
+		if ( pGroupCounts[iButton] == nFaceCount )
+		{
+			pGroupButton->SetCheck( TRUE );
+		}
+		else
+		{
+			// Todo: Come up with a better effect here!
+			CDC *pDC = pGroupButton->GetDC();
+			m_Brush.CreateSolidBrush( pDC->GetTextColor() );
+			CRect buttonRect;
+			pGroupButton->GetClientRect( buttonRect );
+			pDC->FillRect( buttonRect, &m_Brush );
+			m_Brush.DeleteObject();
+			pGroupButton->ReleaseDC( pDC );
+		}
+	}
+
+	m_bColorOverride = false;
+}
+
+//=============================================================================
+
+BEGIN_MESSAGE_MAP( CFaceSmoothingVisualDlg, CDialog )
+	//{{AFX_MSG_MAP(CFaceSmoothingVisualDlg)
+	ON_COMMAND_EX( ID_SG_VISUAL_1, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_2, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_3, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_4, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_5, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_6, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_7, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_8, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_9, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_10, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_11, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_12, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_13, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_14, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_15, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_16, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_17, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_18, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_19, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_20, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_21, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_22, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_23, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_24, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_25, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_26, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_27, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_28, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_29, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_30, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_31, OnButtonGroup )
+	ON_COMMAND_EX( ID_SG_VISUAL_32, OnButtonGroup )
+	//}}AFX_MSG_MAP
+END_MESSAGE_MAP()
+
+//-----------------------------------------------------------------------------
+// Purpose: Constructor
+//-----------------------------------------------------------------------------
+CFaceSmoothingVisualDlg::CFaceSmoothingVisualDlg( CWnd *pParent ) : CDialog( CFaceSmoothingDlg::IDD, pParent )
+{
+	InitButtonIDs();
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: Deconstructor
+//-----------------------------------------------------------------------------
+CFaceSmoothingVisualDlg::~CFaceSmoothingVisualDlg()
+{
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: 
+//-----------------------------------------------------------------------------
+BOOL CFaceSmoothingVisualDlg::OnInitDialog( void )
+{
+	CDialog::OnInitDialog();
+
+//	SetWindowPos( &wndTop, m_DialogPosRect.left, m_DialogPosRect.top,
+//		m_DialogPosRect.Width(), m_DialogPosRect.Height(), SWP_NOZORDER );
+
+	return TRUE;
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: 
+//-----------------------------------------------------------------------------
+BOOL CFaceSmoothingVisualDlg::OnButtonGroup( UINT uCmd )
+{
+	int iGroup = GetSmoothingGroup( uCmd );
+	if ( iGroup == -1 )
+		return FALSE;
+
+	CMapDoc *pDoc = CMapDoc::GetActiveMapDoc();
+	if ( pDoc )
+	{
+		pDoc->SetSmoothingGroupVisual( iGroup );
+	}
+
+	return TRUE;
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: 
+//-----------------------------------------------------------------------------
+void CFaceSmoothingVisualDlg::InitButtonIDs( void )
+{
+	m_ButtonIDs[0]  = 0;						// Default - no button!
+	m_ButtonIDs[1]  = ID_SG_VISUAL_1;
+	m_ButtonIDs[2]  = ID_SG_VISUAL_2;
+	m_ButtonIDs[3]  = ID_SG_VISUAL_3;
+	m_ButtonIDs[4]  = ID_SG_VISUAL_4;
+	m_ButtonIDs[5]  = ID_SG_VISUAL_5;
+	m_ButtonIDs[6]  = ID_SG_VISUAL_6;
+	m_ButtonIDs[7]  = ID_SG_VISUAL_7;
+	m_ButtonIDs[8]  = ID_SG_VISUAL_8;
+	m_ButtonIDs[9]  = ID_SG_VISUAL_9;
+	m_ButtonIDs[10] = ID_SG_VISUAL_10;
+	m_ButtonIDs[11] = ID_SG_VISUAL_11;
+	m_ButtonIDs[12] = ID_SG_VISUAL_12;
+	m_ButtonIDs[13] = ID_SG_VISUAL_13;
+	m_ButtonIDs[14] = ID_SG_VISUAL_14;
+	m_ButtonIDs[15] = ID_SG_VISUAL_15;
+	m_ButtonIDs[16] = ID_SG_VISUAL_16;
+	m_ButtonIDs[17] = ID_SG_VISUAL_17;
+	m_ButtonIDs[18] = ID_SG_VISUAL_18;
+	m_ButtonIDs[19] = ID_SG_VISUAL_19;
+	m_ButtonIDs[20] = ID_SG_VISUAL_20;
+	m_ButtonIDs[21] = ID_SG_VISUAL_21;
+	m_ButtonIDs[22] = ID_SG_VISUAL_22;
+	m_ButtonIDs[23] = ID_SG_VISUAL_23;
+	m_ButtonIDs[24] = ID_SG_VISUAL_24;
+	m_ButtonIDs[25] = ID_SG_VISUAL_25;
+	m_ButtonIDs[26] = ID_SG_VISUAL_26;
+	m_ButtonIDs[27] = ID_SG_VISUAL_27;
+	m_ButtonIDs[28] = ID_SG_VISUAL_28;
+	m_ButtonIDs[29] = ID_SG_VISUAL_29;
+	m_ButtonIDs[30] = ID_SG_VISUAL_30;
+	m_ButtonIDs[31] = ID_SG_VISUAL_31;
+	m_ButtonIDs[32] = ID_SG_VISUAL_32;
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: 
+//-----------------------------------------------------------------------------
+UINT CFaceSmoothingVisualDlg::GetSmoothingGroup( UINT uCmd )
+{
+	for ( int iGroup = 1; iGroup <= SMOOTHING_GROUP_MAX_COUNT; ++iGroup )
+	{
+		if ( m_ButtonIDs[iGroup] == uCmd )
+			return iGroup;
+	}
+
+	return 0xFFFFFFFF;
+}