Multichannel-Analyzer/MCA/ControlView.cpp

// ControlView.cpp: 实现文件
//

#include <math.h>
#include <chrono>
#include <fstream>
#include <sstream>
#include <iostream>

#include "pch.h"

#include "MCA.h"
#include "MCADoc.h"
#include "MainFrm.h"
#include "ControlView.h"

// CControlView

IMPLEMENT_DYNCREATE(CControlView, CFormView)

CControlView::CControlView()
	: CFormView(IDD_DIALOG_CONTROL)
    , m_nCond(100)
    , m_nT0(0)
    , m_nTime(0)
    , m_nTimeStr((CString)'0')
    , m_nTotalCount(0)
    , m_nPartCount(0)
    , m_nMaxCount(0)
    , m_nLC(4)
    , m_nCursor1(0)
    , m_nCursor2(1023)
    , m_nCursorROI(0)
    , m_nCursorROICount(0)
{
}

CControlView::~CControlView()
{
}

void CControlView::DoDataExchange(CDataExchange* pDX)
{
    CFormView::DoDataExchange(pDX);
    DDX_Text(pDX, IDC_COND, m_nCond);
    DDX_Text(pDX, IDC_TIME, m_nTimeStr);
    DDX_Text(pDX, IDC_TCOUNT, m_nTotalCount);
    DDX_Text(pDX, IDC_PCOUNT, m_nPartCount);
    DDX_Text(pDX, IDC_MCOUNT, m_nMaxCount);
    DDX_Control(pDX, IDC_MODE, m_ComboMode);
    DDX_Control(pDX, IDC_AXIS, m_ComboAxis);
    DDX_Text(pDX, IDC_RANGE, m_nLC);
    DDX_Text(pDX, IDC_CURSOR1, m_nCursor1);
    DDX_Text(pDX, IDC_CURSOR2, m_nCursor2);
    DDX_Text(pDX, IDC_CURSORROI, m_nCursorROI);
    DDX_Text(pDX, IDC_CURSORROI_COUNT, m_nCursorROICount);
}

BEGIN_MESSAGE_MAP(CControlView, CFormView)
	ON_WM_SIZE()
    ON_WM_TIMER()

	ON_COMMAND(ID_FILE_OPEN, &CControlView::OnFileOpen)
    ON_COMMAND(ID_FILE_SAVE, &CControlView::OnFileSave)
    ON_COMMAND(ID_STA_START, &CControlView::OnStaStart)
    ON_COMMAND(ID_STA_STOP, &CControlView::OnStaStop)
    ON_COMMAND(ID_STA_CLEAR, &CControlView::OnStaClear)
    ON_COMMAND(ID_SIMU_CO, &CControlView::OnSimuCo)
    ON_COMMAND(ID_SIMU_CS, &CControlView::OnSimuCs)
    ON_COMMAND(ID_SIMU_NA, &CControlView::OnSimuNa)
    ON_COMMAND(ID_AXIS_LINEAR, &CControlView::OnAxisLinear)
    ON_COMMAND(ID_AXIS_LOG, &CControlView::OnAxisLog)
    ON_COMMAND(ID_RANGE_AUTO, &CControlView::OnRangeAuto)
    ON_COMMAND(ID_RANGE_D4, &CControlView::OnRangeD4)
    ON_COMMAND(ID_RANGE_D2, &CControlView::OnRangeD2)
    ON_COMMAND(ID_RANGE_M2, &CControlView::OnRangeM2)
    ON_COMMAND(ID_RANGE_M4, &CControlView::OnRangeM4)
    ON_COMMAND(ID_DATA_ORIGIN, &CControlView::OnDataOrigin)
    ON_COMMAND(ID_DATA_3, &CControlView::OnData3)
    ON_COMMAND(ID_DATA_5, &CControlView::OnData5)

    ON_UPDATE_COMMAND_UI(ID_STA_START, &CControlView::OnUpdateStaStart)
    ON_UPDATE_COMMAND_UI(ID_STA_STOP, &CControlView::OnUpdateStaStop)
    ON_UPDATE_COMMAND_UI(ID_STA_CLEAR, &CControlView::OnUpdateStaClear)
    ON_UPDATE_COMMAND_UI(ID_SIMU_CO, &CControlView::OnUpdateSimuCo)
    ON_UPDATE_COMMAND_UI(ID_SIMU_CS, &CControlView::OnUpdateSimuCs)
    ON_UPDATE_COMMAND_UI(ID_SIMU_NA, &CControlView::OnUpdateSimuNa)
    ON_UPDATE_COMMAND_UI(ID_RANGE_AUTO, &CControlView::OnUpdateRangeAuto)
    ON_UPDATE_COMMAND_UI(ID_DATA_ORIGIN, &CControlView::OnUpdateDataOrigin)
    ON_UPDATE_COMMAND_UI(ID_DATA_3, &CControlView::OnUpdateData3)
    ON_UPDATE_COMMAND_UI(ID_DATA_5, &CControlView::OnUpdateData5)
