123

Editor/MainTools.cpp

@@ -0,0 +1,246 @@
+// MainTools.cpp : implementation file
+//
+
+#include "stdafx.h"
+#include "MainTools.h"
+#include "ObjectCreateTool.h"
+
+#ifdef _DEBUG
+#define new DEBUG_NEW
+#undef THIS_FILE
+static char THIS_FILE[] = __FILE__;
+#endif
+
+/////////////////////////////////////////////////////////////////////////////
+// CMainTools dialog
+
+
+CMainTools::CMainTools(CWnd* pParent /*=NULL*/)
+	: CDialog(CMainTools::IDD, pParent)
+{
+	//{{AFX_DATA_INIT(CMainTools)
+	//}}AFX_DATA_INIT
+
+	m_lastPressed = -1;
+	m_nTimer = 0;
+}
+
+CMainTools::~CMainTools()
+{
+	ReleaseButtons();
+}
+
+
+void CMainTools::DoDataExchange(CDataExchange* pDX)
+{
+	CDialog::DoDataExchange(pDX);
+	//{{AFX_DATA_MAP(CMainTools)
+	//}}AFX_DATA_MAP
+}
+
+
+BEGIN_MESSAGE_MAP(CMainTools, CDialog)
+	//{{AFX_MSG_MAP(CMainTools)
+	ON_WM_TIMER()
+	ON_WM_DESTROY()
+	//}}AFX_MSG_MAP
+	ON_WM_SIZE()
+END_MESSAGE_MAP()
+
+/////////////////////////////////////////////////////////////////////////////
+// CMainTools message handlers
+
+BOOL CMainTools::OnCmdMsg(UINT nID, int nCode, void* pExtra, AFX_CMDHANDLERINFO* pHandlerInfo) 
+{
+	// Route the commands to the view
+	if (nID == ID_INSERT_ENTITY && AfxGetMainWnd())
+	{
+		AfxGetMainWnd()->OnCmdMsg(nID, nCode, pExtra, pHandlerInfo);
+	}
+	//((CFrameWnd *) (AfxGetMainWnd()))->OnCmdMsg(nID, nCode, pExtra, pHandlerInfo);
+	
+	return CDialog::OnCmdMsg(nID, nCode, pExtra, pHandlerInfo);
+}
+
+void CMainTools::UncheckAll()
+{
+	m_lastPressed = -1;
+	for (int i = 0; i < m_buttons.size(); i++)
+	{
+		m_buttons[i]->SetCheck( 0 );
+	}
+}
+
+BOOL CMainTools::OnCommand(WPARAM wParam, LPARAM lParam) 
+{
+	// TODO: Add your specialized code here and/or call the base class
+	if (HIWORD(wParam) == BN_CLICKED)
+	{
+		int ctrlId = LOWORD(wParam);
+
+		OnButtonPressed( ctrlId );
+		return TRUE;
+	}
+	
+	return CDialog::OnCommand(wParam, lParam);
+}
+
+BOOL CMainTools::OnInitDialog() 
+{
+	CDialog::OnInitDialog();
+
+	CreateButtons();
+	
+	return FALSE;  // return TRUE unless you set the focus to a control
+	              // EXCEPTION: OCX Property Pages should return FALSE
+}
+
+//////////////////////////////////////////////////////////////////////////
+void CMainTools::CreateButtons()
+{
+	int xOffset = 5;
+	int yOffset = 4;
+
+	int m_buttonHeight = 18;
+
+	int row = 0;
+	int col = 0;
+
+	CRect rc;
+	GetClientRect( rc );
+
+	int buttonWidth = ((rc.right - rc.left) - 3*xOffset)/2;
+
+	std::vector<CString> categories;
+	GetIEditor()->GetObjectManager()->GetClassCategories( categories );
+
+	int newHeight = 0;
+
+	// Get class categories.
+	for (int i = 0; i < categories.size(); i++)
+	{
+		CString category = categories[i];
+
+		//CColorButton *button = new CColorButton;
+		//CColorButton *button = new CColorButton;
+
+		CColorCheckBox *button = new CColorCheckBox;
+		button->SetPushedBkColor( RGB(255,255,0) );
+		//button->SetBevel(1);
+		//button->SetBkColor( RGB(255,255,0) );
+		CRect brc;
+
+		brc.left = xOffset + col*(buttonWidth + xOffset);
+		brc.right = brc.left + buttonWidth;
+
+		brc.top = yOffset + row*(m_buttonHeight + 4);
+		brc.bottom = brc.top + m_buttonHeight;
+
+		newHeight = brc.bottom + 4;
+
+		//button->Create( category,WS_CHILD|WS_VISIBLE|WS_TABSTOP|BS_AUTORADIOBUTTON|BS_PUSHLIKE,brc,this,i );
+		//button->Create( category,WS_CHILD|WS_VISIBLE|WS_TABSTOP|BS_AUTOCHECKBOX,brc,this,i );
+		button->Create( category,WS_CHILD|WS_VISIBLE|WS_TABSTOP|BS_PUSHBUTTON,brc,this,i );
+		button->SetFont( CFont::FromHandle( (HFONT)::GetStockObject(DEFAULT_GUI_FONT)) );
+		//button->ModifyStyleEx( 0,WS_EX_STATICEDGE,SWP_FRAMECHANGED );
+		button->SetCheck(0);
+		m_buttons.push_back( button );
+
+		col++;
+		brc.left = xOffset + col*(buttonWidth + xOffset);
+		brc.right = brc.left + buttonWidth;
+		if (brc.right > rc.right-4)
+		{
+			col = 0;
+			row++;
+		}
+	}
+
+	SetWindowPos( NULL,0,0,rc.right-rc.left,newHeight+4,SWP_NOMOVE );
+}
+
+//////////////////////////////////////////////////////////////////////////
+void CMainTools::ReleaseButtons()
+{
+	for (int i = 0; i < m_buttons.size(); i++)
+	{
+		delete m_buttons[i];
+	}
+	m_buttons.clear();
+}
+
+void CMainTools::OnButtonPressed( int i )
+{
+	ASSERT( i >= 0 && i < m_buttons.size() );
+
+	if (i == m_lastPressed)
+	{
+		UncheckAll();
+		if (GetIEditor()->GetEditTool())
+			GetIEditor()->SetEditTool(0);
+		return;
+	}
+	
+	UncheckAll();
+
+	m_lastPressed = i;
+	m_buttons[i]->SetCheck(1);
+
+	CString category;
+	m_buttons[i]->GetWindowText( category );
+	
+	// Create browse mode for this category.
+	CObjectCreateTool *tool = new CObjectCreateTool;
+	GetIEditor()->SetEditTool( tool );
+	tool->SelectCategory( category );
+
+	// Start monitoring button state.
+	StartTimer();
+}
+
+void CMainTools::OnTimer(UINT_PTR nIDEvent) 
+{
+	CDialog::OnTimer(nIDEvent);
+
+	CEditTool *tool = GetIEditor()->GetEditTool();
+	if (!tool || !(tool->GetRuntimeClass() == RUNTIME_CLASS(CObjectCreateTool)))
+	{
+		UncheckAll();
+		StopTimer();
+	}
+}
+
+void CMainTools::StartTimer()
+{
+	StopTimer();
+	m_nTimer = SetTimer(1,500,NULL);
+}
+	
+void CMainTools::StopTimer()
+{
+	if (m_nTimer != 0)
+		KillTimer(m_nTimer);
+	m_nTimer = 0;
+}
+
+void CMainTools::OnDestroy() 
+{
+	StopTimer();
+	CDialog::OnDestroy();
+}
+
+void CMainTools::OnSize(UINT nType, int cx, int cy)
+{
+	CDialog::OnSize(nType, cx, cy);
+
+	/*
+	static bIgnoreSize = false;
+	if (m_buttons.size() > 0 && !bIgnoreSize)
+	{
+		bIgnoreSize = true;
+		ReleaseButtons();
+		CreateButtons();
+		bIgnoreSize = false;
+	}
+	*/
+}