#include "stdafx.h"
#include "SpectrumCompareEngine.h"
#include "OTSSTDLibFileMgr.h"
#include "OTSClassifyOnSpectrumCompEng.h"
#include "ExpressionParseEngine/ParticleSTD.h"
namespace OTSClassifyEngine
{
bool CurveCompareEngine::Init()
{
	
	
	CSTDLibFileMgrPtr pLibFileMgr = CSTDLibFileMgrPtr(new CSTDLibFileMgr(m_StrName));

	
	myLib->LoadSpectrumSTDFromDB(m_StrName);

	ParticleSTDPtr m_std = ParticleSTDPtr(new ParticleSTD());
	if (!pLibFileMgr->LoadPartSTD(m_std))
	{
		return FALSE;
	}

	for (int i = 0; i < myLib->GetSTDItemCount(); i++)
	{
		auto itm = myLib->GetSTDItem(i);
		auto expstditm = m_std->GetSTDItemById(itm->GetID());
		if (expstditm != nullptr)
		{
			itm->SetName(expstditm->GetName().c_str());

			DWORD r, g, b;
			sscanf(expstditm->GetColor().c_str(), "#%2X%2X%2X", &r, &g, &b);
			COLORREF rgb = RGB(r, g, b);
			itm->SetColor(rgb);

			//itm->SetElementsList(expstditm->GetKeyElementList());

			itm->SetBSEValue(expstditm->GetBSE());
			
			itm->SetDensity(expstditm->GetDensity());
			
			itm->SetFormula(expstditm->GetFormula().c_str());
			
		}
	}
	m_Engine = new CClassifyOnSpectrumCompEng(0,myLib);



	return TRUE;
}



bool CurveCompareEngine::ClassifyBySpectrum(COTSParticlePtr particle, CPosXrayPtr xray)
{
	// similarity
	double dSim = 0;
	// classify Particles
	
	auto stditm = m_Engine->GetMatchingSTD( xray, dSim);

	if (dSim > 0.8)
	{
		if (stditm != nullptr)
		{
			particle->SetType(OTS_PARTICLE_TYPE::IDENTIFIED);
			particle->SetClassifyId(stditm->GetID());
			particle->SetClassifyName(stditm->GetName().GetBuffer());
			particle->SetColor(std::to_string(stditm->GetColor()));
		}
		else
		{
			particle->SetType(OTS_PARTICLE_TYPE::NOT_IDENTIFIED);
			particle->SetClassifyName("Not Identified");
		}
	}
	else
	{
		particle->SetType(OTS_PARTICLE_TYPE::NOT_IDENTIFIED);
		particle->SetClassifyName("Not Identified");
	}
	
	

	return true;
}




bool CurveCompareEngine::IfNeedMaxEDS(COTSParticlePtr particle, CPosXrayPtr xray, double& MaxEDSTime)
{
	throw std::logic_error("The method or operation is not implemented.");
}


bool CurveCompareEngine::ClassifyExpression(COTSParticlePtr particle, CPosXrayPtr xray)
{
	return false;
}

bool CurveCompareEngine::ClassifyByExpressionTemporarySpectrum(COTSParticlePtr particle, CPosXrayPtr xray)
{
	return false;
}

bool CurveCompareEngine::ClassifyIncA(COTSParticlePtr particle, int SteelTech, CPosXrayPtr xray)
{
	return false;
}

bool CurveCompareEngine::ZeroElementProcess(COTSParticlePtr particle)
{
	return false;
}


OTSClassifyEngine::CLEEnginePtr GetCurveCompareEngine(std::string libName)
{
	static CLEEnginePtr engine;
	if (engine == NULL)
	{
		engine = CLEEnginePtr(new CurveCompareEngine(libName));
		if (!engine->Init())
		{
			return NULL;
		}
	}
	return engine;
}
}