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

源码开发语言/平台
当前位置：首页 > 源码/资料 > 图形图像 > 图片显示 > 查看源码
DIBDoc.cpp
资源名称：QuickImage.rar　[点击查看]
上传用户：gzboli
上传日期：2013-04-10
资源大小：471k
文件大小：19k
源码类别：

图片显示

开发平台：

Visual C++

DIBDoc.cpp：源码内容
  1. // DIBDoc.cpp : implementation of the CDIBDoc class
  2. //
  4. #include "stdafx.h"
  5. #include "QuickImage.h"
  7. #include "DIBDoc.h"
  8. #include "DlgHistogram.h"
  9. #include "raw.h"
  10. #include "importsjpegfile.h"
  11. #include "mainfrm.h"
  12. #include "ConvoluteDlg.h"
  13. #include "ImgProcess.h"
  14. #include "importsgifapi.h"
  15. #include "Global.h"
  17. #ifdef _DEBUG
  18. #define new DEBUG_NEW
  19. #undef THIS_FILE
  20. static char THIS_FILE[] = __FILE__;
  21. #endif
  23. /////////////////////////////////////////////////////////////////////////////
  24. // CDIBDoc
  25. extern CString g_strCurrentDir;
  27. IMPLEMENT_DYNCREATE(CDIBDoc, CDocument)
  29. BEGIN_MESSAGE_MAP(CDIBDoc, CDocument)
  30. //{{AFX_MSG_MAP(CDIBDoc)
  31. //}}AFX_MSG_MAP
  32. ON_MESSAGE(WM_REALIZEPAL, OnRealizePal)
  33. ON_COMMAND_RANGE(ID_IMAGE_HISTOGRAM, ID_IMAGE_THIN, OnImageProcess)
  34. ON_UPDATE_COMMAND_UI_RANGE(ID_IMAGE_HISTOGRAM, ID_IMAGE_THIN, OnUpdateImageProcess)
  36. END_MESSAGE_MAP()
  38. /////////////////////////////////////////////////////////////////////////////
  39. // CDIBDoc construction/destruction
  41. CDIBDoc::CDIBDoc()
  42. {
  43. m_hDIB = NULL;
  44. m_palDIB = NULL;
  45. m_sizeDoc = CSize(1,1);     // dummy value to make CScrollView happy
  47. // m_hRightDIB = NULL;
  48. // *m_strRightFile = '';
  49. }
  51. CDIBDoc::~CDIBDoc()
  52. {
  53. if (m_hDIB != NULL)
  54. {
  55. ::GlobalFree((HGLOBAL) m_hDIB);
  56. m_hDIB = NULL;
  57. }
  58. if (m_palDIB != NULL)
  59. {
  60. delete m_palDIB;
  61. m_palDIB = NULL;
  62. }
  63. /* if (m_hRightDIB != NULL)
  64. {
  65. ::GlobalFree((HGLOBAL) m_hRightDIB);
  66. }*/
  67. }
  69. BOOL CDIBDoc::OnNewDocument()
  70. {
  71. if (!CDocument::OnNewDocument())
  72. return FALSE;
  74. // TODO: add reinitialization code here
  75. // (SDI documents will reuse this document)
  77. return TRUE;
  78. }
  80. void CDIBDoc::InitDIBData()
  81. {
  82. if (m_palDIB != NULL)
  83. {
  84. delete m_palDIB;
  85. m_palDIB = NULL;
  86. }
  87. if (m_hDIB == NULL)
  88. {
  89. return;
  90. }
  91. // Set up document size
  92. LPSTR lpDIB = (LPSTR) ::GlobalLock((HGLOBAL) m_hDIB);
  93. if (::DIBWidth(lpDIB) > INT_MAX ||abs(::DIBHeight(lpDIB)) > INT_MAX)
  94. {
  95. ::GlobalUnlock((HGLOBAL) m_hDIB);
  96. ::GlobalFree((HGLOBAL) m_hDIB);
  97. m_hDIB = NULL;
  98. CString strMsg;
  99. strMsg.LoadString(IDS_DIB_TOO_BIG);
  100. MessageBox(NULL, strMsg, NULL, MB_ICONINFORMATION | MB_OK);
  101. return;
  102. }
  103. m_sizeDoc = CSize((int) ::DIBWidth(lpDIB), (int) abs(::DIBHeight(lpDIB)));
  104. ::GlobalUnlock((HGLOBAL) m_hDIB);
  105. // Create copy of palette
  106. if(::DIBBitCount(lpDIB) < 24)
  107. {
  108. m_palDIB = new CPalette;
  109. if (m_palDIB == NULL)
  110. {
  111. // we must be really low on memory
  112. ::GlobalFree((HGLOBAL) m_hDIB);
  113. m_hDIB = NULL;
  114. return;
  115. }
  116. if (::CreateDIBPalette(m_hDIB, m_palDIB) == NULL)
  117. {
  118. // DIB may not have a palette
  119. delete m_palDIB;
  120. m_palDIB = NULL;
  121. return;
  122. }
  123. }
  124. }
  126. void CDIBDoc::ReplaceHDIB(HDIB hDIB)
  127. {
  128. if (m_hDIB != NULL)
  129. {
  130. ::GlobalFree((HGLOBAL) m_hDIB);
  131. }
  132. m_hDIB = hDIB;
  133. }
  136. /////////////////////////////////////////////////////////////////////////////
  137. // CDIBDoc serialization
  139. void CDIBDoc::Serialize(CArchive& ar)
  140. {
  141. if (ar.IsStoring())
  142. {
  143. // TODO: add storing code here
  144. }
  145. else
  146. {
  147. // TODO: add loading code here
  148. }
  149. }
  151. /////////////////////////////////////////////////////////////////////////////
  152. // CDIBDoc diagnostics
  154. #ifdef _DEBUG
  155. void CDIBDoc::AssertValid() const
  156. {
  157. CDocument::AssertValid();
  158. }
  160. void CDIBDoc::Dump(CDumpContext& dc) const
  161. {
  162. CDocument::Dump(dc);
  163. }
  164. #endif //_DEBUG
  166. /////////////////////////////////////////////////////////////////////////////
  167. // CDIBDoc commands
  168. BOOL CDIBDoc::OnOpenDocument(LPCTSTR lpszPathName) 
  169. {
  170. BeginWaitCursor();
  172. CMultiDocTemplate * pDocTemplate = ((CQuickImageApp*)AfxGetApp())->m_pImageTemplate;
  174. POSITION pos = pDocTemplate->GetFirstDocPosition();
  175. CDIBDoc* pDoc = (CDIBDoc*)pDocTemplate->GetNextDoc(pos);
  176. if(NULL != pDoc && pDoc != this)
  177. {
  178. if(pDoc->IsModified())
  179. {
  180. CString strMsg = _T("Save change to") + pDoc->GetTitle() + _T("?");
  181. if(AfxMessageBox(strMsg, MB_ICONWARNING | MB_YESNO) == IDYES)
  182. {
  183. pDoc->SaveModified();//>OnSaveDocument(pDoc->GetPathName());
  184. }
  185. }
  186. pDoc->OnCloseDocument();
  187. }
  189. char *pExt=strrchr(lpszPathName,'.');
  190. char cExt[4];
  191. strcpy(cExt,++pExt);
  192. _strlwr(cExt);
  193. DeleteContents();
  195. if (strcmp(cExt,"raw")==0)
  196. {
  197. CRaw raw;
  198. if(!raw.ReadFromFile(lpszPathName))
  199. {
  200. return FALSE;
  201. }
  202. m_hDIB = raw.Show();
  203. }
  204. else if(strcmp(cExt,"bmp")==0
  205. || strcmp(cExt,"dib")==0
  206. || strcmp(cExt,"ddb")==0)
  207. {
  208. CFile file;
  209. CFileException fe;
  210. if (!file.Open(lpszPathName, CFile::modeRead | CFile::shareDenyWrite, &fe))
  211. {
  212. fe.ReportError();
  213. return NULL;
  214. }
  215. // replace calls to Serialize with ReadDIBFile function
  216. TRY
  217. {
  218. m_hDIB = ::ReadDIBFile(file);
  219. }
  220. CATCH (CFileException, eLoad)
  221. {
  222. file.Abort(); // will not throw an exception
  223. eLoad->ReportError();
  224. m_hDIB = NULL;
  225. return NULL;
  226. }
  227. END_CATCH
  228. if (m_hDIB == NULL)
  229. {
  230. // may not be DIB format
  231. CString strMsg;
  232. strMsg.LoadString(IDS_CANNOT_LOAD_DIB);
  233. MessageBox(NULL, strMsg, NULL, MB_ICONINFORMATION | MB_OK);
  234. return NULL;
  235. }
  236. }
  237. else if(strcmp(cExt,"jpg")==0 || strcmp(cExt,"jpeg")==0)
  238. {
  239. UINT height,width;
  240. BYTE *pData = NULL;
  241. pData = JpegFile::JpegFileToRGB(lpszPathName, &width, &height);
  242. if(NULL == pData)return NULL;
  243. JpegFile::BGRFromRGB(pData, width, height);
  244. JpegFile::VertFlipBuf(pData, width * 3, height);
  246. UINT widthDW;
  247. BYTE *pDIB = JpegFile::MakeDwordAlignedBuf(pData, width, height, &widthDW);
  249. delete []pData;
  250. if(NULL == pDIB)
  251. {
  252. char strErr[100];
  253. sprintf(strErr, "Error: %d", GetLastError());
  254. MessageBox(AfxGetMainWnd()->GetSafeHwnd(), strErr, "Read File Error",
  255. MB_OK|MB_ICONERROR);
  256. }
  257. m_hDIB = ::ShowJPEG(width, height, pDIB);
  258. delete []pDIB;
  259. }
  260. else if(strcmp(cExt, "gif") == 0)
  261. {
  262. m_hDIB = ReadGIF(lpszPathName);
  263. }
  264. else if(strcmp(cExt,"pcx") == 0)
  265. {
  266. m_hDIB = ::ReadPCX256(lpszPathName);
  267. }
  268. else
  269. {
  270. // CString strMsg;
  271. // strMsg.Format("%sn不是图像文件!", lpszPathName);
  272. AfxMessageBox("不识别的文件格式!");
  273. return FALSE;
  274. }
  276. if(NULL != m_hDIB)
  277. {
  278. InitDIBData();
  279. }
  280. else
  281. {
  282. return FALSE;
  283. }
  284. SetPathName(lpszPathName);
  285. SetModifiedFlag(FALSE);
  286. EndWaitCursor();
  288. // if(g_strCurrentDir.IsEmpty())
  289. {
  290. CString strCurDir = GetPathName();
  291. char *pTitle = strrchr(strCurDir, '\');
  292. pTitle[1] = '';
  293. CGlobal::ChangeDirectory(strCurDir);
  294. }
  295. return TRUE;
  296. }
  298. BOOL CDIBDoc::OnSaveDocument(LPCTSTR lpszPathName) 
  299. {
  300. char *pExt=strrchr(lpszPathName,'.');
  301. char cExt[4];
  302. strcpy(cExt,++pExt);
  303. _strlwr(cExt);
  304. BeginWaitCursor();
  305. BOOL bSuccess = FALSE;
  306. // CTime t_Start = CTime::GetCurrentTime();
  308. if(strcmp(cExt,"bmp")==0
  309. || strcmp(cExt,"dib")==0
  310. || strcmp(cExt,"ddb")==0)
  311. {
  312. CFile file;
  313. CFileException fe;
  314. if (!file.Open(lpszPathName, CFile::modeCreate |
  315. CFile::modeReadWrite | CFile::shareExclusive, &fe))
  316. {
  317. ReportSaveLoadException(lpszPathName, &fe,
  318. TRUE, AFX_IDP_INVALID_FILENAME);
  319. return FALSE;
  320. }
  321. // replace calls to Serialize with SaveDIB function
  322. TRY
  323. {
  324. bSuccess = ::SaveDIB(m_hDIB, file);
  325. file.Close();
  326. }
  327. CATCH (CException, eSave)
  328. {
  329. file.Abort(); // will not throw an exception
  330. EndWaitCursor();
  331. ReportSaveLoadException(lpszPathName, eSave,
  332. TRUE, AFX_IDP_FAILED_TO_SAVE_DOC);
  333. }
  334. END_CATCH
  335. }
  336. else if (strcmp(cExt,"raw")==0)
  337. {
  338. bSuccess = CRaw::SaveRaw(lpszPathName, m_hDIB);
  339. }
  340. else if(strcmp(cExt,"pcx") == 0)
  341. {
  342. LPSTR lpDIB;
  343. LPSTR    lpDIBBits;
  344. lpDIB = (LPSTR) ::GlobalLock(m_hDIB);
  345. lpDIBBits = ::FindDIBBits(lpDIB);
  347. // 判断是否是8-bpp位图（这里为了方便，只处理8-bpp位图的行程编码）
  348. if (::DIBNumColors(lpDIB) != 256)
  349. {
  350. // 提示用户
  351. AfxMessageBox("目前只支持256色位图的行程编码！", MB_ICONINFORMATION | MB_OK);
  352. ::GlobalUnlock(m_hDIB);
  353. return FALSE;
  354. }
  355. bSuccess = ::DIBToPCX256(lpDIB, lpszPathName);
  356. GlobalUnlock(m_hDIB);
  357. }
  358. else
  359. {
  360. // ASSERT(FALSE);
  361. }
  363. SetPathName(lpszPathName);
  364. SetModifiedFlag(FALSE);
  365. // CTimeSpan t_Span = CTime::GetCurrentTime() - t_Start;
  366. // ((CMainFrame*)AfxGetMainWnd())->SetStatusText(
  367. // t_Span.Format("Elapsed time- %H : %M : %S"));
  368. EndWaitCursor();
  370. return bSuccess;
  371. return CDocument::OnSaveDocument(lpszPathName);
  372. }
  374. void CDIBDoc::OnRealizePal(WPARAM wParam, LPARAM lParam)
  375. {
  376. // ASSERT(wParam != NULL);
  378. if (NULL == m_hDIB)
  379. return;  // must be a new document
  381. if (m_palDIB != NULL)
  382. {
  383. CWnd* pAppFrame = AfxGetApp()->m_pMainWnd;
  385. CClientDC appDC(pAppFrame);
  386. // All views but one should be a background palette.
  387. // wParam contains a handle to the active view, so the SelectPalette
  388. // bForceBackground flag is FALSE only if wParam == m_hWnd (this view)
  389. CPalette* oldPalette = appDC.SelectPalette(m_palDIB, FALSE);
  391. if (oldPalette != NULL)
  392. {
  393. UINT nColorsChanged = appDC.RealizePalette();
  394. if (nColorsChanged > 0)
  395. UpdateAllViews(NULL);
  396. appDC.SelectPalette(oldPalette, TRUE);
  397. }
  398. else
  399. {
  400. TRACE0("tSelectPalette failed!n");
  401. }
  402. }
  403. }
  405. void CDIBDoc::OnImageProcess(UINT uID)
  406. {
  407. CTime t_Start = CTime::GetCurrentTime();
  409. if(ID_IMAGE_HISTOGRAM == uID)
  410. {
  411. CDlgHistogram dlg(m_hDIB);
  412. dlg.DoModal();
  413. }
  414. else if(ID_IMAGE_HISTOGRAM_EQUAL == uID)
  415. {
  416. HDIB hDIB = CImage::HistoEqualize(m_hDIB);
  417. if(NULL != hDIB)
  418. {
  419. ReplaceHDIB(hDIB);
  420. InitDIBData();
  421. SetModifiedFlag();
  422. UpdateAllViews(NULL);
  423. }
  424. }
  425. else if(ID_IMAGE_MEDIANFILTER == uID)
  426. {
  427. if (CImgProcess::MedianFilterDIB(m_hDIB))
  428. {
  429. InitDIBData();
  430. SetModifiedFlag(TRUE);
  431. // OnRealizePal((WPARAM)0,0);  // realize the new palette
  432. UpdateAllViews(NULL);
  433. }
  434. }
  435. else if(ID_IMAGE_SMOTH_LF1 == uID)
  436. {
  437. CConvoluteDlg FD(1, 10, 3);
  438. if (FD.DoModal() == IDOK)
  439. {
  440. if(CImgProcess::LowPassDIB(m_hDIB, FD.m_nStreng, FILTER1))
  441. {
  442. InitDIBData();
  443. SetModifiedFlag(TRUE);
  444. // OnRealizePal((WPARAM)0, 0);  // realize the new palette
  445. UpdateAllViews(NULL);
  446. }
  447. }
  448. }
  449. else if(ID_IMAGE_SMOTH_LF2 == uID)
  450. {
  451. CConvoluteDlg FD(1, 10, 3);
  452. if (FD.DoModal() == IDOK)
  453. {
  454. if(CImgProcess::LowPassDIB(m_hDIB, FD.m_nStreng, FILTER2))
  455. {
  456. InitDIBData();
  457. SetModifiedFlag(TRUE);
  458. // OnRealizePal((WPARAM)0, 0);  // realize the new palette
  459. UpdateAllViews(NULL);
  460. }
  461. }
  462. }
  463. else if(ID_IMAGE_SMOTH_LF3 == uID)
  464. {
  465. CConvoluteDlg FD(1, 10, 3);
  466. if (FD.DoModal() == IDOK)
  467. {
  468. if(CImgProcess::LowPassDIB(m_hDIB, FD.m_nStreng, FILTER3))
  469. {
  470. InitDIBData();
  471. SetModifiedFlag(TRUE);
  472. // OnRealizePal((WPARAM)0, 0);  // realize the new palette
  473. UpdateAllViews(NULL);
  474. }
  475. }
  476. }
  477. else if(ID_IMAGE_SHARP_HF1 == uID)
  478. {
  479. CConvoluteDlg FD(1, 10, 3);
  480. if (FD.DoModal() == IDOK)
  481. {
  482. if(CImgProcess::HighPassDIB(m_hDIB, FD.m_nStreng, FILTER1))
  483. {
  484. InitDIBData();
  485. SetModifiedFlag(TRUE);
  486. // OnRealizePal((WPARAM)0,0);  // realize the new palette
  487. UpdateAllViews(NULL);
  488. }
  489. }
  490. }
  491. else if(ID_IMAGE_SHARP_HF2 == uID)
  492. {
  493. CConvoluteDlg FD(1, 10, 3);
  494. if (FD.DoModal() == IDOK)
  495. {
  496. if(CImgProcess::HighPassDIB(m_hDIB, FD.m_nStreng, FILTER2))
  497. {
  498. InitDIBData();
  499. SetModifiedFlag(TRUE);
  500. // OnRealizePal((WPARAM)0,0);  // realize the new palette
  501. UpdateAllViews(NULL);
  502. }
  503. }
  504. }
  505. else if(ID_IMAGE_SHARP_HF3 == uID)
  506. {
  507. CConvoluteDlg FD(1, 10, 3);
  508. if (FD.DoModal() == IDOK)
  509. {
  510. if(CImgProcess::HighPassDIB(m_hDIB, FD.m_nStreng, FILTER3))
  511. {
  512. InitDIBData();
  513. SetModifiedFlag(TRUE);
  514. // OnRealizePal((WPARAM)0,0);  // realize the new palette
  515. UpdateAllViews(NULL);
  516. }
  517. }
  518. }
  519. else if(ID_IMAGE_EDGE_H == uID)
  520. {
  521. EdgeEnhance(HORZ);
  522. }
  523. else if(ID_IMAGE_EDGE_V == uID)
  524. {
  525. EdgeEnhance(VERT);
  526. }
  527. else if(ID_IMAGE_EDGE_VH == uID)
  528. {
  529. EdgeEnhance(VERTHORZ);
  530. }
  531. else if(ID_IMAGE_EDGE_N == uID)
  532. {
  533. EdgeEnhance(NORTH);
  534. }
  535. else if(ID_IMAGE_EDGE_NE == uID)
  536. {
  537. EdgeEnhance(NORTHEAST);
  538. }
  539. else if(ID_IMAGE_EDGE_E == uID)
  540. {
  541. EdgeEnhance(EAST);
  542. }
  543. else if(ID_IMAGE_EDGE_SE == uID)
  544. {
  545. EdgeEnhance(SOUTHEAST);
  546. }
  547. else if(ID_IMAGE_EDGE_S == uID)
  548. {
  549. EdgeEnhance(SOUTH);
  550. }
  551. else if(ID_IMAGE_EDGE_SW == uID)
  552. {
  553. EdgeEnhance(SOUTHWEST);
  554. }
  555. else if(ID_IMAGE_EDGE_W == uID)
  556. {
  557. EdgeEnhance(WEST);
  558. }
  559. else if(ID_IMAGE_EDGE_NW == uID)
  560. {
  561. EdgeEnhance(NORTHWEST);
  562. }
  563. else if(ID_IMAGE_EDGE_LP1 == uID)
  564. {
  565. EdgeEnhance(LAP1);
  566. }
  567. else if(ID_IMAGE_EDGE_LP2 == uID)
  568. {
  569. EdgeEnhance(LAP2);
  570. }
  571. else if(ID_IMAGE_EDGE_LP3 == uID)
  572. {
  573. EdgeEnhance(LAP3);
  574. }
  575. else if(ID_IMAGE_EDGE_LP4 == uID)
  576. {
  577. EdgeEnhance(LAP4);
  578. }
  579. else if(ID_IMAGE_CONTOUR_H == uID)
  580. {
  581. if(CImgProcess::ContourDIB(m_hDIB, TRUE))
  582. {
  583. InitDIBData();
  584. SetModifiedFlag(TRUE);
  585. // OnRealizePal((WPARAM)0, 0);  // realize the new palette
  586. UpdateAllViews(NULL);
  587. }
  588. }
  589. else if(ID_IMAGE_CONTOUR_V == uID)
  590. {
  591. if(CImgProcess::ContourDIB(m_hDIB, FALSE))
  592. {
  593. InitDIBData();
  594. SetModifiedFlag(TRUE);
  595. // OnRealizePal((WPARAM)0, 0);  // realize the new palette
  596. UpdateAllViews(NULL);
  597. }
  598. }
  599. else if(ID_IMAGE_CONTOUR_SOBER == uID)
  600. {
  601. EdgeEnhance(SOBEL);
  602. }
  603. else if(ID_IMAGE_CONTOUR_HOUGH == uID)
  604. {
  605. EdgeEnhance(HOUGH);
  606. }
  607. else if(ID_IMAGE_REVERSE == uID)
  608. {
  609. if(CImgProcess::ReverseDIB(m_hDIB))
  610. {
  611. InitDIBData();
  612. SetModifiedFlag(TRUE);
  613. // OnRealizePal((WPARAM)0, 0);  // realize the new palette
  614. UpdateAllViews(NULL);
  615. }
  616. }
  617. else if(ID_IMAGE_FFT == uID)
  618. {
  619. if(CImgProcess::FFTDIB(m_hDIB))
  620. {
  621. InitDIBData();
  622. SetModifiedFlag(TRUE);
  623. // OnRealizePal((WPARAM)0, 0);  // realize the new palette
  624. UpdateAllViews(NULL);
  625. }
  626. }
  627. else if(ID_IMAGE_DCT == uID)
  628. {
  629. if(CImgProcess::DCTDIB(m_hDIB))
  630. {
  631. InitDIBData();
  632. SetModifiedFlag(TRUE);
  633. // OnRealizePal((WPARAM)0, 0);  // realize the new palette
  634. UpdateAllViews(NULL);
  635. }
  636. }
  637. else if(ID_IMAGE_WALH == uID)
  638. {
  639. if(CImgProcess::WALhDIB(m_hDIB))
  640. {
  641. InitDIBData();
  642. SetModifiedFlag(TRUE);
  643. // OnRealizePal((WPARAM)0, 0);  // realize the new palette
  644. UpdateAllViews(NULL);
  645. }
  646. }
  647. else if(ID_IMAGE_EROSION_H == uID)
  648. {
  649. if(CImgProcess::ErosionDIB(m_hDIB, TRUE))
  650. {
  651. InitDIBData();
  652. SetModifiedFlag(TRUE);
  653. // OnRealizePal((WPARAM)0, 0);  // realize the new palette
  654. UpdateAllViews(NULL);
  655. }
  656. }
  657. else if(ID_IMAGE_EROSION_V == uID)
  658. {
  659. if(CImgProcess::ErosionDIB(m_hDIB, FALSE))
  660. {
  661. InitDIBData();
  662. SetModifiedFlag(TRUE);
  663. // OnRealizePal((WPARAM)0, 0);  // realize the new palette
  664. UpdateAllViews(NULL);
  665. }
  666. }
  667. else if(ID_IMAGE_DILATION_H == uID)
  668. {
  669. if(CImgProcess::DilationDIB(m_hDIB, TRUE))
  670. {
  671. InitDIBData();
  672. SetModifiedFlag(TRUE);
  673. // OnRealizePal((WPARAM)0, 0);  // realize the new palette
  674. UpdateAllViews(NULL);
  675. }
  676. }
  677. else if(ID_IMAGE_DILATION_V == uID)
  678. {
  679. if(CImgProcess::DilationDIB(m_hDIB, FALSE))
  680. {
  681. InitDIBData();
  682. SetModifiedFlag(TRUE);
  683. // OnRealizePal((WPARAM)0, 0);  // realize the new palette
  684. UpdateAllViews(NULL);
  685. }
  686. }
  687. else if(ID_IMAGE_CLOSE_H == uID)
  688. {
  689. if(CImgProcess::MorphCloseDIB(m_hDIB, TRUE))
  690. {
  691. InitDIBData();
  692. SetModifiedFlag(TRUE);
  693. // OnRealizePal((WPARAM)0, 0);  // realize the new palette
  694. UpdateAllViews(NULL);
  695. }
  696. }
  697. else if(ID_IMAGE_CLOSE_V == uID)
  698. {
  699. if(CImgProcess::MorphCloseDIB(m_hDIB, FALSE))
  700. {
  701. InitDIBData();
  702. SetModifiedFlag(TRUE);
  703. // OnRealizePal((WPARAM)0, 0);  // realize the new palette
  704. UpdateAllViews(NULL);
  705. }
  706. }
  707. else if(ID_IMAGE_OPEN_H == uID)
  708. {
  709. if(CImgProcess::MorphOpenDIB(m_hDIB, TRUE))
  710. {
  711. InitDIBData();
  712. SetModifiedFlag(TRUE);
  713. // OnRealizePal((WPARAM)0, 0);  // realize the new palette
  714. UpdateAllViews(NULL);
  715. }
  716. }
  717. else if(ID_IMAGE_OPEN_V == uID)
  718. {
  719. if(CImgProcess::MorphOpenDIB(m_hDIB, FALSE))
  720. {
  721. InitDIBData();
  722. SetModifiedFlag(TRUE);
  723. // OnRealizePal((WPARAM)0, 0);  // realize the new palette
  724. UpdateAllViews(NULL);
  725. }
  726. }
  727. else if(ID_IMAGE_THIN == uID)
  728. {
  729. if(CImgProcess::ThinningDIB(m_hDIB))
  730. {
  731. InitDIBData();
  732. SetModifiedFlag(TRUE);
  733. // OnRealizePal((WPARAM)0, 0);  // realize the new palette
  734. UpdateAllViews(NULL);
  735. }
  736. }
  737. else
  738. {
  739. AfxMessageBox("Hi, what are you doing?");
  740. }
  742. CTimeSpan t_Span = CTime::GetCurrentTime() - t_Start;
  743. CString strStatus = t_Span.Format("Elapsed time- %H : %M : %S");
  744. ((CMainFrame*)AfxGetMainWnd())->SetStatusText(0, strStatus);
  745. }
  747. void CDIBDoc::OnUpdateImageProcess(CCmdUI *pCmdUI)
  748. {
  749. pCmdUI->Enable(NULL != m_hDIB);
  750. }
  752. void CDIBDoc::EdgeEnhance(int nAlgorithm)
  753. {
  754. CConvoluteDlg FD(1, 10, 3);
  755. if (FD.DoModal() == IDOK)
  756. {
  757. if(CImgProcess::EdgeEnhanceDIB(m_hDIB, FD.m_nStreng, nAlgorithm))
  758. {
  759. InitDIBData();
  760. SetModifiedFlag(TRUE);
  761. // OnRealizePal((WPARAM)0, 0);  // realize the new palette
  762. UpdateAllViews(NULL);
  763. }
  764. }
  765. }
  767. BOOL CDIBDoc::EnumFile(LPCTSTR lpszPathName)
  768. {
  769. ASSERT(strlen(lpszPathName) > 0);
  771. BeginWaitCursor();
  773. char *pExt=strrchr(lpszPathName,'.');
  774. char cExt[4];
  775. strcpy(cExt,++pExt);
  776. _strlwr(cExt);
  777. DeleteContents();
  778. HDIB hDIB = NULL;
  780. if (strcmp(cExt,"raw")==0)
  781. {
  782. CRaw raw;
  783. if(!raw.ReadFromFile(lpszPathName))
  784. {
  785. return FALSE;
  786. }
  787. hDIB = raw.Show();
  788. }
  789. else if(strcmp(cExt,"bmp")==0
  790. || strcmp(cExt,"dib")==0
  791. || strcmp(cExt,"ddb")==0)
  792. {
  793. CFile file;
  794. CFileException fe;
  795. if (!file.Open(lpszPathName, CFile::modeRead | CFile::shareDenyWrite, &fe))
  796. {
  797. fe.ReportError();
  798. return NULL;
  799. }
  800. // replace calls to Serialize with ReadDIBFile function
  801. TRY
  802. {
  803. hDIB = ::ReadDIBFile(file);
  804. }
  805. CATCH (CFileException, eLoad)
  806. {
  807. file.Abort(); // will not throw an exception
  808. eLoad->ReportError();
  809. hDIB = NULL;
  810. return NULL;
  811. }
  812. END_CATCH
  813. if (hDIB == NULL)
  814. {
  815. // may not be DIB format
  816. CString strMsg;
  817. strMsg.LoadString(IDS_CANNOT_LOAD_DIB);
  818. MessageBox(NULL, strMsg, NULL, MB_ICONINFORMATION | MB_OK);
  819. return NULL;
  820. }
  821. }
  822. else if(strcmp(cExt,"jpg")==0 || strcmp(cExt,"jpeg")==0)
  823. {
  824. UINT height,width;
  825. BYTE *pData = NULL;
  826. pData = JpegFile::JpegFileToRGB(lpszPathName, &width, &height);
  827. if(NULL == pData)return NULL;
  828. JpegFile::BGRFromRGB(pData, width, height);
  829. JpegFile::VertFlipBuf(pData, width * 3, height);
  831. UINT widthDW;
  832. BYTE *pDIB = JpegFile::MakeDwordAlignedBuf(pData, width, height, &widthDW);
  834. delete []pData;
  835. if(NULL == pDIB)
  836. {
  837. char strErr[100];
  838. sprintf(strErr, "Error: %d", GetLastError());
  839. MessageBox(AfxGetMainWnd()->GetSafeHwnd(), strErr, "Read File Error",
  840. MB_OK|MB_ICONERROR);
  841. }
  842. hDIB = ::ShowJPEG(width, height, pDIB);
  843. delete []pDIB;
  844. }
  845. else if(strcmp(cExt, "gif") == 0)
  846. {
  847. hDIB = ReadGIF(lpszPathName);
  848. }
  849. else if(strcmp(cExt,"pcx") == 0)
  850. {
  851. hDIB = ::ReadPCX256(lpszPathName);
  852. }
  853. else
  854. {
  855. // CString strMsg;
  856. // strMsg.Format("%sn不是图像文件!", lpszPathName);
  857. AfxMessageBox("不识别的文件格式!");
  858. return FALSE;
  859. }
  861. if(NULL != hDIB)
  862. {
  863. ReplaceHDIB(hDIB);
  864. InitDIBData();
  865. }
  866. else
  867. {
  868. return FALSE;
  869. }
  870. SetPathName(lpszPathName);
  871. char *pTitle = strrchr(lpszPathName, '\');
  872. SetTitle(++pTitle);
  873. SetModifiedFlag(FALSE);
  874. EndWaitCursor();
  875. return TRUE;
  876. }