#include "pch.h" #include "framework.h" #include "app.h" #include "level2DataDlg.h" #include "afxdialogex.h" #include "GUITool.h" #include "ExcelUtil.h" #ifdef _DEBUG #define new DEBUG_NEW #endif #include #ifdef _DEBUG #define new DEBUG_NEW #endif #include "JsonUtil.h" #include "tool.h" Level2CaptureInfo Clevel2DataDlg::level2Data; Clevel2DataDlg::Clevel2DataDlg(CWnd* pParent /*=nullptr*/) : CDialogEx(IDD_LEVEL2_DATA, pParent) { m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME); } void Clevel2DataDlg::DoDataExchange(CDataExchange* pDX) { CDialogEx::DoDataExchange(pDX); DDX_Control(pDX, IDC_LIST1, m_list); } BEGIN_MESSAGE_MAP(Clevel2DataDlg, CDialogEx) ON_WM_SYSCOMMAND() ON_WM_PAINT() ON_WM_QUERYDRAGICON() ON_LBN_SELCHANGE(IDC_LIST1, &Clevel2DataDlg::OnLbnSelchangeList1) ON_BN_CLICKED(IDC_BUTTON1, &Clevel2DataDlg::OnBnClickedButton1) ON_BN_CLICKED(IDC_BUTTON2, &Clevel2DataDlg::OnBnClickedButton2) END_MESSAGE_MAP() // Clevel2DataDlg 消息处理程序 BOOL Clevel2DataDlg::OnInitDialog() { CDialogEx::OnInitDialog(); // 将“关于...”菜单项添加到系统菜单中。 // IDM_ABOUTBOX 必须在系统命令范围内。 ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX); ASSERT(IDM_ABOUTBOX < 0xF000); CMenu* pSysMenu = GetSystemMenu(FALSE); if (pSysMenu != nullptr) { BOOL bNameValid; CString strAboutMenu; bNameValid = strAboutMenu.LoadString(IDS_ABOUTBOX); ASSERT(bNameValid); if (!strAboutMenu.IsEmpty()) { pSysMenu->AppendMenu(MF_SEPARATOR); pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu); } } // 设置此对话框的图标。 当应用程序主窗口不是对话框时,框架将自动 // 执行此操作 SetIcon(m_hIcon, TRUE); // 设置大图标 SetIcon(m_hIcon, FALSE); // 设置小图标 // TODO: 在此添加额外的初始化代码 CString st; st.Format(_T("总共%d条数据,耗时%d毫秒"), level2Data.dataList.size(), level2Data.time); GetDlgItem(IDC_STATIC)->SetWindowTextW(st); //设置list的数据 list::iterator ele; m_list.InsertColumn(0, _T("序号"), LVCFMT_CENTER, 40); m_list.InsertColumn(1, _T("时间"), LVCFMT_LEFT, 80); m_list.InsertColumn(2, _T("价格"), LVCFMT_CENTER, 80); m_list.InsertColumn(3, _T("手数"), LVCFMT_CENTER, 50); m_list.InsertColumn(4, _T("类型"), LVCFMT_RIGHT, 40); int index = 0; for (ele = level2Data.dataList.begin();ele != level2Data.dataList.end();++ele) { index++; TradeData *data = *ele; CString time; time = data->time.c_str(); CString operate; switch (data->operateType) { case OPERATE_BUY: operate = _T("买");break; case OPERATE_BUY_CANCEL: operate = _T("买撤");break; case OPERATE_SELL: operate = _T("卖");break; case OPERATE_SELL_CANCEL: operate = _T("卖撤");break; default: break; } char chCode[20]; sprintf(chCode, "%.2lf", stod(data->price)); std::string price(chCode); if (data->limitPrice == LIMIT_PRICE_NORMAL) { price.append("(正常)"); } else if (data->limitPrice == LIMIT_PRICE_UP) { price.append("(涨停)"); } else if (data->limitPrice == LIMIT_PRICE_DOWN) { price.append("(跌停)"); } string t=data->time; if (data->cancelTime > 0) { t.append(" ").append(to_string( data->cancelTime)); string cancelTimeUnit; if (data->cancelTimeUnit == TIME_SECOND) { cancelTimeUnit = "s"; } else if (data->cancelTimeUnit == TIME_MINITE) { cancelTimeUnit = "m"; } else if (data->cancelTimeUnit == TIME_HOUR) { cancelTimeUnit = "h"; } t.append(cancelTimeUnit); } m_list.InsertItem(index-1, NULL); m_list.SetItemText(index - 1, 0, CString(to_string( index).c_str())); m_list.SetItemText(index - 1, 1, CString(t.c_str())); m_list.SetItemText(index - 1, 2, CString(price.c_str())); m_list.SetItemText(index - 1, 3, CString(to_string(data->num).c_str())); m_list.SetItemText(index - 1, 4, operate); } return TRUE; // 除非将焦点设置到控件,否则返回 TRUE } // 如果向对话框添加最小化按钮,则需要下面的代码 // 来绘制该图标。 对于使用文档/视图模型的 MFC 应用程序, // 这将由框架自动完成。 void Clevel2DataDlg::OnPaint() { if (IsIconic()) { CPaintDC dc(this); // 用于绘制的设备上下文 SendMessage(WM_ICONERASEBKGND, reinterpret_cast(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 { CDialogEx::OnPaint(); } } void Clevel2DataDlg::OnLbnSelchangeList1() { // TODO: 在此添加控件通知处理程序代码 } void Clevel2DataDlg::OnBnClickedButton1() { CString root = GUITool::selectFolder(_T("请选择导出的目录")); string path = (CW2A(root.GetString())); if (path.length() > 0) { path.append("\\").append(to_string(time(0))).append(".xls"); ExcelUtil::save(level2Data.dataList, path); AfxMessageBox(_T("保存成功")); } } //导出为json void Clevel2DataDlg::OnBnClickedButton2() { string json = JsonUtil::toJsonStr( JsonUtil::toJson(level2Data.dataList)); CString root = GUITool::selectFolder(_T("请选择导出的目录")); string path = (CW2A(root.GetString())); if (path.length() > 0) { path.append("\\").append(to_string(time(0))).append(".json"); Tool::saveText(path, json); AfxMessageBox(_T("保存成功")); } }