ZeroElementRulesDataDB.cpp 6.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267
  1. #include "stdafx.h"
  2. #include "ZeroElementRulesDataDB.h"
  3. #include "ZeroElementRulesDataTable.h"
  4. namespace OTSClassifyEngine
  5. {
  6. using namespace OTSSQLITE;
  7. CZeroElementRulesDataDB::CZeroElementRulesDataDB(CDBStoreBasePtr a_datastore)
  8. {
  9. m_tableInfo.reset(new CZeroElementRulesDataTable());
  10. myDB = CreateNewSQLiteDB(a_datastore, m_tableInfo);
  11. }
  12. CZeroElementRulesDataDB::~CZeroElementRulesDataDB()
  13. {
  14. }
  15. CDBTableBasePtr CZeroElementRulesDataDB::GetTableInfo()
  16. {
  17. /*if (!m_tableInfo)
  18. {
  19. m_tableInfo.reset(new CSTDDataTable);
  20. }*/
  21. return m_tableInfo;
  22. }
  23. BOOL CZeroElementRulesDataDB::Init(const BOOL a_bClean /*= FALSE*/)
  24. {
  25. return myDB->Init(a_bClean);
  26. }
  27. BOOL CZeroElementRulesDataDB::CreateTable(const BOOL a_bForce /*= FALSE*/)
  28. {
  29. return myDB->CreateTable(a_bForce);
  30. }
  31. BOOL CZeroElementRulesDataDB::DeleteTable()
  32. {
  33. return myDB->DeleteTable();
  34. }
  35. BOOL CZeroElementRulesDataDB::RemoveAllRows()
  36. {
  37. return myDB->RemoveAllRows();
  38. }
  39. BOOL CZeroElementRulesDataDB::IsDBExist()
  40. {
  41. return myDB->IsDBExist();
  42. }
  43. OTSSQLITE::CDBStoreBasePtr CZeroElementRulesDataDB::GetDatastore()
  44. {
  45. return myDB->GetDatastore();
  46. }
  47. OTSSQLITE::CDBQueryBasePtr CZeroElementRulesDataDB::GetTableQuery(LPCTSTR a_sOrderColumnName /*= nullptr*/)
  48. {
  49. return myDB->GetTableQuery(a_sOrderColumnName);
  50. }
  51. ZeroElementRuleList CZeroElementRulesDataDB::GetZeroElementRulesLib(BOOL bForce /*= FALSE*/)
  52. {
  53. auto query = GetTableQuery("ZeroElement");
  54. ASSERT(query);
  55. auto ZeroElementRulelist = ReadZeroElementRuleList(query);
  56. return ZeroElementRulelist;
  57. }
  58. ZeroElementRuleList CZeroElementRulesDataDB::ReadZeroElementRuleList(CDBQueryBasePtr a_query)
  59. {
  60. ZeroElementRuleList listSTDItem;
  61. int nRowId = 0;
  62. int nWrongItems = 0;
  63. while (!a_query->IsEOF())
  64. {
  65. if (!ReadZeroElementRuleItem(a_query)) //current x-ray point
  66. {
  67. LogErrorTrace(__FILE__, __LINE__, _T("Read xray point failed"));
  68. nWrongItems++;
  69. break;
  70. }
  71. listSTDItem.push_back(m_ZeroElementRuleItem);
  72. a_query->NextRow();
  73. nRowId++;
  74. }
  75. return listSTDItem;
  76. }
  77. BOOL CZeroElementRulesDataDB::ReadZeroElementRuleItem(CDBQueryBasePtr a_query)
  78. {
  79. int nCol;
  80. m_ZeroElementRuleItem = ZeroElementRulePtr(new ZeroElementRule());
  81. CString sZeroElement;
  82. nCol = (int)CZeroElementRulesDataTable::ColumnID::S_ZeroElement - (int)CZeroElementRulesDataTable::ColumnID::MIN;
  83. sZeroElement = a_query->GetColStringValue(nCol, _T(""));
  84. m_ZeroElementRuleItem->SetZeroElementName(sZeroElement.GetString());
  85. CString sUsingElementList;
  86. nCol = (int)CZeroElementRulesDataTable::ColumnID::S_UsingElementList - (int)CZeroElementRulesDataTable::ColumnID::MIN;
  87. sUsingElementList = a_query->GetColStringValue(nCol, _T(""));
  88. CElementsList elelist;
  89. std::vector < std::string> UsingElementStrlist;
  90. xmls::SplitString(sUsingElementList.GetBuffer(), UsingElementStrlist, ",");
  91. for (auto s : UsingElementStrlist)
  92. {
  93. CElementPtr ele = CElementPtr(new CElement(s.c_str()));
  94. elelist.push_back(ele);
  95. }
  96. m_ZeroElementRuleItem->SetUsingElementList(elelist);
  97. CString sUsingImgPropertyList;
  98. nCol = (int)CZeroElementRulesDataTable::ColumnID::S_UsingImgPropertyList - (int)CZeroElementRulesDataTable::ColumnID::MIN;
  99. sUsingImgPropertyList = a_query->GetColStringValue(nCol, _T(""));
  100. std::vector < std::string> UsingImgPropertyStrlist;
  101. xmls::SplitString(sUsingImgPropertyList.GetBuffer(), UsingImgPropertyStrlist, ",");
  102. m_ZeroElementRuleItem->SetImgPropertyList(UsingImgPropertyStrlist);
  103. CString sUsingOtherPropertyList;
  104. nCol = (int)CZeroElementRulesDataTable::ColumnID::S_UsingOtherPropertyList - (int)CZeroElementRulesDataTable::ColumnID::MIN;
  105. sUsingOtherPropertyList = a_query->GetColStringValue(nCol, _T(""));
  106. std::vector < std::string> UsingOtherPropertyStrlist;
  107. xmls::SplitString(sUsingOtherPropertyList.GetBuffer(), UsingOtherPropertyStrlist, ",");
  108. m_ZeroElementRuleItem->SetOtherpropertyList(UsingOtherPropertyStrlist);
  109. CString sExpression;
  110. nCol = (int)CZeroElementRulesDataTable::ColumnID::S_Expression - (int)CZeroElementRulesDataTable::ColumnID::MIN;
  111. sExpression = a_query->GetColStringValue(nCol, _T(""));
  112. m_ZeroElementRuleItem->SetExpressionStr(sExpression.GetString());
  113. return TRUE;
  114. }
  115. #pragma region Region_1
  116. //CSTDItemPtr CZeroElementRulesDataDB::GetSTDItemById(const long a_nIncAId)
  117. //{
  118. // CSTDItemPtr pSTDItem;
  119. // if (!m_listSTDItem.empty())
  120. // {
  121. // for (auto pItem : m_listSTDItem)
  122. // {
  123. // if (pItem->GetSTDId() == (DWORD)a_nIncAId)
  124. // {
  125. // pSTDItem = pItem;
  126. // }
  127. // }
  128. // }
  129. // else
  130. // {
  131. // auto tableQuery = GetQueryById(a_nIncAId);
  132. // ASSERT(tableQuery);
  133. // if (!tableQuery)
  134. // {
  135. // return pSTDItem;
  136. // }
  137. // if (!ReadSTDItem(tableQuery))
  138. // {
  139. // LogErrorTrace(__FILE__, __LINE__, _T("read x-ray failed."));
  140. // return pSTDItem;
  141. // }
  142. // pSTDItem = CSTDItemPtr(new CSTDItem(*m_pSTDItem.get()));
  143. // }
  144. // return pSTDItem;
  145. //}
  146. //CSTDItemsList& CZeroElementRulesDataDB::GetSTDItemList(const BOOL a_bForce/* = FALSE*/)
  147. //{
  148. // if (a_bForce)
  149. // {
  150. // m_listSTDItem.clear();
  151. // }
  152. // if (m_listSTDItem.size() == 0)
  153. // {
  154. // ReadSTDItemList();
  155. // }
  156. // return m_listSTDItem;
  157. //}
  158. //BOOL CZeroElementRulesDataDB::ReadSTDItemList()
  159. //{
  160. // auto tableInfoPtr = GetTableInfo();
  161. // ASSERT(tableInfoPtr);
  162. // if (!tableInfoPtr)
  163. // {
  164. // return FALSE;
  165. // }
  166. // auto query = GetTableQuery();
  167. // ASSERT(query);
  168. // if (!query)
  169. // {
  170. // return FALSE;
  171. // }
  172. // m_listSTDItem = ReadSTDItemList(query);
  173. // return TRUE;
  174. //}
  175. //CDBQueryBasePtr CZeroElementRulesDataDB::GetQueryById(const long a_nIncAId)
  176. //{
  177. // CDBQueryBasePtr query;
  178. // auto datastorePtr = GetDatastore();
  179. // ASSERT(datastorePtr);
  180. // if (!datastorePtr)
  181. // {
  182. // return query;
  183. // }
  184. // auto tableInfoPtr = GetTableInfo();
  185. // ASSERT(tableInfoPtr);
  186. // if (!tableInfoPtr)
  187. // {
  188. // return query;
  189. // }
  190. // CString sIncAIdColumnName = tableInfoPtr->GetColumnName((int)CSTDEditorDataTable::ColumnID::N_STDId - (int)CSTDEditorDataTable::ColumnID::MIN);
  191. // CString sSQLCommand;
  192. // sSQLCommand.Format(_T("SELECT * FROM \'%s\' WHERE %s = %d AND %s = %d;"),
  193. // (LPCTSTR)tableInfoPtr->GetTableName(),
  194. // (LPCTSTR)sIncAIdColumnName,
  195. // a_nIncAId);
  196. // query = datastorePtr->QueryByCommand(sSQLCommand);
  197. // ASSERT(query);
  198. // if (!query || !query->IsValid())
  199. // {
  200. // LogErrorTrace(__FILE__, __LINE__, _T("Invalid quary command (%s)."), (LPCTSTR)sSQLCommand);
  201. // ASSERT(FALSE);
  202. // return (CDBQueryBasePtr());
  203. // }
  204. // // do the table related valid checking
  205. // if (query->GetColCount() != GetTableInfo()->GetColumnCount())
  206. // {
  207. // LogErrorTrace(__FILE__, __LINE__, _T("query col num value is %d, but not %d"), query->GetColCount(), GetTableInfo()->GetColumnCount());
  208. // ASSERT(FALSE);
  209. // return (CDBQueryBasePtr());
  210. // }
  211. // return query;
  212. //}
  213. #pragma endregion Region_1
  214. }