3d-treedlg.cpp

此文档是用vistual studio 2005 开发的用来描述3D-tree 的生长过程

// 3D-treeDlg.cpp : 实现文件
//

#include "stdafx.h"
#include <GL/glut.h>
#include "3D-tree.h"
#include "3D-treeDlg.h"
#include "Land.h"
#include "MyTree.h"
#include "afxwin.h"
#include <sstream>

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


// 用于应用程序“关于”菜单项的 CAboutDlg 对话框

class CAboutDlg : public CDialog
{
public:
	CAboutDlg();

	// 对话框数据
	enum { IDD = IDD_ABOUTBOX };

protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV 支持

	// 实现
protected:
	DECLARE_MESSAGE_MAP()
public:
	afx_msg void OnBnClickedOk();
public:
	CEdit m_message;
public:
	afx_msg void OnPaint();
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	DDX_Control(pDX, IDC_EDIT_MESSAGE, m_message);
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
	ON_BN_CLICKED(IDOK, &CAboutDlg::OnBnClickedOk)
	ON_WM_PAINT()
END_MESSAGE_MAP()


// CMy3DtreeDlg 对话框




CMy3DtreeDlg::CMy3DtreeDlg(CWnd* pParent /*=NULL*/)
: CDialog(CMy3DtreeDlg::IDD, pParent)
{
	PixelFormat=0;
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CMy3DtreeDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CMy3DtreeDlg, CDialog)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	//}}AFX_MSG_MAP
	ON_WM_SIZE()
	ON_WM_TIMER()
	ON_WM_MOUSEMOVE()
	ON_WM_LBUTTONDOWN()
	ON_WM_LBUTTONUP()
	ON_WM_WINDOWPOSCHANGED()
	ON_COMMAND(ID_MENUITEM_ABOUT, &CMy3DtreeDlg::OnMenuitemAbout)
	ON_COMMAND(ID_MENUITEM_EXIT, &CMy3DtreeDlg::OnMenuitemExit)
	ON_COMMAND(ID_MUNUITEM_HELP, &CMy3DtreeDlg::OnMunuitemHelp)
	ON_BN_CLICKED(IDCANCEL, &CMy3DtreeDlg::OnBnClickedCancel)
	ON_COMMAND(ID_MENUITEM_CONTROL, &CMy3DtreeDlg::OnMenuitemControl)
	ON_COMMAND(ID_FULLSCREEN, &CMy3DtreeDlg::OnFullscreen)
	ON_WM_CLOSE()
	ON_WM_RBUTTONUP()
	ON_COMMAND(ID_ORTHO_VIEWER, &CMy3DtreeDlg::OnOrthoViewer)
	ON_COMMAND(ID_OVERLOOK_VIEWER, &CMy3DtreeDlg::OnOverlookViewer)
	ON_COMMAND(ID_SIDE_VIEWER, &CMy3DtreeDlg::OnSideViewer)
	ON_COMMAND(ID_PERSPECTIVE_VIEWER, &CMy3DtreeDlg::OnPerspectiveViewer)
	ON_COMMAND(ID_MENUITEM_INFO, &CMy3DtreeDlg::OnMenuitemInfo)
	ON_WM_NCLBUTTONDBLCLK()
	ON_WM_DESTROY()
	ON_COMMAND(ID_MENUITEM_AUTOMOVE, &CMy3DtreeDlg::OnMenuitemAutomove)
END_MESSAGE_MAP()


// CMy3DtreeDlg 消息处理程序

BOOL CMy3DtreeDlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	ini = new ChIni("user.ini");
	startX = 0, startY = 0, endX = 0, endY = 0;
	showTips = ini->readValueAsBoolean( "Config", "ShowTips" );
	rotateX = ini->readValueAsDouble( "Config", "rotateX" );
	rotateY = ini->readValueAsDouble( "Config", "rotateY" );
	translateX = ini->readValueAsDouble( "Config", "translateX" );
	translateY = ini->readValueAsDouble( "Config", "translateY" );
	translateZ = ini->readValueAsDouble( "Config", "translateZ" );
	changeStep = 0.1f;
	isDrag = false;
	showGrowProcess = false;
	isAutoRotate = false;
	isDrawForest = false;

	//CDC* m_pDC;
	//m_pDC = GetDC();//得到当前窗体的设备上下文对象；
	//int m_pixelsX = m_pDC->GetDeviceCaps(HORZRES);//获取屏幕的宽度；
	//int m_pixelsY = m_pDC->GetDeviceCaps(VERTRES);//获取屏幕的高度；
	//mainRect.top = 0;
	//mainRect.left = 0;
	//mainRect.bottom = m_pixelsY;
	//mainRect.right = m_pixelsX;

