InclutionClassifyEngine.cpp 4.3 KB

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