mytreeview.cpp

一个非常不错的MFC ODBC范例

// MyTreeView.cpp : implementation of the CMyTreeView class
//

#include "stdafx.h"
#include "MainFrm.h"
#include "MyMFCODBCSmpl.h"
#include "MyMFCODBCSmplDoc.h"
#include "MyTreeView.h"
#include "resource.h"
#include "CategorySet.h"
#include "ProductSet.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CMyTreeView

IMPLEMENT_DYNCREATE(CMyTreeView, CTreeView)

BEGIN_MESSAGE_MAP(CMyTreeView, CTreeView)
	//{{AFX_MSG_MAP(CMyTreeView)
	ON_COMMAND(ID_DEL_CATEG, OnDelCateg)
	ON_NOTIFY_REFLECT(TVN_SELCHANGED, OnSelchanged)
	//}}AFX_MSG_MAP
	// Standard printing commands
	ON_COMMAND(ID_FILE_PRINT, CTreeView::OnFilePrint)
	ON_COMMAND(ID_FILE_PRINT_DIRECT, CTreeView::OnFilePrint)
	ON_COMMAND(ID_FILE_PRINT_PREVIEW, CTreeView::OnFilePrintPreview)
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CMyTreeView construction/destruction

CMyTreeView::CMyTreeView()
{

}

CMyTreeView::~CMyTreeView()
{
}

BOOL CMyTreeView::PreCreateWindow(CREATESTRUCT& cs)
{
	return CTreeView::PreCreateWindow(cs);
}

/////////////////////////////////////////////////////////////////////////////
// CMyTreeView drawing

void CMyTreeView::OnDraw(CDC* pDC)
{
	CMyMFCODBCSmplDoc* pDoc = GetDocument();
	ASSERT_VALID(pDoc);
}

void CMyTreeView::OnInitialUpdate()
{
	CTreeView::OnInitialUpdate();
	//建立产品信息树
	CreateMyTree();
}

/////////////////////////////////////////////////////////////////////////////
// CMyTreeView printing

BOOL CMyTreeView::OnPreparePrinting(CPrintInfo* pInfo)
{
	// default preparation
	return DoPreparePrinting(pInfo);
}

void CMyTreeView::OnBeginPrinting(CDC* /*pDC*/, CPrintInfo* /*pInfo*/)
{
	// TODO: add extra initialization before printing
}

void CMyTreeView::OnEndPrinting(CDC* /*pDC*/, CPrintInfo* /*pInfo*/)
{
	// TODO: add cleanup after printing
}

/////////////////////////////////////////////////////////////////////////////
// CMyTreeView diagnostics

#ifdef _DEBUG
void CMyTreeView::AssertValid() const
{
	CTreeView::AssertValid();
}

void CMyTreeView::Dump(CDumpContext& dc) const
{
	CTreeView::Dump(dc);
}

CMyMFCODBCSmplDoc* CMyTreeView::GetDocument() // non-debug version is inline
{
	ASSERT(m_pDocument->IsKindOf(RUNTIME_CLASS(CMyMFCODBCSmplDoc)));
	return (CMyMFCODBCSmplDoc*)m_pDocument;
}
#endif //_DEBUG

/////////////////////////////////////////////////////////////////////////////
// CMyTreeView message handlers