	// 将“关于...”菜单项添加到系统菜单中。

	// IDM_ABOUTBOX 必须在系统命令范围内。
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);
	if (pSysMenu != NULL)
	{
		CString strAboutMenu;
		strAboutMenu.LoadString(IDS_ABOUTBOX);
		if (!strAboutMenu.IsEmpty())
		{
			pSysMenu->AppendMenu(MF_SEPARATOR);
			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
		}
	}

	// 设置此对话框的图标。当应用程序主窗口不是对话框时，框架将自动
	//  执行此操作
	SetIcon(m_hIcon, TRUE);			// 设置大图标
	SetIcon(m_hIcon, FALSE);		// 设置小图标

	// TODO: 在此添加额外的初始化代码

	pMenu = GetMenu();
	CMy3DtreeDlg::ShowWindow( SW_SHOWMAXIMIZED );
	//CMy3DtreeDlg::ShowWindow( SW_HIDE );
	this->GetClientRect( &mainRect );//获得主窗口大小

	controlPanel = new CControlDialog(this);
	controlPanel->Create( IDD_DIALOG_CONTROL, this );
	controlPanel->GetClientRect( &controlRect );
	GetDlgItem(IDC_RENDER)->SetWindowPos(NULL,0,0,mainRect.Width() - controlRect.Width(),mainRect.Height(),SWP_SHOWWINDOW);
	controlPanel->SetWindowPos( NULL, mainRect.right - controlRect.Width(), 0, controlRect.Width(), mainRect.Height(), SWP_SHOWWINDOW );

	CWnd *wnd=GetDlgItem(IDC_RENDER);
	hrenderDC=::GetDC(wnd->m_hWnd);
	if(SetWindowPixelFormat(hrenderDC)==FALSE) {
		return 0;
	}

	if(CreateViewGLContext(hrenderDC)==FALSE) {
		return 0;
	}

	glMatrixMode( GL_PROJECTION );
	glLoadIdentity();
	CRect rect;
	wnd->GetClientRect( &rect );
	gluPerspective( 45.0f, ( float )( rect.Width() ) / ( float )( rect.Height() ), 0.1f, 100.0f );
	glMatrixMode( GL_MODELVIEW );
	glLoadIdentity();
	InitGL();

	tree = new MyTree( 10, 1.0f, 0.1f, 70.0f, 25.0f, 8, 0.1 );
	land = new Land( 50.0f, "ground.bmp" );

	controlPanel->setControlTree( tree, land, &showGrowProcess, &rotateX, &rotateY, 
								&translateX, &translateY, &translateZ, &changeStep, 
								&isAutoRotate, &isDrawForest, &reGenerateRandomTree );
	CMy3DtreeDlg::ShowWindow( SW_SHOWMAXIMIZED );
	infoDlg = new CInfoDlg();
	if ( showTips == true ) {
		SetTimer( 2, 100, 0 );
	}
	return TRUE;  // 除非将焦点设置到控件，否则返回 TRUE
}

void CMy3DtreeDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	{
		CAboutDlg dlgAbout;
		KillTimer(1);
		dlgAbout.DoModal();
		SetTimer( 1, 10, 0 );

	}
	else
	{
		CDialog::OnSysCommand(nID, lParam);
	}
}

bool CMy3DtreeDlg::InitGL() {
	float light_ambient[] = { 0.0, 0.0, 0.0, 1.0 };
	float light_diffuse[] = { 1.0, 1.0, 1.0, 1.0 };
	float light_specular[] = { 1.0, 1.0, 1.0, 1.0 };
	float light_position[] = { 1.0, 1.0, 1.0, 0.0 };

	glShadeModel( GL_SMOOTH );
	glClearColor( 0.45f, 0.75f, 0.95f, 0.0f );
	glClearDepth( 1.0f );

	glEnable( GL_TEXTURE_2D ); /* Enable Texture Mapping */
	glPixelStorei( GL_UNPACK_ALIGNMENT, 1 ); /* Pixel Storage Mode To Byte Alignment */
	glHint( GL_PERSPECTIVE_CORRECTION_HINT, GL_NICEST );

	glLightfv( GL_LIGHT0, GL_AMBIENT, light_ambient );
	glLightfv( GL_LIGHT0, GL_DIFFUSE, light_diffuse );
	glLightfv( GL_LIGHT0, GL_SPECULAR, light_specular );
	glLightfv( GL_LIGHT0, GL_POSITION, light_position );
	glEnable( GL_LIGHTING );
	glEnable( GL_LIGHT0 );

	glEnable( GL_COLOR_MATERIAL );

	glEnable( GL_DEPTH_TEST );
	glDepthFunc( GL_LEQUAL );
	//glEnable(GL_LINE_SMOOTH); // 启用反走样
	SetTimer( 1, 10, 0 );
	return true;
}

