#pragma once #include "resource.h" class CModelLibCopyTreeDlg : public CDialog { public: DECLARE_DYNAMIC(CModelLibCopyTreeDlg) enum { IDD = IDD_EQU_LIB_COPY_TREE_DLG }; CModelLibCopyTreeDlg(CImageList &img , CUIISQLiteServer &db , LPCTSTR pszTreeRoot , LPCTSTR pszTablePrefix , CWnd *pParent = NULL); ~CModelLibCopyTreeDlg(); protected: virtual void DoDataExchange(CDataExchange* pDX); virtual BOOL OnInitDialog(); DECLARE_MESSAGE_MAP() public: afx_msg void OnBnClickedOk(); afx_msg void OnBnClickedCancel(); afx_msg void OnNewFolder(); int getClassId() { return m_iClassId; } private: CImageList &m_imageList; CUIISQLiteServer &m_sqlDB; CTreeCtrl m_ctrlDirTree; int m_iClassId; CString m_sTreeRoot; CString m_sTablePrefix; };