void CMyTreeView::CreateMyTree()
{
	CTreeCtrl& PrdtTreeCtrl = GetTreeCtrl();
	// 建立树图标链
	CImageList  *pimagelist= new CImageList;
	pimagelist->Create(MAKEINTRESOURCE(IDB_PRDT_BMP),16,1,RGB(255,255,255));
	PrdtTreeCtrl.SetImageList(pimagelist, TVSIL_NORMAL);
	//结点基本信息
	TV_INSERTSTRUCT tvinsert;
	tvinsert.hParent = NULL;
	tvinsert.hInsertAfter = TVI_LAST;
	tvinsert.item.mask = TVIF_IMAGE | TVIF_SELECTEDIMAGE| TVIF_TEXT;
	tvinsert.item.hItem = NULL;
	tvinsert.item.state = 0;
	tvinsert.item.stateMask = 0;
	tvinsert.item.cchTextMax = 6;
	tvinsert.item.iSelectedImage = 0;
	tvinsert.item.cChildren = 0;
	tvinsert.item.lParam = 0;
	//从数据库中读出相关信息并插入到树上
	CDatabase db;
	CCategorySet* pCateSet = new CCategorySet(&db);
	if(!pCateSet->IsOpen())pCateSet->Open();
	if(!pCateSet->IsBOF())pCateSet->MoveFirst();
	while(!pCateSet->IsEOF())
	{
		CString strTemp = pCateSet->m_CategoryName;
		tvinsert.item.pszText = strTemp.GetBuffer(strTemp.GetLength());
		tvinsert.item.iImage = 0;
		tvinsert.item.iSelectedImage = 0;
		tvinsert.hParent = NULL;
		HTREEITEM hTop = PrdtTreeCtrl.InsertItem(&tvinsert);
		PrdtTreeCtrl.SetItemData(hTop,pCateSet->m_CategoryID*10+0);
		//读出每个分类的产品
		CProductSet* pPrdtSet = new CProductSet(&db);
		char strID[10];
		_itoa(pCateSet->m_CategoryID,strID,10);
		//过滤
		pPrdtSet->m_strFilter = CString("CategoryID = ") + strID ;
		if(!pPrdtSet->IsOpen())pPrdtSet->Open();
		if(!pPrdtSet->IsBOF())pPrdtSet->MoveFirst();
		while(!pPrdtSet->IsEOF())
		{
			strTemp = pPrdtSet->m_ProductName;
			tvinsert.item.pszText = strTemp.GetBuffer(strTemp.GetLength());
			tvinsert.item.iImage = 1;
			tvinsert.item.iSelectedImage = 1;
			tvinsert.hParent = hTop;
			HTREEITEM hPrdt = PrdtTreeCtrl.InsertItem(&tvinsert);
			PrdtTreeCtrl.SetItemData(hPrdt,pPrdtSet->m_ProductID*10+1);
			pPrdtSet->MoveNext();
		}
		pPrdtSet->Close();
		delete pPrdtSet;
		pCateSet->MoveNext();
	}

	pCateSet->Close();
	delete pCateSet;
}
//删除产品分类
void CMyTreeView::OnDelCateg() 
{
	if( nCurCategID <= 0 )	return;
	CDatabase db;
	char ch[8];
	_itoa(nCurCategID,ch,10);
	if( !db.IsOpen() ) db.Open(_T("MyNWind"));
	//调用存储过程
	db.ExecuteSQL("EXEC SP_DEL_CATEG " +CString(ch));	
	db.Close();
	CTreeCtrl& tc = GetTreeCtrl();
	tc.DeleteItem(tc.GetSelectedItem());
	UpdateWindow();
}

void CMyTreeView::OnSelchanged(NMHDR* pNMHDR, LRESULT* pResult) 
{
	NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;

	CTreeCtrl& PrdtTreeCtrl = GetTreeCtrl();
	CPoint		ptAction;
	UINT		nFlags;
	GetCursorPos(&ptAction);
	PrdtTreeCtrl.ScreenToClient(&ptAction);
	HTREEITEM hItem = PrdtTreeCtrl.HitTest(ptAction,&nFlags);
	if(hItem==NULL)return;
	LPARAM para = PrdtTreeCtrl.GetItemData(hItem);
	//计算当前活动的产品分类ID
	nCurCategID = para/10;
	CMainFrame * pMain = (CMainFrame*)AfxGetMainWnd();
	CDocument * pDoc = pMain->GetActiveDocument();
	//发送消息
	if( pDoc )
		pDoc->UpdateAllViews(NULL,para);
	
	*pResult = 0;
}