#include "stdafx.h"
|
#include "ThresDotProcess.h"
|
#include "VisionRecipe.h"
|
#include "VisionBufferPro.h"
|
#include "FileRecipe.h"
|
#include "SoftVisionApp.h"
|
|
CThresDotProcess::CThresDotProcess(){
|
/* code */
|
m_nUse = 0;
|
m_id = 0;
|
m_strName.Format(_T("Point%d"), m_id);
|
|
m_minThres = 60;
|
m_maxThres = 255;
|
|
m_minSizeX = 50;
|
m_minSizeY = 50;
|
m_eDir = DIMENSION_NONE;
|
m_nPoseX = 0;
|
m_nPoseY = 0;
|
m_width = 0;
|
m_height = 0;
|
m_nSmoothSize = 2;
|
m_poseType = 0;
|
}
|
|
CThresDotProcess::CThresDotProcess(DimensionDir eDir, int id) {
|
/* code */
|
m_eDir = eDir;
|
m_nUse = 0;
|
m_id = id;
|
switch (eDir) {
|
case GLOBAL_DEFINE::DIMENSION_A:
|
m_strName.Format(_T("TopA_T%d"), m_id);
|
break;
|
case GLOBAL_DEFINE::DIMENSION_B:
|
m_strName.Format(_T("TopB_T%d"), m_id);
|
break;
|
case GLOBAL_DEFINE::DIMENSION_C:
|
m_strName.Format(_T("TopC_T%d"), m_id);
|
break;
|
case GLOBAL_DEFINE::DIMENSION_D:
|
m_strName.Format(_T("TopD_T%d"), m_id);
|
break;
|
case GLOBAL_DEFINE::DIMENSION_A_DN:
|
m_strName.Format(_T("BotA_T%d"), m_id);
|
break;
|
case GLOBAL_DEFINE::DIMENSION_B_DN:
|
m_strName.Format(_T("BotB_T%d"), m_id);
|
break;
|
case GLOBAL_DEFINE::DIMENSION_C_DN:
|
m_strName.Format(_T("BotC_T%d"), m_id);
|
break;
|
case GLOBAL_DEFINE::DIMENSION_D_DN:
|
m_strName.Format(_T("BotD_T%d"), m_id);
|
break;
|
case GLOBAL_DEFINE::DIMENSION_NONE:
|
m_strName.Format(_T("P_T%d"), m_id);
|
break;
|
default:
|
m_strName.Format(_T("P_T%d"), m_id);
|
break;
|
}
|
|
m_minThres = 60;
|
m_maxThres = 255;
|
m_nSmoothSize = 2;
|
|
m_minSizeX = 50;
|
m_minSizeY = 50;
|
m_nPoseX = 0;
|
m_nPoseY = 0;
|
m_width = 0;
|
m_height = 0;
|
m_poseType = 0;
|
}
|
|
CThresDotProcess::~CThresDotProcess(){
|
/* code */
|
m_nNgInfo.clear();
|
}
|
|
bool CThresDotProcess::ExecChip(DimensionDir eDir) {
|
/* code */
|
CBlSideData *pSideData = CVisionRecipe::getInstance()->getSideData(m_eDir);
|
if (NULL == pSideData) return false;
|
if (0 == m_poseType) {
|
if (!pSideData->m_bTopMark_Find) return false;
|
if (!pSideData->m_bBotMark_Find) return false;
|
}
|
else {
|
if (!pSideData->m_bTopPoint_Find) return false;
|
if (!pSideData->m_bBotPoint_Find) return false;
|
}
|
|
int x = 0;
|
int y = 0;
|
int width = 0;
|
int height = 0;
|
getRegion(x, y, width, height);
|
|
int x1 = x;
|
int y1 = y;
|
int x2 = x1 + width - 1;
|
int y2 = y1 + height - 1;
|
m_ptLeftTop.x = x1;
|
m_ptLeftTop.y = y1;
|
// Log_GetDebug()->TraceInfo("Trace ROI: (%d, %d, %d, %d), %d", x1, y1, x2, y2, __LINE__);
|
|
if (nullptr != pSideData) { //ÇøÓòµÄÏÔʾ½á¹û
|
DispVisionResult inf;
|
inf.nType = 1;
|
inf.strName = m_strName;
|
inf.eDir = (int)(m_eDir);
|
inf.eVision = (int)(VISION_THRES);
|
inf.rectX1 = x1;
|
inf.rectY1 = y1;
|
inf.rectX2 = x2;
|
inf.rectY2 = y2;
|
pSideData->m_vDispVisionResult.push_back(inf);
|
|
inf.nType = 4;
|
inf.strName = m_strName;
|
inf.eDir = (int)(m_eDir);
|
inf.eVision = (int)(VISION_THRES);
|
inf.pointX = x1;
|
inf.pointY = y1;
|
pSideData->m_vDispVisionResult.push_back(inf);
|
}
|
|
HalconCpp::HObject hImage;
|
if (!CVisionBufferPro::getImageROI(eDir, hImage, x1, y1, x2, y2)) return false;
|
|
//HalconCpp::HWindow hDispWin(0, 0, m_width, m_height);
|
//hDispWin.SetPart(0, 0, m_height - 1, m_width - 1);
|
//hDispWin.SetColored(6);
|
//hDispWin.DispObj(hImage);
|
//hDispWin.Click();
|
|
|
int fzX = (int)(m_minSizeX / m_xMmvsp + 0.5f);
|
int fzY = (int)(m_minSizeY / m_yMmvsp + 0.5f);
|
|
HalconCpp::HObject hResultRegion;
|
HalconCpp::GenEmptyObj(&hResultRegion);
|
if (m_minThres > 0) {
|
HalconCpp::HObject hRegion, hConRegion, hSelRegionX, hSelRegionY, hConcatRegion, hOpenRegion;
|
HalconCpp::Threshold(hImage, &hRegion, 0, m_minThres);
|
if (m_nSmoothSize > 0) {
|
double fSmoothSize = m_nSmoothSize + 0.5;
|
HalconCpp::OpeningCircle(hRegion, &hOpenRegion, fSmoothSize);
|
}
|
else {
|
hOpenRegion = hRegion;
|
}
|
HalconCpp::Connection(hOpenRegion, &hConRegion);
|
HalconCpp::SelectShape(hConRegion, &hSelRegionX, "width", "and", fzX, MAXINT);
|
HalconCpp::SelectShape(hConRegion, &hSelRegionY, "height", "and", fzY, MAXINT);
|
HalconCpp::ConcatObj(hSelRegionX, hSelRegionY, &hConcatRegion);
|
HalconCpp::Union1(hConcatRegion, &hResultRegion);
|
}
|
|
if (m_maxThres < 255) {
|
HalconCpp::HObject hRegion, hConRegion, hSelRegionX, hSelRegionY, hConcatRegion, hOpenRegion, hUnionRegion;
|
HalconCpp::Threshold(hImage, &hRegion, m_maxThres, 255);
|
if (m_nSmoothSize > 0) {
|
double fSmoothSize = m_nSmoothSize + 0.5;
|
HalconCpp::OpeningCircle(hRegion, &hOpenRegion, fSmoothSize);
|
}
|
else {
|
hOpenRegion = hRegion;
|
}
|
HalconCpp::Connection(hOpenRegion, &hConRegion);
|
HalconCpp::SelectShape(hConRegion, &hSelRegionX, "width", "and", fzX, MAXINT);
|
HalconCpp::SelectShape(hConRegion, &hSelRegionY, "height", "and", fzY, MAXINT);
|
HalconCpp::ConcatObj(hSelRegionX, hSelRegionY, &hConcatRegion);
|
HalconCpp::Union1(hConcatRegion, &hUnionRegion);
|
HalconCpp::GenEmptyObj(&hResultRegion);
|
hResultRegion.Clear();
|
hResultRegion = hUnionRegion;
|
}
|
|
HalconCpp::HObject hConRegion;
|
HalconCpp::Connection(hResultRegion, &hConRegion);
|
//hDispWin.DispObj(hConRegion);
|
//hDispWin.Click();
|
|
int nCount = ClsVision::ObjectNumber(hConRegion);
|
if (nCount < 1) return false;
|
|
getResult(hConRegion);
|
|
return true;
|
}
|
|
void CThresDotProcess::getResult(HalconCpp::HObject &hResultRegion) {
|
/* code */
|
int nCount = ClsVision::ObjectNumber(hResultRegion);
|
if (nCount < 1) return;
|
|
NgDotInfo ng;
|
ng.eDir = m_eDir;
|
ng.id = m_id;
|
ng.isRes = false;
|
ng.minValue = m_minSizeX;
|
ng.maxValue = m_minSizeY;
|
ng.strName = m_strName;
|
ng.ngType = DefectLoc_Chip; //ȱÏÝÀàÐÍ
|
ng.eVision = VISION_THRES;
|
|
CBlSideData *pSideData = CVisionRecipe::getInstance()->getSideData(m_eDir);
|
HalconCpp::HTuple hvArea, hvY, hvX;
|
HalconCpp::HTuple hvY1, hvX1, hvY2, hvX2;
|
HalconCpp::AreaCenter(hResultRegion, &hvArea, &hvY, &hvX);
|
HalconCpp::SmallestRectangle1(hResultRegion, &hvY1, &hvX1, &hvY2, &hvX2);
|
for (int i = 0; i < nCount; i++) {
|
ng.result = hvArea[i].I();
|
ng.xResult = (int)(m_xMmvsp * (hvX2[i].I() - hvX1[i].I() + 1));
|
ng.yResult = (int)(m_yMmvsp * (hvY2[i].I() - hvY1[i].I() + 1));
|
ng.xPosPxl = (int)(0.5 * (hvX2[i].I() + hvX1[i].I())) + m_ptLeftTop.x;
|
ng.yPosPxl = (int)(0.5 * (hvY1[i].I() + hvY2[i].I())) + m_ptLeftTop.y;
|
ng.x1 = hvX1[i].I() + m_ptLeftTop.x - 1;
|
ng.y1 = hvY1[i].I() + m_ptLeftTop.y - 1;
|
ng.x2 = hvX2[i].I() + m_ptLeftTop.x + 1;
|
ng.y2 = hvY2[i].I() + m_ptLeftTop.y + 1;
|
|
if (nullptr != pSideData) { //ÇøÓòµÄÏÔʾ½á¹û
|
DispVisionResult inf;
|
inf.nType = 1;
|
inf.strName = m_strName;
|
inf.eDir = (int)(m_eDir);
|
inf.eVision = (int)(VISION_THRES);
|
inf.rectX1 = ng.x1;
|
inf.rectY1 = ng.y1;
|
inf.rectX2 = ng.x2;
|
inf.rectY2 = ng.y2;
|
pSideData->m_vDispVisionResult.push_back(inf);
|
}
|
|
m_nNgInfo.push_back(ng);
|
}
|
}
|
|
void CThresDotProcess::getRegion(int &xPos, int &yPos, int &width, int &height) {
|
/* code */
|
CBlSideData* pSideData = CVisionRecipe::getInstance()->getSideData(m_eDir);
|
if (nullptr == pSideData) return;
|
|
//»ñÈ¡×ø±ê
|
Point2I offset, result;
|
offset.x = m_nPoseX;
|
offset.y = m_nPoseY;
|
if (!CVisionBufferPro::TransformToReal(m_eDir, offset, m_poseType, result)) return;
|
|
xPos = result.x;
|
yPos = result.y;
|
width = (int)(m_width / pSideData->m_dPixelSizeX);
|
height = (int)(m_height / pSideData->m_dPixelSizeY);
|
}
|
|
void CThresDotProcess::setRegion(int xPos, int yPos, int width, int height) {
|
/* code */
|
//ÉèÖÃͼÏñµÄROIÇøÓò
|
CBlSideData* pSideData = CVisionRecipe::getInstance()->getSideData(m_eDir);
|
if (nullptr == pSideData) return;
|
|
//ÉèÖÃͼÏñµÄÇøÓò
|
int x1 = xPos;
|
int y1 = yPos;
|
int x2 = xPos + width - 1;
|
int y2 = yPos + height - 1;
|
// Log_GetDebug()->TraceInfo("Trace Set ROI: (%d, %d, %d, %d), %d", x1, y1, x2, y2, __LINE__);
|
Point2I point;
|
point.x = x1;
|
point.y = y1;
|
|
Point2I pose, ptTop, ptBot;
|
if (!CSoftVisionApp::getInstance()->transformToStandard(m_eDir, point, pose, ptTop, ptBot)) return;
|
|
if (0 == m_poseType) {
|
m_nPoseX = pose.x;
|
m_nPoseY = pose.y;
|
}
|
else if (1 == m_poseType) {
|
m_nPoseX = ptTop.x;
|
m_nPoseY = ptTop.y;
|
}
|
else if (2 == m_poseType) {
|
m_nPoseX = ptBot.x;
|
m_nPoseY = ptBot.y;
|
}
|
m_width = (int)(pSideData->m_dPixelSizeX * width);
|
m_height = (int)(pSideData->m_dPixelSizeY * height);
|
}
|
|
int CThresDotProcess::getID(void) {
|
/* code*/
|
return m_id;
|
}
|
|
void CThresDotProcess::setID(int id) {
|
/* code */
|
m_id = id;
|
}
|
|
void CThresDotProcess::getMmvsp(DimensionDir eDir) {
|
/* code */
|
CBlSideData *pSideData = CVisionRecipe::getInstance()->getSideData(eDir);
|
m_xMmvsp = pSideData->m_dPixelSizeX;
|
m_yMmvsp = pSideData->m_dPixelSizeY;
|
}
|
|
int CThresDotProcess::Execute(DimensionDir eDir, Point2I offset) {
|
/* code */
|
m_nNgInfo.clear();
|
if (1 != m_nUse) return 0;
|
if (m_width < 1 || m_height < 1) return 0;
|
|
//1. »ñÈ¡½âÊͶÈ
|
getMmvsp(eDir);
|
|
//2. Ö´Ðмì²â
|
ExecChip(eDir);
|
|
int nRet = (int)(m_nNgInfo.size());
|
return nRet;
|
}
|
|
Json::Value CThresDotProcess::WriteToJson(void) {
|
/* code */
|
std::string strName = CFileRecipe::toString(m_strName);
|
Json::Value jsValue;
|
jsValue["alg type"] = THRES_VISION_TYPE;
|
jsValue["side"] = (int)(m_eDir);
|
jsValue["id"] = m_id;
|
jsValue["name"] = strName;
|
jsValue["pos x"] = m_nPoseX;
|
jsValue["pos y"] = m_nPoseY;
|
jsValue["width"] = m_width;
|
jsValue["height"] = m_height;
|
jsValue["use"] = m_nUse;
|
jsValue["min thres"] = m_minThres;
|
jsValue["max thres"] = m_maxThres;
|
jsValue["min size x"] = m_minSizeX;
|
jsValue["min size y"] = m_minSizeY;
|
jsValue["smooth size"] = m_nSmoothSize;
|
jsValue["pose type"] = m_poseType;
|
|
return jsValue;
|
}
|
|
void CThresDotProcess::DecodeJson(Json::Value &jsValue) {
|
/* code */
|
int num = (int)(jsValue.size());
|
if (num < 1) return;
|
|
//1. side
|
std::string strName = "side";
|
if (jsValue.isMember(strName.c_str()) && jsValue[strName.c_str()].isInt()) {
|
m_eDir = (DimensionDir)(jsValue[strName.c_str()].asInt());
|
}
|
|
//2. id
|
strName = "id";
|
if (jsValue.isMember(strName.c_str()) && jsValue[strName.c_str()].isInt()) {
|
m_id = jsValue[strName.c_str()].asInt();
|
}
|
|
//3. name
|
strName = "name";
|
if (jsValue.isMember(strName.c_str()) && jsValue[strName.c_str()].isString()) {
|
m_strName = CFileRecipe::toCString(jsValue[strName.c_str()].asString());
|
}
|
|
//4. use
|
strName = "use";
|
if (jsValue.isMember(strName.c_str()) && jsValue[strName.c_str()].isInt()) {
|
m_nUse = jsValue[strName.c_str()].asInt();
|
}
|
|
//5. pos x
|
strName = "pos x";
|
if (jsValue.isMember(strName.c_str()) && jsValue[strName.c_str()].isInt()) {
|
m_nPoseX = jsValue[strName.c_str()].asInt();
|
}
|
|
//6. pos y
|
strName = "pos y";
|
if (jsValue.isMember(strName.c_str()) && jsValue[strName.c_str()].isInt()) {
|
m_nPoseY = jsValue[strName.c_str()].asInt();
|
}
|
|
//8. width
|
strName = "width";
|
if (jsValue.isMember(strName.c_str()) && jsValue[strName.c_str()].isInt()) {
|
m_width = jsValue[strName.c_str()].asInt();
|
}
|
|
//9. height
|
strName = "height";
|
if (jsValue.isMember(strName.c_str()) && jsValue[strName.c_str()].isInt()) {
|
m_height = jsValue[strName.c_str()].asInt();
|
}
|
|
//10. min thres
|
strName = "min thres";
|
if (jsValue.isMember(strName.c_str()) && jsValue[strName.c_str()].isInt()) {
|
m_minThres = jsValue[strName.c_str()].asInt();
|
}
|
|
//11. min
|
strName = "max thres";
|
if (jsValue.isMember(strName.c_str()) && jsValue[strName.c_str()].isInt()) {
|
m_maxThres = jsValue[strName.c_str()].asInt();
|
}
|
|
//12. min size x
|
strName = "min size x";
|
if (jsValue.isMember(strName.c_str()) && jsValue[strName.c_str()].isInt()) {
|
m_minSizeX = jsValue[strName.c_str()].asInt();
|
}
|
|
//13. min size y
|
strName = "min size y";
|
if (jsValue.isMember(strName.c_str()) && jsValue[strName.c_str()].isInt()) {
|
m_minSizeY = jsValue[strName.c_str()].asInt();
|
}
|
|
//14. smooth size
|
strName = "smooth size";
|
if (jsValue.isMember(strName.c_str()) && jsValue[strName.c_str()].isInt()) {
|
m_nSmoothSize = jsValue[strName.c_str()].asInt();
|
}
|
|
//15. smooth size
|
strName = "pose type";
|
if (jsValue.isMember(strName.c_str()) && jsValue[strName.c_str()].isInt()) {
|
m_nSmoothSize = jsValue[strName.c_str()].asInt();
|
}
|
}
|
