1 2 #ifndef _WAIT_DIALOG_ 3 #define _WAIT_DIALOG_ 4 5 6 class wait_dialog : public CDialog 7 { 8 public: 9 DECLARE_DYNAMIC(wait_dialog) 10 wait_dialog(int width, int height, const char *msg, CWnd *parent); // standard constructor 11 virtual ~wait_dialog(); 12 13 enum { IDD = IDD_DIALOG_TEMPLATE }; 14 15 public: SetWorkThread(HANDLE thread)16 void SetWorkThread(HANDLE thread) { m_work_thread = thread; } 17 18 protected: 19 afx_msg int OnCreate(LPCREATESTRUCT lpCreateStruct); 20 afx_msg void OnPaint(); 21 afx_msg void OnShowWindow(BOOL bShow, UINT nStatus); 22 afx_msg LRESULT OnDialogVisible(WPARAM wParam, LPARAM lParam); 23 DECLARE_MESSAGE_MAP() 24 25 void PaintTitleBar(CDC *dc); 26 27 CStatic wait_msg; 28 CString mMessage; 29 int mIconId; 30 int m_width; 31 int m_height; 32 HANDLE m_work_thread; 33 }; 34 35 #endif 36