OTSSTDLibFileMgr.cpp 4.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231
  1. #include "stdafx.h"
  2. #include "OTSSTDLibFileMgr.h"
  3. #include "OTSFileSys.h"
  4. using namespace OTSTools;
  5. namespace OTSClassifyEngine
  6. {
  7. // project file extension
  8. const CString ORETYPE_FILE_EXT = _T("db");
  9. const CString ORETYPE_FILE_FILTER = _T("Particle Type Files (*.db)|*.db|All Files (*.*)|*.*||");
  10. const CString SysSTDLibFilePath = ".\\Config\\SysData\\";
  11. // constructor
  12. CSTDLibFileMgr::CSTDLibFileMgr(std::string fileName)
  13. {
  14. m_strPathName = SysSTDLibFilePath+fileName.c_str();
  15. Init();
  16. }
  17. // destructor
  18. CSTDLibFileMgr::~CSTDLibFileMgr()
  19. {
  20. // cleanup
  21. Cleanup();
  22. }
  23. BOOL CSTDLibFileMgr::CreateSTDLibFile()
  24. {
  25. // check file name
  26. m_strPathName.Trim();
  27. if (m_strPathName.IsEmpty())
  28. {
  29. // error, wrong file name
  30. LogErrorTrace(__FILE__, __LINE__, _T("Empty file path name"));
  31. ASSERT(FALSE);
  32. return FALSE;
  33. }
  34. if (COTSFileSys::Exists(m_strPathName))
  35. {
  36. if (!Open(m_strPathName, FALSE))
  37. {
  38. LogErrorTrace(__FILE__, __LINE__, _T("Open STD file failed."));
  39. ASSERT(FALSE);
  40. return FALSE;
  41. }
  42. }
  43. else
  44. {
  45. if (!Create(m_strPathName))
  46. {
  47. LogErrorTrace(__FILE__, __LINE__, _T("Create X-ray file failed."));
  48. ASSERT(FALSE);
  49. return FALSE;
  50. }
  51. }
  52. return TRUE;
  53. }
  54. BOOL CSTDLibFileMgr::LoadPartSTD(ParticleSTDPtr lib, BOOL bClear /*= TRUE*/)
  55. {
  56. //AFX_MANAGE_STATE(AfxGetStaticModuleState());
  57. if (!COTSFileSys::Exists(m_strPathName))
  58. {
  59. LogErrorTrace(__FILE__, __LINE__, _T("GenerateSTDLib: there is no STD lib file."+ m_strPathName));
  60. return FALSE;
  61. }
  62. if (!CreateSTDLibFile())
  63. {
  64. LogErrorTrace(__FILE__, __LINE__, _T("Create or open STD file failed."));
  65. return FALSE;
  66. }
  67. GetParticleSTDDB();
  68. if (m_pParticleSTDDB->GetPartSTDLib(lib,bClear) == NULL)
  69. {
  70. return false;
  71. }
  72. return TRUE;
  73. }
  74. BOOL CSTDLibFileMgr::LoadZeroElementRulesData(ParticleSTDPtr lib, BOOL bClear /*= TRUE*/)
  75. {
  76. //AFX_MANAGE_STATE(AfxGetStaticModuleState());
  77. if (!COTSFileSys::Exists(m_strPathName))
  78. {
  79. LogErrorTrace(__FILE__, __LINE__, _T("GenerateSTDLib: there is no STD lib file."));
  80. return FALSE;
  81. }
  82. if (!CreateSTDLibFile())
  83. {
  84. LogErrorTrace(__FILE__, __LINE__, _T("Create or open STD file failed."));
  85. return FALSE;
  86. }
  87. GetZeroElementRulesDataDB();
  88. auto rules= m_pZeroElementRulesDataDB->GetZeroElementRulesLib(bClear);
  89. lib->setZeroElementRuleList(rules);
  90. return TRUE;
  91. }
  92. BOOL CSTDLibFileMgr::LoadMaxEDSRulesData(ParticleSTDPtr lib, BOOL bClear /*= TRUE*/)
  93. {
  94. if (!COTSFileSys::Exists(m_strPathName))
  95. {
  96. LogErrorTrace(__FILE__, __LINE__, _T("GenerateSTDLib: there is no STD lib file."));
  97. return FALSE;
  98. }
  99. if (!CreateSTDLibFile())
  100. {
  101. LogErrorTrace(__FILE__, __LINE__, _T("Create or open STD file failed."));
  102. return FALSE;
  103. }
  104. m_pMaxEDSRulesDataDB=GetMaxEDSRulesDataDB();
  105. auto maxedsrls = m_pMaxEDSRulesDataDB->GetMaxEDSRulesLib(bClear);
  106. lib->setMaxEDSRuleList(maxedsrls);
  107. return TRUE;
  108. }
  109. std::string CSTDLibFileMgr::LoadConstantsData( BOOL bClear /*= TRUE*/)
  110. {
  111. //AFX_MANAGE_STATE(AfxGetStaticModuleState());
  112. if (!COTSFileSys::Exists(m_strPathName))
  113. {
  114. LogErrorTrace(__FILE__, __LINE__, _T("GenerateSTDLib: there is no STD lib file."));
  115. return FALSE;
  116. }
  117. if (!CreateSTDLibFile())
  118. {
  119. LogErrorTrace(__FILE__, __LINE__, _T("Create or open STD file failed."));
  120. return FALSE;
  121. }
  122. GetConstantsDataDb();
  123. return m_pConstantsDataDb->GetConstantsLib(bClear);
  124. //return TRUE;
  125. }
  126. CParticleSTDDBPtr CSTDLibFileMgr::GetParticleSTDDB()
  127. {
  128. if (!m_pParticleSTDDB)
  129. {
  130. auto datastorePtr = GetDatastore();
  131. if (datastorePtr)
  132. {
  133. m_pParticleSTDDB = std::make_shared<CParticleSTDDB>(datastorePtr);
  134. }
  135. }
  136. ASSERT(m_pParticleSTDDB);
  137. return m_pParticleSTDDB;
  138. }
  139. OTSClassifyEngine::CConstantsDataDbPtr CSTDLibFileMgr::GetConstantsDataDb()
  140. {
  141. if (!m_pConstantsDataDb)
  142. {
  143. auto datastorePtr = GetDatastore();
  144. if (datastorePtr)
  145. {
  146. m_pConstantsDataDb = std::make_shared<CConstantsDataDb>(datastorePtr);
  147. }
  148. }
  149. ASSERT(m_pConstantsDataDb);
  150. return m_pConstantsDataDb;
  151. }
  152. OTSClassifyEngine::CMaxEDSRulesDataDBPtr CSTDLibFileMgr::GetMaxEDSRulesDataDB()
  153. {
  154. if (!m_pMaxEDSRulesDataDB)
  155. {
  156. auto datastorePtr = GetDatastore();
  157. if (datastorePtr)
  158. {
  159. m_pMaxEDSRulesDataDB = std::make_shared<CMaxEDSRulesDataDB>(datastorePtr);
  160. }
  161. }
  162. ASSERT(m_pMaxEDSRulesDataDB);
  163. return m_pMaxEDSRulesDataDB;
  164. }
  165. OTSClassifyEngine::CZeroElementRulesDataDBPtr CSTDLibFileMgr::GetZeroElementRulesDataDB()
  166. {
  167. if (!m_pZeroElementRulesDataDB)
  168. {
  169. auto datastorePtr = GetDatastore();
  170. if (datastorePtr)
  171. {
  172. m_pZeroElementRulesDataDB = std::make_shared<CZeroElementRulesDataDB>(datastorePtr);
  173. }
  174. }
  175. ASSERT(m_pZeroElementRulesDataDB);
  176. return m_pZeroElementRulesDataDB;
  177. }
  178. void CSTDLibFileMgr::Init()
  179. {
  180. }
  181. void CSTDLibFileMgr::Cleanup()
  182. {
  183. }
  184. }