1 files changed, 316 insertions, 0 deletions
diff --git a/hammer/editgroups.cpp b/hammer/editgroups.cpp
new file mode 100644
index 0000000..d783311
--- /dev/null
+++ b/hammer/editgroups.cpp
@@ -0,0 +1,316 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+//
+// Purpose: A dialog for adding, deleting, and renaming visgroups.
+//
+//=============================================================================//
+
+#include "stdafx.h"
+#include "hammer.h"
+#include "EditGroups.h"
+#include "MainFrm.h"
+#include "MapWorld.h"
+#include "CustomMessages.h"
+#include "GlobalFunctions.h"
+#include "VisGroup.h"
+
+// memdbgon must be the last include file in a .cpp file!!!
+#include <tier0/memdbgon.h>
+
+static const unsigned int g_uSelChangeMsg = ::RegisterWindowMessage(GROUPLIST_MSG_SEL_CHANGE);
+
+
+BEGIN_MESSAGE_MAP(CEditGroups, CDialog)
+	//{{AFX_MSG_MAP(CEditGroups)
+	ON_BN_CLICKED(IDC_COLOR, OnColor)
+	ON_EN_CHANGE(IDC_NAME, OnChangeName)
+	ON_BN_CLICKED(IDC_NEW, OnNew)
+	ON_BN_CLICKED(IDC_REMOVE, OnRemove)
+	ON_WM_CLOSE()
+	ON_REGISTERED_MESSAGE(g_uSelChangeMsg, OnSelChangeGroupList)
+	//}}AFX_MSG_MAP
+END_MESSAGE_MAP()
+
+
+//-----------------------------------------------------------------------------
+// Purpose: Constructor.
+// Input  : pParent - Parent window.
+//-----------------------------------------------------------------------------
+CEditGroups::CEditGroups(CWnd* pParent /*=NULL*/)
+	: CDialog(CEditGroups::IDD, pParent)
+{
+	//{{AFX_DATA_INIT(CEditGroups)
+		// NOTE: the ClassWizard will add member initialization here
+	//}}AFX_DATA_INIT
+}
+
+
+//-----------------------------------------------------------------------------
+// Purpose: Exchanges data between controls and data members.
+// Input  : pDX - 
+//-----------------------------------------------------------------------------
+void CEditGroups::DoDataExchange(CDataExchange* pDX)
+{
+	CDialog::DoDataExchange(pDX);
+	//{{AFX_DATA_MAP(CEditGroups)
+	DDX_Control(pDX, IDC_NAME, m_cName);
+	//}}AFX_DATA_MAP
+}
+
+
+//-----------------------------------------------------------------------------
+// Purpose: Sets the object's color as the visgroup color if the object belongs
+//			to the given visgroup.
+// Input  : pObject - Object to evaluate.
+//			pGroup - Visgroup to check against.
+// Output : Returns TRUE to continue enumerating.
+//-----------------------------------------------------------------------------
+static BOOL UpdateObjectColor(CMapClass *pObject, CVisGroup *pGroup)
+{
+	pObject->UpdateObjectColor();
+	return(TRUE);
+}
+
+
+//-----------------------------------------------------------------------------
+// Purpose: Invokes the color picker dialog to modify the selected visgroup.
+//-----------------------------------------------------------------------------
+void CEditGroups::OnColor(void) 
+{
+	CVisGroup *pGroup = m_cGroupList.GetSelectedVisGroup();
+
+	if (pGroup != NULL)
+	{
+		color32 rgbColor = pGroup->GetColor();
+		CColorDialog dlg(RGB(rgbColor.r, rgbColor.g, rgbColor.b), CC_FULLOPEN);
+
+		if (dlg.DoModal() == IDOK)
+		{
+			// change group color
+			pGroup->SetColor(GetRValue(dlg.m_cc.rgbResult), GetGValue(dlg.m_cc.rgbResult), GetBValue(dlg.m_cc.rgbResult));
+			m_cColorBox.SetColor(dlg.m_cc.rgbResult, TRUE);
+
+			// change all object colors
+			GetActiveWorld()->EnumChildren(ENUMMAPCHILDRENPROC(UpdateObjectColor), DWORD(pGroup));
+
+			CMapDoc::GetActiveMapDoc()->UpdateAllViews( MAPVIEW_UPDATE_COLOR );
+		}
+	}
+}
+
+
+//-----------------------------------------------------------------------------
+// Purpose: Called when the contents of the name edit control changes. Renames
+//			the currently selected group with the new edit control contents.
+//-----------------------------------------------------------------------------
+void CEditGroups::OnChangeName(void)
+{
+	CVisGroup *pGroup = m_cGroupList.GetSelectedVisGroup();
+
+	CString szName;
+	m_cName.GetWindowText(szName);
+	pGroup->SetName(szName);
+	m_cGroupList.UpdateVisGroup(pGroup);
+}
+
+
+//-----------------------------------------------------------------------------
+// Purpose: Creates a new visgroup and adds it to the list.
+//-----------------------------------------------------------------------------
+void CEditGroups::OnNew(void)
+{
+	CMapDoc *pDoc = CMapDoc::GetActiveMapDoc();
+	CVisGroup *pGroup = pDoc->VisGroups_AddGroup("new group");
+	pGroup->SetVisible(VISGROUP_SHOWN);
+	UpdateGroupList();
+	m_cGroupList.SelectVisGroup(pGroup);
+	m_cName.EnableWindow(TRUE);
+	m_cName.SetActiveWindow();
+}
+
+
+//-----------------------------------------------------------------------------
+// Purpose: Called when the remove button is pressed from the visgroup editor.
+//			Deletes the selected visgroup and removes all references to it.
+//-----------------------------------------------------------------------------
+void CEditGroups::OnRemove(void) 
+{
+	CVisGroup *pGroup = m_cGroupList.GetSelectedVisGroup();
+	if (!pGroup)
+		return;
+	//Don't allow user to delete autovisgroups.
+	if ( pGroup->IsAutoVisGroup() )
+		return;
+
+	CMapDoc *pDoc = CMapDoc::GetActiveMapDoc();
+	if (pDoc != NULL)
+	{
+		pDoc->VisGroups_RemoveGroup(pGroup);
+
+		if (pGroup->GetVisible() != VISGROUP_SHOWN)
+		{
+			pDoc->VisGroups_UpdateAll();
+			pDoc->UpdateVisibilityAll();
+			pDoc->UpdateAllViews( MAPVIEW_UPDATE_OBJECTS );
+		}
+		else
+		{
+			pDoc->UpdateAllViews( MAPVIEW_UPDATE_COLOR );
+		}
+	}
+
+	UpdateGroupList();
+}
+
+
+//-----------------------------------------------------------------------------
+// Purpose: Handles selection change in the visgroup list. Updates the static
+//			text controls with the name and colot of the selected visgroup.
+//-----------------------------------------------------------------------------
+LRESULT CEditGroups::OnSelChangeGroupList(WPARAM wParam, LPARAM lParam)
+{
+	CVisGroup *pVisGroup = m_cGroupList.GetSelectedVisGroup();
+	if (!pVisGroup)
+		return 0;
+
+	UpdateControlsForVisGroup(pVisGroup);
+	return 0;
+}
+
+
+//-----------------------------------------------------------------------------
+// Purpose: 
+//-----------------------------------------------------------------------------
+void CEditGroups::UpdateControlsForVisGroup(CVisGroup *pVisGroup)
+{
+	if (!pVisGroup)
+		return;
+
+	//
+	// Update the name and color controls.
+	//
+	m_cName.SetWindowText(pVisGroup->GetName());
+	color32 rgbColor = pVisGroup->GetColor();
+	m_cColorBox.SetColor(RGB(rgbColor.r, rgbColor.g, rgbColor.b), TRUE);
+}
+
+
+//-----------------------------------------------------------------------------
+// Purpose: Sets up initial state of dialog.
+//-----------------------------------------------------------------------------
+BOOL CEditGroups::OnInitDialog(void)
+{
+	CDialog::OnInitDialog();
+
+	m_cGroupList.SubclassDlgItem(IDC_GROUPS, this);
+	m_cColorBox.SubclassDlgItem(IDC_COLORBOX, this);
+
+	//
+	// Fill the listbox with the visgroup names.
+	//
+	UpdateGroupList();
+
+	//
+	// Disable the edit name window if there are no visgroups in the list.
+	//
+	if (m_cGroupList.GetVisGroupCount())
+	{
+		CVisGroup *pVisGroup = m_cGroupList.GetVisGroup(0);
+		m_cGroupList.SelectVisGroup(pVisGroup);
+		UpdateControlsForVisGroup(pVisGroup);
+	}
+	else
+	{
+		m_cName.EnableWindow(FALSE);
+	}
+
+	return(TRUE);
+}
+
+
+//-----------------------------------------------------------------------------
+// Purpose: 
+//-----------------------------------------------------------------------------
+void CEditGroups::UpdateGroupList()
+{
+	if (!IsWindow(m_hWnd))
+	{
+		return;
+	}
+
+	m_cGroupList.SetRedraw(false);
+	m_cGroupList.DeleteAllItems();
+
+	CMapDoc *pDoc = CMapDoc::GetActiveMapDoc();
+	if (pDoc != NULL)
+	{
+		int nCount = pDoc->VisGroups_GetCount();
+		for (int i = 0; i < nCount; i++)
+		{
+			CVisGroup *pGroup = pDoc->VisGroups_GetVisGroup(i);
+			if (!pGroup->GetParent())
+			{
+				m_cGroupList.AddVisGroup(pGroup);
+			}
+		}
+	}
+
+	m_cGroupList.ExpandAll();
+	m_cGroupList.SetRedraw(true);
+	m_cGroupList.Invalidate();
+}
+
+
+//-----------------------------------------------------------------------------
+// Purpose: Called when the dialog is closing. Sends a message to update the
+//			visgroup dialog bar in case any visgroup changes were made.
+//-----------------------------------------------------------------------------
+void CEditGroups::OnClose(void)
+{
+	GetMainWnd()->GlobalNotify(WM_MAPDOC_CHANGED);
+	CDialog::OnClose();
+}
+
+
+//-----------------------------------------------------------------------------
+// Purpose: Called when the dialog is closing. Sends a message to update the
+//			visgroup dialog bar in case any visgroup changes were made.
+//-----------------------------------------------------------------------------
+BOOL CEditGroups::DestroyWindow(void)
+{
+	GetMainWnd()->GlobalNotify(WM_MAPDOC_CHANGED);
+	return(CDialog::DestroyWindow());
+}
+
+
+BEGIN_MESSAGE_MAP(CColorBox, CStatic)
+	ON_WM_PAINT()
+END_MESSAGE_MAP()
+
+
+//-----------------------------------------------------------------------------
+// Purpose: Sets the color of the color box.
+// Input  : c - RGB color to set.
+//			bRedraw - TRUE repaints, FALSE does now.
+//-----------------------------------------------------------------------------
+void CColorBox::SetColor(COLORREF c, BOOL bRedraw)
+{
+	m_c = c;
+	if (bRedraw)
+	{
+		RedrawWindow(NULL, NULL, RDW_INVALIDATE | RDW_UPDATENOW);
+	}
+}
+
+
+//-----------------------------------------------------------------------------
+// Purpose: Fills the colorbox window with the current color.
+//-----------------------------------------------------------------------------
+void CColorBox::OnPaint(void)
+{
+	CPaintDC dc(this);
+	CRect r;
+	GetClientRect(r);
+	CBrush brush(m_c);
+	dc.FillRect(r, &brush);
+}
+