www.pudn.com > Sterren_ASe_Explorer.rar > NodeView.cpp

/* Copyright (C) William van der Sterren, 2002.  
 * All rights reserved worldwide. 
 * 
 * This software is provided "as is" without express or implied 
 * warranties. You may freely copy and compile this source into 
 * applications you distribute provided that the copyright text 
 * below is included in the resulting source code, for example: 
 * "Portions Copyright (C) William van der Sterren, 2002" 
 */ 
 
/* Copyright (C) James Matthews, 2001.  
 * All rights reserved worldwide. 
 * 
 * This software is provided "as is" without express or implied 
 * warranties. You may freely copy and compile this source into 
 * applications you distribute provided that the copyright text 
 * below is included in the resulting source code, for example: 
 * "Portions Copyright (C) James Matthews, 2001" 
 */ 
// LeftView.cpp : implementation of the CNodeView class 
// 
 
#include "stdafx.h" 
#include "ase.h" 
 
#include "aseDoc.h" 
#include "NodeView.h" 
 
#ifdef _DEBUG 
#define new DEBUG_NEW 
#undef THIS_FILE 
static char THIS_FILE[] = __FILE__; 
#endif 
 
///////////////////////////////////////////////////////////////////////////// 
// CNodeView 
 
IMPLEMENT_DYNCREATE(CNodeView, CTreeView) 
 
BEGIN_MESSAGE_MAP(CNodeView, CTreeView) 
	//{{AFX_MSG_MAP(CNodeView) 
	ON_NOTIFY_REFLECT(TVN_SELCHANGED, OnSelChanged) 
	//}}AFX_MSG_MAP 
END_MESSAGE_MAP() 
 
///////////////////////////////////////////////////////////////////////////// 
// CNodeView construction/destruction 
 
TVSORTCB CNodeView::m_tvSortData; 
 
CNodeView::CNodeView() 
{ 
	m_cImageList.Create(IDB_NODEICONS,16,16,RGB(255,0,255)); 
 
	m_cTahoma.CreatePointFont(80, "Verdana"); 
} 
 
CNodeView::~CNodeView() 
{ 
} 
 
BOOL CNodeView::PreCreateWindow(CREATESTRUCT& cs) 
{ 
	// TODO: Modify the Window class or styles here by modifying 
	//  the CREATESTRUCT cs 
 
	return CTreeView::PreCreateWindow(cs); 
} 
 
///////////////////////////////////////////////////////////////////////////// 
// CNodeView drawing 
 
void CNodeView::OnDraw(CDC* pDC) 
{ 
	CAseDoc* pDoc = GetDocument(); 
	ASSERT_VALID(pDoc); 
 
	// TODO: add draw code for native data here 
} 
 
 
void CNodeView::OnInitialUpdate() 
{ 
	CTreeView::OnInitialUpdate(); 
 
	CTreeCtrl &ctrl = GetTreeCtrl(); 
 
	ctrl.SetFont(&m_cTahoma, true); 
	ctrl.SetImageList(&m_cImageList, TVSIL_NORMAL); 
	ctrl.DeleteAllItems(); 
 
	m_tvSortData.hParent = NULL; 
	m_tvSortData.lpfnCompare = SortByF; 
	m_tvSortData.lParam = 0; 
} 
 
///////////////////////////////////////////////////////////////////////////// 
// CNodeView diagnostics 
 
#ifdef _DEBUG 
void CNodeView::AssertValid() const 
{ 
	CTreeView::AssertValid(); 
} 
 
void CNodeView::Dump(CDumpContext& dc) const 
{ 
	CTreeView::Dump(dc); 
} 
 
CAseDoc* CNodeView::GetDocument() // non-debug version is inline 
{ 
	ASSERT(m_pDocument->IsKindOf(RUNTIME_CLASS(CAseDoc))); 
	return (CAseDoc*)m_pDocument; 
} 
#endif //_DEBUG 
 
///////////////////////////////////////////////////////////////////////////// 
// CNodeView message handlers 
 
BOOL CNodeView::Create(LPCTSTR lpszClassName, LPCTSTR lpszWindowName, DWORD dwStyle, const RECT& rect, CWnd* pParentWnd, UINT nID, CCreateContext* pContext)  
{ 
	dwStyle |= TVS_HASLINES | TVS_LINESATROOT | TVS_HASBUTTONS | TVS_SHOWSELALWAYS; 
	 
	return CWnd::Create(lpszClassName, lpszWindowName, dwStyle, rect, pParentWnd, nID, pContext); 
} 
 
