InclutionClassifyEngine.cpp 4.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180
  1. #pragma once
  2. #include "stdafx.h"
  3. #include "InclutionClassifyEngine.h"
  4. #include "IClassifyEngine.h"
  5. #include "InclutionEngine/STDFileMgr.h"
  6. #include "InclutionEngine/OTSClassifyEng.h"
  7. namespace OTSClassifyEngine
  8. {
  9. bool InclutionClassifyEngine::Init()
  10. {
  11. CSTDFileMgrPtr stdfile = CSTDFileMgrPtr(new CSTDFileMgr());
  12. if (!stdfile->Load())
  13. {
  14. return false;
  15. }
  16. myStd = stdfile-> GenerateSTDLib(IDENTIFY_KIND::ALL);
  17. if (myStd!=nullptr)
  18. {
  19. m_Engine = new COTSClassifyEng(myStd);
  20. return true;
  21. }
  22. else
  23. {
  24. return false;
  25. }
  26. }
  27. bool InclutionClassifyEngine::ClassifyIncA(COTSParticlePtr particle, int SteelTech, CPosXrayPtr xray)
  28. {
  29. if (particle == nullptr || xray == nullptr) return false;
  30. int Id;
  31. int grpId;
  32. auto result = m_Engine->ClassifyXray((OTSDATA::STEEL_TECHNOLOGY)SteelTech, xray->GetElementQuantifyData(), Id, grpId);
  33. if (grpId == (int)OTS_PARTICLE_TYPE::INVALID)
  34. {
  35. particle->SetType(OTS_PARTICLE_TYPE::INVALID);
  36. particle->SetClassifyName("Invalid");
  37. particle->SetColor("#000000");
  38. particle->SetClassifyId(Id);
  39. particle->SetGroupId(grpId);
  40. std::string grpName;
  41. std::string grpColor;
  42. m_Engine->GetGroupNameAndColorById(grpId, grpName, grpColor);
  43. particle->SetGroupName(grpName);
  44. particle->SetGroupColor(grpColor);
  45. }
  46. else if (grpId == (int)OTS_PARTICLE_TYPE::NOT_IDENTIFIED)
  47. {
  48. particle->SetType(OTS_PARTICLE_TYPE::NOT_IDENTIFIED);
  49. particle->SetClassifyName("Not Identified");
  50. particle->SetColor("#000000");
  51. particle->SetClassifyId((int)OTS_PARTICLE_TYPE::NOT_IDENTIFIED);
  52. particle->SetGroupId(grpId);
  53. std::string grpName;
  54. std::string grpColor;
  55. m_Engine->GetGroupNameAndColorById(grpId, grpName, grpColor);
  56. particle->SetGroupName(grpName);
  57. particle->SetGroupColor(grpColor);
  58. }
  59. else if (grpId == (int)OTS_PARTICLE_TYPE::ISNOT_INCLUTION)
  60. {
  61. if (Id == (int)NOT_INCLUTION_ID::SiC)
  62. {
  63. particle->SetType(OTS_PARTICLE_TYPE::IDENTIFIED);
  64. particle->SetClassifyName("SiC");
  65. particle->SetClassifyId((int)NOT_INCLUTION_ID::SiC);
  66. particle->SetColor("#7B68EE");
  67. }
  68. else if (Id == (int)NOT_INCLUTION_ID::FeO)
  69. {
  70. particle->SetType(OTS_PARTICLE_TYPE::IDENTIFIED);
  71. particle->SetClassifyName("FeO");
  72. particle->SetClassifyId((int)NOT_INCLUTION_ID::FeO);
  73. particle->SetColor("#FF6347");
  74. }
  75. std::string grpName;
  76. std::string grpColor;
  77. m_Engine->GetGroupNameAndColorById(grpId, grpName, grpColor);
  78. particle->SetGroupId(grpId);
  79. particle->SetGroupName(grpName);
  80. particle->SetGroupColor(grpColor);
  81. }
  82. else
  83. {
  84. auto stditm = myStd->GetSTDItemById(Id);
  85. if (stditm != nullptr)
  86. {
  87. particle->SetType(OTS_PARTICLE_TYPE::IDENTIFIED);
  88. particle->SetClassifyId(Id);
  89. particle->SetColor(stditm->GetColor().GetBuffer());
  90. particle->SetClassifyName(stditm->GetName().GetBuffer());
  91. particle->SetGroupId(grpId);
  92. std::string grpName;
  93. std::string grpColor;
  94. m_Engine->GetGroupNameAndColorById(grpId, grpName, grpColor);
  95. particle->SetGroupName(grpName);
  96. particle->SetGroupColor(grpColor);
  97. }
  98. else
  99. {
  100. particle->SetType(OTS_PARTICLE_TYPE::NOT_IDENTIFIED);
  101. particle->SetClassifyName("Not Identified");
  102. particle->SetColor("#000000");
  103. particle->SetGroupId(grpId);
  104. std::string grpName;
  105. std::string grpColor;
  106. m_Engine->GetGroupNameAndColorById(grpId, grpName, grpColor);
  107. particle->SetGroupName(grpName);
  108. particle->SetGroupColor(grpColor);
  109. }
  110. }
  111. return result;
  112. }
  113. bool InclutionClassifyEngine::ZeroElementProcess(COTSParticlePtr particle)
  114. {
  115. auto chems = particle->GetXrayInfo()->GetElementQuantifyData();
  116. CElementChemistriesList chemsOut;
  117. m_Engine->NominateElChemsList(chems, chemsOut);
  118. particle->GetXrayInfo()->SetElementQuantifyData(chemsOut);
  119. return true;
  120. }
  121. bool InclutionClassifyEngine::IfNeedMaxEDS(COTSParticlePtr particle, CPosXrayPtr xray, double& MaxEDSTime)
  122. {
  123. throw std::logic_error("The method or operation is not implemented.");
  124. }
  125. bool InclutionClassifyEngine::ClassifyBySpectrum(COTSParticlePtr particle, CPosXrayPtr xray)
  126. {
  127. return false;
  128. }
  129. bool InclutionClassifyEngine::ClassifyExpression(COTSParticlePtr particle, CPosXrayPtr xray)
  130. {
  131. return false;
  132. }
  133. bool InclutionClassifyEngine::ClassifyByExpressionTemporarySpectrum(COTSParticlePtr particle, CPosXrayPtr xray)
  134. {
  135. return false;
  136. }
  137. OTSClassifyEngine::CLEEnginePtr CreateInclutionEngine()
  138. {
  139. return CLEEnginePtr(new InclutionClassifyEngine());
  140. }
  141. OTSClassifyEngine::CLEEnginePtr GetInclutionEngine()
  142. {
  143. static CLEEnginePtr engine;
  144. if (engine == NULL)
  145. {
  146. engine = CLEEnginePtr( new InclutionClassifyEngine());
  147. if (!engine->Init())
  148. {
  149. return NULL;
  150. }
  151. }
  152. return engine;
  153. }
  154. }