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

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

金融证券系统

开发平台：

Visual C++

DownloadDlg.cpp：源码内容
  1. // DownloadDlg.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  6. #include "DownloadDlg.h"
  7. #include "UpgradeDlg.h"
  9. #ifdef _DEBUG
  10. #define new DEBUG_NEW
  11. #undef THIS_FILE
  12. static char THIS_FILE[] = __FILE__;
  13. #endif
  15. /////////////////////////////////////////////////////////////////////////////
  16. // CDownloadDlg dialog
  18. HANDLE CDownloadDlg::m_hEventKillDownloadThread = NULL;
  19. HANDLE CDownloadDlg::m_hEventDownloadThreadKilled = NULL;
  22. BOOL CALLBACK DownloadCallback(DWORD dwCode, DWORD dwProgress,
  23. LPCTSTR lpszMsg, void *cookie)
  24. {
  25. HWND hMainWnd = (HWND)cookie;
  27. if( PROG_PROGRESS == dwCode && ::IsWindow(hMainWnd) )
  28. {
  29. ::SendMessage( hMainWnd, WM_USER_DOWNLOAD_PROGRESS, dwProgress, (LPARAM)lpszMsg );
  30. }
  31. else if( ::IsWindow(hMainWnd) )
  32. {
  33. ::SendMessage( hMainWnd, WM_USER_DOWNLOAD_MESSAGE, dwCode, (LPARAM)lpszMsg );
  34. }
  36. // User wants to quit program
  37. if(WAIT_OBJECT_0 == WaitForSingleObject(CDownloadDlg::m_hEventKillDownloadThread,0) )
  38. {
  39. ::PostMessage( hMainWnd, WM_USER_DOWNLOAD_END, FALSE, 0 );
  40. SetEvent(CDownloadDlg::m_hEventDownloadThreadKilled);
  41. AfxEndThread( 0, TRUE );
  42. return FALSE;
  43. }
  44. return TRUE;
  45. }
  47. UINT DownloadMain(LPVOID pParam)
  48. {
  49. LPDOWNLOAD_INFO pInfo=(LPDOWNLOAD_INFO)pParam;
  51. int nCountDay = 0;
  52. int nCountBase = 0;
  53. int nCountXDR = 0;
  54. int nCountMin5 = 0;
  56. if( pInfo->bDay )
  57. {
  58. ::SendMessage( pInfo->hMainWnd, WM_USER_DOWNLOAD_MESSAGE, PROG_NEWGROUP, IDS_DLMSG_DAY );
  59. ::SendMessage( pInfo->hMainWnd, WM_USER_DOWNLOAD_PROGRESS, 0, NULL );
  60. CSPTime tmFrom( pInfo->tmDayFrom );
  61. CSPTime tmTo( pInfo->tmDayTo );
  62. tmFrom = CSPTime( tmFrom.GetYear(), tmFrom.GetMonth(), tmFrom.GetDay(), 0, 0, 0 );
  63. tmTo = CSPTime( tmTo.GetYear(), tmTo.GetMonth(), tmTo.GetDay(), 23, 59, 59 );
  64. nCountDay = pInfo->pNetDB->NetloadDayKLine( NULL, &tmFrom, &tmTo, DownloadCallback, pInfo->hMainWnd );
  65. }
  67. // User wants to quit program
  68. if(WAIT_OBJECT_0 == WaitForSingleObject(CDownloadDlg::m_hEventKillDownloadThread,0) )
  69. {
  70. ::PostMessage( pInfo->hMainWnd, WM_USER_DOWNLOAD_END, FALSE, 0 );
  71. SetEvent(CDownloadDlg::m_hEventDownloadThreadKilled);
  72. AfxEndThread( 0, TRUE );
  73. return 0;
  74. }
  76. if( pInfo->bBase )
  77. {
  78. ::SendMessage( pInfo->hMainWnd, WM_USER_DOWNLOAD_MESSAGE, PROG_NEWGROUP, IDS_DLMSG_BASE );
  79. ::SendMessage( pInfo->hMainWnd, WM_USER_DOWNLOAD_PROGRESS, 0, NULL );
  80. nCountBase = pInfo->pNetDB->NetloadBasetable( NULL, NULL, NULL, DownloadCallback, pInfo->hMainWnd );
  81. }
  83. // User wants to quit program
  84. if(WAIT_OBJECT_0 == WaitForSingleObject(CDownloadDlg::m_hEventKillDownloadThread,0) )
  85. {
  86. ::PostMessage( pInfo->hMainWnd, WM_USER_DOWNLOAD_END, FALSE, 0 );
  87. ::SendMessage( pInfo->hMainWnd, WM_USER_DOWNLOAD_PROGRESS, 0, NULL );
  88. SetEvent(CDownloadDlg::m_hEventDownloadThreadKilled);
  89. AfxEndThread( 0, TRUE );
  90. return 0;
  91. }
  93. if( pInfo->bXDR )
  94. {
  95. ::SendMessage( pInfo->hMainWnd, WM_USER_DOWNLOAD_MESSAGE, PROG_NEWGROUP, IDS_DLMSG_XDR );
  96. ::SendMessage( pInfo->hMainWnd, WM_USER_DOWNLOAD_PROGRESS, 0, NULL );
  97. nCountXDR = pInfo->pNetDB->NetloadDRData( NULL, NULL, NULL, DownloadCallback, pInfo->hMainWnd );
  98. }
  100. // User wants to quit program
  101. if(WAIT_OBJECT_0 == WaitForSingleObject(CDownloadDlg::m_hEventKillDownloadThread,0) )
  102. {
  103. ::PostMessage( pInfo->hMainWnd, WM_USER_DOWNLOAD_END, FALSE, 0 );
  104. SetEvent(CDownloadDlg::m_hEventDownloadThreadKilled);
  105. AfxEndThread( 0, TRUE );
  106. return 0;
  107. }
  109. if( pInfo->bMin5 )
  110. {
  111. ::SendMessage( pInfo->hMainWnd, WM_USER_DOWNLOAD_MESSAGE, PROG_NEWGROUP, IDS_DLMSG_MIN5 );
  112. ::SendMessage( pInfo->hMainWnd, WM_USER_DOWNLOAD_PROGRESS, 0, NULL );
  113. CSPTime tmFrom( pInfo->tmMin5From );
  114. CSPTime tmTo( pInfo->tmMin5To );
  115. tmFrom = CSPTime( tmFrom.GetYear(), tmFrom.GetMonth(), tmFrom.GetDay(), 0, 0, 0 );
  116. tmTo = CSPTime( tmTo.GetYear(), tmTo.GetMonth(), tmTo.GetDay(), 23, 59, 59 );
  117. nCountMin5 = pInfo->pNetDB->Netload5MinKLine( NULL, &tmFrom, &tmTo, DownloadCallback, pInfo->hMainWnd );
  118. }
  120. if( pInfo->bMin5 && nCountMin5 == 0 )
  121. {
  122. CString strNote;
  123. strNote.LoadString( IDS_DLMIN5_NOTSUPPORT );
  124. ::SendMessage( pInfo->hMainWnd, WM_USER_DOWNLOAD_END, TRUE, (LPARAM)strNote.GetBuffer(strNote.GetLength()+1) );
  125. strNote.ReleaseBuffer();
  126. }
  127. else if( pInfo->bBase && nCountBase == 0 )
  128. {
  129. CString strNote;
  130. strNote.LoadString( IDS_DLBASE_NOTSUPPORT );
  131. ::SendMessage( pInfo->hMainWnd, WM_USER_DOWNLOAD_END, TRUE, (LPARAM)strNote.GetBuffer(strNote.GetLength()+1) );
  132. strNote.ReleaseBuffer();
  133. }
  134. else
  135. {
  136. ::SendMessage( pInfo->hMainWnd, WM_USER_DOWNLOAD_END, TRUE, 0 );
  137. }
  139. AfxEndThread( 0, TRUE );
  140. return 0;
  141. }
  143. CDownloadDlg::CDownloadDlg(CWnd* pParent /*=NULL*/)
  144. : CDialog(CDownloadDlg::IDD, pParent)
  145. {
  146. //{{AFX_DATA_INIT(CDownloadDlg)
  147. m_tmDayFrom = 0;
  148. m_tmDayTo = 0;
  149. m_tmMin5From = 0;
  150. m_tmMin5To = 0;
  151. //}}AFX_DATA_INIT
  153. m_pDownloadInfo = NULL;
  154. m_bDBModified = FALSE;
  156. CSPTime tmLatestOld = 0;
  157. if( m_netdb.SetRootPath( AfxGetProfile().GetSelfDBPath(), IStStore::dbtypeSelfDB ) )
  158. m_netdb.GetTimeLocalRange( &tmLatestOld, NULL, NULL );
  159. m_tmLatestOld = tmLatestOld.GetTime();
  161. m_bInitDownloadTime = FALSE;
  162. m_bAutoRun = FALSE;
  164. m_bReportFinished = TRUE;
  165. }
  167. CDownloadDlg::~CDownloadDlg()
  168. {
  169. if( m_pDownloadInfo )
  170. {
  171. delete m_pDownloadInfo;
  172. m_pDownloadInfo = NULL;
  173. }
  174. }
  176. void CDownloadDlg::SetInitDownloadTime( CSPTime tmDLBegin, CSPTime tmDLEnd, BOOL bAutoRun )
  177. {
  178. m_bInitDownloadTime = TRUE;
  179. m_tmDLBegin = tmDLBegin;
  180. m_tmDLEnd = tmDLEnd;
  181. m_bAutoRun = TRUE;
  182. }
  184. void CDownloadDlg::DoDataExchange(CDataExchange* pDX)
  185. {
  186. CDialog::DoDataExchange(pDX);
  187. //{{AFX_DATA_MAP(CDownloadDlg)
  188. DDX_Control(pDX, IDC_STATIC_LATEST, m_staticLatest);
  189. DDX_Control(pDX, IDC_STATIC_PROGRESS, m_staticProgress);
  190. DDX_Control(pDX, IDC_CHECK_MIN5, m_btnCheckMin5);
  191. DDX_Control(pDX, IDC_CHECK_REPORT, m_btnCheckReport);
  192. DDX_Control(pDX, IDC_PROXY_SET, m_btnProxySet);
  193. DDX_Control(pDX, IDCANCEL, m_btnCancel);
  194. DDX_Control(pDX, IDC_STATIC_STATUS, m_staticStatus);
  195. DDX_Control(pDX, IDOK, m_btnOK);
  196. DDX_Control(pDX, IDC_STOP_REFRESH, m_btnStopRefresh);
  197. DDX_Control(pDX, IDC_STOP, m_btnStop);
  198. DDX_Control(pDX, IDC_REFRESH_SERVERS, m_btnRefreshServers);
  199. DDX_Control(pDX, IDC_CHECK_XDR, m_btnCheckXDR);
  200. DDX_Control(pDX, IDC_CHECK_DAY, m_btnCheckDay);
  201. DDX_Control(pDX, IDC_CHECK_BASE, m_btnCheckBase);
  202. DDX_Control(pDX, IDC_TIME_DAYTO, m_dtctrlDayTo);
  203. DDX_Control(pDX, IDC_TIME_DAYFROM, m_dtctrlDayFrom);
  204. DDX_Control(pDX, IDC_DL_PROGRESS, m_ctrlProgress);
  205. DDX_Control(pDX, IDC_STATIC_REFRESH, m_staticRefreshServer);
  206. DDX_Control(pDX, IDC_COMBO_SERVERS, m_comboServers);
  207. DDX_DateTimeCtrl(pDX, IDC_TIME_DAYFROM, m_tmDayFrom);
  208. DDX_DateTimeCtrl(pDX, IDC_TIME_DAYTO, m_tmDayTo);
  209. DDX_DateTimeCtrl(pDX, IDC_TIME_MIN5FROM, m_tmMin5From);
  210. DDX_DateTimeCtrl(pDX, IDC_TIME_MIN5TO, m_tmMin5To);
  211. //}}AFX_DATA_MAP
  212. }
  215. BEGIN_MESSAGE_MAP(CDownloadDlg, CDialog)
  216. //{{AFX_MSG_MAP(CDownloadDlg)
  217. ON_BN_CLICKED(IDC_REFRESH_SERVERS, OnRefreshServers)
  218. ON_BN_CLICKED(IDC_STOP_REFRESH, OnStopRefresh)
  219. ON_BN_CLICKED(IDC_STOP, OnStop)
  220. ON_NOTIFY(NM_SETFOCUS, IDC_TIME_DAYFROM, OnSetfocusTimeDay)
  221. ON_BN_CLICKED(IDC_PROXY_SET, OnProxySet)
  222. ON_NOTIFY(NM_SETFOCUS, IDC_TIME_DAYTO, OnSetfocusTimeDay)
  223. ON_NOTIFY(NM_SETFOCUS, IDC_TIME_MIN5FROM, OnSetfocusTimeMin5)
  224. ON_NOTIFY(NM_SETFOCUS, IDC_TIME_MIN5TO, OnSetfocusTimeMin5)
  225. //}}AFX_MSG_MAP
  226. ON_MESSAGE(WM_USER_DOWNLOAD_MESSAGE, OnDownloadMessage)
  227. ON_MESSAGE(WM_USER_DOWNLOAD_PROGRESS, OnDownloadProgress)
  228. ON_MESSAGE(WM_USER_DOWNLOAD_END, OnDownloadEnd)
  229. ON_MESSAGE(WM_APP_STKRECEIVER_DATA, OnStkReceiverData)
  230. END_MESSAGE_MAP()
  232. /////////////////////////////////////////////////////////////////////////////
  233. // CDownloadDlg message handlers
  235. BOOL CDownloadDlg::OnInitDialog() 
  236. {
  237. CDialog::OnInitDialog();
  239. // TODO: Add extra initialization here
  240. m_staticRefreshServer.SetWindowPos( NULL, 0,0,0,0, SWP_NOSIZE|SWP_NOMOVE|SWP_NOZORDER|SWP_HIDEWINDOW);
  241. m_staticStatus.SetWindowPos( NULL, 0,0,0,0, SWP_NOSIZE|SWP_NOMOVE|SWP_NOZORDER|SWP_HIDEWINDOW);
  242. m_ctrlProgress.SetWindowPos( NULL, 0,0,0,0, SWP_NOSIZE|SWP_NOMOVE|SWP_NOZORDER|SWP_HIDEWINDOW);
  244. m_btnStopRefresh.EnableWindow( FALSE );
  245. m_btnStop.EnableWindow( FALSE );
  247. InitComboServers();
  249. m_ctrlProgress.SetRange( 0, STKLIB_MAX_PROGRESS );
  250. m_ctrlProgress.SetPos( 0 );
  251. m_btnCheckReport.SetCheck( AfxGetStkReceiver().NetEngineIsWorking() );
  252. m_btnCheckReport.EnableWindow( AfxGetStkReceiver().NetEngineIsWorking() );
  253. m_btnCheckDay.SetCheck( TRUE );
  255. // Set Time
  256. CSPTime tmCurrent = CSPTime::GetCurrentTime();
  257. if( tmCurrent.GetHour() < 15 )
  258. tmCurrent -= CSPTimeSpan(1,0,0,0);
  260. if( tmCurrent.GetMonth() == 1 && tmCurrent.GetDay() == 1 )
  261. tmCurrent -= CSPTimeSpan(1,0,0,0);
  262. if( tmCurrent.GetMonth() == 5 && tmCurrent.GetDay() == 1 )
  263. tmCurrent -= CSPTimeSpan(1,0,0,0);
  264. if( tmCurrent.GetMonth() == 10 && tmCurrent.GetDay() == 1 )
  265. tmCurrent -= CSPTimeSpan(1,0,0,0);
  266. if( tmCurrent.GetDayOfWeek() == 1 )
  267. tmCurrent -= CSPTimeSpan(2,0,0,0);
  268. if( tmCurrent.GetDayOfWeek() == 7 )
  269. tmCurrent -= CSPTimeSpan(1,0,0,0);
  271. m_tmDayFrom = m_netdb.GetTimeInitial().GetTime();
  272. m_tmDayTo = tmCurrent;
  273. m_tmMin5From = m_tmMin5To = tmCurrent;
  275. CSPTime tmLatest;
  276. if( m_netdb.SetRootPath( AfxGetProfile().GetSelfDBPath(), IStStore::dbtypeSelfDB )
  277. && m_netdb.GetTimeLocalRange( &tmLatest, NULL, NULL ) )
  278. {
  279. CString strTimeLatest = (LPCTSTR)AfxGetTimeString( tmLatest.GetTime(), "%Y-%m-%d", TRUE ); //tmLatest.Format( "%Y-%m-%d,%A" );
  280. m_staticLatest.SetWindowText( strTimeLatest );
  282. m_tmDayFrom = CSPTime(tmLatest.GetTime()) + CSPTimeSpan(1,0,0,0 );
  284. if( m_tmDayFrom.GetMonth() == 1 && m_tmDayFrom.GetDay() == 1 )
  285. m_tmDayFrom += CSPTimeSpan(1,0,0,0);
  286. if( m_tmDayFrom.GetMonth() == 5 && m_tmDayFrom.GetDay() == 1 )
  287. m_tmDayFrom += CSPTimeSpan(1,0,0,0);
  288. if( m_tmDayFrom.GetMonth() == 10 && m_tmDayFrom.GetDay() == 1 )
  289. m_tmDayFrom += CSPTimeSpan(1,0,0,0);
  290. if( m_tmDayFrom.GetDayOfWeek() == 1 )
  291. m_tmDayFrom += CSPTimeSpan(1,0,0,0);
  292. if( m_tmDayFrom.GetDayOfWeek() == 7 )
  293. m_tmDayFrom += CSPTimeSpan(2,0,0,0);
  295. if( m_tmDayFrom > m_tmDayTo )
  296. m_tmDayTo = m_tmDayFrom;
  297. }
  299. // Initial Download
  300. if( m_bInitDownloadTime )
  301. {
  302. m_tmDayFrom = m_tmDLBegin;
  303. m_tmDayTo = m_tmDLEnd;
  304. }
  306. UpdateData( FALSE );
  308. if( m_bAutoRun )
  309. {
  310. UpdateWindow( );
  311. ShowWindow( SW_SHOW );
  312. OnOK( );
  313. }
  315. return TRUE;  // return TRUE unless you set the focus to a control
  316.               // EXCEPTION: OCX Property Pages should return FALSE
  317. }
  319. void CDownloadDlg::InitComboServers( )
  320. {
  321. for( int i=m_comboServers.GetCount()-1; i>=0; i-- )
  322. m_comboServers.DeleteString(i);
  323. CString strMain = AfxGetProfile().GetMainServer();
  324. m_comboServers.AddString( strMain );
  326. CSPStringArray & astr = AfxGetProfile().GetDownloadServers();
  327. for( i=0; i<astr.GetSize(); i++ )
  328. {
  329. if( 0 != strMain.CompareNoCase((LPCTSTR)astr.ElementAt(i)) )
  330. m_comboServers.AddString( (LPCTSTR)astr.ElementAt(i) );
  331. }
  332. m_comboServers.SelectString( 0, AfxGetProfile().GetDefaultServer() );
  333. }
  335. BOOL CDownloadDlg::PreTranslateMessage(MSG* pMsg) 
  336. {
  337. // TODO: Add your specialized code here and/or call the base class
  339. return CDialog::PreTranslateMessage(pMsg);
  340. }
  342. void CDownloadDlg::OnOK() 
  343. {
  344. UpdateData( );
  346. // clear error pac msg
  347. m_astrErrorPac.RemoveAll();
  349. // disable and enble buttons
  350. m_btnOK.EnableWindow( FALSE );
  351. m_btnCancel.EnableWindow( FALSE );
  352. m_btnProxySet.EnableWindow( FALSE );
  353. m_btnRefreshServers.EnableWindow( FALSE );
  354. m_btnStop.EnableWindow( TRUE );
  356. BOOL bOK = TRUE;
  358. // Set Message
  359. SendMessage( WM_USER_DOWNLOAD_MESSAGE, PROG_NEWGROUP, IDS_DLMSG_CONNECTING );
  360. m_staticStatus.SetWindowPos( NULL, 0,0,0,0, SWP_NOSIZE|SWP_NOMOVE|SWP_NOZORDER|SWP_SHOWWINDOW);
  361. MSG msg;
  362. while (::PeekMessage(&msg, NULL, NULL, NULL, PM_NOREMOVE) )
  363. AfxGetApp()->PumpMessage();
  365. // set server info
  366. if( bOK && !m_netdb.SetRootPath( AfxGetProfile().GetSelfDBPath(), IStStore::dbtypeSelfDB ) )
  367. {
  368. AfxMessageBox( IDS_ROOTPATH_ERROR, MB_OK | MB_ICONINFORMATION );
  369. bOK = FALSE;
  370. }
  371. CString strServer;
  372. m_comboServers.GetWindowText( strServer );
  373. CDownloadServer server;
  374. if( !server.FromString( strServer ) )
  375. {
  376. AfxMessageBox( IDS_NOSERVER_DEFINED, MB_OK | MB_ICONINFORMATION );
  377. bOK = FALSE;
  378. }
  379. AfxGetProfile().SetDefaultServer( strServer );
  381. if( bOK && !m_netdb.SetServer( server.m_strAddress, server.m_nPort ) )
  382. {
  383. AfxMessageBox( IDS_NOSERVER_DEFINED, MB_OK | MB_ICONINFORMATION );
  384. bOK = FALSE;
  385. }
  386. // set access type
  387. if( bOK && !m_netdb.SetAccessType( AfxGetQSProfile().GetAccessType(), AfxGetQSProfile().GetProxyType(),
  388. AfxGetQSProfile().GetProxyAddress(), AfxGetQSProfile().GetProxyPort(),
  389. AfxGetQSProfile().GetProxyUser(), AfxGetQSProfile().GetProxyPasswd() ) )
  390. {
  391. AfxMessageBox( IDS_NOSERVER_DEFINED, MB_OK | MB_ICONINFORMATION );
  392. bOK = FALSE;
  393. }
  395. // login
  396. if( bOK && !m_netdb.Login( AfxGetSView().GetS(), AfxGetSView().GetU(),
  397.    NULL, AfxGetProfile().GetVersion() ) )
  398. {
  399. char szErr[1024];
  400. if( m_netdb.GetLastErrorMessage(szErr,1023) )
  401. AfxMessageBox( szErr, MB_OK | MB_ICONINFORMATION );
  402. else
  403. AfxMessageBox( IDS_LOGIN_FAILED, MB_OK | MB_ICONINFORMATION );
  404. bOK = FALSE;
  405. }
  406. // load package info
  407. if( bOK && ! m_netdb.NetloadPackageInfo( NULL, NULL ) )
  408. {
  409. char szErr[1024];
  410. if( m_netdb.GetLastErrorMessage(szErr,1023) )
  411. AfxMessageBox( szErr, MB_OK | MB_ICONINFORMATION );
  412. else
  413. AfxMessageBox( IDS_DOWNLOAD_FAILED, MB_OK | MB_ICONINFORMATION );
  414. bOK = FALSE;
  415. }
  417. if( bOK )
  418. {
  419. // show message
  420. m_staticStatus.SetWindowPos( NULL, 0,0,0,0, SWP_NOSIZE|SWP_NOMOVE|SWP_NOZORDER|SWP_SHOWWINDOW);
  421. m_ctrlProgress.SetWindowPos( NULL, 0,0,0,0, SWP_NOSIZE|SWP_NOMOVE|SWP_NOZORDER|SWP_SHOWWINDOW);
  422. CString strMsg;
  423. strMsg.LoadString( IDS_DATA_DOWNLOADING );
  424. m_staticStatus.SetWindowText( strMsg );
  426. CDownloadDlg::m_hEventKillDownloadThread = CreateEvent(NULL,FALSE,FALSE,NULL);
  427. CDownloadDlg::m_hEventDownloadThreadKilled = CreateEvent(NULL,FALSE,FALSE,NULL);
  429. if( NULL == m_pDownloadInfo )
  430. m_pDownloadInfo = new DOWNLOAD_INFO;
  431. memset( m_pDownloadInfo, 0, sizeof(DOWNLOAD_INFO) );
  432. m_pDownloadInfo->hMainWnd = GetSafeHwnd();
  433. m_pDownloadInfo->pNetDB = &m_netdb;
  434. m_pDownloadInfo->bReport = m_btnCheckReport.GetCheck();
  435. m_pDownloadInfo->bDay = m_btnCheckDay.GetCheck();
  436. m_pDownloadInfo->tmDayFrom = m_tmDayFrom.GetTime();
  437. m_pDownloadInfo->tmDayTo = m_tmDayTo.GetTime();
  438. m_pDownloadInfo->bMin5 = m_btnCheckMin5.GetCheck();
  439. m_pDownloadInfo->tmMin5From = m_tmMin5From.GetTime();
  440. m_pDownloadInfo->tmMin5To = m_tmMin5To.GetTime();
  441. m_pDownloadInfo->bBase = m_btnCheckBase.GetCheck();
  442. m_pDownloadInfo->bXDR = m_btnCheckXDR.GetCheck();
  444. if( m_pDownloadInfo->bReport )
  445. {
  446. ::SendMessage( GetSafeHwnd(), WM_USER_DOWNLOAD_MESSAGE, PROG_NEWGROUP, IDS_DLMSG_REPORT );
  447. ::SendMessage( GetSafeHwnd(), WM_USER_DOWNLOAD_PROGRESS, 0, NULL );
  448. m_nReportCount = 0;
  449. m_bReportFinished = FALSE;
  450. AfxGetStkReceiver().AddRcvDataWnd( GetSafeHwnd() );
  451. CStockContainer & container = AfxGetStockContainer();
  452. container.Lock();
  453. m_nReportTotal = container.GetSize();
  454. m_nReportRequestSent = AfxGetStkReceiver().RequestStockData( CStock::dataReport, container.GetData(), min(160,container.GetSize()), 0, 0 );
  455. container.UnLock();
  456. }
  457. else
  458. {
  459. AfxBeginThread( DownloadMain, (LPVOID)m_pDownloadInfo,THREAD_PRIORITY_NORMAL);
  460. }
  461. }
  462. else
  463. {
  464. // disable and enble buttons
  465. m_btnOK.EnableWindow( TRUE );
  466. m_btnCancel.EnableWindow( TRUE );
  467. m_btnProxySet.EnableWindow( TRUE );
  468. m_btnRefreshServers.EnableWindow( TRUE );
  469. m_btnStop.EnableWindow( FALSE );
  470. }
  471. }
  473. void CDownloadDlg::OnCancel() 
  474. {
  475. // TODO: Add extra cleanup here
  476. m_btnCancel.EnableWindow( FALSE );
  477. AfxGetStkReceiver().RemoveRcvDataWnd( GetSafeHwnd() );
  479. AfxReleaseDB();
  480. char szErr[1024];
  481. if( !AfxInitializeDB( szErr, sizeof(szErr) ) ) AfxMessageBox( szErr, MB_OK|MB_ICONINFORMATION );
  483. CDialog::OnCancel( );
  485. if( m_bDBModified )
  486. {
  487. BOOL bRecalculate = FALSE;
  488. CSPTime tmLatest = 0;
  489. if( ( m_netdb.SetRootPath( AfxGetProfile().GetSelfDBPath(), IStStore::dbtypeSelfDB )
  490. && m_netdb.GetTimeLocalRange( &tmLatest, NULL, NULL )
  491. && tmLatest.GetTime() > m_tmLatestOld.GetTime() ) )
  492. {
  493. bRecalculate = TRUE;
  494. }
  495. if( m_bAutoRun )
  496. bRecalculate = FALSE;
  497. AfxGetStkUIApp()->OnDBChanged( FALSE /*bRecalculate*/ );
  498. }
  500. //if( !m_bAutoRun )
  501. //{
  502. // CNetInfoDlg dlg;
  503. // dlg.DoModal( );
  505. // if( AfxGetProfile().HasNewVersion()
  506. // && IDYES == AfxMessageBox( IDS_DOWNLOAD_UPGRADENOW, MB_YESNO ) )
  507. // {
  508. // CUpgradeDlg dlg;
  509. // dlg.SetAutoRun( TRUE );
  510. // dlg.DoModal();
  511. // }
  512. //}
  514. CDialog::PostNcDestroy();
  515. }
  517. void CDownloadDlg::OnRefreshServers() 
  518. {
  519. UpdateData( );
  521. // disable and enble buttons
  522. m_btnRefreshServers.EnableWindow( FALSE );
  523. m_btnOK.EnableWindow( FALSE );
  524. m_btnStop.EnableWindow( FALSE );
  525. m_btnCancel.EnableWindow( FALSE );
  526. m_btnProxySet.EnableWindow( FALSE );
  527. m_btnStopRefresh.EnableWindow( TRUE );
  529. // show message
  530. m_staticRefreshServer.SetWindowPos( NULL, 0,0,0,0, SWP_NOSIZE|SWP_NOMOVE|SWP_NOZORDER|SWP_SHOWWINDOW);
  531. CString strMsg;
  532. strMsg.LoadString( IDS_SERVERS_REFRESHING );
  533. m_staticRefreshServer.SetWindowText( strMsg );
  535. m_bStopRefresh = FALSE;
  536. BOOL bOK = TRUE;
  538. // refresh servers
  539. CString strServer = AfxGetProfile().GetMainServer();
  540. CDownloadServer server;
  541. if( !server.FromString( strServer ) )
  542. {
  543. AfxMessageBox( IDS_NOSERVER_DEFINED, MB_OK | MB_ICONINFORMATION );
  544. bOK = FALSE;
  545. }
  546. if( bOK && !m_netdb.SetServer( server.m_strAddress, server.m_nPort ) )
  547. {
  548. AfxMessageBox( IDS_NOSERVER_DEFINED, MB_OK | MB_ICONINFORMATION );
  549. bOK = FALSE;
  550. }
  551. // set access type
  552. if( bOK && !m_netdb.SetAccessType( AfxGetQSProfile().GetAccessType(), AfxGetQSProfile().GetProxyType(),
  553. AfxGetQSProfile().GetProxyAddress(), AfxGetQSProfile().GetProxyPort(),
  554. AfxGetQSProfile().GetProxyUser(), AfxGetQSProfile().GetProxyPasswd() ) )
  555. {
  556. AfxMessageBox( IDS_NOSERVER_DEFINED, MB_OK | MB_ICONINFORMATION );
  557. bOK = FALSE;
  558. }
  559. // login
  560. if( bOK && !m_netdb.Login( AfxGetSView().GetS(), AfxGetSView().GetU(),
  561. NULL, AfxGetProfile().GetVersion() ) ) 
  562. {
  563. char szErr[1024];
  564. if( m_netdb.GetLastErrorMessage(szErr,1023) )
  565. AfxMessageBox( szErr, MB_OK | MB_ICONINFORMATION );
  566. else
  567. AfxMessageBox( IDS_LOGIN_FAILED, MB_OK | MB_ICONINFORMATION );
  568. bOK = FALSE;
  569. }
  571. // load server info
  572. if( bOK && ! m_netdb.NetloadServerInfo( NULL, NULL ) )
  573. {
  574. char szErr[1024];
  575. if( m_netdb.GetLastErrorMessage(szErr,1023) )
  576. AfxMessageBox( szErr, MB_OK | MB_ICONINFORMATION );
  577. else
  578. AfxMessageBox( IDS_REFRESHSERVERS_FAILED, MB_OK | MB_ICONINFORMATION );
  579. }
  581. // refresh server combo list
  582. if( bOK )
  583. {
  584. CServers & servers = m_netdb.GetServers();
  585. servers.SaveToProfile();
  586. InitComboServers();
  587. }
  589. // logout
  590. m_netdb.Logout();
  592. // set message and enable buttons
  593. if( bOK )
  594. strMsg.LoadString( IDS_SERVERS_REFRESHED );
  595. else
  596. strMsg.LoadString( IDS_REFRESHSERVERS_PAUSED );
  597. m_staticRefreshServer.SetWindowText( strMsg );
  599. m_btnRefreshServers.EnableWindow( TRUE );
  600. m_btnOK.EnableWindow( TRUE );
  601. m_btnCancel.EnableWindow( TRUE );
  602. m_btnProxySet.EnableWindow( TRUE );
  603. m_btnStop.EnableWindow( FALSE );
  604. m_btnStopRefresh.EnableWindow( FALSE );
  605. }
  607. void CDownloadDlg::OnStopRefresh() 
  608. {
  609. // TODO: Add your control notification handler code here
  610. // set message and enable buttons
  611. m_bStopRefresh = TRUE;
  612. }
  614. void CDownloadDlg::OnStop()
  615. {
  616. if( !m_bReportFinished )
  617. {
  618. m_bReportFinished = TRUE;
  619. ::PostMessage( GetSafeHwnd(), WM_USER_DOWNLOAD_END, FALSE, 0 );
  620. }
  622. if( m_hEventKillDownloadThread )
  623. SetEvent( m_hEventKillDownloadThread );
  624. // if( m_hEventDownloadThreadKilled )
  625. // WaitForSingleObject( m_hEventDownloadThreadKilled, INFINITE );
  626. }
  628. LRESULT CDownloadDlg::OnStkReceiverData(WPARAM wParam, LPARAM lParam)
  629. {
  630. PCOMMPACKET pCommPacket = (PCOMMPACKET)lParam;
  632. switch( wParam )
  633. {
  634. case CStock::dataReport:
  635. if( pCommPacket && CStock::dataReport == pCommPacket->m_dwDataType 
  636. && pCommPacket->m_dwCount > 0 )
  637. {
  638. m_nReportCount += pCommPacket->m_dwCount;
  639. for( DWORD i=0; i<pCommPacket->m_dwCount; i++ )
  640. {
  641. CKData kdata;
  642. kdata.SetKType( CKData::ktypeDay );
  643. KDATA kd;
  644. if( UpdateKDATAByREPORT( kd, &(pCommPacket->m_pReport[i]) ) )
  645. {
  646. kdata.Add( kd );
  647. AfxGetDB().InstallKData( kdata, FALSE );
  648. }
  649. }
  651. CStockContainer & container = AfxGetStockContainer();
  652. if( m_nReportRequestSent - m_nReportCount < 64 && m_nReportRequestSent < (UINT)container.GetSize() )
  653. {
  654. container.Lock();
  655. m_nReportRequestSent += AfxGetStkReceiver().RequestStockData( CStock::dataReport, container.GetData()+m_nReportRequestSent, min(160,container.GetSize()-m_nReportRequestSent), 0, 0 );
  656. container.UnLock();
  657. }
  659. if( !m_bReportFinished && m_nReportCount > 0.95 * m_nReportTotal )
  660. {
  661. m_bReportFinished = TRUE;
  662. ::SendMessage( GetSafeHwnd(), WM_USER_DOWNLOAD_PROGRESS, STKLIB_MAX_PROGRESS, NULL );
  663. AfxBeginThread( DownloadMain, (LPVOID)m_pDownloadInfo,THREAD_PRIORITY_NORMAL);
  664. }
  665. else if( !m_bReportFinished && m_nReportTotal > 0 )
  666. {
  667. DWORD dwProgress = STKLIB_MAX_PROGRESS * m_nReportCount / m_nReportTotal;
  668. ::SendMessage( GetSafeHwnd(), WM_USER_DOWNLOAD_PROGRESS, dwProgress, NULL );
  669. }
  670. }
  671. break;
  672. }
  673. return 0L;
  674. }
  676. LRESULT CDownloadDlg::OnDownloadMessage(WPARAM wParam, LPARAM lParam)
  677. {
  678. // set message and enable buttons
  679. CString strNewMsg;
  680. CString strOldMsg;
  681. m_staticStatus.GetWindowText( strOldMsg );
  683. if( PROG_NEWGROUP == wParam )
  684. {
  685. strNewMsg.LoadString( lParam );
  686. }
  687. else if( PROG_ERRORPAC == wParam )
  688. {
  689. for( int i=0; i<m_astrErrorPac.GetSize(); i++ )
  690. {
  691. if( 0 == m_astrErrorPac.ElementAt(i).Compare( (LPCTSTR)lParam ) )
  692. break;
  693. }
  694. if( i == m_astrErrorPac.GetSize() )
  695. m_astrErrorPac.Add( (LPCTSTR)lParam );
  697. CString strMsg = CNetDatabase::GetProgressMessage( wParam );
  698. strMsg += (LPCTSTR)lParam;
  699. if( strMsg.GetLength() > 0 )
  700. {
  701. strNewMsg = strOldMsg;
  702. int nIndex = strOldMsg.Find( " -- " );
  703. if( -1 != nIndex )
  704. strNewMsg = strOldMsg.Left( nIndex );
  705. strNewMsg += " -- " + strMsg;
  706. }
  707. }
  708. else
  709. {
  710. CString strMsg = CNetDatabase::GetProgressMessage( wParam );
  711. if( strMsg.GetLength() > 0 )
  712. {
  713. strNewMsg = strOldMsg;
  714. int nIndex = strOldMsg.Find( " -- " );
  715. if( -1 != nIndex )
  716. strNewMsg = strOldMsg.Left( nIndex );
  717. strNewMsg += " -- " + strMsg;
  718. }
  719. }
  721. if( strNewMsg.GetLength() > 0 )
  722. {
  723. if( 0 != strNewMsg.Compare( strOldMsg ) )
  724. m_staticStatus.SetWindowText( strNewMsg );
  725. }
  727. return 1;
  728. }
  730. LRESULT CDownloadDlg::OnDownloadProgress(WPARAM wParam, LPARAM lParam)
  731. {
  732. // set message and enable buttons
  733. if( wParam < 0 )
  734. wParam = 0;
  735. if( wParam > STKLIB_MAX_PROGRESS )
  736. wParam = STKLIB_MAX_PROGRESS;
  737. m_ctrlProgress.SetPos( wParam );
  739. CString strProgressNew;
  740. strProgressNew.Format( "%.2f%%", 100.*wParam/STKLIB_MAX_PROGRESS );
  741. CString strProgressOld;
  742. m_staticProgress.GetWindowText( strProgressOld );
  743. if( 0 != strProgressNew.Compare( strProgressOld ) )
  744. m_staticProgress.SetWindowText( strProgressNew );
  745. return 1;
  746. }
  748. LRESULT CDownloadDlg::OnDownloadEnd(WPARAM wParam, LPARAM lParam)
  749. {
  750. // logout
  751. m_netdb.Logout();
  753. m_ctrlProgress.SetPos( STKLIB_MAX_PROGRESS );
  754. CString strProgressNew;
  755. strProgressNew.Format( "%.2f%%", 100. );
  756. m_staticProgress.SetWindowText( strProgressNew );
  758. // set message and enable buttons
  759. CString strMsg;
  760. if( wParam )
  761. strMsg.LoadString( IDS_DOWNLOAD_FINISHED );
  762. else
  763. strMsg.LoadString( IDS_DOWNLOAD_PAUSED );
  765. CString strComma;
  766. strComma.LoadString( IDS_COMMA );
  767. CString strFullStop;
  768. strFullStop.LoadString( IDS_FULLSTOP );
  769. for( int i=0; i<m_astrErrorPac.GetSize(); i++ )
  770. {
  771. if( 0 == i )
  772. strMsg += CNetDatabase::GetProgressMessage( PROG_ERRORPAC );
  773. else
  774. strMsg += strComma;
  775. strMsg += m_astrErrorPac.ElementAt(i);
  776. }
  778. if( m_astrErrorPac.GetSize() > 0 )
  779. strMsg += strFullStop;
  780. if( lParam )
  781. strMsg += (LPCTSTR)lParam;
  783. m_staticStatus.SetWindowText( strMsg );
  785. m_btnOK.EnableWindow( TRUE );
  786. m_btnCancel.EnableWindow( TRUE );
  787. m_btnProxySet.EnableWindow( TRUE );
  788. m_btnRefreshServers.EnableWindow( TRUE );
  789. m_btnStop.EnableWindow( FALSE );
  791. if( CDownloadDlg::m_hEventKillDownloadThread )
  792. CloseHandle(CDownloadDlg::m_hEventKillDownloadThread);
  793. if( CDownloadDlg::m_hEventDownloadThreadKilled )
  794. CloseHandle(CDownloadDlg::m_hEventDownloadThreadKilled);
  795. CDownloadDlg::m_hEventKillDownloadThread = NULL;
  796. CDownloadDlg::m_hEventDownloadThreadKilled = NULL;
  798. CSPTime tmLatest;
  799. if( m_netdb.SetRootPath( AfxGetProfile().GetSelfDBPath(), IStStore::dbtypeSelfDB )
  800. && m_netdb.GetTimeLocalRange( &tmLatest, NULL, NULL ) )
  801. {
  802. CString strTimeLatest = tmLatest.Format( "%Y-%m-%d,%A" );
  803. m_staticLatest.SetWindowText( strTimeLatest );
  804. }
  806. m_bDBModified = TRUE;
  808. if( m_bAutoRun && wParam )
  809. OnCancel( );
  811. return 1;
  812. }
  814. void CDownloadDlg::OnSetfocusTimeDay(NMHDR* pNMHDR, LRESULT* pResult) 
  815. {
  816. // TODO: Add your control notification handler code here
  817. m_btnCheckDay.SetCheck( 1 );
  818. *pResult = 0;
  819. }
  821. void CDownloadDlg::OnSetfocusTimeMin5(NMHDR* pNMHDR, LRESULT* pResult) 
  822. {
  823. // TODO: Add your control notification handler code here
  824. m_btnCheckMin5.SetCheck( 1 );
  825. *pResult = 0;
  826. }
  828. void CDownloadDlg::OnProxySet() 
  829. {
  830. AfxDoProxySetting();
  831. }