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

源码开发语言/平台
当前位置：首页 > 源码/资料 > 图形图像 > 图形图像处理 > 查看源码
视频编解码器View.cpp
资源名称：263.rar　[点击查看]
上传用户：szklck
上传日期：2007-01-22
资源大小：925k
文件大小：22k
源码类别：

图形图像处理

开发平台：

Visual C++

视频编解码器View.cpp：源码内容
  1. // 视频编解码器View.cpp : implementation of the CMyView class
  2. //
  4. #include "stdafx.h"
  5. #include <math.h>                   
  6. #include "视频编解码器.h"
  8. #include "视频编解码器Doc.h"
  9. #include "视频编解码器View.h"
  10. #include "OpenFiles.h"
  11. #include "Global.h"
  12. #include "ProgressBar.h"
  13. #include "Display.h"
  14. #include "PropertyDlg.h"
  16. #ifdef _DEBUG
  17. #define new DEBUG_NEW
  18. #undef THIS_FILE
  19. static char THIS_FILE[] = __FILE__;
  20. #endif
  22. int pels,lines,mv_outside_frame,long_vectors,cpels;
  23. CFile streamfile,psnrfile;
  24. static void SeekPsnr(PictImage *curr,PictImage *recon,int width,int height,double psnr[3]);
  26. #define OFFLINE_RATE_CONTROL//jwp
  27. /////////////////////////////////////////////////////////////////////////////
  28. // CMyView
  30. IMPLEMENT_DYNCREATE(CMyView, CView)
  32. BEGIN_MESSAGE_MAP(CMyView, CView)
  33. //{{AFX_MSG_MAP(CMyView)
  34. ON_COMMAND(ID_Encode, OnEncode)
  35. ON_COMMAND(ID_Decode, OnDecode)
  36. ON_COMMAND(ID_FILE_OPEN, OnFileOpen)
  37. //}}AFX_MSG_MAP
  38. // Standard printing commands
  39. ON_COMMAND(ID_FILE_PRINT, CView::OnFilePrint)
  40. ON_COMMAND(ID_FILE_PRINT_DIRECT, CView::OnFilePrint)
  41. ON_COMMAND(ID_FILE_PRINT_PREVIEW, CView::OnFilePrintPreview)
  42. ON_COMMAND(ID_Encode, OnEncode)
  43. ON_COMMAND(ID_Decode, OnDecode)
  44. //}}AFX_MSG_MAP
  45. END_MESSAGE_MAP()
  46. // CMyView construction/destruction
  48. CMyView::CMyView()
  50. {
  51. m_font.CreateFont(20,0,0,0,FW_THIN,FALSE,FALSE,FALSE,GB2312_CHARSET,OUT_DEFAULT_PRECIS,CLIP_DEFAULT_PRECIS,DEFAULT_QUALITY,FIXED_PITCH|FF_MODERN,"宋体");
  52. bFlag=0;
  53. }
  55. CMyView::~CMyView()
  56. {
  57. m_font.DeleteObject();
  59. }
  61. BOOL CMyView::PreCreateWindow(CREATESTRUCT& cs)
  62. {
  63. // TODO: Modify the Window class or styles here by modifying
  64. //  the CREATESTRUCT cs
  66. return CView::PreCreateWindow(cs);
  67. }
  69. /////////////////////////////////////////////////////////////////////////////
  70. // CMyView drawing
  72. void CMyView::OnDraw(CDC* pDC)
  73. {
  74. CBitmap bitmap;
  75. CDC dcMemory;
  76. bitmap.LoadBitmap(IDB_car);
  77. dcMemory.CreateCompatibleDC(pDC);
  78. dcMemory.SelectObject(&bitmap);
  79. pDC->SetTextColor(0x64e896);
  81. if(bFlag==1)//编码
  82. {
  83.    CFont *pOldFont;
  84.    pOldFont=pDC->SelectObject(&m_font);
  85.    pDC->SetBkMode(TRANSPARENT);
  87. pDC->StretchBlt(0,0,1024*3/4,768*3/4,&dcMemory,0,0,1024,768,SRCCOPY);
  88.         CString kk;
  89. kk.Format("正在编码!共%d帧",MaxFrame);
  90. pDC->TextOut(60,60,kk);
  92. kk.Format("每%d帧进行一次帧内编码，进度情况请看状态栏!",Pbetween);
  93.     pDC->TextOut(60,100,kk);
  94. }
  95.     else if(bFlag==2)//解码
  96. {
  97. CBrush Brush(RGB(100,200,150));//190,190,190
  98. CBrush* oldBrush=pDC->SelectObject(&Brush);
  99.     CPen pen(0,2,RGB(100,200,150));
  100. CPen *oldPen=pDC->SelectObject(&pen);
  101. pDC->Rectangle(40,40,440,370);//画矩形
  102. pDC->MoveTo(40,370); // (wide,height)
  103. pDC->LineTo(620,370);
  104. pDC->MoveTo(620,40);        
  105. pDC->LineTo(620,370);
  106. pDC->MoveTo(40,40);        
  107. pDC->LineTo(620,40);
  108. pDC->SelectObject(oldPen);
  109. pDC->SelectObject(oldBrush);
  111. //显示解码图象
  112. CDisplay m_display;
  113. m_display.play_movie(pDC,DecfileName);
  114. conclusion=m_display.conclusion;
  116. //输出解码结果
  117. pDC->SetTextColor(RGB(200,100,150));
  118. int dd=conclusion.Find(",",0);
  119. CString kk=conclusion.Right(conclusion.GetLength()-dd-1);
  120. int zz=kk.Find(",",0);
  121. pDC->TextOut(470,180,conclusion.Left(dd));
  122. pDC->TextOut(470,200,kk.Left(zz));
  123. pDC->TextOut(500,220,kk.Right(kk.GetLength()-zz-1));
  124. }
  125. }
  127. /////////////////////////////////////////////////////////////////////////////
  128. // CMyView printing
  130. BOOL CMyView::OnPreparePrinting(CPrintInfo* pInfo)
  131. {
  132. // default preparation
  133. return DoPreparePrinting(pInfo);
  134. }
  136. void CMyView::OnBeginPrinting(CDC* /*pDC*/, CPrintInfo* /*pInfo*/)
  137. {
  138. // TODO: add extra initialization before printing
  139. }
  141. void CMyView::OnEndPrinting(CDC* /*pDC*/, CPrintInfo* /*pInfo*/)
  142. {
  143. // TODO: add cleanup after printing
  144. }
  146. /////////////////////////////////////////////////////////////////////////////
  147. // CMyView diagnostics
  149. #ifdef _DEBUG
  150. void CMyView::AssertValid() const
  151. {
  152. CView::AssertValid();
  153. }
  155. void CMyView::Dump(CDumpContext& dc) const
  156. {
  157. CView::Dump(dc);
  158. }
  160. CMyDoc* CMyView::GetDocument() // non-debug version is inline
  161. {
  162. ASSERT(m_pDocument->IsKindOf(RUNTIME_CLASS(CMyDoc)));
  163. return (CMyDoc*)m_pDocument;
  164. }
  165. #endif //_DEBUG
  167. /////////////////////////////////////////////////////////////////////////////
  168. // CMyView message handlers
  170. void CMyView::OnEncode() 
  171. {
  173. CPropertyDlg properties("编码属性设置",this,0);
  174. int ff;
  175. CString dd;
  176. if(properties.DoModal()==IDOK)
  177. {
  178. m_szFilePathName=properties.encodeDlg.m_InitDir;
  179. m_szFileName=m_szFilePathName;
  180. m_Type=properties.encodeDlg.typeindex;
  181. MaxFrame=properties.encodeDlg.m_MaxFrame;
  182.         QP=properties.encodeDlg.m_QP;
  183. QPI=properties.encodeDlg.m_QPI;
  184. ifPsnr=properties.encodeDlg.m_ifPsnr;
  185. Pbetween=properties.encodeDlg.m_Pbetween+1;
  186.         
  187. ratecontrol=properties.m_page1.m_ifratecontrol;
  188. targetrate=properties.m_page1.m_bitrate;
  189. switch(properties.m_page1.m_framerate)
  190. {
  191. case 1:
  192.             ref_frame_rate=30.0;break;
  193. case 0:
  194.             ref_frame_rate=25.0;break;
  195. case 2:
  196. ref_frame_rate=10.0;break;
  197. default:
  198. ref_frame_rate=25.0;
  199. }
  201. int kk=targetrate;
  202. ff=properties.m_page1.m_ifratecontrol;
  203. dd.Format("采用率控制，码率是%d，帧率是%f。",kk,ref_frame_rate);
  204.         if(ff)
  205. MessageBox(dd);
  207. if(m_Type==2)
  208. CodeYUV();
  209.     else
  210. CodeBmps();
  211. }
  212. else
  213. return;
  215.   
  217. }
  219. void CMyView::CodeYUV()
  220. { 
  221. //率控制
  222.   #ifndef OFFLINE_RATE_CONTROL
  223.     float DelayBetweenFramesInSeconds;
  224.     int CommBacklog;
  225.   #else
  226.     PictImage *stored_image = NULL;
  227.     int start_rate_control;
  228.   #endif
  230. CMyDoc* pDoc = GetDocument();
  231. ASSERT_VALID(pDoc);
  232. //初始化
  233. CProgressBar bar(_T("Encode Progress"), 50, MaxFrame);
  234.     m_pImageData=new unsigned char[352*288*3/2];
  235. CFile file;
  236. if((file.Open(m_szFilePathName,CFile::modeRead|CFile::shareDenyNone))==NULL)
  237. {
  238. AfxMessageBox("Can not open the yuv file!");
  239. return;
  240. }
  242. PictImage *prev_image = NULL;
  243.     PictImage *curr_image = NULL;
  244.     PictImage *curr_recon = NULL;
  245.     PictImage *prev_recon = NULL;
  247. int frame_no,first_frameskip=0;  
  248.     int start=1;//从第1帧到第MaxFrame帧
  249.     int orig_frameskip=1;//输入图像原始偏移
  250.     int frameskip=1;//非发送帧
  252. long_vectors =0;//帧间编码的参数
  253. mv_outside_frame=0;
  255. Pict *pic = (Pict *)malloc(sizeof(Pict));
  256. Bits *bits = (Bits *)malloc(sizeof(Bits));
  257. //率控制
  258. Bits *total_bits = (Bits *)malloc(sizeof(Bits));
  259.     Bits *intra_bits = (Bits *)malloc(sizeof(Bits));
  261. pic->BQUANT = DEF_BQUANT;
  262.     pic->seek_dist = DEF_SEEK_DIST;
  263.     pic->use_gobsync = DEF_INSERT_SYNC;//=0
  264.     pic->PB = 0;
  265.     pic->TR = 0;
  266.     pic->QP_mean = (float)0.0;
  267.     pic->unrestricted_mv_mode = 0;
  268. pic->picture_coding_type =0; // PCT_INTRA;
  269. m_orgHeight=288;
  270. m_orgWidth=352;
  271.     pic->source_format = SF_CIF;
  272. switch (pic->source_format) {
  273.     case (SF_SQCIF):
  274.       fprintf(stdout, "Encoding format: SQCIF (128x96)n");
  275.       pels = 128;
  276.       lines = 96;
  277.       break;
  278.     case (SF_QCIF):
  279.       fprintf(stdout, "Encoding format: QCIF (176x144)n");
  280.       pels = 176;
  281.       lines = 144;
  282.       break;
  283.     case (SF_CIF):
  284.       fprintf(stdout, "Encoding format: CIF (352x288)n");
  285.       pels = 352;
  286.       lines = 288;
  287.       break;
  288.     case (SF_4CIF):
  289.       fprintf(stdout, "Encoding format: 4CIF (704x576)n");
  290.       pels = 704;
  291.       lines = 576;
  292.       break;
  293.     case (SF_16CIF):
  294.       fprintf(stdout, "Encoding format: 16CIF (1408x1152)n");
  295.       pels = 1408;
  296.       lines = 1152;
  297.       break;
  298.     default:
  299.       fprintf(stderr,"Illegal coding formatn");
  300.       exit(-1);
  301. }
  302.     cpels = pels/2;
  303. curr_recon = InitImage(pels*lines);
  305. //率控制
  306.   #ifndef OFFLINE_RATE_CONTROL
  307.   // rate control variables 
  308.   pic->bit_rate = targetrate;
  309.   pic->src_frame_rate = (int)(ref_frame_rate / orig_frameskip);
  310.   DelayBetweenFramesInSeconds = (float) 1.0/(float)pic->src_frame_rate;
  311.   InitializeRateControl();
  312.   #endif
  313.   #ifdef OFFLINE_RATE_CONTROL
  314.   start_rate_control = 0;//DEF_START_RATE_CONTROL;
  315.   #else
  316.   pic->target_frame_rate = (float)DEF_TARGET_FRAME_RATE;
  317.   #endif
  319. //建立输出文件
  320. CString outfilename=m_szFileName.Left(m_szFileName.GetLength()-4);
  321. CFileDialog dlg(FALSE,".263",outfilename,OFN_OVERWRITEPROMPT,"263 Files(*.263)|*.263|",NULL);
  322. if (dlg.DoModal()==IDOK)
  323. {
  324. bFlag=1;
  325. pDoc->UpdateAllViews(NULL);
  326.  
  327. CString tempname;
  328. tempname=dlg.GetPathName();
  329. outfilename=tempname.Left(tempname.GetLength()-4);
  330.     if((streamfile.Open(tempname,CFile::modeWrite|CFile::modeCreate))==FALSE)
  331. AfxMessageBox("Can't create file!"); 
  332. streamfile.SeekToBegin();
  333.       initbits ();
  336. CTime StartTime=CTime::GetCurrentTime();
  337.         CTimeSpan ElapsedTime;
  339.     file.Read(m_pImageData,sizeof(BYTE)*352*288*3/2);
  340. pic->QUANT=QPI;
  341. curr_image = FillImage(m_pImageData);
  342. curr_recon = CodeOneIntra(curr_image, QPI, bits, pic);
  343. bits->header += alignbits (); // pictures shall be byte aligned 
  345. //率控制
  346.   AddBitsPicture(bits);
  347.   memcpy(intra_bits,bits,sizeof(Bits));
  348.   ZeroBits(total_bits);
  349.     //* number of seconds to encode *
  350.   int chosen_frameskip=1;//jwp
  351.     //* compute first frameskip *
  352.   #ifdef OFFLINE_RATE_CONTROL
  353.    float seconds = (MaxFrame - start + chosen_frameskip) * orig_frameskip/ ref_frame_rate;
  354.    first_frameskip = chosen_frameskip;
  355.    frameskip = chosen_frameskip;
  356.   #else
  357.   CommBacklog = intra_bits->total -(int)(DelayBetweenFramesInSeconds * pic->bit_rate);
  359.   if (pic->bit_rate == 0) {
  360.     frameskip = chosen_frameskip;
  361.   }
  362.   else {  //* rate control is used *
  363.     frameskip = 1;
  364.     while ( (int)(DelayBetweenFramesInSeconds*pic->bit_rate) <= CommBacklog) {
  365.       CommBacklog -= (int) ( DelayBetweenFramesInSeconds * pic->bit_rate );
  366.       frameskip += 1;
  367.     }
  368.   }
  369.   first_frameskip = frameskip;
  370.   #endif
  372. CString kk,m_Spsnr;
  373.     if(ifPsnr)
  374. {
  375. if(psnrfile.Open(outfilename+".doc",CFile::modeWrite|CFile::modeCreate)==FALSE)
  376.     MessageBox("Cannot create the output psnr file!", "Error",MB_ICONERROR | MB_OK);
  377. SeekPsnr(curr_image,curr_recon,352,288,psnrs);
  378. frame_no=1;
  379. kk.Format("第%d帧的lum峰值信噪比为%6.4fdBn",frame_no,psnrs[0]);
  380. m_Spsnr=kk;
  381. kk.Format("第%d帧的Cb峰值信噪比为%6.4fdBn",frame_no,psnrs[1]);
  382. m_Spsnr+=kk;
  383. kk.Format("第%d帧的Cr峰值信噪比为%6.4fdBn",frame_no,psnrs[2]);
  384. m_Spsnr+=kk;
  385. MessageBox(m_Spsnr);
  386. psnrfile.SeekToBegin();
  387. psnrfile.Write(m_Spsnr,m_Spsnr.GetLength());
  388. }
  389.  //第二帧 
  390. pic->QUANT=QP;
  391.    for(frame_no=start+first_frameskip;frame_no<=MaxFrame;frame_no+=frameskip)
  392. {
  393. file.Seek((frame_no-1)*352*288*3/2,SEEK_SET);
  394.     file.Read(m_pImageData,sizeof(BYTE)*352*288*3/2);
  395. if(m_pImageData==NULL)
  396.   return;
  397.    
  398.      pic->picture_coding_type =1; // PCT_INTER;
  400.   if(!ratecontrol)
  401.   pic->QUANT=QP;
  402.   
  403.       prev_image=curr_image;
  404.   prev_recon=curr_recon;
  405.       curr_image = FillImage(m_pImageData);
  406.   pic->TR+=(((frameskip+(pic->PB?98:0))*orig_frameskip)%256);
  407.   if(frameskip+(pic->PB?98:0)>256)
  408. MessageBox("Warning:frameskip>256");
  410.      streamfile.SeekToEnd();
  411.     if(((frame_no-1)%Pbetween)==0)
  412.  {
  413.    pic->picture_coding_type =0; // PCT_INTRA;
  414.    pic->QUANT=QPI;
  415.        curr_recon = CodeOneIntra(curr_image, QPI, bits, pic);
  416.    AddBitsPicture(bits);
  417.    memcpy(intra_bits,bits,sizeof(Bits));
  419.  }
  420.     else
  421. { 
  422.   CodeOneInter(prev_image,curr_image,prev_recon,curr_recon,pic->QUANT,frameskip,bits,pic);
  423.   AddBitsPicture(bits); 
  424. }
  425.    bits->header += alignbits (); //* pictures shall be byte aligned *
  427.  
  428.    if(ifPsnr)
  429.    { SeekPsnr(curr_image,curr_recon,352,288,psnrs);
  430. kk.Format("第%d帧的lum峰值信噪比为%6.4fn",frame_no,psnrs[0]);
  431. m_Spsnr=kk;
  432. kk.Format("第%d帧的Cb峰值信噪比为%6.4fn",frame_no,psnrs[1]);
  433. m_Spsnr+=kk;
  434. kk.Format("第%d帧的Cr峰值信噪比为%6.4fn",frame_no,psnrs[2]);
  435. m_Spsnr+=kk;
  436. psnrfile.SeekToEnd();
  437. psnrfile.Write(m_Spsnr,m_Spsnr.GetLength());
  438.    }
  439. //率控制
  441.    AddBits(total_bits, bits);
  442. #ifndef OFFLINE_RATE_CONTROL
  443.     if (pic->bit_rate != 0 && pic->PB)
  444.       CommBacklog -= (int)
  445.       ( DelayBetweenFramesInSeconds*pic->bit_rate ) * pdist;
  447.     if (pic->bit_rate != 0) {
  448.       UpdateRateControl(bits->total);
  450.       CommBacklog += bits->total;
  451.       frameskip = 1;
  452.       CommBacklog -= (int)
  453.         (frameskip * DelayBetweenFramesInSeconds *pic->bit_rate);
  455.       while ( (int)(DelayBetweenFramesInSeconds*pic->bit_rate) <= CommBacklog)
  456.       {
  457.         CommBacklog -= (int) ( DelayBetweenFramesInSeconds * pic->bit_rate );
  458.         frameskip += 1;
  459.       }
  460.     }
  461. #else
  462.     //* Aim for the targetrate with a once per frame rate control scheme *
  463.     if (targetrate != 0)
  464.       if (frame_no - start > (MaxFrame - start) * start_rate_control/100.0)
  466.         pic->QUANT = FrameUpdateQP(total_bits->total + intra_bits->total,
  467.            bits->total / (pic->PB?2:1),
  468.            (MaxFrame-frame_no) / chosen_frameskip ,
  469.            pic->QUANT, targetrate, seconds);
  470.    frameskip = chosen_frameskip;
  471. #endif
  472.   CString kkk;
  473.   kkk.Format("编码率为%d,%d帧的total_bits->total 为%d,intra_bits->total 为 %d, bits->total 为%d,new quant is %d.",
  474.   targetrate,frame_no,total_bits->total,intra_bits->total,bits->total,pic->QUANT);
  475.   if(ifPsnr)
  476.   {
  477.   psnrfile.SeekToEnd();
  478.       psnrfile.Write(kkk,kkk.GetLength());
  479.   }
  481. //显示进度信息
  482. CString str;
  483. str.Format("%d%% complete", frame_no*100/MaxFrame);
  484. bar.SetText(str);
  485. bar.StepIt();
  486.     PeekAndPump(); //调用函数实现消息的转发
  487.    }//end for frame_no
  489. //  pDoc->SetModifiedFlag(TRUE);
  490.   file.Close();
  491.   if(ifPsnr)
  492.   psnrfile.Close();
  493.   streamfile.Close();
  494.   delete prev_image;
  495.   delete prev_recon;
  496.   delete curr_image;
  497.   curr_recon=NULL;
  498. //  delete curr_recon;
  499.   free(bits);
  500.   free(pic);
  501.   
  503.     long hours,minutes,second;//计算所用的时间
  504.     ElapsedTime = CTime::GetCurrentTime() - StartTime;
  505. bFlag=0;    
  506. hours = ElapsedTime.GetTotalHours();
  507.     minutes = ElapsedTime.GetTotalMinutes();
  508.     second = ElapsedTime.GetTotalSeconds();
  509.     second = second + 60*minutes + 3600*hours;
  510.     csTimeElapse.Format("编码%d帧视频序列,耗时:%d秒！",frame_no-1,second);
  511.     MessageBox(csTimeElapse);
  512. pDoc->UpdateAllViews(NULL);
  513. }
  514. else
  515.     MessageBox("No file is saved.","系统提示",MB_OK);
  516. // delete m_pImageData;
  518. }
  520. void CMyView::CodeBmps()
  521. {
  522.    //初始化
  523. CMyDoc* pDoc = GetDocument();
  524. ASSERT_VALID(pDoc);
  525. CProgressBar bar(_T("Encode Progress"), 50, MaxFrame);
  527.     m_pImageData=new unsigned char[352*288+352*288/2];
  528. m_pImageData = OpenImageFile(m_szFilePathName, &m_orgWidth, &m_orgHeight, m_Type);
  530.     PictImage *prev_image = NULL;
  531.     PictImage *curr_image = NULL;
  532.     PictImage *curr_recon = NULL;
  533.     PictImage *prev_recon = NULL;
  535. int frame_no,first_frameskip=0;  
  536.     int start=2;//从第1帧到第MaxFrame帧
  537.     int orig_frameskip=1;//输入图像原始偏移
  538.     int frameskip=1;//非发送帧
  540. long_vectors =0;//帧间编码的参数
  541. mv_outside_frame=0;
  543. Pict *pic = (Pict *)malloc(sizeof(Pict));
  544. Bits *bits = (Bits *)malloc(sizeof(Bits));
  545. pic->BQUANT = DEF_BQUANT;
  546.     pic->seek_dist = DEF_SEEK_DIST;
  547.     pic->use_gobsync = DEF_INSERT_SYNC;//=0
  548.     pic->PB = 0;
  549.     pic->TR = 0;
  550.     pic->QP_mean = (float)0.0;
  551.     pic->unrestricted_mv_mode = 0;
  552. pic->picture_coding_type =0; // PCT_INTRA;
  553. if((m_orgHeight==288)&&(m_orgWidth==352))
  554.        pic->source_format = SF_CIF;
  555. switch (pic->source_format) {
  556.     case (SF_SQCIF):
  557.       fprintf(stdout, "Encoding format: SQCIF (128x96)n");
  558.       pels = 128;
  559.       lines = 96;
  560.       break;
  561.     case (SF_QCIF):
  562.       fprintf(stdout, "Encoding format: QCIF (176x144)n");
  563.       pels = 176;
  564.       lines = 144;
  565.       break;
  566.     case (SF_CIF):
  567.       fprintf(stdout, "Encoding format: CIF (352x288)n");
  568.       pels = 352;
  569.       lines = 288;
  570.       break;
  571.     case (SF_4CIF):
  572.       fprintf(stdout, "Encoding format: 4CIF (704x576)n");
  573.       pels = 704;
  574.       lines = 576;
  575.       break;
  576.     case (SF_16CIF):
  577.       fprintf(stdout, "Encoding format: 16CIF (1408x1152)n");
  578.       pels = 1408;
  579.       lines = 1152;
  580.       break;
  581.     default:
  582.       fprintf(stderr,"Illegal coding formatn");
  583.       exit(-1);
  584. }
  585.     cpels = pels/2;
  586. curr_recon = InitImage(pels*lines);
  588. //建立输出文件
  589. CString outfilename=m_szFileName.Left(m_szFileName.GetLength()-4);
  590. CFileDialog dlg(FALSE,".263",outfilename,OFN_OVERWRITEPROMPT,"263 Files(*.263)|*.263|",NULL);
  591. if (dlg.DoModal()==IDOK)
  592. {  
  593. bFlag=1;
  594. pDoc->UpdateAllViews(NULL);
  596.         CString tempname;
  597. tempname=dlg.GetPathName();
  598. outfilename=tempname.Left(tempname.GetLength()-4);
  600.     if((streamfile.Open(tempname,CFile::modeWrite|CFile::modeCreate))==FALSE)
  601. AfxMessageBox("Can't create file!"); 
  602. streamfile.SeekToBegin();
  603.       initbits ();
  605. CTime StartTime=CTime::GetCurrentTime();
  606.         CTimeSpan ElapsedTime;
  608. pic->QUANT=QPI;
  609. curr_image = FillImage(m_pImageData);
  610. curr_recon = CodeOneIntra(curr_image, QPI, bits, pic);
  611. bits->header += alignbits (); /* pictures shall be byte aligned */
  613. CString kk,m_Spsnr;
  614.    if(ifPsnr)
  615.    {
  616. if(psnrfile.Open(outfilename+".doc",CFile::modeWrite|CFile::modeCreate)==FALSE)
  617.     MessageBox("Cannot create the output psnr file!", "Error",MB_ICONERROR | MB_OK);
  618. SeekPsnr(curr_image,curr_recon,352,288,psnrs);
  619. frame_no=1;
  620. kk.Format("第%d帧的lum峰值信噪比为%6.4fn",frame_no,psnrs[0]);
  621. m_Spsnr=kk;
  622. kk.Format("第%d帧的Cb峰值信噪比为%6.4fn",frame_no,psnrs[1]);
  623. m_Spsnr+=kk;
  624. kk.Format("第%d帧的Cr峰值信噪比为%6.4fn",frame_no,psnrs[2]);
  625. m_Spsnr+=kk;
  626. MessageBox(m_Spsnr);
  627. psnrfile.SeekToBegin();
  628. psnrfile.Write(m_Spsnr,m_Spsnr.GetLength());
  629.    }
  631.  //第二帧 
  634.     for(frame_no=start+first_frameskip;frame_no<=MaxFrame;frame_no+=frameskip)
  635. {
  636.      m_szFilePathName = GetNextFileName(m_szFilePathName,1);
  637.       m_pImageData = OpenImageFile(m_szFilePathName, &m_orgWidth, &m_orgHeight, m_Type);
  638.   if(m_pImageData==NULL)
  639.   return;
  640.    
  641.      pic->picture_coding_type =1; // PCT_INTER;
  642.   pic->QUANT=QP;
  643.       prev_image=curr_image;
  644.   prev_recon=curr_recon;
  645.       curr_image = FillImage(m_pImageData);
  646.   pic->TR+=(((frameskip+(pic->PB?98:0))*orig_frameskip)%256);
  647.   if(frameskip+(pic->PB?98:0)>256)
  648. MessageBox("Warning:frameskip>256");
  650.      streamfile.SeekToEnd();
  651.      if(((frame_no-1)%Pbetween)==0)
  652.  {
  653.    pic->picture_coding_type =0; // PCT_INTRA;
  654.    pic->QUANT=QPI;
  655.        curr_recon = CodeOneIntra(curr_image, QPI, bits, pic);
  656.  }
  657.     else
  658.     CodeOneInter(prev_image,curr_image,prev_recon,curr_recon,QP,frameskip,bits,pic);
  659.     bits->header += alignbits (); /* pictures shall be byte aligned */
  661.    if(ifPsnr)
  662.    {
  663. SeekPsnr(curr_image,curr_recon,352,288,psnrs);
  664. kk.Format("第%d帧的lum峰值信噪比为%6.4fn",frame_no,psnrs[0]);
  665. m_Spsnr=kk;
  666. kk.Format("第%d帧的Cb峰值信噪比为%6.4fn",frame_no,psnrs[1]);
  667. m_Spsnr+=kk;
  668. kk.Format("第%d帧的Cr峰值信噪比为%6.4fn",frame_no,psnrs[2]);
  669. m_Spsnr+=kk;
  670. psnrfile.SeekToEnd();
  671. psnrfile.Write(m_Spsnr,m_Spsnr.GetLength());
  672.    }
  674. //显示进度信息
  675. CString str;
  676. str.Format("%d%% complete", frame_no*100/MaxFrame);
  677. bar.SetText(str);
  678. bar.StepIt();
  679.     PeekAndPump(); //调用函数实现消息的转发
  681.    }//end for frame_no
  683. //  pDoc->SetModifiedFlag(TRUE);
  684.   if(ifPsnr)
  685.   psnrfile.Close();
  687.   streamfile.Close();
  688.   delete prev_image;
  689.   delete prev_recon;
  690.   delete curr_image;
  691.   curr_recon=NULL;
  692. //  delete curr_recon;
  693.   free(bits);
  694.   free(pic);
  696.     long hours,minutes,second;//计算所用的时间
  697.     ElapsedTime = CTime::GetCurrentTime() - StartTime;
  698. bFlag=0;
  699.     hours = ElapsedTime.GetTotalHours();
  700.     minutes = ElapsedTime.GetTotalMinutes();
  701.     second = ElapsedTime.GetTotalSeconds();
  702.     second = second + 60*minutes + 3600*hours;
  704.     csTimeElapse.Format("编码%d帧图像序列,耗时:%d秒！",frame_no-1,second);
  705.     MessageBox(csTimeElapse);
  706. pDoc->UpdateAllViews(NULL);
  708. }
  709. else
  710.     MessageBox("No file is saved.","系统提示",MB_OK);
  711. delete m_pImageData;
  712. }
  714. BOOL CMyView::PeekAndPump()
  715. {
  717. static MSG msg;
  718. while (::PeekMessage(&msg,NULL,0,0,PM_NOREMOVE)) 
  719. {
  720. if (!AfxGetApp()->PumpMessage()) 
  721. {
  722. ::PostQuitMessage(0);
  723. return FALSE;
  724. }
  725. }
  726. return TRUE;
  728. }
  730. void CMyView::OnDecode() 
  731. {
  732. CMyDoc* pDoc = GetDocument();
  733. ASSERT_VALID(pDoc);
  734. CDC *pDC=GetDC();
  736. CFileDialog decfile(TRUE,NULL,NULL,OFN_HIDEREADONLY,"263 Files(*.263)|*.263|");
  737. decfile.m_ofn.lpstrTitle="打开263文件";
  738. if(decfile.DoModal()==IDOK)
  739.    DecfileName=decfile.GetPathName();//获取263文件路径
  740. bFlag=2;
  741. pDoc->UpdateAllViews(NULL);
  742. }
  744. void SeekPsnr(PictImage *curr, PictImage *recon, int width,int height, double psnr[3])
  745. {
  746. int i;
  747. double MSE1=0;
  748. double MSE2=0;
  750.     for(i=0;i<width*height;i++)
  751.     MSE1+=((double)recon->lum[i]-(double)curr->lum[i])
  752.     *((double)recon->lum[i]-(double)curr->lum[i]);
  753. MSE1/=(width*height);
  754. if(MSE1==0)
  755. psnr[0]=0;
  756. else
  757. psnr[0]=10*log10(255*255/MSE1);
  759. MSE1=0;
  760. for(i=0;i<((width*height)>>2);i++)
  761. {
  762. MSE1+=((double)recon->Cb[i]-(double)curr->Cb[i])
  763.     *((double)recon->Cb[i]-(double)curr->Cb[i]);
  764. MSE2+=((double)recon->Cr[i]-(double)curr->Cr[i])
  765.     *((double)recon->Cr[i]-(double)curr->Cr[i]);
  766. }  
  767. MSE1/=(width*height)>>2;
  768. MSE2/=(width*height)>>2;
  769. if(MSE1==0)
  770. psnr[1]=0;
  771. else
  772. psnr[1]=10*log10(255*255/MSE1);
  773. if(MSE2==0)
  774. psnr[2]=0;
  775. else
  776. psnr[2]=10*log10(255*255/MSE2);
  778. }
  782. void CMyView::OnFileOpen() 
  783. {
  784. CDC* pDC;
  785.     pDC=GetDC();
  787. CFile look_result;
  788. CString resultshow="";
  789. CString kk;
  790.     CFileDialog lookresult(TRUE,NULL,NULL,OFN_HIDEREADONLY,"文本 Files(*.txt;*.doc)|*.txt;*.doc|");
  791. lookresult.m_ofn.lpstrTitle="打开编码结果文件";
  792. lookresult.m_ofn.lpstrInitialDir="F:\standard_pictures\MISSUSA_raw";
  793. if(lookresult.DoModal()==IDOK)
  794. {
  795.     resultshow=lookresult.GetPathName();
  796.     if (look_result.Open(resultshow,CFile::modeRead)==NULL)
  797. {
  798.     AfxMessageBox("Can not open the result file!");
  799.     return;
  800. }
  801.     look_result.SeekToBegin(); 
  802. }
  804. }