// MCADoc.cpp: CMCADoc 类的实现
//

#include "pch.h"
#include "framework.h"

// SHARED_HANDLERS 可以在实现预览、缩略图和搜索筛选器句柄的
// ATL 项目中进行定义，并允许与该项目共享文档代码。
#ifndef SHARED_HANDLERS
#include "MCA.h"
#endif

#include "MCADoc.h"

#include <stdio.h>
#include <fstream>

#ifdef _DEBUG
#define new DEBUG_NEW
#endif

// CMCADoc

IMPLEMENT_DYNCREATE(CMCADoc, CDocument)

BEGIN_MESSAGE_MAP(CMCADoc, CDocument)
END_MESSAGE_MAP()


// CMCADoc 构造/析构
CMCADoc::CMCADoc() noexcept {
    FILE *stream_tmp;
    int channel_id, channel_count;

    srand((unsigned)time(0));

	for (int i = 0; i < 1024; i++)   m_nChannelCount[i] = 0;
	for (int i = 0; i < 1024; i++)   m_nChannelSmooth[i] = 0;

	std::ifstream probeFile("data/Co60.txt");
	if (!probeFile.good()) {
		AfxMessageBox((CString)"未搜索到数据文件！");
		throw "FileNotFoundError";
	}

    freopen_s(&stream_tmp, "data/Co60.txt", "r", stdin);
    for (int i = 0; i < 1024; i++)
    {
        scanf_s("%d,%d", &channel_id, &channel_count);
        es_Co60[i] = channel_count;
        if (i > 0)   es_Co60[i] += es_Co60[i-1];
    }
    fclose(stdin);
    freopen_s(&stream_tmp, "data/Cs137.txt", "r", stdin);
    for (int i = 0; i < 1024; i++)
    {
        scanf_s("%d,%d", &channel_id, &channel_count);
        es_Cs137[i] = channel_count;
        if (i > 0)   es_Cs137[i] += es_Cs137[i-1];
    }
    fclose(stdin);
    freopen_s(&stream_tmp, "data/Na22.txt", "r", stdin);
    for (int i = 0; i < 1024; i++)
    {
        scanf_s("%d,%d", &channel_id, &channel_count);
        es_Na22[i] = channel_count;
        if (i > 0)   es_Na22[i] += es_Na22[i-1];
    }
    fclose(stdin);
    for (int i = 0; i < 1024; i++)
    {
        es_Co60[i] /= es_Co60[1023];
        es_Cs137[i] /= es_Cs137[1023];
        es_Na22[i] /= es_Na22[1023];
    }
}

CMCADoc::~CMCADoc() {
}

BOOL CMCADoc::OnNewDocument() {
	if (!CDocument::OnNewDocument())
		return FALSE;

	// TODO: 在此添加重新初始化代码
	// (SDI 文档将重用该文档)
	SetTitle(TEXT("MultiChannel Analyzer"));

	return TRUE;
}

// CMCADoc 序列化
void CMCADoc::Serialize(CArchive& ar) {
	if (ar.IsStoring())
	{
		// TODO: 在此添加存储代码
	}
	else
	{
		// TODO: 在此添加加载代码
	}
}

#ifdef SHARED_HANDLERS
// 缩略图的支持
void CMCADoc::OnDrawThumbnail(CDC& dc, LPRECT lprcBounds) {
	// 修改此代码以绘制文档数据
	dc.FillSolidRect(lprcBounds, RGB(255, 255, 255));

	CString strText = _T("TODO: implement thumbnail drawing here");
	LOGFONT lf;

	CFont* pDefaultGUIFont = CFont::FromHandle((HFONT) GetStockObject(DEFAULT_GUI_FONT));
	pDefaultGUIFont->GetLogFont(&lf);
	lf.lfHeight = 36;

	CFont fontDraw;
	fontDraw.CreateFontIndirect(&lf);

	CFont* pOldFont = dc.SelectObject(&fontDraw);
	dc.DrawText(strText, lprcBounds, DT_CENTER | DT_WORDBREAK);
	dc.SelectObject(pOldFont);
}

// 搜索处理程序的支持
void CMCADoc::InitializeSearchContent() {
	CString strSearchContent;
	// 从文档数据设置搜索内容。
	// 内容部分应由“;”分隔

	// 例如:     strSearchContent = _T("point;rectangle;circle;ole object;")；
	SetSearchContent(strSearchContent);
}

void CMCADoc::SetSearchContent(const CString& value) {
	if (value.IsEmpty())
	{
		RemoveChunk(PKEY_Search_Contents.fmtid, PKEY_Search_Contents.pid);
	}
	else
	{
		CMFCFilterChunkValueImpl *pChunk = nullptr;
		ATLTRY(pChunk = new CMFCFilterChunkValueImpl);
		if (pChunk != nullptr)
		{
			pChunk->SetTextValue(PKEY_Search_Contents, value, CHUNK_TEXT);
			SetChunkValue(pChunk);
		}
	}
}

#endif // SHARED_HANDLERS

// CMCADoc 诊断
#ifdef _DEBUG
void CMCADoc::AssertValid() const {
	CDocument::AssertValid();
}

void CMCADoc::Dump(CDumpContext& dc) const {
	CDocument::Dump(dc);
}
#endif //_DEBUG

// CMCADoc 命令
double CMCADoc::Uniform() {
    int x = rand();
    double y = (x % 10000) / 10000.0;
    return y;
}

void CMCADoc::RandomPeak(CString name, int n) {
	__int64 K;
	double eps;

	for (int i = 0; i < n; i++) {
		eps = Uniform();

		if (name == "Co60") K = std::lower_bound(es_Co60, es_Co60 + 1024, eps) - es_Co60;
		else if (name == "Cs137") K = std::lower_bound(es_Cs137, es_Cs137 + 1024, eps) - es_Cs137;
		else K = std::lower_bound(es_Na22, es_Na22 + 1024, eps) - es_Na22;

		m_nChannelCount[K] += 1;
	}
}

int CMCADoc::GetTotal(int L, int R) {
	int cSum = 0;
	for (int i = L; i <= min(1023, R); i++) cSum += m_nChannelCount[i];
	return cSum;
}

int CMCADoc::GetMax(int L, int R) {
	int cMax = 0;
	for (int i = L; i <= min(1023, R); i++)
		if (m_nChannelCount[i] > cMax)
			cMax = m_nChannelCount[i];
	return cMax;
}

CView* CMCADoc::GetView(CRuntimeClass* pClass) {
	CView* pView{};
	POSITION pos = GetFirstViewPosition();
	while (pos != NULL) {
		pView = GetNextView(pos);
		if (pView->IsKindOf(pClass))   break;
	}
	if (!pView->IsKindOf(pClass)) {
		AfxMessageBox((CString)"Cannot Locate the View!");
		return nullptr;
	}
	return pView;
}