开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 行业应用 > 金融证券系统 > 查看源码
StkUI.cpp
资源名称：clking_src.rar　[点击查看]
上传用户：zhanglf88
上传日期：2013-11-19
资源大小：6036k
文件大小：25k
源码类别：

金融证券系统

开发平台：

Visual C++

StkUI.cpp：源码内容
  1. // StkUI.cpp : Defines the class behaviors for the application.
  2. //
  4. #include "stdafx.h"
  5. #include "StkUI.h"
  6. #include <io.h>
  7. #include <afxsock.h>
  9. #include "MainFrm.h"
  10. #include "ChildFrm.h"
  11. #include "StaticDoc.h"
  12. #include "View/SimuView.h"
  13. #include "View/InfoView.h"
  14. #include "Dialog/SetPrpt.h"
  15. #include "Dialog/SelectStk.h"
  16. #include "Dialog/SetRate.h"
  17. #include "Dialog/SetRule.h"
  18. #include "Dialog/SimuReport.h"
  19. #include "Dialog/SimuRealOp.h"
  20. #include "Dialog/CloseStrategyDlg.h"
  22. /////////////////////////////////////////////////////////////////////////////
  23. // CStkUIApp
  25. BEGIN_MESSAGE_MAP(CStkUIApp, CWinApp)
  26. //{{AFX_MSG_MAP(CStkUIApp)
  27. ON_COMMAND(ID_APP_ABOUT, OnAppAbout)
  28. ON_COMMAND(ID_FILE_NEW, OnFileNew)
  29. ON_COMMAND(ID_STRATEGY_NEW, OnStrategyNew)
  30. ON_COMMAND(ID_FILE_OPEN, OnFileOpen)
  31. ON_COMMAND(ID_STRATEGY_SAVE, OnStrategySave)
  32. ON_UPDATE_COMMAND_UI(ID_STRATEGY_SAVE, OnUpdateStrategySave)
  33. ON_COMMAND(ID_STRATEGY_SAVE_AS, OnStrategySaveAs)
  34. ON_UPDATE_COMMAND_UI(ID_STRATEGY_SAVE_AS, OnUpdateStrategySaveAs)
  35. ON_COMMAND(ID_STRATEGY_CLOSE, OnStrategyClose)
  36. ON_UPDATE_COMMAND_UI(ID_STRATEGY_CLOSE, OnUpdateStrategyClose)
  37. ON_COMMAND(ID_STRATEGY_SAVEALL, OnStrategySaveall)
  38. ON_UPDATE_COMMAND_UI(ID_STRATEGY_SAVEALL, OnUpdateStrategySaveall)
  39. //}}AFX_MSG_MAP
  40. END_MESSAGE_MAP()
  42. /////////////////////////////////////////////////////////////////////////////
  43. // CStkUIApp construction
  45. CStkUIApp::CStkUIApp()
  46. {
  47. m_pStaticDoc = NULL;
  48. m_bAutoUpdateViews = TRUE;
  49. }
  51. UINT LoadKDataCacheMain(LPVOID pParam)
  52. {
  53. AfxGetDB().LoadKDataCache(AfxGetStockContainer(), NULL, NULL, 0, STKLIB_MAX_PROGRESS);
  54. AfxGetStockContainer().OnDataChanged();
  56. AfxGetStkReceiver().RefreshStockContainer(AfxGetStockContainer(), TRUE);
  58. AfxReloadStockMain();
  60. ::PostMessage(AfxGetMainWnd()->GetSafeHwnd(), WM_USER_UPDATESLISTVIEW,0, 0);
  61. ::PostMessage(AfxGetMainWnd()->GetSafeHwnd(), WM_USER_INITDATES, 0, 0);
  63. AfxEndThread(0, TRUE);
  64. return 0;
  65. }
  67. BOOL CALLBACK LoadProgram(HWND hWnd, int nMinProgress, int nMaxProgress)
  68. {
  69. int nMin = nMinProgress;
  70. int nMax = nMaxProgress;
  71. int nRange = nMaxProgress-nMinProgress;
  72. CStkUIApp * pApp = AfxGetStkUIApp();
  73. if (NULL == pApp)
  74. return FALSE;
  76. CStaticDoc * pStaticDoc = pApp->GetStaticDoc();
  77. CMainFrame * pMainFrame = AfxGetMainFrame();
  79. HINSTANCE hInstanceOld = AfxGetResourceHandle();
  80. AfxSetResourceHandle(AfxGetInstanceHandle());
  82. SendMessage(hWnd, WM_USER_STARTUP_PROGRESS, nMin, 0);
  84. AfxGetStockContainer().Load(&AfxGetDB(), NULL, NULL);
  85. AfxGetDomainContainer().Load(AfxGetProfile().GetDomainFile());
  86. AfxGetGroupContainer().Load(AfxGetProfile().GetGroupFile());
  87. SendMessage(hWnd, WM_USER_STARTUP_PROGRESS, nMin+nRange*4/30, 0);
  89. pMainFrame->m_SearchBox.InitStocks(TRUE, TRUE, TRUE);
  90. SendMessage(hWnd, WM_USER_STARTUP_PROGRESS, nMin+nRange*8/30, 0);
  92. // Open Simulation Strategy
  93. if (NULL == pApp->GetFirstStrategyPosition())
  94. {
  95. if (0 == pApp->OpenLastOpenedStrategy())
  96. pApp->OpenDefaultStrategy();
  97. }
  98. SendMessage(hWnd, WM_USER_STARTUP_PROGRESS, nMin+nRange*12/30, 0);
  100. // Show Views
  101. //if (!pStaticDoc->GetViewIfExist(RUNTIME_CLASS(CWizardView)))
  102. // pStaticDoc->ShowStaticView(RUNTIME_CLASS(CWizardView), TRUE);
  103. //SendMessage(hWnd, WM_USER_STARTUP_PROGRESS, nMin+nRange*14/30, 0);
  104. //if (!pStaticDoc->GetViewIfExist(RUNTIME_CLASS(CSimuView)))
  105. // pStaticDoc->ShowStaticView(RUNTIME_CLASS(CSimuView), TRUE);
  106. //SendMessage(hWnd, WM_USER_STARTUP_PROGRESS, nMin+nRange*16/30, 0);
  107. //if (!pStaticDoc->GetViewIfExist(RUNTIME_CLASS(CSListView)))
  108. // pStaticDoc->ShowStaticView(RUNTIME_CLASS(CSListView), TRUE);
  109. //SendMessage(hWnd, WM_USER_STARTUP_PROGRESS, nMin+nRange*18/30, 0);
  110. //if (!pStaticDoc->GetViewIfExist(RUNTIME_CLASS(CRealTimeView)))
  111. // pStaticDoc->ShowStaticView(RUNTIME_CLASS(CRealTimeView), TRUE);
  112. //SendMessage(hWnd, WM_USER_STARTUP_PROGRESS, nMin+nRange*20/30, 0);
  113. //if (!pStaticDoc->GetViewIfExist(RUNTIME_CLASS(CMultiSortView)))
  114. // pStaticDoc->ShowStaticView(RUNTIME_CLASS(CMultiSortView), TRUE);
  115. //SendMessage(hWnd, WM_USER_STARTUP_PROGRESS, nMin+nRange*22/30, 0);
  116. //if (!pStaticDoc->GetViewIfExist(RUNTIME_CLASS(CGraphView)))
  117. // pStaticDoc->ShowStaticView(RUNTIME_CLASS(CGraphView), TRUE);
  118. //SendMessage(hWnd, WM_USER_STARTUP_PROGRESS, nMin+nRange*24/30, 0);
  119. //if (!pStaticDoc->GetViewIfExist(RUNTIME_CLASS(CInfoView)))
  120. // pStaticDoc->ShowStaticView(RUNTIME_CLASS(CInfoView), TRUE);
  121. //SendMessage(hWnd, WM_USER_STARTUP_PROGRESS, nMin+nRange*28/30, 0);
  122. //if (!pStaticDoc->GetViewIfExist(RUNTIME_CLASS(CSelectorView)))
  123. // pStaticDoc->ShowStaticView(RUNTIME_CLASS(CSelectorView), TRUE);
  125. //AfxSwitchToStaticView(RUNTIME_CLASS(CSListView));
  127. AfxGetProfile().SetCurrentStock(STKLIB_CODE_SZZS, FALSE);
  128. pMainFrame->m_SearchBox.SetCurrentWindowText();
  130. AfxBeginThread(LoadKDataCacheMain, NULL, THREAD_PRIORITY_NORMAL);
  132. SendMessage(hWnd, WM_USER_STARTUP_PROGRESS, nMax, 0);
  134. AfxSetResourceHandle(hInstanceOld);
  136. return TRUE;
  137. }
  139. /////////////////////////////////////////////////////////////////////////////
  140. // The one and only CStkUIApp object
  142. CStkUIApp theApp;
  144. /////////////////////////////////////////////////////////////////////////////
  145. // CStkUIApp initialization
  146. TCHAR szRegKeyCompanyOld[] = _T("Balang Software");
  147. TCHAR szRegKeyAppOld[] = _T("StockAna");
  149. #ifdef CLKLAN_ENGLISH_US
  150. TCHAR szRegKeyCompany[] = _T("Ninebulls-Stock Software");
  151. TCHAR szRegKeyApp[] = _T("Ninebulls China Stock");
  152. #elif defined CLKVER_OEM
  153. TCHAR szRegKeyCompany[] = OEM_REGKEY_COMPANY;
  154. TCHAR szRegKeyApp[] = OEM_REGKEY_APP;
  155. #else
  156. TCHAR szRegKeyCompany[] = _T("Ninebulls-Stock Software");
  157. TCHAR szRegKeyApp[] = _T("CLKing");
  158. #endif
  160. BOOL CStkUIApp::InitInstance()
  161. {
  162. // Standard initialization
  163. // If you are not using these features and wish to reduce the size
  164. //  of your final executable, you should remove from the following
  165. //  the specific initialization routines you do not need.
  166. AfxSocketInit();
  168. AfxOleInit();
  170. AfxGetProfile().LoadProfile();
  171. ::SetCurrentDirectory(AfxGetProfile().GetWorkDirectory());
  172. AfxGetSView().Load(AfxGetProfile().GetSViewFile());
  174. // Change the registry key under which our settings are stored.
  175. // TODO: You should modify this string to be something appropriate
  176. // such as the name of your company or organization.
  177. // SetRegistryKey(szRegKeyCompany);
  178. AfxRegSetValue(szRegKeyCompany, szRegKeyApp, "Install", "Path", AfxGetProfile().GetWorkDirectory());
  179. AfxRegSetValue(szRegKeyCompany, szRegKeyApp, "Install", "Version", AfxGetProfile().GetVersion());
  181. // change profile (INI file)
  182. BOOL bEnable = AfxEnableMemoryTracking(FALSE);
  183. if (m_pszProfileName) free((void*)m_pszProfileName);
  184. m_pszProfileName = _tcsdup(AfxGetProfile().GetWorkDirectory()+m_pszExeName+".ini");
  185. AfxEnableMemoryTracking(bEnable);
  187. LoadStdProfileSettings(64);  // Load standard INI file options (including MRU)
  189. // Register the application's document templates.  Document templates
  190. //  serve as the connection between documents, frame windows and views.
  192. // Enable DDE Execute open
  193. EnableShellOpen();
  194. AfxUnregisterShellFileTypes();
  195. // AfxRegisterShellFileTypes(FALSE);
  197. // Parse command line for standard shell commands, DDE, file open
  198. CCommandLineInfo cmdInfo;
  199. ParseCommandLine(cmdInfo);
  201. // Dispatch commands specified on the command line
  202. if (CCommandLineInfo::AppUnregister == cmdInfo.m_nShellCommand)
  203. {
  204. AfxUnregisterShellFileTypes();
  205. if (!cmdInfo.m_bRunEmbedded)
  206. AfxMessageBox(AFX_IDP_UNREG_DONE, MB_OK | MB_ICONINFORMATION);
  207. if (m_pCmdInfo == NULL)
  208. {
  209. m_pCmdInfo = new CCommandLineInfo;
  210. m_pCmdInfo->m_nShellCommand = CCommandLineInfo::AppUnregister;
  211. }
  212. return FALSE;
  213. }
  215. {
  216. // Empty Temp Directory
  217. CNetDatabase netdb;
  218. if (netdb.SetRootPath(AfxGetProfile().GetSelfDBPath(), IStStore::dbtypeSelfDB))
  219. netdb.EmptyTempDirectory();
  220. }
  221. // Load Data
  222. AfxSetDB(&AfxGetNetDB());
  223. char szErr[1024];
  224. if (!AfxInitializeDB(szErr, sizeof(szErr))) AfxMessageBox(szErr, MB_OK|MB_ICONINFORMATION);
  226. // Create MainFrame and Static Doc and child frames and static views
  227. CStaticDoc * pStaticDoc = CStaticDoc::OpenDocumentFile(NULL);
  228. if (NULL == pStaticDoc)
  229. {
  230. AfxMessageBox(IDS_STATICDOC_FAILED, MB_OK | MB_ICONINFORMATION);
  231. return FALSE;
  232. }
  233. pStaticDoc->m_bAutoDelete = FALSE;
  234. CString strAppTitle;
  235. strAppTitle.LoadString(AFX_IDS_APP_TITLE);
  236. pStaticDoc->SetTitle(strAppTitle);
  237. SetStaticDoc(pStaticDoc);
  239. // 建立主窗口
  240. CMainFrame* pMainFrame = CMainFrame::CreateNewFrame();
  241. if (pMainFrame == NULL)
  242. {
  243. AfxMessageBox(IDS_STATICDOC_FAILED, MB_OK | MB_ICONINFORMATION);
  244. delete pStaticDoc;
  245. SetStaticDoc(NULL);
  246. return FALSE;
  247. }
  248. SetMainWnd(pMainFrame);
  250. pStaticDoc->ShowStaticView(RUNTIME_CLASS(CSListView), TRUE);
  251. AfxSwitchToStaticView(RUNTIME_CLASS(CSListView));
  253. // 装载主窗口最后状态
  254. WINDOWPLACEMENT wp;
  255. if (AfxGetProfile().GetWindowPlacement(&wp))
  256. m_pMainWnd->SetWindowPlacement(&wp);
  257. else
  258. m_pMainWnd->ShowWindow(SW_SHOWMAXIMIZED);
  260. // 股票行情接收
  261. AfxGetStkReceiver().CreateReceiver(pMainFrame);
  262. AfxGetStkReceiver().NetEngineSetAutoReport(TRUE);
  264. // 显示启动窗口
  265. CStartupDlg startup;
  266. CBitmap bmp;
  267. bmp.LoadBitmap(IDB_DLGLEFTLOGO);
  268. startup.SetBitmap((HBITMAP)bmp.GetSafeHandle());
  269. startup.SetLoadProgramFunc(LoadProgram);
  270. if (IDOK != startup.DoModal())
  271. {
  272. //pMainFrame->SendMessage(WM_CLOSE);
  273. //return FALSE;
  274. }
  276. if (CCommandLineInfo::FileOpen == cmdInfo.m_nShellCommand)
  277. {
  278. OpenStrategyFile(cmdInfo.m_strFileName);
  279. AfxSwitchToStaticView(RUNTIME_CLASS(CSimuView));
  280. }
  282. return TRUE;
  283. }
  285. BOOL CStkUIApp::OnDBChanged(BOOL bRecalculate)
  286. {
  287. CWaitCursor waitcursor;
  289. // Reload DB
  290. AfxGetStockContainer().Load(&AfxGetDB(), NULL, NULL);
  291. AfxGetDomainContainer().Load(AfxGetProfile().GetDomainFile());
  292. AfxGetGroupContainer().Load(AfxGetProfile().GetGroupFile());
  293. AfxGetDB().LoadKDataCache(AfxGetStockContainer(), NULL, NULL, 0, STKLIB_MAX_PROGRESS);
  294. AfxGetStockContainer().OnDataChanged();
  296. AfxGetStkReceiver().RefreshStockContainer(AfxGetStockContainer(), TRUE);
  298. // Reload AfxGetStockMain()
  299. AfxReloadStockMain();
  301. // Recalcuate Yield
  302. if (bRecalculate && IDYES == AfxMessageBox(IDS_INSTALLPAC_IFRECALCULATE, MB_YESNO|MB_ICONINFORMATION))
  303. {
  304. AfxRecalculateYield(AfxGetProfile().GetYieldAverageDays(), TRUE);
  305. AfxGetStockContainer().ReloadBase(&AfxGetDB());
  306. }
  308. // Update SListBar
  309. CMainFrame * pMainFrame = AfxGetMainFrame();
  310. if (pMainFrame)
  311. {
  312. pMainFrame->m_SearchBox.InitStocks(TRUE, TRUE, TRUE);
  313. pMainFrame->m_SearchBox.SetCurrentWindowText();
  314. }
  316. // Update Views
  317. AfxGetSListStockContainer().ReRetrieveFromStatic(AfxGetActiveStrategy(), TRUE);
  318. AfxGetStaticDoc()->UpdateAllViews(NULL, UPDATE_HINT_SLISTVIEW, NULL);
  319. AfxGetStaticDoc()->UpdateAllViews(NULL, UPDATE_HINT_GRAPHVIEW, NULL);
  320. AfxGetStaticDoc()->UpdateAllViews(NULL, UPDATE_HINT_BASEVIEW, NULL);
  321. AfxGetStaticDoc()->UpdateAllViews(NULL, UPDATE_HINT_SIMUVIEW_REREALRUN, NULL);
  322. AfxGetStaticDoc()->UpdateAllViews(NULL, UPDATE_HINT_SELECTORVIEW, NULL);
  324. return TRUE;
  325. }
  327. void CStkUIApp::SaveLastOpenedStrategy()
  328. {
  329. CSPStringArray astr;
  330. POSITION pos = GetFirstStrategyPosition();
  331. while(pos != NULL)
  332. {
  333. CStrategy* pStrategy = GetNextStrategy(pos);
  334. if (pStrategy)
  335. astr.Add(pStrategy->GetPathName());
  336. }
  338. AfxGetProfile().SetLastOpenedStrategy(astr);
  339. }
  341. int CStkUIApp::OpenLastOpenedStrategy()
  342. {
  343. CSPStringArray & astrStrategyToOpen = AfxGetProfile().GetLastOpenedStrategy();
  344. CSPString strLastActive = AfxGetProfile().GetLastActiveStrategy();
  345. if (0 != access(strLastActive,0))
  346. strLastActive.Empty();
  347. int nCount = 0;
  348. for(int i=0; i<astrStrategyToOpen.GetSize(); i++)
  349. {
  350. if (0 != access(astrStrategyToOpen.ElementAt(i),0))
  351. continue;
  352. if (NULL != OpenStrategyFile(astrStrategyToOpen.ElementAt(i)))
  353. {
  354. nCount ++;
  355. if (strLastActive.IsEmpty())
  356. strLastActive = astrStrategyToOpen.ElementAt(i);
  357. }
  358. }
  360. if (!strLastActive.IsEmpty())
  361. OpenStrategyFile(strLastActive);
  363. return nCount;
  364. }
  366. int CStkUIApp::OpenDefaultStrategy()
  367. {
  368. CSPStringArray astrStrategyToOpen;
  370. // Recent
  371. if (m_pRecentFileList)
  372. {
  373. for(int i=m_pRecentFileList->GetSize()-1; i>=0; i--)
  374. {
  375. CString strPath = (*m_pRecentFileList)[i];
  376. if (!strPath.IsEmpty())
  377. astrStrategyToOpen.Add(strPath);
  378. }
  379. }
  381. // projects
  382. CString strExt = AfxGetStrategyFileExt();
  383. if (!strExt.IsEmpty())
  384. {
  385. CFileFind finder;
  386. BOOL bWorking = finder.FindFile(AfxGetFilePath((LPCTSTR)AfxGetProfile().GetProjectPath(), LPCTSTR("*" + strExt)));
  387. while(bWorking)
  388. {
  389. bWorking = finder.FindNextFile();
  390. CString strPath = finder.GetFilePath();
  392. for(int i=0; i<astrStrategyToOpen.GetSize(); i++)
  393. {
  394. if (0 == strPath.Compare(astrStrategyToOpen.ElementAt(i)))
  395. break;
  396. }
  397. if (i == astrStrategyToOpen.GetSize())
  398. astrStrategyToOpen.Add(strPath);
  399. }
  400. finder.Close();
  401. }
  403. AfxGetProfile().SetLastOpenedStrategy(astrStrategyToOpen);
  404. return OpenLastOpenedStrategy();
  405. }
  407. CStrategy * CStkUIApp::GetActiveStrategy()
  408. {
  409. CStrategy * pStrategy = NULL;
  411. //CStrategyView * pStrategyView = AfxGetStrategyView();
  412. //if (pStrategyView)
  413. // pStrategy = pStrategyView->GetActiveStrategy();
  415. return pStrategy;
  416. }
  418. void CStkUIApp::AddStrategy(CStrategy* pStrategy)
  419. {
  420. if (NULL == pStrategy)
  421. return;
  422. ASSERT(m_strategyList.Find(pStrategy, NULL) == NULL); // must not be in list
  423. m_strategyList.AddTail(pStrategy);
  425. if (m_bAutoUpdateViews)
  426. {
  427. //::SendMessage(AfxGetStrategyView()->GetSafeHwnd(), WM_USER_ADDSTRATEGY, 0, 0);
  428. AfxGetStaticDoc()->UpdateAllViews(NULL, UPDATE_HINT_WIZARDVIEW, NULL);
  429. // AfxGetMainFrame()->ShowWorkspBar();
  430. }
  431. }
  433. void CStkUIApp::RemoveStrategy(CStrategy* pStrategy)
  434. {
  435. POSITION pos = m_strategyList.Find(pStrategy);
  436. ASSERT(NULL != pos);
  437. if (pos)
  438. m_strategyList.RemoveAt(pos);
  440. if (m_bAutoUpdateViews)
  441. {
  442. //::SendMessage(AfxGetStrategyView()->GetSafeHwnd(), WM_USER_REMOVESTRATEGY, 0, 0);
  443. AfxGetStaticDoc()->UpdateAllViews(NULL, UPDATE_HINT_WIZARDVIEW, NULL);
  444. }
  445. }
  447. POSITION CStkUIApp::GetFirstStrategyPosition() const
  448. {
  449. return m_strategyList.GetHeadPosition();
  450. }
  452. CStrategy* CStkUIApp::GetNextStrategy(POSITION& rPos) const
  453. {
  454. return (CStrategy*)m_strategyList.GetNext(rPos);
  455. }
  457. BOOL CStkUIApp::ExistStrategy(CStrategy * pStrategy)
  458. {
  459. POSITION pos = GetFirstStrategyPosition();
  460. while(NULL != pos)
  461. {
  462. CStrategy * p = GetNextStrategy(pos);
  463. if (p == pStrategy)
  464. return TRUE;
  465. }
  466. return FALSE;
  467. }
  469. BOOL CStkUIApp::IsFileOpened(LPCTSTR lpszFileName)
  470. {
  471. POSITION pos = GetFirstStrategyPosition();
  472. while (pos != NULL)
  473. {
  474. CStrategy* pStrategy = GetNextStrategy(pos);
  475. if (0 == pStrategy->GetPathName().CompareNoCase(lpszFileName))
  476. {
  477. return TRUE;
  478. }
  479. }
  480. return FALSE;
  481. }
  483. CString CStkUIApp::GetLastStrategyDirectory()
  484. {
  485. CString strLastStrategyDirectory;
  486. if (m_pRecentFileList && m_pRecentFileList->GetSize() > 0)
  487. {
  488. strLastStrategyDirectory = (*m_pRecentFileList)[0];
  489. strLastStrategyDirectory = AfxGetFileDirectoryExist(strLastStrategyDirectory, "");
  490. }
  492. if (strLastStrategyDirectory.IsEmpty() || 0 != access(strLastStrategyDirectory,0))
  493. {
  494. strLastStrategyDirectory = AfxGetProfile().GetProjectPath();
  495. }
  496. return strLastStrategyDirectory;
  497. }
  499. CString CStkUIApp::GetNextNewStrategyTitle(CString & strExtBuffer, CString strPath)
  500. {
  501. CString string;
  502. VERIFY(string.LoadString(IDS_STRATEGY_NONAME));
  504. CString strExt = AfxGetStrategyFileExt();
  505. if (!strExt.IsEmpty())
  506. strExtBuffer = strExt;
  508. CStringArray astrExistTitle;
  509. POSITION pos = GetFirstStrategyPosition();
  510. while(NULL != pos)
  511. {
  512. CStrategy * pStrategy = GetNextStrategy(pos);
  513. astrExistTitle.Add(AfxGetFileTitleNoExt(pStrategy->GetPathName()));
  514. }
  516. CFileFind finder;
  517. BOOL bWorking = finder.FindFile(AfxGetFilePath((LPCTSTR)strPath, LPCTSTR("*" + strExt)));
  518. while(bWorking)
  519. {
  520. bWorking = finder.FindNextFile();
  521. astrExistTitle.Add(finder.GetFileTitle());
  522. }
  523. finder.Close();
  525. for(int i=1; ; i++)
  526. {
  527. CString strTemp;
  528. strTemp.Format("%s(%d)", string, i);
  530. BOOL bHas = FALSE;
  531. for(int k=0; k<astrExistTitle.GetSize(); k++)
  532. {
  533. if (0 == strTemp.CompareNoCase(astrExistTitle.ElementAt(k)))
  534. {
  535. bHas = TRUE;
  536. break;
  537. }
  538. }
  540. if (!bHas)
  541. {
  542. string = strTemp;
  543. break;
  544. }
  545. }
  546. return string;
  547. }
  549. CStrategy* CStkUIApp::OpenStrategyFile(LPCTSTR lpszPathName)
  550. {
  551. // Resolve File Name
  552. TCHAR szPath[_MAX_PATH];
  553. ASSERT(lstrlen(lpszPathName) < sizeof(szPath));
  554. if (NULL != lpszPathName)
  555. {
  556. TCHAR szTemp[_MAX_PATH];
  557. if (lpszPathName[0] == '"')
  558. ++lpszPathName;
  559. lstrcpyn(szTemp, lpszPathName, _MAX_PATH);
  560. LPTSTR lpszLast = _tcsrchr(szTemp, '"');
  561. if (lpszLast != NULL)
  562. *lpszLast = 0;
  563. AfxFullPath(szPath, szTemp);
  564. TCHAR szLinkName[_MAX_PATH];
  565. if (AfxResolveShortcut(AfxGetMainWnd(), szPath, szLinkName, _MAX_PATH))
  566. lstrcpy(szPath, szLinkName);
  567. }
  569. if (NULL != lpszPathName)
  570. {
  571. POSITION pos = GetFirstStrategyPosition();
  572. while (pos != NULL)
  573. {
  574. CStrategy* pStrategy = GetNextStrategy(pos);
  575. if (0 == pStrategy->GetPathName().CompareNoCase(szPath))
  576. {
  577. if (m_bAutoUpdateViews)
  578. {
  579. //::SendMessage(AfxGetStrategyView()->GetSafeHwnd(), WM_USER_ACTIVATESTRATEGY, DWORD(pStrategy), 0);
  580. AfxGetStaticDoc()->UpdateAllViews(NULL, UPDATE_HINT_WIZARDVIEW, NULL);
  581. // AfxGetMainFrame()->ShowWorkspBar();
  582. }
  583. return pStrategy;
  584. }
  585. }
  586. }
  588. CStrategy   * pStrategy = new CStrategy();
  589. if (NULL == pStrategy)
  590. return NULL;
  591. char szErr[256];
  592. if (!pStrategy->OpenStrategyFile(szPath, szErr, sizeof(szErr)-1))
  593. {
  594. AfxMessageBox(szErr, MB_OK|MB_ICONINFORMATION);
  595. delete pStrategy;
  596. return NULL;
  597. }
  599. AddStrategy(pStrategy);
  600. return pStrategy;
  601. }
  603. void CStkUIApp::CloseAllStrategy(BOOL bEndSession)
  604. {
  605. POSITION pos = GetFirstStrategyPosition();
  606. while (pos != NULL)
  607. {
  608. CStrategy* pStrategy = GetNextStrategy(pos);
  609. pStrategy->OnClose();
  610. delete pStrategy;
  611. }
  612. m_strategyList.RemoveAll();
  614. if (! bEndSession && m_bAutoUpdateViews)
  615. {
  616. //::SendMessage(AfxGetStrategyView()->GetSafeHwnd(), WM_USER_REMOVESTRATEGY, 0, 0);
  617. AfxGetStaticDoc()->UpdateAllViews(NULL, UPDATE_HINT_WIZARDVIEW, NULL);
  618. }
  619. }
  621. CRecentFileList * CStkUIApp::GetRecentFileList()
  622. {
  623. return m_pRecentFileList;
  624. }
  626. /////////////////////////////////////////////////////////////////////////////
  627. // CAboutDlg dialog used for App About
  629. class CAboutDlg : public CDialog
  630. {
  631. public:
  632. CAboutDlg();
  634. // Dialog Data
  635. //{{AFX_DATA(CAboutDlg)
  636. enum { IDD = IDD_ABOUTBOX };
  637. //}}AFX_DATA
  638. CButton m_btnOK;
  639. protected:
  640. //{{AFX_VIRTUAL(CAboutDlg)
  641. virtual void DoDataExchange(CDataExchange* pDX);
  642. //}}AFX_VIRTUAL
  644. // Implementation
  645. protected:
  646. //{{AFX_MSG(CAboutDlg)
  647. virtual BOOL OnInitDialog();
  648. //}}AFX_MSG
  649. DECLARE_MESSAGE_MAP()
  650. };
  652. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  653. {
  654. }
  656. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  657. {
  658. CDialog::DoDataExchange(pDX);
  660. DDX_Control(pDX, IDOK, m_btnOK);
  661. }
  663. BOOL CAboutDlg::OnInitDialog() 
  664. {
  665. return TRUE;
  666. }
  668. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  669. END_MESSAGE_MAP()
  671. // App command to run the dialog
  672. void CStkUIApp::OnAppAbout()
  673. {
  674. CAboutDlg aboutDlg;
  675. aboutDlg.DoModal();
  676. }
  678. /////////////////////////////////////////////////////////////////////////////
  679. // CStkUIApp message handlers
  682. int CStkUIApp::ExitInstance() 
  683. {
  684. // TODO: Add your specialized code here and/or call the base class
  685. SaveLastOpenedStrategy();
  686. CloseAllStrategy(TRUE);
  688. AfxGetDB().StoreCodetable(AfxGetStockContainer());
  689. AfxGetProfile().StoreProfile();
  690. AfxGetSView().Store(AfxGetProfile().GetSViewFile());
  691. AfxReleaseDB();
  693. if (m_pStaticDoc)
  694. {
  695. m_pStaticDoc->OnCloseDocument();
  696. delete m_pStaticDoc;
  697. m_pStaticDoc = NULL;
  698. }
  699. return CWinApp::ExitInstance();
  700. }
  702. void CStkUIApp::OnFileNew() 
  703. {
  704. // TODO: Add your command handler code here
  705. // OpenStrategyFile(NULL);
  706. }
  708. void CStkUIApp::OnFileOpen() 
  709. {
  710. // TODO: Add your command handler code here
  711. CString newName;
  712. if (m_pRecentFileList && m_pRecentFileList->GetSize() > 0)
  713. newName = (*m_pRecentFileList)[0];
  715. if (!AfxDoPromptFileName(newName, IDS_OPENFILE,
  716.   OFN_HIDEREADONLY | OFN_FILEMUSTEXIST | OFN_EXPLORER | OFN_ENABLESIZING, TRUE))
  717. return; // open cancelled
  719. if (NULL != OpenStrategyFile(newName))
  720. AfxSwitchToStaticView(RUNTIME_CLASS(CSimuView));
  721. // if returns NULL, the user has already been alerted
  722. }
  724. BOOL CStkUIApp::SaveAllModified() 
  725. {
  726. // TODO: Add your specialized code here and/or call the base class
  727. CStrategy * pActiveStrategy = AfxGetActiveStrategy();
  728. if (pActiveStrategy)
  729. pActiveStrategy->DoFileSave();
  730. /*
  731. POSITION pos = GetFirstStrategyPosition();
  732. while (pos != NULL)
  733. {
  734. CStrategy* pStrategy = GetNextStrategy(pos);
  735. if (!pStrategy->DoFileSave()) // >SaveModified())
  736. return FALSE;
  737. }
  738. */
  739. return TRUE;
  740. }
  742. void CStkUIApp::OnStrategyNew() 
  743. {
  744. // TODO: Add your command handler code here
  745. CStrategy * pStrategy = new CStrategy();
  746. if (NULL == pStrategy)
  747. {
  748. AfxMessageBox(IDS_STRATEGY_FAILED, MB_OK | MB_ICONINFORMATION);
  749. return;
  750. }
  752. CSetPrpt firstPage(NULL, IDS_STRATEGY_SETPRPT, NULL);
  753. firstPage.m_dwButtonFlags = PSWIZB_NEXT | PSWIZB_DISABLEDFINISH;
  754. firstPage.SetStrategy(pStrategy);
  756. CSelectStk selectStk(NULL, IDS_STRATEGY_SELECTSTK, NULL);
  757. selectStk.m_dwButtonFlags = PSWIZB_BACK | PSWIZB_NEXT | PSWIZB_DISABLEDFINISH;
  759. CSetRule setrule(NULL, IDS_STRATEGY_SETRULE, NULL);
  760. setrule.m_dwButtonFlags = PSWIZB_BACK | PSWIZB_NEXT | PSWIZB_DISABLEDFINISH;
  761. setrule.SetTechParameters(&(pStrategy->GetTechParametersBuy()), &(pStrategy->GetSelectedTechsBuy()));
  763. CSetRate setrate(NULL, IDS_STRATEGY_SETRATE, NULL);
  764. setrate.m_dwButtonFlags = PSWIZB_BACK | PSWIZB_FINISH;
  765. setrate.SetStrategy(pStrategy);
  767. CPropertySheet sht;
  768. sht.AddPage(&firstPage);
  769. sht.AddPage(&selectStk);
  770. sht.AddPage(&setrule);
  771. sht.AddPage(&setrate);
  772. sht.SetWizardMode();
  774. sht.m_psh.dwFlags |= (PSH_WIZARD97 | PSH_WIZARDHASFINISH /* | PSH_HEADER /*Header Bitmap*/);
  775. //CBitmap bmp;
  776. //bmp.LoadBitmap(IDB_BALANG);
  777. //sht.SetBitmap((HBITMAP)bmp.GetSafeHandle());
  779. if (ID_WIZFINISH == sht.DoModal())
  780. {
  781. pStrategy->SetStocks(selectStk.m_domainTemp);
  782. pStrategy->SaveStrategyFile(firstPage.GetPathName());
  783. AddStrategy(pStrategy);
  784. AfxSwitchToStaticView(RUNTIME_CLASS(CSimuView));
  785. }
  786. else
  787. {
  788. delete pStrategy;
  789. }
  790. }
  792. void CStkUIApp::OnStrategySave() 
  793. {
  794. // TODO: Add your command handler code here
  795. // CStrategy * pStrategy = GetActiveStrategy();
  796. // if (pStrategy)
  797. // pStrategy->DoFileSave();
  798. }
  800. void CStkUIApp::OnUpdateStrategySave(CCmdUI* pCmdUI) 
  801. {
  802. // TODO: Add your command update UI handler code here
  803. CStrategy * pActive = GetActiveStrategy();
  804. pCmdUI->Enable(FALSE); // NULL != pActiveStrategy);
  805. }
  807. void CStkUIApp::OnStrategySaveAs() 
  808. {
  809. // TODO: Add your command handler code here
  810. CStrategy * pStrategy = GetActiveStrategy();
  811. if (pStrategy)
  812. {
  813. CString newName = AfxGetFileTitleNoExt(pStrategy->GetPathName());
  814. // check for dubious filename
  815. int iBad = newName.FindOneOf(_T(" #%;/\"));
  816. if (iBad != -1)
  817. newName.ReleaseBuffer(iBad);
  819. // append the default suffix if there is one
  820. newName += AfxGetStrategyFileExt();
  822. if (!AfxDoPromptFileName(newName, IDS_SAVEFILE,
  823. OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT | OFN_PATHMUSTEXIST | OFN_EXPLORER | OFN_ENABLESIZING, FALSE))
  824. return;
  826. pStrategy->SetPathName(newName);
  827. pStrategy->SetName(AfxGetFileTitleNoExt(newName));
  828. pStrategy->DoFileSave();
  829. //AfxOnStrategyUpdated(pStrategy, FALSE);
  830. }
  831. }
  833. void CStkUIApp::OnUpdateStrategySaveAs(CCmdUI* pCmdUI) 
  834. {
  835. // TODO: Add your command update UI handler code here
  837. pCmdUI->Enable(NULL != GetActiveStrategy());
  838. }
  840. void CStkUIApp::OnStrategySaveall() 
  841. {
  842. // TODO: Add your command handler code here
  843. SaveAllModified();
  844. }
  846. void CStkUIApp::OnUpdateStrategySaveall(CCmdUI* pCmdUI) 
  847. {
  848. // TODO: Add your command update UI handler code here
  849. pCmdUI->Enable(NULL != GetFirstStrategyPosition());
  850. }
  852. void CStkUIApp::OnStrategyClose() 
  853. {
  854. // TODO: Add your command handler code here
  855. CSimuView * pSimuView = AfxGetSimuView();
  856. CStrategy * pActive = AfxGetActiveStrategy();
  857. //CStrategyView * pStrategyView = AfxGetStrategyView();
  859. CCloseStrategyDlg dlg;
  860. if (IDOK == dlg.DoModal())
  861. {
  862. BOOL bOldValue = m_bAutoUpdateViews;
  863. m_bAutoUpdateViews = FALSE;
  865. POSITION pos = dlg.GetFirstCloseStrategyPosition();
  866. while(NULL != pos)
  867. {
  868. CStrategy * pStrategy = dlg.GetNextCloseStrategy(pos);
  869. if (pSimuView && pStrategy == pActive && !pSimuView->CanChangeActiveStrategy())
  870. continue;
  871. //if (pStrategy == pActive && pStrategyView)
  872. //{
  873. // pStrategyView->SetActiveStrategy(NULL);
  874. // if (pSimuView)
  875. // pSimuView->OnActiveStrategyChanged();
  876. //}
  877. //RemoveStrategy(pStrategy);
  878. pStrategy->OnClose();
  879. delete pStrategy;
  880. }
  882. m_bAutoUpdateViews = bOldValue;
  883. if (m_bAutoUpdateViews)
  884. {
  885. //::SendMessage(AfxGetStrategyView()->GetSafeHwnd(), WM_USER_REMOVESTRATEGY, 0, 0);
  886. AfxGetStaticDoc()->UpdateAllViews(NULL, UPDATE_HINT_WIZARDVIEW, NULL);
  887. }
  888. }
  889. }
  891. void CStkUIApp::OnUpdateStrategyClose(CCmdUI* pCmdUI) 
  892. {
  893. // TODO: Add your command update UI handler code here
  894. pCmdUI->Enable(NULL != GetFirstStrategyPosition());
  895. }
  897. void CStkUIApp::WinHelp(DWORD dwData, UINT nCmd) 
  898. {
  899. // TODO: Add your specialized code here and/or call the base class
  900. // CWinApp::WinHelp(dwData, nCmd);
  901. }