OTS/OTSCPP/OTSClassifyEngine/ParticleEngine/ParticleSTDDB.cpp

#include "stdafx.h"
#include "ParticleSTDDB.h"
#include "ParticleSTDTable.h"
#include "ParticleSTD.h"
namespace OTSClassifyEngine
{
	using namespace OTSSQLITE;
	CParticleSTDDB::CParticleSTDDB(CDBStoreBasePtr a_datastore)
	{
		m_tableInfo.reset(new CParticleSTDTable());
		myDB = CreateNewSQLiteDB(a_datastore, m_tableInfo);
	}

	CParticleSTDDB::~CParticleSTDDB()
	{
	}


	CDBTableBasePtr CParticleSTDDB::GetTableInfo()
	{
		/*if (!m_tableInfo)
		{
			m_tableInfo.reset(new CSTDDataTable);
		}*/

		return m_tableInfo;
	}

	BOOL CParticleSTDDB::Init(const BOOL a_bClean /*= FALSE*/)
	{
		return myDB->Init(a_bClean);
	}

	BOOL CParticleSTDDB::CreateTable(const BOOL a_bForce /*= FALSE*/)
	{
		return myDB->CreateTable(a_bForce);
	}

	BOOL CParticleSTDDB::DeleteTable()
	{
		return myDB->DeleteTable();
	}

	BOOL CParticleSTDDB::RemoveAllRows()
	{
		return myDB->RemoveAllRows();
	}

	BOOL CParticleSTDDB::IsDBExist()
	{
		return myDB->IsDBExist();
	}

	OTSSQLITE::CDBStoreBasePtr CParticleSTDDB::GetDatastore()
	{
		return myDB->GetDatastore();
	}

	OTSSQLITE::CDBQueryBasePtr CParticleSTDDB::GetTableQuery(LPCTSTR a_sOrderColumnName /*= nullptr*/)
	{
		return myDB->GetTableQuery(a_sOrderColumnName);
	}

	ParticleSTDPtr CParticleSTDDB::GetPartSTDLib(ParticleSTDPtr partStd,BOOL bForce/* = FALSE*/)
	{
		
		if (bForce)
		{
			partStd->PartSTDItemListClear();
		}
		auto query = GetTableQuery("STDId");
		ASSERT(query);
		if (!query)
		{
			return FALSE;
		}
		auto partstdlist= ReadParticleSTDList(query);
		partStd->setSTDList(partstdlist);

		//m_pOretype->SetName(_T(""));
		return partStd;
	}
	

	PartSTDItemList CParticleSTDDB::ReadParticleSTDList(CDBQueryBasePtr a_query)
	{
		PartSTDItemList listSTDItem;

		int nRowId = 0;
		int nWrongItems = 0;
		while (!a_query->IsEOF())
		{
			if (!ReadParticleSTDItem(a_query))	//current x-ray point		
			{
				LogErrorTrace(__FILE__, __LINE__, _T("Read xray point failed"));
				nWrongItems++;
				break;
			}

			listSTDItem.push_back(m_pSTDItem);

			a_query->NextRow();
			nRowId++;
		}

		return listSTDItem;
	}

