mytree.cpp

这是本人两年前兼职为某个公司做的石油钻进设计软件

// MyTree.cpp : implementation file
//

#include "stdafx.h"
#include "CVenus.h"
#include "MyTree.h"

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

/////////////////////////////////////////////////////////////////////////////
// CMyTree

static CString csProjectTree[] =
{
	_T("工程"),
	_T("(1)工艺初选"),
	_T("(2)自喷井气水节点分析"),
	_T("(3)气举排水采气节点分析"),
	_T("(4)气举排水采气优化设计"),
	_T("(5)抽油机排水采气优化设计"),
	_T("(6)电潜泵排水采气节点分析"),
	_T("(7)电潜泵排水采气优化设计"),
	_T("(8)射流泵排水采气节点分析"),
	_T("(9)射流泵排水采气优化设计"),
	_T("(10)气井连续排液临界流量"),
	_T("(11)多相流垂管流压力梯度"),
	_T("(12)生产气井压力梯度"),
	_T("(13)注气井压力梯度"),
};

CMyTree::CMyTree()
{
}

CMyTree::~CMyTree()
{
}


BEGIN_MESSAGE_MAP(CMyTree, CTreeCtrl)
	//{{AFX_MSG_MAP(CMyTree)
	ON_NOTIFY_REFLECT(NM_DBLCLK, OnDblclk)
	ON_NOTIFY_REFLECT(NM_RETURN, OnReturn)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CMyTree message handlers

void CMyTree::OnDblclk(NMHDR* pNMHDR, LRESULT* pResult) 
{
	HTREEITEM hTreeItem1 = GetSelectedItem();
	CString str = GetItemText(hTreeItem1);

	if( ((CCVenusApp*)AfxGetApp())->m_pGraphDlg == NULL ) return;

	for(int i = 1; i <= 13; i++)
	{
		if(str.Compare(csProjectTree[i]) == 0)
		{
			((CCVenusApp*)AfxGetApp())->OnProject(IDM_PROJECT_1 + i - 1);
			break;
		}
	}
	
	*pResult = 0;
}

void CMyTree::OnReturn(NMHDR* pNMHDR, LRESULT* pResult) 
{
	HTREEITEM hTreeItem1 = GetSelectedItem();
	CString str = GetItemText(hTreeItem1);

	if( ((CCVenusApp*)AfxGetApp())->m_pGraphDlg == NULL ) return;

	for(int i = 1; i <= 13; i++)
	{
		if(str.Compare(csProjectTree[i]) == 0)
		{
			((CCVenusApp*)AfxGetApp())->OnProject(IDM_PROJECT_1 + i - 1);
			break;
		}
	}
		
	*pResult = 0;
}