int CNodeView::OnNotifyChild(_asNode *parent, _asNode *child, int data, void *pThis) 
{ 
	CNodeView *me = reinterpret_cast(pThis); 
	CTreeCtrl &treectrl = me->GetTreeCtrl(); 
 
	CString str; 
	str.Format("%d,%d: f=% .1f, g=% .1f, h=% .1f", child->GetX(), child->GetY(), child->GetF(), child->GetG(), child->GetH()); 
 
	HTREEITEM hNew; 
	HTREEITEM hTree = (parent) ? HTREEITEM(parent->dataptr) : treectrl.GetRootItem(); 
 
	me->GetDocument()->NodeAdded(child, data); 
 
	int icon = data; 
	hNew = treectrl.InsertItem(str, icon, icon, hTree); 
 
	child->dataptr = (void *)(hNew); 
	treectrl.SetItemData(hNew, DWORD(child)); 
 
	return 0; 
} 
 
int CNodeView::OnNotifyList(_asNode *previous, _asNode *addnode, int data, void *pThis) 
{ 
	CNodeView *me = reinterpret_cast(pThis); 
	 
	HTREEITEM hItem; 
	HTREEITEM hPrevious = (previous) ? HTREEITEM(previous->dataptr) : NULL; 
 
	CString str; 
	CTreeCtrl &treectrl = me->GetTreeCtrl(); 
	 
	if (data < 3) { 
		hItem = treectrl.GetNextItem(treectrl.GetRootItem(), TVGN_NEXT); 
		str.Format("%d,%d: f = % .2f", addnode->GetX(), addnode->GetY(), addnode->GetF()); 
 
		if (data == 0) { 
			hItem = treectrl.InsertItem(str, 2, 2, hItem, hPrevious); 
			treectrl.SetItemData(hItem, DWORD(addnode)); 
		} else if (data == 1) { 
			hItem = treectrl.InsertItem(str, 2, 2, hItem, hItem); 
			treectrl.SetItemData(hItem, DWORD(addnode)); 
		} else if (data == 2) { 
			hItem = treectrl.GetChildItem(hItem); 
			if (hItem) treectrl.DeleteItem(hItem); 
		} 
 
		me->GetDocument()->ListChanges(addnode, 1); 
 
	} else if (data == 3) { 
		hItem = treectrl.GetRootItem(); 
		hItem = treectrl.GetNextItem(hItem, TVGN_NEXT); 
		hItem = treectrl.GetNextItem(hItem, TVGN_NEXT); 
 
		str.Format("%d,%d", addnode->GetX(), addnode->GetY()); 
 
		hItem = treectrl.InsertItem(str, 4, 4, hItem, NULL); 
		treectrl.SetItemData(hItem, DWORD(addnode)); 
		 
		me->GetDocument()->ListChanges(addnode, 2); 
	} 
 
	return 0; 
} 
 
int CNodeView::OnUpdateList(_asNode *previous, _asNode *addnode, int data, void *pThis) 
{ 
	CNodeView *me = reinterpret_cast(pThis); 
 
	if (data < 3) me->GetDocument()->ListChanges(addnode, 1); 
	else if (data == 3) me->GetDocument()->ListChanges(addnode, 2); 
 
	return 0; 
} 
 
void CNodeView::OnSelChanged(NMHDR* pNMHDR, LRESULT* pResult)  
{ 
	NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR; 
 
	HTREEITEM hTreeItem = pNMTreeView->itemNew.hItem; 
	_asNode *node = (_asNode *)(GetTreeCtrl().GetItemData(hTreeItem)); 
 
	GetDocument()->DrawNode(node); 
 
	*pResult = 0; 
} 
 
void CNodeView::OnPreAStar() 
{ 
	m_hGoalNode = NULL; 
	GetTreeCtrl().DeleteAllItems(); 
} 
 
void CNodeView::OnPostAStar(_asNode *end) 
{ 
	ASSERT(end); 
 
	m_hGoalNode = HTREEITEM(end->dataptr); 
	GetTreeCtrl().SetItemState(m_hGoalNode, TVIS_BOLD, TVIS_BOLD); 
} 
 
void CNodeView::DisplayGoalNode() 
{ 
	if (m_hGoalNode) GetTreeCtrl().EnsureVisible(m_hGoalNode); 
} 
 
int CNodeView::SortByF(LPARAM lParam1, LPARAM lParam2, LPARAM lParamSort) 
{ 
	_asNode *item1 = reinterpret_cast<_asNode *>(lParam1); 
	_asNode *item2 = reinterpret_cast<_asNode *>(lParam2); 
 
	if ( item1->GetF() <= item2->GetF() )  
    return -1; 
 
	return 1; 
} 
 
void CNodeView::SortOpen() 
{ 
	HTREEITEM hItem; 
	hItem = GetTreeCtrl().GetNextItem(GetTreeCtrl().GetRootItem(), TVGN_NEXT); 
 
	if (hItem) { 
		m_tvSortData.hParent = hItem; 
 
		GetTreeCtrl().SortChildrenCB(&m_tvSortData); 
	} 
}