开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 数值算法/人工智能 > 生物技术 > 查看源码
CellView.cpp
资源名称：celldetect.rar　[点击查看]
上传用户：mjs2008
上传日期：2021-05-16
资源大小：5089k
文件大小：64k
源码类别：

生物技术

开发平台：

Visual C++

CellView.cpp：源码内容
  1. // CellView.cpp : implementation of the CCellView class
  2. //
  3. #include "stdafx.h"
  4. #include "Cell.h"
  6. #include "CellDoc.h"
  7. #include "CellView.h"
  9. #include "HSI.h"
  10. #include "HSIDlg.h"
  11. #include "HistogramDlg.h"
  12. #include "Set.h"
  13. #include  <math.h>
  15. #ifdef _DEBUG
  16. #define new DEBUG_NEW
  17. #undef THIS_FILE
  18. static char THIS_FILE[] = __FILE__;
  19. #endif
  21. #define MAX_HOLE 625
  23. #define BITMAP_ID 0x4D42 // universal id for a bitmap
  25. #define DISTANCE(x0,y0,x1,y1) sqrt((x0-x1)*(x0-x1)+(y0-y1)*(y0-y1))
  27. /////////////////////////////
  28. // GLOBALS
  29. HBITMAP g_hBitmap=NULL;
  30. CString g_csFileName; // current open file name
  31. int g_nMapWidth=0;
  32. int g_nMapHeight=0;
  33. //bool g_bImgBufferChanged;
  34. RGB spec_rgb = { 0xff, 0x00, 0x00 }; // 标识色
  35. //RGB * g_pOrgImgBuffer; // 全局的原始图像数据
  36. RGB * g_pImgBuffer; // 全局的图像数据
  37. //RGB * g_pImgBufferBack; // 全局的图像数据 用于redo
  38. BYTE * g_pSobelResult; // Soble运算的结果
  39. HSI * g_pHSIBuffer; // 全局的HSI数据
  40. FLAGS * g_pFlags; // 全局的标志量
  41. FLAGS * g_pFlagsBack; // 全局的标志量
  43. bool g_bDir4Ero=false;
  44. bool g_bDir4Dil=false;
  46. long g_nCellCount=0;
  47. long g_nCellTotArea=0;
  50. /////////////////////////////////////////////////////////////////////////////
  51. // 用于某些特定函数中的全局
  52. CPoint scroll_lefttop; // scroll bar 左上角
  53. double huegap,intgap,satgap; // 在FindInVectorHSI中使用.提高效率
  54. long tot_area,tot_x,tot_y,max_radius; // 用于递归
  55. vector<CENTER_POINT> points_temp; // 用于临时存储CENTER_POINT
  57. int *qh; //queue handle
  58. int *qs, *qst, *qr; //queue save, start, read
  59. long qSz; //queue size (physical)
  60. int xt, yt; //temporary x and y locations
  61. /////////////////////////////////////////////////////////////////////////////
  62. // CCellView
  64. IMPLEMENT_DYNCREATE(CCellView, CScrollView)
  66. BEGIN_MESSAGE_MAP(CCellView, CScrollView)
  67. //{{AFX_MSG_MAP(CCellView)
  68. ON_COMMAND(ID_FILE_OPEN, OnFileOpen)
  69. ON_WM_LBUTTONDOWN()
  70. ON_WM_MOUSEMOVE()
  71. ON_WM_LBUTTONUP()
  72. ON_COMMAND(ID_PROC_HSI, OnProcHsi)
  73. ON_WM_SETCURSOR()
  74. ON_COMMAND(ID_EDIT_UNDO, OnEditUndo)
  75. ON_COMMAND(ID_PROC_SOBEL, OnProcSobel)
  76. ON_COMMAND(ID_DISP_SOBEL, OnDispSobel)
  77. ON_COMMAND(ID_PROC_HISTOGRAM, OnProcHistogram)
  78. ON_COMMAND(ID_PROC_SPEC_COLOR, OnProcSpecColor)
  79. ON_COMMAND(ID_DISP_EDGE, OnDispEdge)
  80. ON_COMMAND(ID_DISP_REGION, OnDispRegion)
  81. ON_COMMAND(ID_PROC_RELOAD, OnProcReload)
  82. ON_COMMAND(ID_PROC_FINDCENTER, OnProcFindCenter)
  83. ON_COMMAND(ID_PROC_SOBEL_CORRECT, OnProcSobelCorrect)
  84. ON_COMMAND(ID_PROC_FORCE_KILL, OnProcForceKill)
  85. ON_COMMAND(ID_PROC_FORCE_SELE, OnProcForceSele)
  86. ON_COMMAND(ID_PROC_DILATION, OnProcDilation)
  87. ON_COMMAND(ID_PROC_EROSION, OnProcErosion)
  88. ON_COMMAND(ID_PROC_SMOOTH, OnProcSmooth)
  89. ON_COMMAND(ID_PROC_STAT, OnProcStat)
  90. ON_COMMAND(ID_PROC_FILLHOLE, FillHoles)
  91. //}}AFX_MSG_MAP
  92. // Standard printing commands
  93. ON_COMMAND(ID_FILE_PRINT, CScrollView::OnFilePrint)
  94. ON_COMMAND(ID_FILE_PRINT_DIRECT, CScrollView::OnFilePrint)
  95. ON_COMMAND(ID_FILE_PRINT_PREVIEW, CScrollView::OnFilePrintPreview)
  96. END_MESSAGE_MAP()
  98. /////////////////////////////////////////////////////////////////////////////
  99. // CCellView construction/destruction
  101. CCellView::CCellView()
  102. {
  103. // TODO: add construction code here
  104. m_DragRectSize.cx = 1;
  105. m_DragRectSize.cy = 1;
  107. g_csFileName = "";
  108. g_pImgBuffer = NULL;
  109. // g_pOrgImgBuffer = NULL;
  110. // g_pImgBufferBack = NULL;
  111. g_pHSIBuffer = NULL;
  112. g_pFlags = NULL;
  113. g_pFlagsBack = NULL;
  114. g_pSobelResult= NULL;
  116. // 所有布尔量初始化
  117. // g_bImgBufferChanged=false;
  119. m_bDrag=false;
  120. m_bProcHsi=false;
  121. m_bDispSobel=false;
  122. m_bIsDispEdge=false;
  123. m_bFullEdge=false;
  124. m_bForceKill=false;
  125. m_bForceAdd=false;
  127. CSize sz(g_nMapWidth,g_nMapHeight);
  128. SetScrollSizes(MM_TEXT, sz);
  129. }
  131. CCellView::~CCellView()
  132. {
  133. if(g_hBitmap)
  134. DeleteObject(g_hBitmap);
  135. if(g_pImgBuffer)
  136. delete[] g_pImgBuffer;
  137. // if (g_pOrgImgBuffer)
  138. // delete[] g_pOrgImgBuffer;
  139. // if (g_pImgBufferBack)
  140. // delete[] g_pImgBufferBack;
  141. if(g_pHSIBuffer)
  142. delete[] g_pHSIBuffer;
  143. if (g_pSobelResult)
  144. delete[] g_pSobelResult;
  145. if(g_pFlags)
  146. delete[] g_pFlags;
  147. if (g_pFlagsBack)
  148. delete[] g_pFlagsBack;
  149. }
  151. BOOL CCellView::PreCreateWindow(CREATESTRUCT& cs)
  152. {
  153. // TODO: Modify the Window class or styles here by modifying
  154. //  the CREATESTRUCT cs
  156. return CView::PreCreateWindow(cs);
  157. }
  159. /////////////////////////////////////////////////////////////////////////////
  160. // CCellView drawing
  162. void CCellView::OnDraw(CDC* pDC)
  163. {
  164. int x,y;
  165. CCellDoc* pDoc = GetDocument();
  166. ASSERT_VALID(pDoc);
  167. // TODO: add draw code for native data here
  168. if(g_hBitmap)
  169. {
  170. scroll_lefttop=GetScrollPosition();
  171. RECT rect;
  172. GetClientRect(&rect);
  173. int drawwd,drawht;
  174. if (g_nMapWidth>rect.right-rect.left)
  175. drawwd=rect.right-rect.left+1;
  176. else drawwd=g_nMapWidth;
  177. if (g_nMapHeight>rect.bottom-rect.top)
  178. drawht=rect.bottom-rect.top+1;
  179. else drawht=g_nMapHeight;
  181. BITMAPINFO bitmapinfo;
  182. CDC *pdc = GetDC();
  183. HDC dc = pdc->m_hDC;
  184. HDC memdc = ::CreateCompatibleDC(dc);
  185. ::SelectObject(memdc, g_hBitmap);
  186. if (m_bDispSobel) // 显示Sobel
  187. {
  188. BYTE *cur_pos=g_pSobelResult;
  189. for(y = 0; y < g_nMapHeight; y++)
  190. {
  191. for(x = 0; x < g_nMapWidth; x++)
  192. {
  193. if (x-scroll_lefttop.x>=0 && x-scroll_lefttop.x<=drawwd && y-scroll_lefttop.y>=0 && y-scroll_lefttop.y<=drawht)
  194. SetPixel(memdc,x,y,RGB(*cur_pos,*cur_pos,*cur_pos));
  195. cur_pos++;
  196. }
  197. }
  198. // g_bImgBufferChanged=false;
  199. }
  200. else
  201. {
  202. bitmapinfo.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
  203. bitmapinfo.bmiHeader.biWidth = g_nMapWidth;
  204. bitmapinfo.bmiHeader.biHeight = g_nMapHeight;
  205. bitmapinfo.bmiHeader.biPlanes = 1;
  206. bitmapinfo.bmiHeader.biBitCount = 24;
  207. bitmapinfo.bmiHeader.biCompression = BI_RGB;
  208. GetDIBits(memdc, g_hBitmap, 0, g_nMapHeight, NULL,
  209. &bitmapinfo, DIB_RGB_COLORS);
  210. FlipBitmapData(g_pImgBuffer); // flip it
  211. SetDIBits(memdc, g_hBitmap, 0, g_nMapHeight, g_pImgBuffer,
  212. &bitmapinfo, DIB_RGB_COLORS);
  213. FlipBitmapData(g_pImgBuffer); // flip back
  214. /* if (m_bIsDispEdge)
  215. {
  216. FlipBitmapData(g_pImgBuffer); // flip it
  217. SetDIBits(memdc, g_hBitmap, 0, g_nMapHeight, g_pImgBuffer,
  218. &bitmapinfo, DIB_RGB_COLORS);
  219. FlipBitmapData(g_pImgBuffer); // flip back
  220. }
  221. else
  222. {
  223. FlipBitmapData(g_pImgBuffer); // flip it
  224. SetDIBits(memdc, g_hBitmap, 0, g_nMapHeight, g_pImgBuffer,
  225. &bitmapinfo, DIB_RGB_COLORS);
  226. FlipBitmapData(g_pImgBuffer); // flip back
  227. }
  228. */ }
  229. ::BitBlt(dc, 0, 0, drawwd, drawht, memdc, scroll_lefttop.x, scroll_lefttop.y, SRCCOPY);
  231. FLAGS* cur_flag;
  232. if(m_bIsDispEdge)
  233. {
  234. for(y = 0; y < g_nMapHeight; y++)
  235. {
  236. cur_flag=&g_pFlags[y*g_nMapWidth];
  237. for(x = 0; x < g_nMapWidth; x++)
  238. {
  239. if(cur_flag->edged)
  240. if (x-scroll_lefttop.x>=0 && x-scroll_lefttop.x<=drawwd && y-scroll_lefttop.y>=0 && y-scroll_lefttop.y<=drawht)
  241. pdc->SetPixel(x-scroll_lefttop.x, y-scroll_lefttop.y, RGB(spec_rgb.r, spec_rgb.g, spec_rgb.b));
  242. cur_flag++;
  243. }
  244. }
  245. }
  246. else // region
  247. {
  248. for(y = 0; y < g_nMapHeight; y++)
  249. {
  250. cur_flag=&g_pFlags[y*g_nMapWidth];
  251. for(x = 0; x < g_nMapWidth; x++)
  252. {
  253. if(cur_flag->marked)
  254. if (x-scroll_lefttop.x>=0 && x-scroll_lefttop.x<=drawwd && y-scroll_lefttop.y>=0 && y-scroll_lefttop.y<=drawht)
  255. pdc->SetPixel(x-scroll_lefttop.x, y-scroll_lefttop.y, RGB(spec_rgb.r, spec_rgb.g, spec_rgb.b));
  256. cur_flag++;
  257. }
  258. }
  259. }
  260. CENTER_POINT centerp;
  261. CPen pen;
  262. pen.CreatePen(PS_SOLID, 1, RGB(spec_rgb.r, spec_rgb.g, spec_rgb.b));
  263. pdc->SelectObject(pen);
  264. for (int i=0;i<m_vCenterPoints.size();i++)
  265. {
  266. centerp=m_vCenterPoints.at(i);
  267. Arc(pdc->m_hDC,
  268. centerp.x-scroll_lefttop.x-centerp.radius,
  269. centerp.y-scroll_lefttop.y-centerp.radius,
  270. centerp.x-scroll_lefttop.x+centerp.radius,
  271. centerp.y-scroll_lefttop.y+centerp.radius,
  272. centerp.x-scroll_lefttop.x+centerp.radius,
  273. centerp.y-scroll_lefttop.y,
  274. centerp.x-scroll_lefttop.x+centerp.radius,
  275. centerp.y-scroll_lefttop.y
  276. );
  277. }
  278. DeleteObject(pen);
  280. ReleaseDC(pdc);
  281. ::DeleteDC(memdc);
  282. }
  283. }
  285. /////////////////////////////////////////////////////////////////////////////
  286. // CCellView printing
  288. BOOL CCellView::OnPreparePrinting(CPrintInfo* pInfo)
  289. {
  290. // default preparation
  291. return DoPreparePrinting(pInfo);
  292. }
  294. void CCellView::OnBeginPrinting(CDC* /*pDC*/, CPrintInfo* /*pInfo*/)
  295. {
  296. // TODO: add extra initialization before printing
  297. }
  299. void CCellView::OnEndPrinting(CDC* /*pDC*/, CPrintInfo* /*pInfo*/)
  300. {
  301. // TODO: add cleanup after printing
  302. }
  304. /////////////////////////////////////////////////////////////////////////////
  305. // CCellView diagnostics
  307. #ifdef _DEBUG
  308. void CCellView::AssertValid() const
  309. {
  310. CView::AssertValid();
  311. }
  313. void CCellView::Dump(CDumpContext& dc) const
  314. {
  315. CView::Dump(dc);
  316. }
  318. CCellDoc* CCellView::GetDocument() // non-debug version is inline
  319. {
  320. ASSERT(m_pDocument->IsKindOf(RUNTIME_CLASS(CCellDoc)));
  321. return (CCellDoc*)m_pDocument;
  322. }
  323. #endif //_DEBUG
  325. /////////////////////////////////////////////////////////////////////////////
  326. // CCellView message handlers
  328. void CCellView::OnFileOpen() 
  329. {
  330. // 打开文件对话框
  331. char szFilter[] = "位图文件 (*.bmp)|*.bmp|所有文件(*.*)|*.*||";
  332. CString szFilename;
  333. BITMAP bmp;
  334. CFileDialog *FileDlg = new CFileDialog(TRUE, NULL, NULL, OFN_OVERWRITEPROMPT, szFilter, NULL);
  336. if(FileDlg->DoModal() == IDOK)
  337. {
  338. szFilename = FileDlg->GetPathName();
  339. if(!szFilename.IsEmpty())
  340. {
  341. HANDLE handle;
  342. handle = LoadImage(theApp.m_hInstance, szFilename, IMAGE_BITMAP, 0, 0, LR_LOADFROMFILE);
  343. if(handle)
  344. {
  345. if(g_hBitmap)
  346. DeleteObject(g_hBitmap);
  347. InvalidateRect(0, TRUE);
  348. g_hBitmap = (struct HBITMAP__ *) handle;
  349. g_csFileName = FileDlg->GetFileName();
  350. theApp.m_pMainWnd->SetWindowText((LPCTSTR) (g_csFileName + " - Cell"));
  351. }
  352. else
  353. {
  354. MessageBox("无法打开文件");
  355. return;
  356. }
  357. m_vAllSelected.clear();
  358. // 所有布尔量初始化
  359. m_bDrag=false;
  360. m_bProcHsi=false;
  361. m_bForceKill=false;
  362. m_bForceAdd=false;
  363. m_bDispSobel=false;
  364. AfxGetApp()->m_pMainWnd->GetMenu()->CheckMenuItem(ID_DISP_SOBEL, MF_UNCHECKED);
  365. m_bIsDispEdge=false;
  366. AfxGetApp()->m_pMainWnd->GetMenu()->CheckMenuItem(ID_DISP_REGION, MF_CHECKED);
  367. AfxGetApp()->m_pMainWnd->GetMenu()->CheckMenuItem(ID_DISP_EDGE, MF_UNCHECKED);
  368. m_vCenterPoints.clear(); // 清除!
  370. CDC *pdc = GetDC();
  371. HDC dc = pdc->m_hDC;
  372. HDC memdc = ::CreateCompatibleDC(dc);
  373. CBitmap::FromHandle(g_hBitmap)->GetBitmap(&bmp);
  374. ::SelectObject(memdc, g_hBitmap);
  376. g_nMapWidth=bmp.bmWidth;
  377. g_nMapHeight=bmp.bmHeight;
  379. // 设置滚动条
  380. CSize sz(g_nMapWidth,g_nMapHeight);
  381. SetScrollSizes(MM_TEXT, sz);
  383. // fill g_pImgBuffer
  384. if (LoadImageBuffer(memdc,(LPSTR)(LPCTSTR)g_csFileName)==0)
  385. return;
  387. // 初始化Flags
  388. if(g_pFlags)
  389. delete[] g_pFlags;
  390. g_pFlags = new FLAGS[g_nMapHeight * g_nMapWidth];
  391. memset(g_pFlags, 0, g_nMapHeight * g_nMapWidth * sizeof(FLAGS));
  392. if (g_pFlagsBack)
  393. delete[] g_pFlagsBack;
  394. g_pFlagsBack = new FLAGS[g_nMapHeight * g_nMapWidth];
  395. memset(g_pFlagsBack, 0, g_nMapHeight * g_nMapWidth * sizeof(FLAGS));
  397. GenHSIData();
  399. OnProcSobel(); // 预先生成Sobel信息
  401. ::BitBlt(dc, 0, 0, g_nMapWidth, g_nMapHeight, memdc, 0, 0, SRCCOPY);
  402. ReleaseDC(pdc);
  403. ::DeleteDC(memdc);
  404. }
  405. }
  407. delete FileDlg;
  408. }
  410. void CCellView::OnLButtonDown(UINT nFlags, CPoint point) 
  411. {
  412. m_LBDnPoint = point;
  413. m_LastPoint = point;
  414. m_bDrag = true;
  416. // draw selection box
  417. RECT currect;
  418. currect.top = m_LBDnPoint.y;
  419. currect.left = m_LBDnPoint.x;
  420. currect.right = point.x;
  421. currect.bottom = point.y;
  423. CDC *pdc = GetDC();
  424. pdc->DrawDragRect(&currect, m_DragRectSize, NULL, m_DragRectSize, NULL, NULL);
  425. ReleaseDC(pdc);
  427. CView::OnLButtonDown(nFlags, point);
  428. }
  430. void CCellView::OnMouseMove(UINT nFlags, CPoint point) 
  431. {
  432. if(m_bDrag)
  433. {
  434. RECT lastrect, currect;
  435. lastrect.top = m_LBDnPoint.y > m_LastPoint.y ? m_LastPoint.y : m_LBDnPoint.y;
  436. lastrect.left = m_LBDnPoint.x > m_LastPoint.x ? m_LastPoint.x : m_LBDnPoint.x;
  437. lastrect.right = m_LBDnPoint.x > m_LastPoint.x ? m_LBDnPoint.x : m_LastPoint.x;
  438. lastrect.bottom = m_LBDnPoint.y > m_LastPoint.y ? m_LBDnPoint.y : m_LastPoint.y;
  439. currect.top = m_LBDnPoint.y > point.y ? point.y : m_LBDnPoint.y;
  440. currect.left = m_LBDnPoint.x > point.x ? point.x : m_LBDnPoint.x;
  441. currect.right = m_LBDnPoint.x > point.x ? m_LBDnPoint.x : point.x;
  442. currect.bottom = m_LBDnPoint.y > point.y ? m_LBDnPoint.y : point.y;
  444. SIZE sz;
  445. sz.cx = 1;
  446. sz.cy = 1;
  448. CDC *pdc = GetDC();
  449. pdc->DrawDragRect(&currect, m_DragRectSize, &lastrect, m_DragRectSize, NULL, NULL);
  450. ReleaseDC(pdc);
  452. m_LastPoint = point;
  453. }
  455. CView::OnMouseMove(nFlags, point);
  456. }
  458. void CCellView::OnLButtonUp(UINT nFlags, CPoint point) 
  459. {
  460. if (!m_bDrag) // 安全措施
  461. {
  462. CView::OnLButtonUp(nFlags, point);
  463. return;
  464. }
  466. int exchange;
  467. m_bDrag = false;
  469. // 画选择框
  470. m_SelectedRect.left = m_LBDnPoint.x;
  471. m_SelectedRect.top = m_LBDnPoint.y;
  472. m_SelectedRect.right = point.x;
  473. m_SelectedRect.bottom = point.y;
  475. if(m_SelectedRect.left > m_SelectedRect.right)
  476. {
  477. exchange = m_SelectedRect.left;
  478. m_SelectedRect.left = m_SelectedRect.right;
  479. m_SelectedRect.right = exchange;
  480. }
  482. if(m_SelectedRect.top > m_SelectedRect.bottom)
  483. {
  484. exchange = m_SelectedRect.top;
  485. m_SelectedRect.top = m_SelectedRect.bottom;
  486. m_SelectedRect.bottom = exchange;
  487. }
  489. SIZE zero;
  490. zero.cx = 0;
  491. zero.cy = 0;
  493. CDC *pdc = GetDC();
  494. pdc->DrawDragRect(&m_SelectedRect, zero, &m_SelectedRect, m_DragRectSize, NULL, NULL);
  495. ReleaseDC(pdc);
  497. m_LastPoint.x = 0; // 设置为0
  499. if(g_hBitmap)
  500. {
  501. if(m_bProcHsi)
  502. {
  503. if(m_SelectedRect.left >= m_SelectedRect.right || m_SelectedRect.top >= m_SelectedRect.bottom)
  504. MessageBox("选择的范围不正确.请重新选择!");
  505. else
  506. {
  507. m_bProcHsi = false;
  508. CHSIDlg hsiDlg(&m_SelectedRect);
  509. if(hsiDlg.DoModal() == IDOK)
  510. ProcHSI(hsiDlg.m_bEx);
  511. }
  512. }
  513. else if (m_bForceKill)
  514. {
  515. if(m_SelectedRect.left >= m_SelectedRect.right || m_SelectedRect.top >= m_SelectedRect.bottom)
  516. MessageBox("选择的范围不正确.请重新选择!");
  517. else
  518. {
  519. m_bForceKill = false;
  520. m_SelectedRect.left+=scroll_lefttop.x;
  521. m_SelectedRect.right+=scroll_lefttop.x;
  522. m_SelectedRect.top+=scroll_lefttop.y;
  523. m_SelectedRect.bottom+=scroll_lefttop.y;
  524. for(int j = m_SelectedRect.top; j < m_SelectedRect.bottom; j++)
  525. {
  526. for(int i = m_SelectedRect.left; i < m_SelectedRect.right; i++)
  527. {
  528. if (i<0 || i>=g_nMapWidth || j<0 || j>=g_nMapHeight)
  529. continue;
  530. g_pFlags[INDEX(i, j)].marked=0;
  531. }
  532. }
  533. GenEdge();
  534. InvalidateRect(0,TRUE);
  535. }
  536. }
  537. else if (m_bForceAdd)
  538. {
  539. if(m_SelectedRect.left >= m_SelectedRect.right || m_SelectedRect.top >= m_SelectedRect.bottom)
  540. MessageBox("选择的范围不正确.请重新选择!");
  541. else
  542. {
  543. m_bForceAdd = false;
  544. m_SelectedRect.left+=scroll_lefttop.x;
  545. m_SelectedRect.right+=scroll_lefttop.x;
  546. m_SelectedRect.top+=scroll_lefttop.y;
  547. m_SelectedRect.bottom+=scroll_lefttop.y;
  548. for(int j = m_SelectedRect.top; j < m_SelectedRect.bottom; j++)
  549. {
  550. for(int i = m_SelectedRect.left; i < m_SelectedRect.right; i++)
  551. {
  552. if (i<0 || i>=g_nMapWidth || j<0 || j>=g_nMapHeight)
  553. continue;
  554. g_pFlags[INDEX(i, j)].marked=1;
  555. }
  556. }
  557. GenEdge();
  558. InvalidateRect(0,TRUE);
  559. }
  560. }
  561. }
  563. CView::OnLButtonUp(nFlags, point);
  564. }
  566. bool CCellView::LoadImageBuffer(HDC memdc,char *szFileName)
  567. {
  568. BITMAPFILEHEADER bitmapfileheader;
  569. BITMAPINFOHEADER bitmapinfoheader;
  570. BITMAPINFO  bitmapinfo;
  572. // this function opens a bitmap file and loads the data into bitmap
  573. int file_handle;  // the file handle
  574. OFSTRUCT file_data;          // the file data information
  576. // open the file if it exists
  577. if ((file_handle = OpenFile(szFileName,&file_data,OF_READ))==-1)
  578.    return 0;
  580. // now load the bitmap file header
  581. _lread(file_handle, &bitmapfileheader,sizeof(BITMAPFILEHEADER));
  583. // test if this is a bitmap file
  584. if (bitmapfileheader.bfType!=BITMAP_ID)
  585. {
  586. // close the file
  587. _lclose(file_handle);
  588. // return error
  589. return 0;
  590. } // end if
  592. // now we know this is a bitmap, so read in all the sections
  594. // first the bitmap infoheader
  596. // now load the bitmap file header
  597. _lread(file_handle, &bitmapinfoheader, sizeof(BITMAPINFOHEADER));
  599. if (bitmapinfoheader.biWidth%4 != 0) // 宽度不是4的倍数
  600. {
  601. MessageBox("文件宽度不是4的倍数");
  602. _lclose(file_handle);
  603. return 0;
  604. }
  606. if(g_pImgBuffer) // clear it if not null
  607. delete[] g_pImgBuffer;
  608. g_pImgBuffer = new RGB[g_nMapHeight * g_nMapWidth]; // make space
  609. // 申请备份buffer空间
  610. // if (g_pImgBufferBack)
  611. // delete[] g_pImgBufferBack;
  612. // g_pImgBufferBack=new RGB[g_nMapHeight * g_nMapWidth];
  613. // if (g_pOrgImgBuffer)
  614. // delete[] g_pOrgImgBuffer;
  615. // g_pOrgImgBuffer=new RGB[g_nMapHeight * g_nMapWidth];
  617. // now load the color palette if there is one
  618. if (bitmapinfoheader.biBitCount != 24)
  619. {
  620. // set bitmapinfo
  621. bitmapinfo.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
  622. bitmapinfo.bmiHeader.biWidth = g_nMapWidth;
  623. bitmapinfo.bmiHeader.biHeight = g_nMapHeight;
  624. bitmapinfo.bmiHeader.biPlanes = 1;
  625. bitmapinfo.bmiHeader.biBitCount = 24;
  626. bitmapinfo.bmiHeader.biCompression = BI_RGB;
  627. GetDIBits(memdc, g_hBitmap, 0, g_nMapHeight, NULL,
  628. &bitmapinfo, DIB_RGB_COLORS);
  629. GetDIBits(memdc, g_hBitmap, 0, g_nMapHeight, g_pImgBuffer,
  630. &bitmapinfo, DIB_RGB_COLORS);
  631. }
  632. else
  633. {
  634. _llseek(file_handle, bitmapfileheader.bfOffBits,FILE_BEGIN);
  635. _lread(file_handle, g_pImgBuffer, 3*g_nMapHeight*g_nMapWidth);
  636. }
  637. FlipBitmapData(g_pImgBuffer);
  639. // memcpy(g_pOrgImgBuffer,g_pImgBuffer,sizeof(RGB)*g_nMapWidth*g_nMapHeight);
  641. // close the file
  642. _lclose(file_handle);
  644. // return success
  645. return 1;
  646. }
  648. void CCellView::FlipBitmapData(RGB *buffer)
  649. {
  650. RGB *tempmem;
  651. int bytes_per_line=g_nMapWidth*sizeof(RGB);
  653. tempmem=new RGB[g_nMapWidth*g_nMapHeight];
  654. memcpy(tempmem,buffer,bytes_per_line*g_nMapHeight);
  655. for (int i=0;i<g_nMapHeight;i++)
  656.     memcpy(&buffer[((g_nMapHeight-1) - i)*g_nMapWidth],
  657. &tempmem[i*g_nMapWidth], bytes_per_line);
  658. delete[] tempmem;
  659. }
  661. void CCellView::OnProcHsi() 
  662. {
  663. if(g_hBitmap)
  664. {
  665. m_bProcHsi = true;
  666. MessageBox("请选择一块小区域作为颜色选取范围");
  667. }
  668. else
  669. MessageBox("请先打开图像文件!");
  670. }
  672. BOOL CCellView::OnSetCursor(CWnd* pWnd, UINT nHitTest, UINT message) 
  673. {
  674. if(m_bProcHsi || m_bForceKill || m_bForceAdd)
  675. ::SetCursor(AfxGetApp()->LoadStandardCursor(IDC_CROSS));
  676. else
  677. ::SetCursor(AfxGetApp()->LoadStandardCursor(IDC_ARROW));
  678. return TRUE;
  679. }
  681. void CCellView::ProcHSI(bool bEx)
  682. {
  683. m_vHSI.clear();
  685. // fill vector
  686. HSI *hsi;
  688. huegap=256.0*g_nHueGap/100.0;
  689. if (huegap>5.12)
  690. huegap=5.12;
  691. huegap+=0.0001;
  692. intgap=g_nIntGap/100.0;
  693. if (intgap>0.02)
  694. intgap=0.02;
  695. intgap+=0.0001;
  696. satgap=g_nSatGap/100.0;
  697. if (satgap>0.02)
  698. satgap=0.02;
  699. satgap+=0.0001;
  701. m_SelectedRect.left+=scroll_lefttop.x;
  702. m_SelectedRect.right+=scroll_lefttop.x;
  703. m_SelectedRect.top+=scroll_lefttop.y;
  704. m_SelectedRect.bottom+=scroll_lefttop.y;
  705. for(int j = m_SelectedRect.top; j < m_SelectedRect.bottom; j++)
  706. {
  707. for(int i = m_SelectedRect.left; i < m_SelectedRect.right; i++)
  708. {
  709. if (i<0 || i>=g_nMapWidth || j<0 || j>=g_nMapHeight)
  710. continue;
  711. if(!g_pFlags[INDEX(i, j)].marked)
  712. {
  713. hsi=&g_pHSIBuffer[INDEX(i,j)];
  714. if(FindInVectorHSI(m_vHSI, *hsi) == false)
  715. {
  716. m_vHSI.push_back(*hsi);
  717. m_vAllSelected.push_back(*hsi);
  718. }
  719. }
  720. }
  721. }
  723. if(m_vHSI.size())
  724. {
  725. // backup
  726. // memcpy(g_pImgBufferBack,g_pImgBuffer,
  727. // sizeof(RGB)*g_nMapWidth*g_nMapHeight);
  728. memcpy(g_pFlagsBack,g_pFlags,
  729. sizeof(FLAGS)*g_nMapWidth*g_nMapHeight);
  731. // g_bImgBufferChanged=true;
  732. HsiProcess(bEx);
  733. // m_vHSI.clear();
  734. GenEdge();
  735. InvalidateRect(0, TRUE);
  736. }
  737. }
  739. void CCellView::GenHSIData()
  740. {
  741. int wd, ht;
  742. HSI tmp;
  743. RGB    *cur;
  745. if(g_pHSIBuffer)
  746. delete[] g_pHSIBuffer;
  747. g_pHSIBuffer = new HSI[g_nMapWidth * g_nMapHeight];
  748. memset(g_pHSIBuffer, 0, sizeof(HSI) * g_nMapWidth * g_nMapHeight);
  750. HSI *pHueBuffer = g_pHSIBuffer;
  752. cur = g_pImgBuffer;
  753. for(ht = 0; ht < g_nMapHeight; ht++)
  754. {
  755. for(wd = 0; wd < g_nMapWidth; wd++)
  756. {
  757. RgbToHsi(cur, &tmp);
  758. pHueBuffer->Hue = tmp.Hue*255.0/360.0;
  759. pHueBuffer->Intensity = tmp.Intensity;
  760. pHueBuffer->Saturation = tmp.Saturation;
  762. pHueBuffer++;
  763. cur++;
  764. }
  765. }
  766. }
  768. bool CCellView::FindInVectorHSI(vector<HSI> v, HSI c)
  769. {
  770. int size = v.size();
  771. for(int i = 0; i < size; i++)
  772. // if(c.Hue == v.at(i).Hue && c.Intensity == v.at(i).Intensity && c.Saturation == v.at(i).Saturation)
  773. if( fabs(c.Hue - v.at(i).Hue) < huegap &&
  774. fabs(c.Intensity - v.at(i).Intensity) < intgap && 
  775. fabs(c.Saturation - v.at(i).Saturation) < satgap )
  776. return true;
  777. return false;
  778. }
  780. void CCellView::HsiProcess(bool bEx)
  781. {
  782. RGB *cur=g_pImgBuffer;
  783. HSI *pHSI=g_pHSIBuffer;
  785. int size = m_vHSI.size();
  786. double sH, sS, sI, dH, dS, dI;
  787. HSI tmp, vtmp;
  788. int ht, wd;
  789. FLAGS *cur_flag = g_pFlags; // flag
  790. for(ht = 0; ht < g_nMapHeight; ht++)
  791. {
  792. for(wd = 0; wd < g_nMapWidth; wd++)
  793. {
  794. if(cur_flag->marked)
  795. {
  796. cur++;
  797. cur_flag++; // flag
  798. pHSI++;
  799. continue;
  800. }
  801. memcpy(&tmp,pHSI,sizeof(HSI));
  802. pHSI++;
  803. for(int i = 0; i < size; i++)
  804. {
  805. vtmp = m_vHSI.at(i);
  806. sH = vtmp.Hue;
  807. sS = vtmp.Saturation;
  808. sI = vtmp.Intensity;
  809. dH = tmp.Hue;
  810. dS = tmp.Saturation;
  811. dI = tmp.Intensity;
  812. if(!bEx) // 排除法
  813. {
  814. if
  815. (
  816. sH - dH >= 0 - g_nHueGap * 2.55
  817. && sH - dH <= g_nHueGap * 2.55
  818. && sS - dS >= 0 - g_nSatGap / 100.0
  819. && sS - dS <= g_nSatGap / 100.0
  820. && sI - dI >= 0 - g_nIntGap / 100.0
  821. && sI - dI <= g_nIntGap / 100.0
  822. )
  823. {
  824. cur_flag->marked = 1; // set marked
  825. // cur->r=cur->g=cur->b=0;
  826. continue;
  827. }
  828. }
  829. else
  830. {
  831. if
  832. (
  833. (sH - dH <= 0 - g_nHueGap * 2.55 || sH - dH >= g_nHueGap * 2.55)
  834. && (sS - dS <= 0 - g_nSatGap / 100.0 || sS - dS >= g_nSatGap / 100.0)
  835. && (sI - dI <= 0 - g_nIntGap / 100.0 || sI - dI >= g_nIntGap / 100.0)
  836. )
  837. {
  838. cur_flag->marked = 1; // set marked
  839. cur->r=cur->g=cur->b=0;
  840. continue;
  841. }
  842. }
  843. }
  845. cur++;
  846. cur_flag++; // flag
  847. }
  848. }
  849. }
  851. void CCellView::OnEditUndo() 
  852. {
  853. // TODO: Add your command handler code here
  854. // RGB *exch;
  855. FLAGS *exchf;
  856. if (g_pFlagsBack)
  857. {
  858. // exch=g_pImgBufferBack;
  859. // g_pImgBufferBack=g_pImgBuffer;
  860. // g_pImgBuffer=exch;
  861. // g_bImgBufferChanged=true;
  863. // 恢复
  864. if(g_pFlags && g_pFlagsBack)
  865. {
  866. exchf=g_pFlags;
  867. g_pFlags=g_pFlagsBack;
  868. g_pFlagsBack=exchf;
  869. }
  871. InvalidateRect(NULL,TRUE);
  872. }
  873. }
  875. void CCellView::OnProcSobel() 
  876. {
  877. if(!g_hBitmap)
  878. MessageBox("请先打开文件");
  879. else // here we go~
  880. {
  881. double SobelTemplateGx[9];
  882. double SobelTemplateGy[9];
  883. // -1 -2 -1
  884. //  0  0  0
  885. //  1  2  1
  886. SobelTemplateGx[0] = -1;
  887. SobelTemplateGx[1] = -2;
  888. SobelTemplateGx[2] = -1;
  889. SobelTemplateGx[3] = 0;
  890. SobelTemplateGx[4] = 0;
  891. SobelTemplateGx[5] = 0;
  892. SobelTemplateGx[6] = 1;
  893. SobelTemplateGx[7] = 2;
  894. SobelTemplateGx[8] = 1;
  896. // -1 0 1
  897. // -2 0 2
  898. // -1 0 1
  899. SobelTemplateGy[0] = -1;
  900. SobelTemplateGy[1] = 0;
  901. SobelTemplateGy[2] = 1;
  902. SobelTemplateGy[3] = -2;
  903. SobelTemplateGy[4] = 0;
  904. SobelTemplateGy[5] = 2;
  905. SobelTemplateGy[6] = -1;
  906. SobelTemplateGy[7] = 0;
  907. SobelTemplateGy[8] = 1;
  908. if(g_pSobelResult)
  909. delete[] g_pSobelResult;
  910. g_pSobelResult = new BYTE[g_nMapWidth * g_nMapHeight];
  911. if(!g_pSobelResult)
  912. {
  913. MessageBox("not enough memory");
  914. return;
  915. }
  917. RGB *got;
  918. double tempr, tempg, tempb;
  919. double multi;
  920. double result;
  921. int x, y;
  923. BYTE *cur_pos=g_pSobelResult;
  925. for(y = 0; y < g_nMapHeight; y++)
  926. {
  927. for(x = 0; x < g_nMapWidth; x++)
  928. {
  929. // 计算Gx方向
  930. tempr = tempg = tempb = 0;
  932. got = GetBit(x - 1, y - 1);
  933. if(got)
  934. {
  935. multi = SobelTemplateGx[0];
  936. tempr += multi * got->r;
  937. tempg += multi * got->g;
  938. tempb += multi * got->b;
  939. }
  941. got = GetBit(x, y - 1);
  942. if(got)
  943. {
  944. multi = SobelTemplateGx[1];
  945. tempr += multi * got->r;
  946. tempg += multi * got->g;
  947. tempb += multi * got->b;
  948. }
  950. got = GetBit(x + 1, y - 1);
  951. if(got)
  952. {
  953. multi = SobelTemplateGx[2];
  954. tempr += multi * got->r;
  955. tempg += multi * got->g;
  956. tempb += multi * got->b;
  957. }
  959. got = GetBit(x - 1, y);
  960. if(got)
  961. {
  962. multi = SobelTemplateGx[3];
  963. tempr += multi * got->r;
  964. tempg += multi * got->g;
  965. tempb += multi * got->b;
  966. }
  968. got = GetBit(x, y);
  969. if(got)
  970. {
  971. multi = SobelTemplateGx[4];
  972. tempr += multi * got->r;
  973. tempg += multi * got->g;
  974. tempb += multi * got->b;
  975. }
  977. got = GetBit(x + 1, y);
  978. if(got)
  979. {
  980. multi = SobelTemplateGx[5];
  981. tempr += multi * got->r;
  982. tempg += multi * got->g;
  983. tempb += multi * got->b;
  984. }
  986. got = GetBit(x - 1, y + 1);
  987. if(got)
  988. {
  989. multi = SobelTemplateGx[6];
  990. tempr += multi * got->r;
  991. tempg += multi * got->g;
  992. tempb += multi * got->b;
  993. }
  995. got = GetBit(x, y + 1);
  996. if(got)
  997. {
  998. multi = SobelTemplateGx[7];
  999. tempr += multi * got->r;
  1000. tempg += multi * got->g;
  1001. tempb += multi * got->b;
  1002. }
  1004. got = GetBit(x + 1, y + 1);
  1005. if(got)
  1006. {
  1007. multi = SobelTemplateGx[8];
  1008. tempr += multi * got->r;
  1009. tempg += multi * got->g;
  1010. tempb += multi * got->b;
  1011. }
  1013. result = fabs(tempr) + fabs(tempg) + fabs(tempb);
  1015. // 计算Gy方向
  1016. tempr = tempg = tempb = 0;
  1018. got = GetBit(x - 1, y - 1);
  1019. if(got)
  1020. {
  1021. multi = SobelTemplateGy[0];
  1022. tempr += multi * got->r;
  1023. tempg += multi * got->g;
  1024. tempb += multi * got->b;
  1025. }
  1027. got = GetBit(x, y - 1);
  1028. if(got)
  1029. {
  1030. multi = SobelTemplateGy[1];
  1031. tempr += multi * got->r;
  1032. tempg += multi * got->g;
  1033. tempb += multi * got->b;
  1034. }
  1036. got = GetBit(x + 1, y - 1);
  1037. if(got)
  1038. {
  1039. multi = SobelTemplateGy[2];
  1040. tempr += multi * got->r;
  1041. tempg += multi * got->g;
  1042. tempb += multi * got->b;
  1043. }
  1045. got = GetBit(x - 1, y);
  1046. if(got)
  1047. {
  1048. multi = SobelTemplateGy[3];
  1049. tempr += multi * got->r;
  1050. tempg += multi * got->g;
  1051. tempb += multi * got->b;
  1052. }
  1054. got = GetBit(x, y);
  1055. if(got)
  1056. {
  1057. multi = SobelTemplateGy[4];
  1058. tempr += multi * got->r;
  1059. tempg += multi * got->g;
  1060. tempb += multi * got->b;
  1061. }
  1063. got = GetBit(x + 1, y);
  1064. if(got)
  1065. {
  1066. multi = SobelTemplateGy[5];
  1067. tempr += multi * got->r;
  1068. tempg += multi * got->g;
  1069. tempb += multi * got->b;
  1070. }
  1072. got = GetBit(x - 1, y + 1);
  1073. if(got)
  1074. {
  1075. multi = SobelTemplateGy[6];
  1076. tempr += multi * got->r;
  1077. tempg += multi * got->g;
  1078. tempb += multi * got->b;
  1079. }
  1081. got = GetBit(x, y + 1);
  1082. if(got)
  1083. {
  1084. multi = SobelTemplateGy[7];
  1085. tempr += multi * got->r;
  1086. tempg += multi * got->g;
  1087. tempb += multi * got->b;
  1088. }
  1090. got = GetBit(x + 1, y + 1);
  1091. if(got)
  1092. {
  1093. multi = SobelTemplateGy[8];
  1094. tempr += multi * got->r;
  1095. tempg += multi * got->g;
  1096. tempb += multi * got->b;
  1097. }
  1099. result += fabs(tempr) + fabs(tempg) + fabs(tempb);
  1101. if(result > 255 * 8)
  1102. *cur_pos++ = 255;
  1103. else
  1104. *cur_pos++ = result / 8;
  1105. }
  1106. }
  1107. // m_bDispSobel = true;
  1108. // AfxGetApp()->m_pMainWnd->GetMenu()->CheckMenuItem(ID_DISP_SOBEL, MF_CHECKED);
  1109. // InvalidateRect(0, TRUE);
  1110. }
  1111. }
  1113. RGB* CCellView::GetBit(int x, int y)
  1114. {
  1115. if(x < 0 || x >= g_nMapWidth || y < 0 || y >= g_nMapHeight)
  1116. return 0;
  1117. return (g_pImgBuffer + y * g_nMapWidth + x);
  1118. }
  1120. void CCellView::OnProcHistogram() 
  1121. {
  1122. if(g_hBitmap)
  1123. {
  1124. CHistogramDlg hDlg;
  1125. if(hDlg.DoModal() == IDOK)
  1126. {
  1127. // backup
  1128. // memcpy(g_pImgBufferBack,g_pImgBuffer,
  1129. // sizeof(RGB)*g_nMapWidth*g_nMapHeight);
  1130. memcpy(g_pFlagsBack,g_pFlags,
  1131. sizeof(FLAGS)*g_nMapWidth*g_nMapHeight);
  1133. // g_bImgBufferChanged=true;
  1135. FLAGS *cur_flag = g_pFlags;
  1136. RGB *cur=g_pImgBuffer;
  1137. for(int j = 0; j < g_nMapHeight; j++)
  1138. {
  1139. for(int i = 0; i < g_nMapWidth; i++)
  1140. {
  1141. if(!(cur_flag->marked))
  1142. {
  1143. if ( (cur->b >= hDlg.m_Thresh_B && cur->b <= hDlg.m_Thresh_BMax)
  1144. && (cur->r >= hDlg.m_Thresh_R && cur->r <= hDlg.m_Thresh_RMax)
  1145. && (cur->g >= hDlg.m_Thresh_G && cur->g <= hDlg.m_Thresh_GMax))
  1146. {
  1147. cur_flag->marked = 1;
  1148. // cur->r=cur->g=cur->b=0; // 不改变了原图了!
  1149. }
  1150. }
  1151. cur_flag++;
  1152. cur++;
  1153. }
  1154. }
  1155. GenEdge();
  1156. InvalidateRect(0, TRUE);
  1157. }
  1158. }
  1159. else MessageBox("请先打开图像文件!");
  1160. }
  1162. void CCellView::GenEdge() // 八方向生成边界
  1163. {
  1164. FLAGS *cur = g_pFlags;
  1165. int width = g_nMapWidth;
  1166. int height = g_nMapHeight;
  1167. for(int j = 0; j < g_nMapHeight; j++)
  1168. {
  1169. for(int i = 0; i < g_nMapWidth; i++)
  1170. {
  1171. cur->edged = 0;
  1172. if(cur->marked)
  1173. {
  1174. if(j == 0 && i == 0) // left top
  1175. {
  1176. if(!((cur + 1)->marked && (cur + width)->marked && (cur + width + 1)->marked)) cur->edged = 1;
  1177. }
  1178. else if(j == 0 && i == width - 1) // right top
  1179. {
  1180. if(!((cur - 1)->marked && (cur + width)->marked && (cur + width - 1)->marked)) cur->edged = 1;
  1181. }
  1182. else if(j == height - 1 && i == 0) // bottom left
  1183. {
  1184. if(!((cur + 1)->marked && (cur - width)->marked && (cur - width + 1)->marked)) cur->edged = 1;
  1185. }
  1186. else if(j == height - 1 && i == width - 1) // bottom right
  1187. {
  1188. if(!((cur - 1)->marked && (cur - width)->marked && (cur - width - 1)->marked)) cur->edged = 1;
  1189. }
  1190. else if(j == 0) // top
  1191. {
  1192. if
  1193. (
  1194. !((cur - 1)->marked
  1195. && (cur + 1)->marked
  1196. && (cur + width - 1)->marked
  1197. && (cur + width + 1)->marked
  1198. && (cur + width)->marked
  1199. )
  1200. ) cur->edged = 1;
  1201. }
  1202. else if(i == 0) // left
  1203. {
  1204. if
  1205. (
  1206. !((cur - width)->marked
  1207. && (cur - width + 1)->marked
  1208. && (cur + 1)->marked
  1209. && (cur + width)->marked
  1210. && (cur + width + 1)->marked
  1211. )
  1212. ) cur->edged = 1;
  1213. }
  1214. else if(j == height - 1) // bottom
  1215. {
  1216. if
  1217. (
  1218. !((cur - 1)->marked
  1219. && (cur + 1)->marked
  1220. && (cur - width - 1)->marked
  1221. && (cur - width + 1)->marked
  1222. && (cur - width)->marked
  1223. )
  1224. ) cur->edged = 1;
  1225. }
  1226. else if(i == width - 1) // right
  1227. {
  1228. if
  1229. (
  1230. !((cur - width)->marked
  1231. && (cur - width - 1)->marked
  1232. && (cur - 1)->marked
  1233. && (cur + width)->marked
  1234. && (cur + width - 1)->marked
  1235. )
  1236. ) cur->edged = 1;
  1237. }
  1238. else // normal
  1239. {
  1240. if
  1241. (
  1242. !((cur - width - 1)->marked
  1243. && (cur - width)->marked
  1244. && (cur - width + 1)->marked
  1245. && (cur - 1)->marked
  1246. && (cur + 1)->marked
  1247. && (cur + width - 1)->marked
  1248. && (cur + width)->marked
  1249. && (cur + width + 1)->marked
  1250. )
  1251. ) cur->edged = 1;
  1252. }
  1253. }
  1254. cur++;
  1255. }
  1256. }
  1257. }
  1259. void CCellView::GenEdge4()// 四方向生成边界
  1260. {
  1261. FLAGS *cur = g_pFlags;
  1262. int width = g_nMapWidth;
  1263. int height = g_nMapHeight;
  1264. for(int j = 0; j < g_nMapHeight; j++)
  1265. {
  1266. for(int i = 0; i < g_nMapWidth; i++)
  1267. {
  1268. cur->edged = 0;
  1269. if(cur->marked)
  1270. {
  1271. if(j == 0 && i == 0) // left top
  1272. {
  1273. if(!((cur + 1)->marked && (cur + width)->marked && (cur + width + 1)->marked)) cur->edged = 1;
  1274. }
  1275. else if(j == 0 && i == width - 1) // right top
  1276. {
  1277. if(!((cur - 1)->marked && (cur + width)->marked && (cur + width - 1)->marked)) cur->edged = 1;
  1278. }
  1279. else if(j == height - 1 && i == 0) // bottom left
  1280. {
  1281. if(!((cur + 1)->marked && (cur - width)->marked && (cur - width + 1)->marked)) cur->edged = 1;
  1282. }
  1283. else if(j == height - 1 && i == width - 1) // bottom right
  1284. {
  1285. if(!((cur - 1)->marked && (cur - width)->marked && (cur - width - 1)->marked)) cur->edged = 1;
  1286. }
  1287. else if(j == 0) // top
  1288. {
  1289. if
  1290. (
  1291. !((cur - 1)->marked
  1292. && (cur + 1)->marked
  1293. && (cur + width - 1)->marked
  1294. && (cur + width + 1)->marked
  1295. && (cur + width)->marked
  1296. )
  1297. ) cur->edged = 1;
  1298. }
  1299. else if(i == 0) // left
  1300. {
  1301. if
  1302. (
  1303. !((cur - width)->marked
  1304. && (cur - width + 1)->marked
  1305. && (cur + 1)->marked
  1306. && (cur + width)->marked
  1307. && (cur + width + 1)->marked
  1308. )
  1309. ) cur->edged = 1;
  1310. }
  1311. else if(j == height - 1) // bottom
  1312. {
  1313. if
  1314. (
  1315. !((cur - 1)->marked
  1316. && (cur + 1)->marked
  1317. && (cur - width - 1)->marked
  1318. && (cur - width + 1)->marked
  1319. && (cur - width)->marked
  1320. )
  1321. ) cur->edged = 1;
  1322. }
  1323. else if(i == width - 1) // right
  1324. {
  1325. if
  1326. (
  1327. !((cur - width)->marked
  1328. && (cur - width - 1)->marked
  1329. && (cur - 1)->marked
  1330. && (cur + width)->marked
  1331. && (cur + width - 1)->marked
  1332. )
  1333. ) cur->edged = 1;
  1334. }
  1335. else // normal
  1336. {
  1337. if
  1338. (
  1339. !((cur - width)->marked
  1340. && (cur - 1)->marked
  1341. && (cur + 1)->marked
  1342. && (cur + width)->marked
  1343. )
  1344. ) cur->edged = 1;
  1345. }
  1346. }
  1347. cur++;
  1348. }
  1349. }
  1350. }
  1352. void CCellView::OnProcSpecColor() 
  1353. {
  1354. CColorDialog cDlg;
  1356. if(cDlg.DoModal() == IDOK)
  1357. {
  1358. COLORREF cr = cDlg.GetColor();
  1359. spec_rgb.b = cr >> 16 & 0xff;
  1360. spec_rgb.g = cr >> 8 & 0xff;
  1361. spec_rgb.r = cr & 0xff;
  1362. InvalidateRect(0, TRUE);
  1363. }
  1364. }
  1366. void CCellView::OnDispEdge() 
  1367. {
  1368. if (g_hBitmap)
  1369. {
  1370. if(!m_bIsDispEdge)
  1371. {
  1372. m_bIsDispEdge = true;
  1373. AfxGetApp()->m_pMainWnd->GetMenu()->CheckMenuItem(ID_DISP_EDGE, MF_CHECKED);
  1374. AfxGetApp()->m_pMainWnd->GetMenu()->CheckMenuItem(ID_DISP_REGION, MF_UNCHECKED);
  1375. InvalidateRect(0, TRUE);
  1376. }
  1377. }
  1378. else
  1379. MessageBox("请先打开文件");
  1380. }
  1382. void CCellView::OnDispRegion() 
  1383. {
  1384. if (g_hBitmap)
  1385. {
  1386. if(m_bIsDispEdge)
  1387. {
  1388. m_bIsDispEdge = false;
  1389. AfxGetApp()->m_pMainWnd->GetMenu()->CheckMenuItem(ID_DISP_REGION, MF_CHECKED);
  1390. AfxGetApp()->m_pMainWnd->GetMenu()->CheckMenuItem(ID_DISP_EDGE, MF_UNCHECKED);
  1391. InvalidateRect(0, TRUE);
  1392. }
  1393. }
  1394. else
  1395. MessageBox("请先打开文件");
  1396. }
  1398. void CCellView::OnDispSobel() 
  1399. {
  1400. if (!g_hBitmap)
  1401. {
  1402. MessageBox("请先打开文件");
  1403. return;
  1404. }
  1405. if (!g_pSobelResult)
  1406. {
  1407. MessageBox("请选择菜单[处理]-[显示→Sobel]来查看梯度信息");
  1408. return;
  1409. }
  1410. m_bDispSobel ^= 1;
  1411. if(m_bDispSobel)
  1412. AfxGetApp()->m_pMainWnd->GetMenu()->CheckMenuItem(ID_DISP_SOBEL, MF_CHECKED);
  1413. else
  1414. AfxGetApp()->m_pMainWnd->GetMenu()->CheckMenuItem(ID_DISP_SOBEL, MF_UNCHECKED);
  1415. // g_bImgBufferChanged=true;
  1416. InvalidateRect(0, TRUE);
  1417. }
  1419. void CCellView::OnProcReload() 
  1420. {
  1421. // TODO: Add your command handler code here
  1422. if (g_hBitmap)
  1423. {
  1424. // g_pFlags置0
  1425. memset(g_pFlags, 0, g_nMapHeight * g_nMapWidth * sizeof(FLAGS));
  1426. // memcpy(g_pImgBuffer,g_pOrgImgBuffer,sizeof(RGB)*g_nMapWidth*g_nMapHeight);
  1427. // g_bImgBufferChanged=true;
  1429. m_bProcHsi=false;
  1430. m_bDispSobel=false;
  1431. AfxGetApp()->m_pMainWnd->GetMenu()->CheckMenuItem(ID_DISP_SOBEL, MF_UNCHECKED);
  1432. m_bIsDispEdge=false;
  1433. AfxGetApp()->m_pMainWnd->GetMenu()->CheckMenuItem(ID_DISP_REGION, MF_CHECKED);
  1434. AfxGetApp()->m_pMainWnd->GetMenu()->CheckMenuItem(ID_DISP_EDGE, MF_UNCHECKED);
  1435. m_vCenterPoints.clear(); // 清除!
  1436. InvalidateRect(0,TRUE);
  1437. }
  1438. else
  1439. MessageBox("请先打开文件");
  1440. }
  1442. void CCellView::OnProcFindCenter() 
  1443. {
  1444. if(g_hBitmap)
  1445. {
  1446. // backup
  1447. memcpy(g_pFlagsBack,g_pFlags,
  1448. sizeof(FLAGS)*g_nMapWidth*g_nMapHeight);
  1450. int pre_shrink_count;
  1451. CSet setdlg;
  1452. if (IDOK==setdlg.DoModal()) // 对话框.输入腐蚀次数
  1453. {
  1454. pre_shrink_count=setdlg.m_nPreCount;
  1455. }
  1456. else return;
  1458. int i,j,x,y;
  1459. FLAGS *cur_flag;
  1460. // 先去掉pre_shrink_count层皮
  1461. GenEdge();
  1462. for (i=0;i<pre_shrink_count;i++)
  1463. {
  1464. cur_flag=g_pFlags;
  1465. for(y = 0; y < g_nMapHeight; y++)
  1466. for(x = 0; x < g_nMapWidth; x++)
  1467. {
  1468. // 去掉边界!
  1469. if (cur_flag->edged)
  1470. cur_flag->marked=0;
  1471. cur_flag++;
  1472. }
  1473. if (i==0)
  1474. GenEdge4();
  1475. else
  1476. GenEdge();
  1477. }
  1478. InvalidateRect(NULL,TRUE);
  1479. MessageBox("去掉了噪声");
  1481. cur_flag=g_pFlags; // 清除visited标志
  1482. for(y = 0; y < g_nMapHeight; y++)
  1483. for(x = 0; x < g_nMapWidth; x++)
  1484. {
  1485. cur_flag->visited=0;
  1486. cur_flag->center=0;
  1487. cur_flag++;
  1488. }
  1490. GenEdge();
  1491. CENTER_POINT pt;
  1492. points_temp.clear();
  1493. bool changed;
  1495. for (i=0;i<40;i++) // 标志中心点的腐蚀
  1496. {
  1497. changed=false;
  1498. // 清除visited标志
  1499. cur_flag=g_pFlags;
  1500. for(y = 0; y < g_nMapHeight; y++)
  1501. for(x = 0; x < g_nMapWidth; x++)
  1502. {
  1503. cur_flag->visited=0;
  1504. cur_flag++;
  1505. }
  1506. cur_flag=g_pFlags;
  1507. for(y = 0; y < g_nMapHeight; y++)
  1508. for(x = 0; x < g_nMapWidth; x++)
  1509. {
  1510. if (y>0 && y<g_nMapHeight-1 && x>0 && x<g_nMapWidth-1) // 最边上的不用处理
  1511. {
  1512. m_bFullEdge=true;
  1513. if (cur_flag->edged && !cur_flag->visited) // 没有访问过的边界
  1514. {
  1515. if ( !(cur_flag-1)->marked &&
  1516.  !(cur_flag+1)->marked &&
  1517.  !(cur_flag+g_nMapWidth)->marked &&
  1518.  !(cur_flag-g_nMapWidth)->marked )
  1519. //  !(cur_flag+g_nMapWidth-1)->marked &&
  1520. //  !(cur_flag+g_nMapWidth+1)->marked &&
  1521. //  !(cur_flag-g_nMapWidth-1)->marked &&
  1522. //  !(cur_flag-g_nMapWidth+1)->marked )
  1523. {
  1524. if (i==0) // 基本上这种是噪音
  1525. {
  1526. cur_flag++;
  1527. continue;
  1528. }
  1529. // 孤立的点
  1530. cur_flag->center=1;
  1531. // 保存一下CENTER_POINT信息
  1532. pt.x=x;
  1533. pt.y=y;
  1534. pt.radius=i+4+pre_shrink_count*2;
  1535. points_temp.push_back(pt);
  1537. cur_flag++;
  1538. continue;
  1539. }
  1540. else
  1541. MarkIt(x,y); // 判断是否需要保存
  1543. if (m_bFullEdge) // 需要保存!
  1544. SaveIt(x,y,i+6);
  1545. }
  1546. }
  1547. cur_flag++;
  1548. }
  1550. cur_flag=g_pFlags;
  1551. for(y = 0; y < g_nMapHeight; y++)
  1552. for(x = 0; x < g_nMapWidth; x++)
  1553. {
  1554. // 去掉边界!
  1555. if (cur_flag->edged)
  1556. {
  1557. changed=true;
  1558. cur_flag->marked=0;
  1559. }
  1560. cur_flag++;
  1561. }
  1562. if (i%2==0)
  1563. GenEdge4();
  1564. else
  1565. GenEdge();
  1567. if (!changed)
  1568. break;
  1569. }
  1571. // 清除visited标志
  1572. cur_flag=g_pFlags;
  1573. for(y = 0; y < g_nMapHeight; y++)
  1574. for(x = 0; x < g_nMapWidth; x++)
  1575. {
  1576. cur_flag->visited=0;
  1577. cur_flag++;
  1578. }
  1579. // 取平均值,获得中心点
  1580. vector<CENTER_POINT> points;
  1581. cur_flag=g_pFlags;
  1582. for(y = 0; y < g_nMapHeight; y++)
  1583. for(x = 0; x < g_nMapWidth; x++)
  1584. {
  1585. if (y>0 && y<g_nMapHeight-1 && x>0 && x<g_nMapWidth-1) // 最边上的不用处理
  1586. {
  1587. if (cur_flag->center)
  1588. {
  1589. if ( !(cur_flag-1)->center && !(cur_flag+1)->center &&
  1590.  !(cur_flag+g_nMapWidth)->center &&
  1591.  !(cur_flag-g_nMapWidth)->center &&
  1592.  !(cur_flag+g_nMapWidth-1)->center &&
  1593.  !(cur_flag+g_nMapWidth+1)->center &&
  1594.  !(cur_flag-g_nMapWidth-1)->center &&
  1595.  !(cur_flag-g_nMapWidth+1)->center )
  1596. {
  1597. // 孤立的点
  1598. pt.x=x;
  1599. pt.y=y;
  1600. points.push_back(pt);
  1601. cur_flag++;
  1602. continue;
  1603. }
  1604. else
  1605. {
  1606. tot_area=0;
  1607. max_radius=0;
  1608. tot_x=0;
  1609. tot_y=0;
  1610. CalcCenterArea(x,y);
  1611. pt.x=tot_x/tot_area;
  1612. pt.y=tot_y/tot_area;
  1613. pt.radius=max_radius;
  1614. g_pFlags[pt.y*g_nMapWidth+pt.x].center=1;
  1615. points.push_back(pt);
  1616. }
  1617. }
  1618. }
  1619. cur_flag++;
  1620. }
  1622. m_vCenterPoints.clear();
  1623. int x0,y0;
  1624. bool adj;
  1625. // 清除center标志
  1626. cur_flag=g_pFlags;
  1627. for(y = 0; y < g_nMapHeight; y++)
  1628. for(x = 0; x < g_nMapWidth; x++)
  1629. {
  1630. cur_flag->center=0;
  1631. cur_flag++;
  1632. }
  1633. // 平均化相近的中心点
  1634. for (i=0;i<points.size();i++)
  1635. {
  1636. x0=points.at(i).x;
  1637. y0=points.at(i).y;
  1638. adj=false;
  1639. for (j=i+1;j<points.size();j++)
  1640. {
  1641. x=points.at(j).x;
  1642. y=points.at(j).y;
  1643. if (abs(x0-x)+abs(y0-y)<10) // 相近
  1644. {
  1645. points.at(j).x=(x+x0)/2;
  1646. points.at(j).y=(y+y0)/2;
  1647. points.at(j).radius=points.at(i).radius;
  1648. points.erase(&points.at(j));
  1649. i--;
  1650. adj=true;
  1651. break;
  1652. }
  1653. }
  1654. if (!adj) // 非相近
  1655. {
  1656. if (points.at(i).radius>5)
  1657. {
  1658. m_vCenterPoints.push_back(points.at(i));
  1659. g_pFlags[points.at(i).y*g_nMapWidth+points.at(i).x].center=1;
  1660. }
  1661. }
  1662. }
  1664. double max_intensity=0.0;
  1665. double hue,inte;
  1666. double min_hue=255.0;
  1667. double max_hue=0.0;
  1668. double overmax;
  1669. double overmin;
  1670. if (m_vAllSelected.size()>0) // 保存着的选项
  1671. {
  1672. for (int i=0;i<m_vAllSelected.size();i++)
  1673. {
  1674. if (m_vAllSelected.at(i).Intensity>max_intensity)
  1675. max_intensity=m_vAllSelected.at(i).Intensity;
  1676. if (m_vAllSelected.at(i).Hue>max_hue)
  1677. max_hue=m_vAllSelected.at(i).Hue;
  1678. if (m_vAllSelected.at(i).Hue<min_hue)
  1679. min_hue=m_vAllSelected.at(i).Hue;
  1680. }
  1682. InvalidateRect(0,TRUE);
  1683. MessageBox("准备开始修正");
  1684. }
  1685. else
  1686. {
  1687. InvalidateRect(0,TRUE);
  1688. return;
  1689. }
  1690. max_hue+=max_hue*0.02;
  1691. min_hue-=min_hue*0.02;
  1692. overmax=max_hue*1.08;
  1693. overmin=min_hue*0.92;
  1695. int r0,r;
  1696. int tx,ty;
  1697. int area,toobad;
  1698. // 去掉被包含的圆
  1699. for (i=0;i<m_vCenterPoints.size();i++)
  1700. {
  1701. x0=m_vCenterPoints.at(i).x;
  1702. y0=m_vCenterPoints.at(i).y;
  1703. r0=m_vCenterPoints.at(i).radius;
  1704. for (j=i+1;j<m_vCenterPoints.size();j++)
  1705. {
  1706. x=m_vCenterPoints.at(j).x;
  1707. y=m_vCenterPoints.at(j).y;
  1708. r=m_vCenterPoints.at(j).radius;
  1709. if (DISTANCE(x0,y0,x,y)<abs(r0-r)+4) // 包含
  1710. {
  1711. CDC *pdc=GetDC();
  1712. CENTER_POINT centerp;
  1713. CPen pen;
  1714. pen.CreatePen(PS_DOT, 1, RGB(255,0,0));
  1715. pdc->SelectObject(pen);
  1716. if (r0>r) // 去掉r0
  1717. centerp=m_vCenterPoints.at(i);
  1718. else
  1719. centerp=m_vCenterPoints.at(j);
  1720. Arc(pdc->m_hDC,
  1721. centerp.x-scroll_lefttop.x-centerp.radius,
  1722. centerp.y-scroll_lefttop.y-centerp.radius,
  1723. centerp.x-scroll_lefttop.x+centerp.radius,
  1724. centerp.y-scroll_lefttop.y+centerp.radius,
  1725. centerp.x-scroll_lefttop.x+centerp.radius,
  1726. centerp.y-scroll_lefttop.y,
  1727. centerp.x-scroll_lefttop.x+centerp.radius,
  1728. centerp.y-scroll_lefttop.y
  1729. );
  1730. DeleteObject(pen);
  1731. if (r0>r) // 去掉r0
  1732. {
  1733. m_vCenterPoints.erase(&m_vCenterPoints.at(i));
  1734. i--;
  1735. }
  1736. else
  1737. m_vCenterPoints.erase(&m_vCenterPoints.at(j));
  1738. }
  1739. }
  1740. }
  1741. vector<CENTER_POINT> tocheck;
  1742. int n,size,total;
  1743. bool isok;
  1744. // 去掉潜在的错误(同两个圆相交,并且不相交的部分是噪声)
  1745. for (i=0;i<m_vCenterPoints.size();i++)
  1746. {
  1747. tocheck.clear();
  1748. x0=m_vCenterPoints.at(i).x;
  1749. y0=m_vCenterPoints.at(i).y;
  1750. r0=m_vCenterPoints.at(i).radius;
  1751. for (j=0;j<m_vCenterPoints.size();j++)
  1752. {
  1753. if (i==j)
  1754. continue;
  1755. x=m_vCenterPoints.at(j).x;
  1756. y=m_vCenterPoints.at(j).y;
  1757. r=m_vCenterPoints.at(j).radius;
  1758. if (DISTANCE(x0,y0,x,y)<abs(r0+r)) // 相交
  1759. {
  1760. pt.x=x; pt.y=y; pt.radius=r;
  1761. tocheck.push_back(pt);
  1762. }
  1763. }
  1764. size=tocheck.size();
  1765. if (size>1) // 同两个以上的圆相交
  1766. {
  1767. area=0;
  1768. total=0;
  1769. toobad=0;
  1770. for (tx=x0-r0;tx<x0+r0;tx++)
  1771. for (ty=y0-r0;ty<y0+r0;ty++)
  1772. {
  1773. if (DISTANCE(x0,y0,tx,ty)<r0) // 所有圆内部的点
  1774. {
  1775. if (tx<0 || tx>g_nMapWidth-1 || ty<0 || ty>g_nMapHeight-1)
  1776. continue;
  1777. isok=true;
  1778. for (n=0;n<size;n++)
  1779. {
  1780. pt=tocheck.at(n);// 取得
  1781. if (DISTANCE(tx,ty,pt.x,pt.y)<pt.radius)
  1782. {
  1783. isok=false;
  1784. break;
  1785. }
  1786. }
  1787. if (isok) // 同所有的圆都不相交的部分
  1788. {
  1789. total++;
  1790. hue=g_pHSIBuffer[ty*g_nMapWidth+tx].Hue;
  1791. inte=g_pHSIBuffer[ty*g_nMapWidth+tx].Intensity;
  1792. if (inte>max_intensity || hue > max_hue || hue < min_hue)
  1793. area++;
  1794. if (hue > overmax || hue < overmin)
  1795. toobad++;
  1796. }
  1797. }
  1798. }
  1799. if (total<r0*r0 || (total<r0*r0*1.5 && area>total*0.5) || toobad>total/6) // need adjust
  1800. {
  1801. CDC *pdc=GetDC();
  1802. CENTER_POINT centerp;
  1803. CPen pen;
  1804. if (total<r0*r0) // 红色 面积过小
  1805. pen.CreatePen(PS_SOLID, 2, RGB(255,0,0));
  1806. else if (toobad>total/6) // 绿色 错误点过多
  1807. pen.CreatePen(PS_SOLID, 2, RGB(0,255,0));
  1808. else // 浅蓝 误差过大
  1809. pen.CreatePen(PS_SOLID, 2, RGB(0,255,255));
  1810. pdc->SelectObject(pen);
  1811. centerp=m_vCenterPoints.at(i);
  1812. Arc(pdc->m_hDC,
  1813. centerp.x-scroll_lefttop.x-centerp.radius,
  1814. centerp.y-scroll_lefttop.y-centerp.radius,
  1815. centerp.x-scroll_lefttop.x+centerp.radius,
  1816. centerp.y-scroll_lefttop.y+centerp.radius,
  1817. centerp.x-scroll_lefttop.x+centerp.radius,
  1818. centerp.y-scroll_lefttop.y,
  1819. centerp.x-scroll_lefttop.x+centerp.radius,
  1820. centerp.y-scroll_lefttop.y
  1821. );
  1822. DeleteObject(pen);
  1823. m_vCenterPoints.erase(&m_vCenterPoints.at(i));
  1824. i--;
  1825. }
  1826. }
  1827. }
  1828. for (i=0;i<m_vCenterPoints.size();i++)
  1829. {
  1830. r0=m_vCenterPoints.at(i).radius;
  1831. if (r0<10)
  1832. {
  1833. x0=m_vCenterPoints.at(i).x;
  1834. y0=m_vCenterPoints.at(i).y;
  1835. area=0;
  1836. toobad=0;
  1837. for (tx=x0-r0;tx<x0+r0;tx++)
  1838. for (ty=y0-r0;ty<y0+r0;ty++)
  1839. if (sqrt((x0-tx)*(x0-tx)+(y0-ty)*(y0-ty))<r0)
  1840. {
  1841. if (tx<0 || tx>g_nMapWidth-1 || ty<0 || ty>g_nMapHeight-1)
  1842. continue;
  1843. hue=g_pHSIBuffer[ty*g_nMapWidth+tx].Hue;
  1844. if (hue > max_hue || hue < min_hue)
  1845. area++;
  1846. if (hue > overmax || hue < overmin)
  1847. toobad++;
  1848. }
  1849. if (area>r0*r0 || toobad>r0*r0/2) // need adjust
  1850. {
  1851. CDC *pdc=GetDC();
  1852. CENTER_POINT centerp;
  1853. CPen pen;
  1854. if (toobad>r0*r0/2)
  1855. pen.CreatePen(PS_DOT, 1, RGB(0,128,0));
  1856. else
  1857. pen.CreatePen(PS_DOT, 1, RGB(0,0,255));
  1858. pdc->SelectObject(pen);
  1859. centerp=m_vCenterPoints.at(i);
  1860. Arc(pdc->m_hDC,
  1861. centerp.x-scroll_lefttop.x-centerp.radius,
  1862. centerp.y-scroll_lefttop.y-centerp.radius,
  1863. centerp.x-scroll_lefttop.x+centerp.radius,
  1864. centerp.y-scroll_lefttop.y+centerp.radius,
  1865. centerp.x-scroll_lefttop.x+centerp.radius,
  1866. centerp.y-scroll_lefttop.y,
  1867. centerp.x-scroll_lefttop.x+centerp.radius,
  1868. centerp.y-scroll_lefttop.y
  1869. );
  1870. DeleteObject(pen);
  1871. m_vCenterPoints.erase(&m_vCenterPoints.at(i));
  1872. i--;
  1873. }
  1874. }
  1875. }
  1876. /* // 去掉潜在的错误(相交,并且包含许多色调范围外的象素)
  1877. for (i=0;i<m_vCenterPoints.size();i++)
  1878. {
  1879. x0=m_vCenterPoints.at(i).x;
  1880. y0=m_vCenterPoints.at(i).y;
  1881. r0=m_vCenterPoints.at(i).radius;
  1882. for (j=i+1;j<m_vCenterPoints.size();j++)
  1883. {
  1884. x=m_vCenterPoints.at(j).x;
  1885. y=m_vCenterPoints.at(j).y;
  1886. r=m_vCenterPoints.at(j).radius;
  1887. if (sqrt((x0-x)*(x0-x)+(y0-y)*(y0-y))<abs(r0+r)) // 相交
  1888. {
  1889. area=0;
  1890. toobad=0;
  1891. for (tx=x0-r0;tx<x0+r0;tx++)
  1892. for (ty=y0-r0;ty<y0+r0;ty++)
  1893. if (sqrt((x0-tx)*(x0-tx)+(y0-ty)*(y0-ty))<r0)
  1894. {
  1895. if (tx<0 || tx>g_nMapWidth-1 || ty<0 || ty>g_nMapHeight-1)
  1896. continue;
  1897. hue=g_pHSIBuffer[ty*g_nMapWidth+tx].Hue;
  1898. if (hue > max_hue || hue < min_hue)
  1899. area++;
  1900. if (hue > overmax || hue < overmin)
  1901. toobad++;
  1902. }
  1903. if (area>r0*r0 || toobad>r0*r0/2) // need adjust
  1904. {
  1905. InvalidateRect(0,TRUE);
  1906. {
  1907. CDC *pdc=GetDC();
  1908. CENTER_POINT centerp;
  1909. CPen pen;
  1910. if (toobad>r0*r0/6)
  1911. pen.CreatePen(PS_SOLID, 3, RGB(255,0,0));
  1912. else
  1913. pen.CreatePen(PS_SOLID, 3, RGB(0,0,0));
  1914. pdc->SelectObject(pen);
  1915. centerp=m_vCenterPoints.at(i);
  1916. Arc(pdc->m_hDC,
  1917. centerp.x-scroll_lefttop.x-centerp.radius,
  1918. centerp.y-scroll_lefttop.y-centerp.radius,
  1919. centerp.x-scroll_lefttop.x+centerp.radius,
  1920. centerp.y-scroll_lefttop.y+centerp.radius,
  1921. centerp.x-scroll_lefttop.x+centerp.radius,
  1922. centerp.y-scroll_lefttop.y,
  1923. centerp.x-scroll_lefttop.x+centerp.radius,
  1924. centerp.y-scroll_lefttop.y
  1925. );
  1926. DeleteObject(pen);
  1927. }
  1928. m_vCenterPoints.erase(&m_vCenterPoints.at(i));
  1929. i--;
  1930. Sleep(2000);
  1931. break;
  1932. }
  1933. }
  1934. }
  1935. }
  1936. */// Sleep(10000);
  1937. // InvalidateRect(0,TRUE);
  1938. }
  1939. else
  1940. MessageBox("请先打开图像文件!");
  1942. }
  1944. void CCellView::MarkIt(int i, int j)
  1945. {
  1946. // if (!m_bFullEdge)
  1947. // return; // 如果已经知道不是FullEdge了,那么不用做了!
  1948. if (i<1 || i>g_nMapWidth-2 || j<1 || j>g_nMapHeight-2)
  1949. {
  1950. return;
  1951. }
  1952. g_pFlags[j*g_nMapWidth+i].visited=1;
  1953. if ( !g_pFlags[j*g_nMapWidth+i-1].visited && // 没有访问过
  1954.   g_pFlags[j*g_nMapWidth+i-1].marked ) // 标志了
  1955. {
  1956. if (g_pFlags[j*g_nMapWidth+i-1].edged ) // 并且是边缘
  1957. MarkIt(i-1,j); // 左边
  1958. else
  1959. m_bFullEdge=false;
  1960. }
  1962. if ( !g_pFlags[j*g_nMapWidth+i+1].visited && // 没有访问过
  1963.   g_pFlags[j*g_nMapWidth+i+1].marked ) // 标志了
  1964. {
  1965. if (g_pFlags[j*g_nMapWidth+i+1].edged ) // 并且是边缘
  1966. MarkIt(i+1,j); // 右边
  1967. else
  1968. m_bFullEdge=false;
  1969. }
  1971. if ( !g_pFlags[(j-1)*g_nMapWidth+i].visited && // 没有访问过
  1972.   g_pFlags[(j-1)*g_nMapWidth+i].marked ) // 标志了
  1973. {
  1974. if (g_pFlags[(j-1)*g_nMapWidth+i].edged ) // 并且是边缘
  1975. MarkIt(i,j-1); // 上面
  1976. else
  1977. m_bFullEdge=false;
  1978. }
  1980. if ( !g_pFlags[(j+1)*g_nMapWidth+i].visited && // 没有访问过
  1981.   g_pFlags[(j+1)*g_nMapWidth+i].marked ) // 标志了
  1982. {
  1983. if (g_pFlags[(j+1)*g_nMapWidth+i].edged ) // 并且是边缘
  1984. MarkIt(i,j+1); // 下面
  1985. else
  1986. m_bFullEdge=false;
  1987. }
  1989. if ( !g_pFlags[(j-1)*g_nMapWidth+i-1].visited && // 没有访问过
  1990.   g_pFlags[(j-1)*g_nMapWidth+i-1].marked ) // 标志了
  1991. {
  1992. if (g_pFlags[(j-1)*g_nMapWidth+i-1].edged ) // 并且是边缘
  1993. MarkIt(i-1,j-1); // 左上
  1994. else
  1995. m_bFullEdge=false;
  1996. }
  1998. if ( !g_pFlags[(j+1)*g_nMapWidth+i-1].visited && // 没有访问过
  1999.   g_pFlags[(j+1)*g_nMapWidth+i-1].marked ) // 标志了
  2000. {
  2001. if (g_pFlags[(j+1)*g_nMapWidth+i-1].edged ) // 并且是边缘
  2002. MarkIt(i-1,j+1); // 左下
  2003. else
  2004. m_bFullEdge=false;
  2005. }
  2007. if ( !g_pFlags[(j-1)*g_nMapWidth+i+1].visited && // 没有访问过
  2008.   g_pFlags[(j-1)*g_nMapWidth+i+1].marked ) // 标志了
  2009. {
  2010. if (g_pFlags[(j-1)*g_nMapWidth+i+1].edged ) // 并且是边缘
  2011. MarkIt(i+1,j-1); // 右上
  2012. else
  2013. m_bFullEdge=false;
  2014. }
  2016. if ( !g_pFlags[(j+1)*g_nMapWidth+i+1].visited && // 没有访问过
  2017.   g_pFlags[(j+1)*g_nMapWidth+i+1].marked ) // 标志了
  2018. {
  2019. if (g_pFlags[(j+1)*g_nMapWidth+i+1].edged ) // 并且是边缘
  2020. MarkIt(i+1,j+1); // 右下
  2021. else
  2022. m_bFullEdge=false;
  2023. }
  2024. }
  2026. void CCellView::SaveIt(int i, int j, int radius)
  2027. {
  2028. if (i<1 || i>g_nMapWidth-2 || j<1 || j>g_nMapHeight-2)
  2029. {
  2030. return;
  2031. }
  2032. CENTER_POINT pt;
  2033. pt.x=i;
  2034. pt.y=j;
  2035. pt.radius=radius;
  2036. points_temp.push_back(pt);
  2038. // marke current point
  2039. g_pFlags[j*g_nMapWidth+i].center=1;
  2040. g_pFlags[j*g_nMapWidth+i].visited=0;
  2042. if ( g_pFlags[j*g_nMapWidth+i-1].visited )
  2043. {
  2044. SaveIt(i-1,j,radius);
  2045. }
  2046. if ( g_pFlags[j*g_nMapWidth+i+1].visited )
  2047. {
  2048. SaveIt(i+1,j,radius);
  2049. }
  2050. if ( g_pFlags[(j-1)*g_nMapWidth+i].visited )
  2051. {
  2052. SaveIt(i,j-1,radius);
  2053. }
  2054. if ( g_pFlags[(j+1)*g_nMapWidth+i].visited )
  2055. {
  2056. SaveIt(i,j+1,radius);
  2057. }
  2059. if ( g_pFlags[(j+1)*g_nMapWidth+i+1].visited )
  2060. {
  2061. SaveIt(i+1,j+1,radius);
  2062. }
  2063. if ( g_pFlags[(j+1)*g_nMapWidth+i-1].visited )
  2064. {
  2065. SaveIt(i-1,j+1,radius);
  2066. }
  2067. if ( g_pFlags[(j-1)*g_nMapWidth+i+1].visited )
  2068. {
  2069. SaveIt(i+1,j-1,radius);
  2070. }
  2071. if ( g_pFlags[(j-1)*g_nMapWidth+i-1].visited )
  2072. {
  2073. SaveIt(i-1,j-1,radius);
  2074. }
  2075. }
  2077. // 利用sobel信息进行修正
  2078. void CCellView::OnProcSobelCorrect() 
  2079. {
  2080. if (!g_hBitmap)
  2081. {
  2082. MessageBox("请先打开图片");
  2083. return;
  2084. }
  2085. // backup
  2086. memcpy(g_pFlagsBack,g_pFlags,
  2087. sizeof(FLAGS)*g_nMapWidth*g_nMapHeight);
  2089. double max_intensity=0.0;
  2090. double min_hue=255.0;
  2091. double max_hue=0.0;
  2092. if (m_vAllSelected.size()) // 还保存着啊!
  2093. for (int i=0;i<m_vAllSelected.size();i++)
  2094. {
  2095. if (m_vAllSelected.at(i).Intensity>max_intensity)
  2096. max_intensity=m_vAllSelected.at(i).Intensity;
  2097. if (m_vAllSelected.at(i).Hue>max_hue)
  2098. max_hue=m_vAllSelected.at(i).Hue;
  2099. if (m_vAllSelected.at(i).Hue<min_hue)
  2100. min_hue=m_vAllSelected.at(i).Hue;
  2101. }
  2102. max_hue-=max_hue*0.1;
  2103. min_hue+=min_hue*0.1;
  2105. int x,y;
  2106. FLAGS *cur_flag=g_pFlags;
  2107. BYTE  *cur_sobel=g_pSobelResult;
  2108. HSI   *cur_hsi=g_pHSIBuffer;
  2109. for(y = 0; y < g_nMapHeight; y++)
  2110. for(x = 0; x < g_nMapWidth; x++)
  2111. {
  2112. if ( cur_flag->marked )
  2113. {
  2114. if (*cur_sobel > 40) // 参数需要调整
  2115. {
  2116. if (max_intensity>0.01)
  2117. if (/*cur_hsi->Intensity>max_intensity &&*/ // 亮度大于最大值并且色调超出范围
  2118. (cur_hsi->Hue>max_hue || cur_hsi->Hue<min_hue) )
  2119. cur_flag->marked=0;
  2120. }
  2121. }
  2123. cur_hsi++;
  2124. cur_flag++;
  2125. cur_sobel++;
  2126. }
  2127. GenEdge();
  2128. InvalidateRect(0,TRUE);
  2129. }
  2131. void CCellView::CalcCenterArea(int i, int j)
  2132. {
  2133. if (i<1 || i>g_nMapWidth-2 || j<1 || j>g_nMapHeight-2)
  2134. {
  2135. return;
  2136. }
  2137. tot_area++;
  2138. tot_x+=i;
  2139. tot_y+=j;
  2140. g_pFlags[j*g_nMapWidth+i].center=0;
  2142. for ( int n=0; n<points_temp.size(); n++ )
  2143. {
  2144. if (points_temp.at(n).x==i && points_temp.at(n).y==j)
  2145. {
  2146. if (points_temp.at(n).radius>max_radius)
  2147. max_radius=points_temp.at(n).radius;
  2148. break;
  2149. }
  2150. }
  2152. if ( g_pFlags[j*g_nMapWidth+i-1].center )
  2153. {
  2154. CalcCenterArea(i-1,j);
  2155. }
  2156. if ( g_pFlags[j*g_nMapWidth+i+1].center )
  2157. {
  2158. CalcCenterArea(i+1,j);
  2159. }
  2160. if ( g_pFlags[(j-1)*g_nMapWidth+i].center )
  2161. {
  2162. CalcCenterArea(i,j-1);
  2163. }
  2164. if ( g_pFlags[(j+1)*g_nMapWidth+i].center )
  2165. {
  2166. CalcCenterArea(i,j+1);
  2167. }
  2169. if ( g_pFlags[(j+1)*g_nMapWidth+i+1].center )
  2170. {
  2171. CalcCenterArea(i+1,j+1);
  2172. }
  2173. if ( g_pFlags[(j+1)*g_nMapWidth+i-1].center )
  2174. {
  2175. CalcCenterArea(i-1,j+1);
  2176. }
  2177. if ( g_pFlags[(j-1)*g_nMapWidth+i+1].center )
  2178. {
  2179. CalcCenterArea(i+1,j-1);
  2180. }
  2181. if ( g_pFlags[(j-1)*g_nMapWidth+i-1].center )
  2182. {
  2183. CalcCenterArea(i-1,j-1);
  2184. }
  2185. }
  2187. void CCellView::OnProcForceKill() 
  2188. {
  2189. // TODO: Add your command handler code here
  2190. if(g_hBitmap)
  2191. {
  2192. m_bForceKill = true;
  2193. }
  2194. else
  2195. MessageBox("请先打开图像文件!");
  2196. }
  2198. void CCellView::OnProcForceSele() 
  2199. {
  2200. // TODO: Add your command handler code here
  2201. if(g_hBitmap)
  2202. {
  2203. m_bForceAdd = true;
  2204. }
  2205. else
  2206. MessageBox("请先打开图像文件!");
  2207. }
  2209. void CCellView::OnProcDilation() 
  2210. {
  2211. if (g_bDir4Dil)
  2212. g_bDir4Dil=false;
  2213. else g_bDir4Dil=true;
  2215. // backup
  2216. memcpy(g_pFlagsBack,g_pFlags,
  2217. sizeof(FLAGS)*g_nMapWidth*g_nMapHeight);
  2219. // 膨胀
  2220. RGB *cur=g_pImgBuffer;
  2221. FLAGS *cur_flag=g_pFlags;
  2223. vector<long> addr;
  2224. for (int ht=0;ht<g_nMapHeight;ht++)
  2225. for (int wd=0;wd<g_nMapWidth;wd++)
  2226. {
  2227. if (!cur_flag->marked)
  2228. {
  2229. if (ht==0 || wd==0 || ht==g_nMapHeight-1 || wd==g_nMapWidth-1)
  2230. {
  2231. cur++;
  2232. cur_flag++;
  2233. continue;
  2234. }
  2235. else if (g_bDir4Dil)
  2236. {
  2237. if (  (cur_flag-1)->marked || // left
  2238.   (cur_flag+1)->marked || // right
  2239.   (cur_flag-g_nMapWidth-1)->marked || // left up
  2240.   (cur_flag-g_nMapWidth+1)->marked || // right up
  2241.   (cur_flag+g_nMapWidth-1)->marked || // left down
  2242.   (cur_flag+g_nMapWidth+1)->marked || // right down
  2243.   (cur_flag-g_nMapWidth)->marked || // up
  2244.   (cur_flag+g_nMapWidth)->marked ) // down
  2245. addr.push_back((long)cur_flag);
  2246. }
  2247. else
  2248. {
  2249. if (  (cur_flag-1)->marked || // left
  2250.   (cur_flag+1)->marked || // right
  2251.   (cur_flag-g_nMapWidth)->marked || // up
  2252.   (cur_flag+g_nMapWidth)->marked ) // down
  2253. addr.push_back((long)cur_flag);
  2254. }
  2255. }
  2256. cur++;
  2257. cur_flag++;
  2258. }
  2260. int size=addr.size();
  2261. for (int i=0;i<size;i++)
  2262. ((FLAGS *)addr.at(i))->marked=1;
  2263. addr.clear();
  2265. GenEdge();
  2267. InvalidateRect(0,TRUE);
  2268. }
  2270. void CCellView::OnProcErosion() 
  2271. {
  2272. if (g_bDir4Ero)
  2273. g_bDir4Ero=false;
  2274. else g_bDir4Ero=true;
  2276. // backup
  2277. memcpy(g_pFlagsBack,g_pFlags,
  2278. sizeof(FLAGS)*g_nMapWidth*g_nMapHeight);
  2280. // 腐蚀
  2281. RGB *cur=g_pImgBuffer;
  2282. FLAGS *cur_flag=g_pFlags;
  2284. vector<long> addr;
  2285. for (int ht=0;ht<g_nMapHeight;ht++)
  2286. for (int wd=0;wd<g_nMapWidth;wd++)
  2287. {
  2288. if (cur_flag->marked)
  2289. {
  2290. if (ht==0 || wd==0 || ht==g_nMapHeight-1 || wd==g_nMapWidth-1)
  2291. {
  2292. cur_flag->marked=0;
  2293. cur++;
  2294. cur_flag++;
  2295. continue;
  2296. }
  2297. else if (g_bDir4Ero)
  2298. {
  2299. if (  !(cur_flag-1)->marked || // left
  2300.   !(cur_flag+1)->marked || // right
  2301.   !(cur_flag-g_nMapWidth-1)->marked || // left up
  2302.   !(cur_flag-g_nMapWidth+1)->marked || // right up
  2303.   !(cur_flag+g_nMapWidth-1)->marked || // left down
  2304.   !(cur_flag+g_nMapWidth+1)->marked || // right down
  2305.   !(cur_flag-g_nMapWidth)->marked || // up
  2306.   !(cur_flag+g_nMapWidth)->marked ) // down
  2307. addr.push_back((long)cur_flag);
  2308. }
  2309. else
  2310. {
  2311. if (  !(cur_flag-1)->marked || // left
  2312.   !(cur_flag+1)->marked || // right
  2313.   !(cur_flag-g_nMapWidth)->marked || // up
  2314.   !(cur_flag+g_nMapWidth)->marked ) // down
  2315. addr.push_back((long)cur_flag);
  2316. }
  2317. }
  2318. cur++;
  2319. cur_flag++;
  2320. }
  2322. int size=addr.size();
  2323. for (int i=0;i<size;i++)
  2324. ((FLAGS *)addr.at(i))->marked=0;
  2325. addr.clear();
  2327. GenEdge();
  2329. InvalidateRect(0,TRUE);
  2330. }
  2332. void CCellView::OnProcSmooth() 
  2333. {
  2334. int wd,ht;
  2335. if(g_hBitmap)
  2336. {
  2337. RGB *g_pTemp=new RGB[g_nMapWidth*g_nMapHeight];
  2338. RGB *ptemp=g_pTemp;
  2339. memset(g_pTemp,0,g_nMapWidth*g_nMapHeight*sizeof(RGB));
  2340. RGB *cur=g_pImgBuffer;
  2341. for (ht=0;ht<g_nMapHeight;ht++)
  2342. for (wd=0;wd<g_nMapWidth;wd++)
  2343. {
  2344. if (ht==0 || wd==0 || ht==g_nMapHeight-1 || wd==g_nMapWidth-1)
  2345. { }
  2346. else
  2347. {
  2348. ptemp->r=( (cur-g_nMapWidth-1)->r +
  2349.  (cur-g_nMapWidth  )->r +
  2350.  (cur-g_nMapWidth+1)->r +
  2351.  (cur-1)->r +
  2352.  cur->r +
  2353.  (cur+1)->r +
  2354.  (cur+g_nMapWidth-1)->r +
  2355.  (cur+g_nMapWidth  )->r +
  2356.  (cur+g_nMapWidth+1)->r )/9;
  2357. ptemp->g=( (cur-g_nMapWidth-1)->g +
  2358.  (cur-g_nMapWidth  )->g +
  2359.  (cur-g_nMapWidth+1)->g +
  2360.  (cur-1)->g +
  2361.  cur->g +
  2362.  (cur+1)->g +
  2363.  (cur+g_nMapWidth-1)->g +
  2364.  (cur+g_nMapWidth  )->g +
  2365.  (cur+g_nMapWidth+1)->g )/9;
  2366. ptemp->b=( (cur-g_nMapWidth-1)->b +
  2367.  (cur-g_nMapWidth  )->b +
  2368.  (cur-g_nMapWidth+1)->b +
  2369.  (cur-1)->b +
  2370.  cur->b +
  2371.  (cur+1)->b +
  2372.  (cur+g_nMapWidth-1)->b +
  2373.  (cur+g_nMapWidth  )->b +
  2374.  (cur+g_nMapWidth+1)->b )/9;
  2375. }
  2376. cur++;
  2377. ptemp++;
  2378. }
  2379. memcpy(g_pImgBuffer,g_pTemp,g_nMapWidth*g_nMapHeight*sizeof(RGB));
  2380. delete[] g_pTemp;
  2381. InvalidateRect(0,TRUE);
  2382. }
  2383. else
  2384. MessageBox("请先打开图像文件!");
  2385. }
  2387. void CCellView::CountSeeds()
  2388. {
  2389. int wd,ht;
  2390. FLAGS *cur_flag;
  2392. g_nCellCount=0;
  2393. g_nCellTotArea=0;
  2395. cur_flag=g_pFlags;
  2396. for (ht=0;ht<g_nMapHeight;ht++)
  2397. for (wd=0;wd<g_nMapWidth;wd++)
  2398. {
  2399. cur_flag->visited=0; // clear visited
  2400. cur_flag++;
  2401. }
  2403. cur_flag=g_pFlags;
  2404. for (ht=0;ht<g_nMapHeight;ht++)
  2405. for (wd=0;wd<g_nMapWidth;wd++)
  2406. {
  2407. if (cur_flag->marked && !cur_flag->visited)
  2408. {
  2409. g_nCellCount++;
  2410. ProcessCountSeeds(wd,ht,cur_flag);
  2411. }
  2412. cur_flag++;
  2413. }
  2415. InvalidateRect(0,TRUE);
  2417. char msg[256];
  2418. double avgarea=(double)g_nCellTotArea/(double)g_nCellCount;
  2419. sprintf(msg,"一共有%d个细胞,平均面积%d象素(大约%d*%d)",
  2420. g_nCellCount,(int)avgarea,(int)sqrt(avgarea),(int)sqrt(avgarea));
  2421. MessageBox(msg);
  2422. }
  2424. void CCellView::ProcessCountSeeds(int wd, int ht, FLAGS *curf)
  2425. {
  2426. if (wd<0 || wd>g_nMapWidth-1 || ht<0 || ht>g_nMapHeight-1)
  2427. return;
  2428. FLAGS *next;
  2429. g_nCellTotArea++;
  2430. curf->visited=1;
  2431. if (ht>0)
  2432. {
  2433. next=curf-g_nMapWidth;
  2434. if (next->marked && !next->visited)
  2435. ProcessCountSeeds(wd,ht-1,next); // up
  2436. }
  2437. if (ht<g_nMapHeight-1)
  2438. {
  2439. next=curf+g_nMapWidth;
  2440. if (next->marked && !next->visited)
  2441. ProcessCountSeeds(wd,ht+1,next); // down
  2442. }
  2443. if (wd>0)
  2444. {
  2445. next=curf-1;
  2446. if (next->marked && !next->visited)
  2447. ProcessCountSeeds(wd-1,ht,next); // left
  2448. }
  2449. if (wd<g_nMapWidth-1)
  2450. {
  2451. next=curf+1;
  2452. if (next->marked && !next->visited)
  2453. ProcessCountSeeds(wd+1,ht,next); // right
  2454. }
  2455. }
  2457. void CCellView::OnProcStat() 
  2458. {
  2459. int result;
  2460. result=MessageBox("是否使用中心点标注信息?",NULL,MB_YESNO);
  2461. if (result==IDYES)
  2462. {
  2463. CounterSeedsCenter();
  2464. }
  2465. else
  2466. {
  2467. CountSeeds();
  2468. }
  2469. }
  2471. void CCellView::CounterSeedsCenter()
  2472. {
  2473. if (m_vCenterPoints.size==0)
  2474. MessageBox("中心点数据似乎没有生成...");
  2475. else
  2476. {
  2477. long tota,totr;
  2478. tota=0; totr=0;
  2479. for (int i=0;i<m_vCenterPoints.size();i++)
  2480. {
  2481. tota+=m_vCenterPoints.at(i).radius*m_vCenterPoints.at(i).radius*3.14;
  2482. totr+=m_vCenterPoints.at(i).radius;
  2483. }
  2484. char msg[256];
  2485. sprintf(msg,"共有%d个细胞,平均半径%d,平均面积%d",
  2486. m_vCenterPoints.size(),
  2487. totr/m_vCenterPoints.size(),
  2488. tota/m_vCenterPoints.size());
  2489. MessageBox(msg);
  2490. }
  2491. }
  2493. void CCellView::FillHoles()
  2494. {
  2495. int wd,ht;
  2496. FLAGS *cur_flag=g_pFlags;
  2498. // backup
  2499. memcpy(g_pFlagsBack,g_pFlags,
  2500. sizeof(FLAGS)*g_nMapWidth*g_nMapHeight);
  2502. qSz=g_nMapWidth*g_nMapHeight*2;
  2503. qh=new int[qSz+1];
  2504. if(!qh) return;
  2506. for (ht=0;ht<g_nMapHeight;ht++)
  2507. for (wd=0;wd<g_nMapWidth;wd++)
  2508. {
  2509. cur_flag->visited=0; // clear visited
  2510. cur_flag++;
  2511. }
  2513. cur_flag=g_pFlags;
  2514. for (ht=0;ht<g_nMapHeight;ht++)
  2515. for (wd=0;wd<g_nMapWidth;wd++)
  2516. {
  2517. if (!cur_flag->marked && !cur_flag->visited)
  2518. {
  2519. g_nCellTotArea=0;
  2520. ProcessFillHoles(wd,ht);
  2521. if (g_nCellTotArea<MAX_HOLE && g_nCellTotArea>0)
  2522. FillTheHole(wd,ht);
  2523. // char msg[128];
  2524. // sprintf(msg,"%d",g_nCellTotArea);
  2525. // InvalidateRect(0,TRUE);
  2526. // MessageBox(msg);
  2527. }
  2528. cur_flag++;
  2529. }
  2531. delete[] qh;
  2532. GenEdge();
  2533. InvalidateRect(0,TRUE);
  2534. }
  2536. void CCellView::ProcessFillHoles(int wd, int ht)
  2537. {
  2538. if (wd<0 || wd>g_nMapWidth-1 || ht<0 || ht>g_nMapHeight-1)
  2539. return;
  2540. qst=qh;
  2541. memset(qst,0,qSz); //Clear the contents
  2542. qs=qr=qst;
  2543. *qs=xt=wd;
  2544. qs++;
  2545. *qs=yt=ht;
  2546. qs++;
  2547. g_pFlags[INDEX(xt,yt)].visited=1;
  2548. g_nCellTotArea++;
  2550. //Main queue loop
  2551. while(qr!=qs)
  2552. {
  2553. //Add new members to queue
  2554. //Above current pixel
  2555. if (yt>0)
  2556. if(!g_pFlags[(yt-1)*g_nMapWidth+xt].visited &&
  2557. !g_pFlags[(yt-1)*g_nMapWidth+xt].marked)
  2558. {
  2559. g_nCellTotArea++;
  2560. *qs=xt;
  2561. qs++;
  2562. *qs=yt-1;
  2563. qs++;
  2564. g_pFlags[(yt-1)*g_nMapWidth+xt].visited=1;
  2565. }
  2566. //Below current pixel
  2567. if (yt<g_nMapHeight-1)
  2568. if(!g_pFlags[(yt+1)*g_nMapWidth+xt].visited &&
  2569. !g_pFlags[(yt+1)*g_nMapWidth+xt].marked)
  2570. {
  2571. g_nCellTotArea++;
  2572. *qs=xt;
  2573. qs++;
  2574. *qs=yt+1;
  2575. qs++;
  2576. g_pFlags[(yt+1)*g_nMapWidth+xt].visited=1;
  2577. }
  2578. //Left of current pixel
  2579. if (xt>0)
  2580. if(!g_pFlags[yt*g_nMapWidth+xt-1].visited &&
  2581. !g_pFlags[yt*g_nMapWidth+xt-1].marked)
  2582. {
  2583. g_nCellTotArea++;
  2584. *qs=xt-1;
  2585. qs++;
  2586. *qs=yt;
  2587. qs++;
  2588. g_pFlags[yt*g_nMapWidth+xt-1].visited=1;
  2589. }
  2590. //Right of current pixel
  2591. if (yt<g_nMapWidth-1)
  2592. if(!g_pFlags[yt*g_nMapWidth+xt+1].visited &&
  2593. !g_pFlags[yt*g_nMapWidth+xt+1].marked)
  2594. {
  2595. g_nCellTotArea++;
  2596. *qs=xt+1;
  2597. qs++;
  2598. *qs=yt;
  2599. qs++;
  2600. g_pFlags[yt*g_nMapWidth+xt+1].visited=1;
  2601. }
  2603. //Retrieve current queue member
  2604. qr+=2;
  2605. xt=*qr;
  2606. yt=*(qr+1);
  2607. } //Back to beginning of loop
  2608. }
  2610. void CCellView::FillTheHole(int wd, int ht)
  2611. {
  2612. if (wd<0 || wd>g_nMapWidth-1 || ht<0 || ht>g_nMapHeight-1)
  2613. return;
  2614. g_nCellTotArea--;
  2615. if (g_nCellTotArea<0)
  2616. return;
  2617. qst=qh;
  2618. memset(qst,0,qSz); //Clear the contents
  2619. qs=qr=qst;
  2620. *qs=xt=wd;
  2621. qs++;
  2622. *qs=yt=ht;
  2623. qs++;
  2624. g_pFlags[INDEX(xt,yt)].marked=1;
  2626. //Main queue loop
  2627. while(qr!=qs)
  2628. {
  2629. //Add new members to queue
  2630. //Above current pixel
  2631. if (yt>0)
  2632. if(g_pFlags[(yt-1)*g_nMapWidth+xt].visited &&
  2633. !g_pFlags[(yt-1)*g_nMapWidth+xt].marked)
  2634. {
  2635. g_nCellTotArea--;
  2636. if (g_nCellTotArea<0)
  2637. return;
  2638. *qs=xt;
  2639. qs++;
  2640. *qs=yt-1;
  2641. qs++;
  2642. g_pFlags[(yt-1)*g_nMapWidth+xt].marked=1;
  2643. }
  2644. //Below current pixel
  2645. if (yt<g_nMapHeight-1)
  2646. if(g_pFlags[(yt+1)*g_nMapWidth+xt].visited &&
  2647. !g_pFlags[(yt+1)*g_nMapWidth+xt].marked)
  2648. {
  2649. g_nCellTotArea--;
  2650. if (g_nCellTotArea<0)
  2651. return;
  2652. *qs=xt;
  2653. qs++;
  2654. *qs=yt+1;
  2655. qs++;
  2656. g_pFlags[(yt+1)*g_nMapWidth+xt].marked=1;
  2657. }
  2658. //Left of current pixel
  2659. if (xt>0)
  2660. if(g_pFlags[yt*g_nMapWidth+xt-1].visited &&
  2661. !g_pFlags[yt*g_nMapWidth+xt-1].marked)
  2662. {
  2663. g_nCellTotArea--;
  2664. if (g_nCellTotArea<0)
  2665. return;
  2666. *qs=xt-1;
  2667. qs++;
  2668. *qs=yt;
  2669. qs++;
  2670. g_pFlags[yt*g_nMapWidth+xt-1].marked=1;
  2671. }
  2672. //Right of current pixel
  2673. if (yt<g_nMapWidth-1)
  2674. if(g_pFlags[yt*g_nMapWidth+xt+1].visited &&
  2675. !g_pFlags[yt*g_nMapWidth+xt+1].marked)
  2676. {
  2677. g_nCellTotArea--;
  2678. if (g_nCellTotArea<0)
  2679. return;
  2680. *qs=xt+1;
  2681. qs++;
  2682. *qs=yt;
  2683. qs++;
  2684. g_pFlags[yt*g_nMapWidth+xt+1].marked=1;
  2685. }
  2687. //Retrieve current queue member
  2688. qr+=2;
  2689. xt=*qr;
  2690. yt=*(qr+1);
  2691. } //Back to beginning of loop
  2692. }