END_MESSAGE_MAP()

// CControlView 诊断

#ifdef _DEBUG
void CControlView::AssertValid() const
{
	CFormView::AssertValid();
}

#ifndef _WIN32_WCE
void CControlView::Dump(CDumpContext& dc) const
{
	CFormView::Dump(dc);
}
#endif
#endif //_DEBUG

int CControlView::roundupPowerof2(int val)
{
    if ((val & (val - 1)) == 0)   return val;
    int andv = 1 << 30;
    while ((andv & val) == 0)   andv = andv >> 1;
    return andv << 1;
}

double CControlView::GetMilliTime()
{
    std::chrono::milliseconds ms = std::chrono::duration_cast<std::chrono::milliseconds>(
        std::chrono::system_clock::now().time_since_epoch()
    );
    return ms.count() / 1000.;
}

void CControlView::UpdateValue()
{
    // 更新信息
    m_nTotalCount = pDoc->GetTotal(0, 1023);
    m_nMaxCount = pDoc->GetMax(0, 1023);
    m_nPartCount = pDoc->GetTotal(m_nCursor1, m_nCursor2);
    m_nCursorROICount = pDoc->m_nChannelCount[m_nCursorROI];

    m_nTime += GetMilliTime() - m_nT0;
    m_nTimeStr.Format(_T("%.1f"), m_nTime);
    m_nT0 = GetMilliTime();

    // 更新屏幕信息
    UpdateData(FALSE);
}

void CControlView::SaveFirst()
{
    int res = MessageBox(TEXT("打开前是否保存？"), TEXT("保存文件"), MB_YESNO | MB_ICONQUESTION);
    if (res == IDYES)   OnFileSave();
}

// CControlView 消息处理程序
void CControlView::OnSize(UINT nType, int cx, int cy)
{
	CFormView::OnSize(nType, cx, cy);
    CRect rect;
    GetClientRect(&rect); // 获取当前客户区view大小

    CSize size;
    size.cx = rect.right - rect.left;
    size.cy = rect.bottom - rect.top;
    SetScrollSizes(MM_HIMETRIC, size); // 将CScrollView的大小设置为当前客户区大小
}

void CControlView::OnInitialUpdate()
{
    CFormView::OnInitialUpdate();

    // 添加运行模式
    m_ComboMode.AddString((CString)"时间");
    m_ComboMode.AddString((CString)"计数");
    m_ComboMode.SetCurSel(0);
    // 添加坐标轴模式
    m_ComboAxis.AddString((CString)"线性");
    m_ComboAxis.AddString((CString)"对数");
    m_ComboAxis.SetCurSel(0);

    m_Menu.LoadMenuW(IDR_MAINFRAME);
    m_wndToolBar.LoadToolBar(IDR_MAINFRAME);
    SetMenu(&m_Menu);
}

void CControlView::OnUpdate(CView* /*pSender*/, LPARAM /*lHint*/, CObject* /*pHint*/)
{
}

