InclutionClassifyEngine.cpp 4.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175
  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. m_Engine->NominateElChemsList(chems, chems);
  117. return true;
  118. }
  119. bool InclutionClassifyEngine::IfNeedMaxEDS(COTSParticlePtr particle, CPosXrayPtr xray, double& MaxEDSTime)
  120. {
  121. throw std::logic_error("The method or operation is not implemented.");
  122. }
  123. bool InclutionClassifyEngine::ClassifyBySpectrum(COTSParticlePtr particle, CPosXrayPtr xray)
  124. {
  125. return false;
  126. }
  127. bool InclutionClassifyEngine::ClassifyExpression(COTSParticlePtr particle, CPosXrayPtr xray)
  128. {
  129. return false;
  130. }
  131. bool InclutionClassifyEngine::ClassifyByExpressionTemporarySpectrum(COTSParticlePtr particle, CPosXrayPtr xray)
  132. {
  133. return false;
  134. }
  135. OTSClassifyEngine::CLEEnginePtr CreateInclutionEngine()
  136. {
  137. return CLEEnginePtr(new InclutionClassifyEngine());
  138. }
  139. OTSClassifyEngine::CLEEnginePtr GetInclutionEngine()
  140. {
  141. static CLEEnginePtr engine;
  142. if (engine == NULL)
  143. {
  144. engine = CLEEnginePtr( new InclutionClassifyEngine());
  145. engine->Init();
  146. }
  147. return engine;
  148. }
  149. }