// 如果向对话框添加最小化按钮，则需要下面的代码
//  来绘制该图标。对于使用文档/视图模型的 MFC 应用程序，
//  这将由框架自动完成。

void CMy3DtreeDlg::OnPaint()
{
	if (IsIconic())
	{
		CPaintDC dc(this); // 用于绘制的设备上下文

		SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);

		// 使图标在工作矩形中居中
		int cxIcon = GetSystemMetrics(SM_CXICON);
		int cyIcon = GetSystemMetrics(SM_CYICON);
		CRect rect;
		GetClientRect(&rect);
		int x = (rect.Width() - cxIcon + 1) / 2;
		int y = (rect.Height() - cyIcon + 1) / 2;

		// 绘制图标
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialog::OnPaint();
	}
}

//当用户拖动最小化窗口时系统调用此函数取得光标显示。
//
HCURSOR CMy3DtreeDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}


void CMy3DtreeDlg::OnSize(UINT nType, int cx, int cy)
{
	CDialog::OnSize(nType, cx, cy);
}

BOOL CMy3DtreeDlg::SetWindowPixelFormat(HDC hDC)
{
	PIXELFORMATDESCRIPTOR pixelDesc;

	pixelDesc.nSize = sizeof(PIXELFORMATDESCRIPTOR);
	pixelDesc.nVersion = 1;

	pixelDesc.dwFlags = PFD_DRAW_TO_WINDOW |
		PFD_SUPPORT_OPENGL |
		PFD_DOUBLEBUFFER |
		PFD_TYPE_RGBA;

	pixelDesc.iPixelType = PFD_TYPE_RGBA;
	pixelDesc.cColorBits = 32;
	pixelDesc.cRedBits = 0;
	pixelDesc.cRedShift = 0;
	pixelDesc.cGreenBits = 0;
	pixelDesc.cGreenShift = 0;
	pixelDesc.cBlueBits = 0;
	pixelDesc.cBlueShift = 0;
	pixelDesc.cAlphaBits = 0;
	pixelDesc.cAlphaShift = 0;
	pixelDesc.cAccumBits = 0;
	pixelDesc.cAccumRedBits = 0;
	pixelDesc.cAccumGreenBits = 0;
	pixelDesc.cAccumBlueBits = 0;
	pixelDesc.cAccumAlphaBits = 0;
	pixelDesc.cDepthBits = 0;
	pixelDesc.cStencilBits = 1;
	pixelDesc.cAuxBuffers = 0;
	pixelDesc.iLayerType = PFD_MAIN_PLANE;
	pixelDesc.bReserved = 0;
	pixelDesc.dwLayerMask = 0;
	pixelDesc.dwVisibleMask = 0;
	pixelDesc.dwDamageMask = 0;

	PixelFormat = ChoosePixelFormat(hDC,&pixelDesc);
	if(PixelFormat==0) // Choose default
	{
		PixelFormat = 1;
		if(DescribePixelFormat(hDC,PixelFormat,
			sizeof(PIXELFORMATDESCRIPTOR),&pixelDesc)==0)
		{
			return FALSE;
		}
	}

	if(SetPixelFormat(hDC,PixelFormat,&pixelDesc)==FALSE)

	{
		return FALSE;
	}

	return TRUE;
}


BOOL CMy3DtreeDlg::CreateViewGLContext(HDC hDC)
{
	hrenderRC = wglCreateContext(hDC);

	if(hrenderRC==NULL)
		return FALSE;

	if(wglMakeCurrent(hDC,hrenderRC)==FALSE)
		return FALSE;
	return TRUE;
}

int rand_locate[ 100 ];
bool firstGenForest = true;
void CMy3DtreeDlg::RenderScene()  
{
	glClear( GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT );
	glLoadIdentity();

	glTranslatef( translateX, translateY, translateZ );
	glRotatef( rotateY, 1.0f, 0.0f, 0.0f );
	glRotatef( rotateX, 0.0f, 1.0f, 0.0f );

	land->draw();

	if ( isDrawForest == true ) {
		if ( reGenerateRandomTree == true ) {
			srand( ( int )time( 0 ) );
			if ( firstGenForest == true ) {