void CControlView::OnTimer(UINT_PTR nIDEvent)
{
    if (pDoc == NULL)   pDoc = (CMCADoc*)GetDocument();

    pDoc->SetModifiedFlag(TRUE);

    // 生成随机信号
    int n = rand() % 50 + 25;
    pDoc->RandomPeak(m_sSource, n);

    // 平滑数据
    for (int i = 0; i < 1024; i++)   pDoc->m_nChannelSmooth[i] = pDoc->m_nChannelCount[i];
    OutputDebugString(m_sSmoothType);
    if (m_sSmoothType == "S3") {
        pDoc->m_nChannelSmooth[0] = (5.0 * pDoc->m_nChannelCount[0] + 2.0 * pDoc->m_nChannelCount[1] - pDoc->m_nChannelCount[2]) / 6.0;
        for (int i = 1; i < 1023; i++)
            pDoc->m_nChannelSmooth[i] = (pDoc->m_nChannelCount[i - 1] + pDoc->m_nChannelCount[i] + pDoc->m_nChannelCount[i + 1]) / 3.0;
        pDoc->m_nChannelSmooth[1023] = (5.0 * pDoc->m_nChannelCount[1023] + 2.0 * pDoc->m_nChannelCount[1022] - pDoc->m_nChannelCount[1021]) / 6.0;
    }
    else if (m_sSmoothType == "S5") {
        pDoc->m_nChannelSmooth[0] = (3.0 * pDoc->m_nChannelCount[0] + 2.0 * pDoc->m_nChannelCount[1] + pDoc->m_nChannelCount[2] - pDoc->m_nChannelCount[4]) / 5.0;
        pDoc->m_nChannelSmooth[1] = (4.0 * pDoc->m_nChannelCount[0] + 3.0 * pDoc->m_nChannelCount[1] + 2 * pDoc->m_nChannelCount[2] + pDoc->m_nChannelCount[3]) / 10.0;
        for (int i = 2; i < 1022; i++)
            pDoc->m_nChannelSmooth[i] = (pDoc->m_nChannelCount[i - 2] + pDoc->m_nChannelCount[i - 1] + pDoc->m_nChannelCount[i] + pDoc->m_nChannelCount[i + 1] + pDoc->m_nChannelCount[i + 2]) / 5.0;
        pDoc->m_nChannelSmooth[1022] = (4.0 * pDoc->m_nChannelCount[1023] + 3.0 * pDoc->m_nChannelCount[1022] + 2 * pDoc->m_nChannelCount[1021] + pDoc->m_nChannelCount[1020]) / 10.0;
        pDoc->m_nChannelSmooth[1023] = (3.0 * pDoc->m_nChannelCount[1023] + 2.0 * pDoc->m_nChannelCount[1022] + pDoc->m_nChannelCount[1021] - pDoc->m_nChannelCount[1019]) / 5.0;
    }
    for (int i = 0; i < 1024; i++)   pDoc->m_nChannelSmooth[i] = max(0, pDoc->m_nChannelSmooth[i]);

    // 更新信息
    UpdateValue();

    // 判断终止
    CString tmp;
    m_ComboMode.GetWindowText(tmp);
    if (tmp == "时间" && m_nTime >= m_nCond)   OnStaStop();
    if (tmp == "计数" && m_nTotalCount >= m_nCond)   OnStaStop();

    // 修改坐标轴
    m_ComboAxis.GetWindowText(tmp);
    if (tmp == "对数") {
        m_bAutoFlag = FALSE;
        m_sAxisMode = "Log";
        m_sRangeMode = "Auto";
    }
    else   m_sAxisMode = "Linear";

    // 判断量程
    if (m_sRangeMode == "Auto" && m_nMaxCount >= m_nLC) m_nLC *= 2;

    pDoc->UpdateAllViews(NULL);
    CFormView::OnTimer(nIDEvent);
}

BOOL CControlView::PreTranslateMessage(MSG* pMsg)
{
    if (pMsg->message == WM_KEYDOWN && pMsg->wParam >= 0x25 && pMsg->wParam <= 0x28) {
        if (pMsg->wParam == VK_LEFT) m_nCursorROI = max(m_nCursor1, m_nCursorROI - 1);
        if (pMsg->wParam == VK_RIGHT) m_nCursorROI = min(m_nCursor2, m_nCursorROI + 1);
        if (pMsg->wParam == VK_UP) {
            m_nLC *= 2;
            m_bAutoFlag = TRUE;
            m_sRangeMode = (CString)"Manual"; 
        }
        if (pMsg->wParam == VK_DOWN) {
            m_nLC = m_nLC > 1 ? m_nLC / 2 : 1;
            m_bAutoFlag = TRUE;
            m_sRangeMode = (CString)"Manual";
        }
        return TRUE;
    }
    return CFormView::PreTranslateMessage(pMsg);
}