	BOOL CParticleSTDDB::ReadParticleSTDItem(CDBQueryBasePtr a_query)
	{
		int nCol;
		m_pSTDItem = PartSTDItemPtr(new PartSTDItem());

		int nIncAIdNow;
		nCol = (int)CParticleSTDTable::ColumnID::N_STDId - (int)CParticleSTDTable::ColumnID::MIN;
		nIncAIdNow = a_query->GetColIntValue(nCol, -1);
		m_pSTDItem->SetID(nIncAIdNow);

		CString sIncAName;
		nCol = (int)CParticleSTDTable::ColumnID::S_StrName - (int)CParticleSTDTable::ColumnID::MIN;
		sIncAName = a_query->GetColStringValue(nCol, _T(""));
		m_pSTDItem->SetName(sIncAName.GetString());

		CString sColor;
		nCol = (int)CParticleSTDTable::ColumnID::S_COLOR - (int)CParticleSTDTable::ColumnID::MIN;
		sColor = a_query->GetColStringValue(nCol, _T(""));
		m_pSTDItem->SetColor(sColor.GetString());

		CString sKeyElementList;
		nCol = (int)CParticleSTDTable::ColumnID::S_KeyElementList - (int)CParticleSTDTable::ColumnID::MIN;
		sKeyElementList = a_query->GetColStringValue(nCol, _T(""));
		std::vector < std::string> keyeleStrlist;
		xmls::SplitString(sKeyElementList.GetBuffer(), keyeleStrlist, ",");
		CElementsList elelist;
		for (auto s : keyeleStrlist)
		{
			CElementPtr ele = CElementPtr(new CElement(s.c_str()));
			elelist.push_back(ele);
		}
		m_pSTDItem->SetKeyElementList(elelist);

		CString sSubElementList;
		nCol = (int)CParticleSTDTable::ColumnID::S_SubElementList - (int)CParticleSTDTable::ColumnID::MIN;
		sSubElementList = a_query->GetColStringValue(nCol, _T(""));
		std::vector < std::string> SubeleStrlist;
		xmls::SplitString(sSubElementList.GetBuffer(), SubeleStrlist, ",");
		elelist.clear();
		for (auto s : SubeleStrlist)
		{
			CElementPtr ele = CElementPtr(new CElement(s.c_str()));
			elelist.push_back(ele);
		}
		m_pSTDItem->SetSubElementList(elelist);


		CString sUsingImgPropertyList;
		nCol = (int)CParticleSTDTable::ColumnID::S_UsingImgPropertyList - (int)CParticleSTDTable::ColumnID::MIN;
		sUsingImgPropertyList = a_query->GetColStringValue(nCol, _T(""));
		std::vector < std::string> UsingImgPropertyeleStrlist;
		xmls::SplitString(sUsingImgPropertyList.GetBuffer(), UsingImgPropertyeleStrlist, ",");
		
		std::vector<std::string> val;
		for (auto s : UsingImgPropertyeleStrlist)
		{
			val.push_back(s);
		}
		m_pSTDItem->SetUsingImgPropertyNameList(val);



		CString sUsingOtherPropertyList;
		nCol = (int)CParticleSTDTable::ColumnID::S_UsingOtherPropertyList - (int)CParticleSTDTable::ColumnID::MIN;
		sUsingOtherPropertyList = a_query->GetColStringValue(nCol, _T(""));

		std::vector < std::string> UsingOtherPropertyStrlist;
		xmls::SplitString(sUsingOtherPropertyList.GetBuffer(), UsingOtherPropertyStrlist, ",");
		val.clear();
		for (auto s : UsingOtherPropertyStrlist)
		{
			val.push_back(s);
		}
		m_pSTDItem->SetUsingOtherpropertyList(val);



		CString sExpression;
		nCol = (int)CParticleSTDTable::ColumnID::S_Expression - (int)CParticleSTDTable::ColumnID::MIN;
		sExpression = a_query->GetColStringValue(nCol, _T(""));
		m_pSTDItem->SetExpressionStr(sExpression.GetString());

		std::string dHardness;
		nCol = (int)CParticleSTDTable::ColumnID::S_Hardness - (int)CParticleSTDTable::ColumnID::MIN;
		dHardness = a_query->GetColStringValue(nCol, _T(""));
		m_pSTDItem->SetHardness(dHardness);

		std::string dDensity;
		nCol = (int)CParticleSTDTable::ColumnID::S_Density - (int)CParticleSTDTable::ColumnID::MIN;
		dDensity = a_query->GetColStringValue(nCol, _T(""));
		m_pSTDItem->SetDensity(dDensity);

		std::string dElectrical_conductivity;
		nCol = (int)CParticleSTDTable::ColumnID::S_Electrical_conductivity - (int)CParticleSTDTable::ColumnID::MIN;
		dElectrical_conductivity = a_query->GetColStringValue(nCol, _T(""));
		m_pSTDItem->SetElectrical_conductivity(dElectrical_conductivity);

		int nBSE;
		nCol = (int)CParticleSTDTable::ColumnID::N_BSE - (int)CParticleSTDTable::ColumnID::MIN;
		nBSE = a_query->GetColIntValue(nCol, -1);
		m_pSTDItem->SetBSE(nBSE);

		CString sFormula;
		nCol = (int)CParticleSTDTable::ColumnID::S_Formula - (int)CParticleSTDTable::ColumnID::MIN;
		sFormula = a_query->GetColStringValue(nCol, _T(""));
		m_pSTDItem->SetFormula(sFormula.GetString());

		CString sElement;
		nCol = (int)CParticleSTDTable::ColumnID::S_Element - (int)CParticleSTDTable::ColumnID::MIN;
		sElement = a_query->GetColStringValue(nCol, _T(""));
		m_pSTDItem->SetElement(sElement.GetString());

		return TRUE;
	}
	
}