void CControlView::OnStaStart()
{
    // 重置结束条件
    m_nT0 = GetMilliTime();
    // 设置状态
    m_bStartFlag = FALSE;
    m_bStopFlag = TRUE;
    m_bClearFlag = FALSE;
    ((CEdit*)GetDlgItem(IDC_COND))->EnableWindow(FALSE);
    // 获取屏幕值至变量
    UpdateData(TRUE);
    // 设置更新
    SetTimer(1, 100, NULL);
}

void CControlView::OnUpdateStaStart(CCmdUI* pCmdUI)
{
    pCmdUI->Enable(m_bStartFlag);
}

void CControlView::OnFileOpen()
{
    if (pDoc == NULL)   pDoc = (CMCADoc*)GetDocument();
    if (pDoc->IsModified())   SaveFirst();
    CString szFilter = (CString)"文本文件(*.txt)|*.txt|";
    CFileDialog fileDlg(TRUE, (CString)"*.txt", NULL, OFN_OVERWRITEPROMPT, szFilter);
    if (IDCANCEL == fileDlg.DoModal())
        return;
    CString szfile = fileDlg.GetPathName();

    int k, cnt = 0;
    int count[1024] = {0};
    std::string line, tmp;
    std::ifstream ifs(szfile);
    
    try
    {
        std::getline(ifs, line);
        m_nTime = 0;
        m_nT0 = GetMilliTime() - std::stoi(line);
        while (std::getline(ifs, line))
        {
            std::stringstream ss(line);
            std::getline(ss, tmp, ',');
            k = std::stoi(tmp);
            std::getline(ss, tmp, ',');
            count[k] = std::stoi(tmp);
            cnt += 1;
        }
        for (int i = 1; i < 1024; i++)   pDoc->m_nChannelCount[i] = count[i];
        UpdateValue();
        pDoc->UpdateAllViews(NULL);
    }
    catch (const std::exception&)
    {
        AfxMessageBox((CString)"文件格式错误");
    }
}

void CControlView::OnFileSave()
{
    CString szFilter = (CString)"文本文件(*.txt)|*.txt|";
    CFileDialog fileDlg(FALSE, (CString)"*.txt", NULL, OFN_OVERWRITEPROMPT, szFilter);
    if (IDCANCEL == fileDlg.DoModal())
        return;
    CString szfile = fileDlg.GetPathName();

    std::ofstream ofs(szfile);
    ofs << m_nTime << std::endl;
    for (int i = 0; i < 1024; i++)
        ofs << i << ", " << pDoc->m_nChannelCount[i] << std::endl;

    AfxMessageBox((CString)"保存成功");

    pDoc->SetModifiedFlag(FALSE);
}

void CControlView::OnStaStop()
{
    // 设置状态
    m_bStartFlag = TRUE;
    m_bStopFlag = FALSE;
    m_bClearFlag = TRUE;
    ((CEdit*)GetDlgItem(IDC_COND))->EnableWindow(TRUE);
    // 停止更新
    KillTimer(1);
}

void CControlView::OnUpdateStaStop(CCmdUI* pCmdUI)
{
    pCmdUI->Enable(m_bStopFlag);
}

void CControlView::OnStaClear()
{
    if (pDoc == NULL)   pDoc = (CMCADoc*)GetDocument();
    // 重置
    m_nLC = 4;
    m_nTime = 0;
    m_nMaxCount = 0;
    m_nPartCount = 0;
    m_nTotalCount = 0;
    m_nTimeStr = (CString)"0.0";
    // 更新屏幕信息
    UpdateData(FALSE);
    // 设置曲线
    for (int i = 0; i < 1024; i++)   pDoc->m_nChannelCount[i] = 0;
    for (int i = 0; i < 1024; i++)   pDoc->m_nChannelSmooth[i] = 0;
    pDoc->UpdateAllViews(NULL);
}

void CControlView::OnUpdateStaClear(CCmdUI* pCmdUI)
{
    pCmdUI->Enable(m_bClearFlag);
}

void CControlView::OnSimuCo()
{
    m_sSource = (CString)"Co60";
    m_bCoFlag = FALSE;
    m_bCsFlag = TRUE;
    m_bNaFlag = TRUE;
    OnStaClear();
    OnStaStart();
}

void CControlView::OnUpdateSimuCo(CCmdUI* pCmdUI)
{
    pCmdUI->Enable(m_bCoFlag);
}

void CControlView::OnSimuCs()
{
    m_sSource = (CString)"Cs137";
    m_bCoFlag = TRUE;
    m_bCsFlag = FALSE;
    m_bNaFlag = TRUE;
    OnStaClear();
    OnStaStart();
}

void CControlView::OnUpdateSimuCs(CCmdUI* pCmdUI)
{
    pCmdUI->Enable(m_bCsFlag);
}

void CControlView::OnSimuNa()
{
    m_sSource = (CString)"Na22";
    m_bCoFlag = TRUE;
    m_bCsFlag = TRUE;
    m_bNaFlag = FALSE;
    OnStaClear();
    OnStaStart();
}

void CControlView::OnUpdateSimuNa(CCmdUI* pCmdUI)
{
    pCmdUI->Enable(m_bNaFlag);
}

void CControlView::OnAxisLinear()
{
    m_sAxisMode = (CString)"Linear";
}

void CControlView::OnAxisLog()
{
    m_sAxisMode = (CString)"Log";
}

void CControlView::OnRangeAuto()
{
    m_sRangeMode = (CString)"Auto";
    m_bAutoFlag = FALSE;
    m_nLC = roundupPowerof2(m_nMaxCount);
}

void CControlView::OnUpdateRangeAuto(CCmdUI* pCmdUI)
{
    pCmdUI->Enable(m_bAutoFlag);
}

void CControlView::OnRangeD4()
{
    m_sRangeMode = (CString)"Manual";
    m_nLC /= 4;
    m_bAutoFlag = TRUE;
}

void CControlView::OnRangeD2()
{
    m_sRangeMode = (CString)"Manual";
    m_nLC /= 2;
    m_bAutoFlag = TRUE;
}

void CControlView::OnRangeM2()
{
    m_sRangeMode = (CString)"Manual";
    m_nLC *= 2;
    m_bAutoFlag = TRUE;
}

void CControlView::OnRangeM4()
{
    m_sRangeMode = (CString)"Manual";
    m_nLC *= 4;
    m_bAutoFlag = TRUE;
}

void CControlView::OnDataOrigin()
{
    m_sSmoothType = (CString)"Origin";
    m_bOriginFlag = FALSE;
    m_bTSmoothFlag = TRUE;
    m_bFSmoothFlag = TRUE;
    pDoc->UpdateAllViews(NULL);
}

void CControlView::OnUpdateDataOrigin(CCmdUI* pCmdUI)
{
    pCmdUI->Enable(m_bOriginFlag);
}

void CControlView::OnData3()
{
    m_sSmoothType = (CString)"S3";
    m_bOriginFlag = TRUE;
    m_bTSmoothFlag = FALSE;
    m_bFSmoothFlag = TRUE;
    pDoc->UpdateAllViews(NULL);
}

void CControlView::OnUpdateData3(CCmdUI* pCmdUI)
{
    pCmdUI->Enable(m_bTSmoothFlag);
}

void CControlView::OnData5()
{
    m_sSmoothType = (CString)"S5";
    m_bOriginFlag = TRUE;
    m_bTSmoothFlag = TRUE;
    m_bFSmoothFlag = FALSE;
    pDoc->UpdateAllViews(NULL);
}

void CControlView::OnUpdateData5(CCmdUI* pCmdUI)
{
    pCmdUI->Enable(m_bFSmoothFlag);
}