TIandSI 3.0

/** * TIandSIpro * (TIandSI Professional) * * 雷霄骅 Lei Xiaohua * leixiaohua1020@126.com * 中国传媒大学/数字电视技术 * Communication University of China / Digital TV Technology * http://blog.csdn.net/leixiaohua1020 * * 本工程可以计算一个压缩视频的时间信息TI（Temporal perceptual Information， * 也可以称时间复杂度）和空间信息SI（Spatial perceptual Information，也可以 * 称空间复杂度）。计算方法出自标准：ITU-R BT.1788 * * Professional: 支持压缩码流（而不是像素数据比如YUV，RGB）作为输入 * * This software can calculate a video bitstream's TI(Temporal perceptual Information) * and SI(Spatial perceptual Information) according to ITU-R BT.1788. * * Professional: Support bitstreams (not raw data such as YUV, RGB, etc.) as Input. * */ #include "stdafx.h" #include "TIandSI.h" #include "TIandSIDlg.h" #include "afxdialogex.h" #include <atlconv.h> #ifdef _DEBUG #define new DEBUG_NEW #endif // 用于应用程序“关于”菜单项的 CAboutDlg 对话框 class CAboutDlg : public CDialogEx { public: CAboutDlg(); // 对话框数据 enum { IDD = IDD_ABOUTBOX }; protected: virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV 支持 // 实现 protected: DECLARE_MESSAGE_MAP() public: // afx_msg void OnDropFiles(HDROP hDropInfo); }; CAboutDlg::CAboutDlg() : CDialogEx(CAboutDlg::IDD) { } void CAboutDlg::DoDataExchange(CDataExchange* pDX) { CDialogEx::DoDataExchange(pDX); } BEGIN_MESSAGE_MAP(CAboutDlg, CDialogEx) // ON_WM_DROPFILES() END_MESSAGE_MAP() // CTIandSIDlg 对话框 CTIandSIDlg::CTIandSIDlg(CWnd* pParent /*=NULL*/) : CDialogEx(CTIandSIDlg::IDD, pParent) { m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME); } void CTIandSIDlg::DoDataExchange(CDataExchange* pDX) { CDialogEx::DoDataExchange(pDX); DDX_Control(pDX, IDC_URLLIST, m_urllist); DDX_Control(pDX, IDC_INPUTURL, m_inputurl); DDX_Control(pDX, IDC_PROGRESS_ALL, m_progressall); DDX_Control(pDX, IDC_PROGRESS_ALL_TEXT, m_progressalltext); DDX_Control(pDX, IDC_PROGRESS_CUR, m_progresscur); DDX_Control(pDX, IDC_PROGRESS_CUR_TEXT, m_progresscurtext); DDX_Control(pDX, IDC_OUTPUT, m_output); DDX_Control(pDX, IDC_DRAW, m_draw); DDX_Control(pDX, IDC_CURVAL, m_curval); } BEGIN_MESSAGE_MAP(CTIandSIDlg, CDialogEx) ON_WM_SYSCOMMAND() ON_WM_PAINT() ON_WM_QUERYDRAGICON() ON_BN_CLICKED(IDOK, &CTIandSIDlg::OnBnClickedOk) ON_BN_CLICKED(IDC_URLLIST_ADD, &CTIandSIDlg::OnBnClickedUrllistAdd) ON_BN_CLICKED(IDC_URLLIST_DELETE, &CTIandSIDlg::OnBnClickedUrllistDelete) ON_BN_CLICKED(IDC_URLLIST_DELETEALL, &CTIandSIDlg::OnBnClickedUrllistDeleteall) ON_BN_CLICKED(IDABOUT, &CTIandSIDlg::OnBnClickedAbout) // ON_BN_CLICKED(IDC_CHART_EDITOR, &CTIandSIDlg::OnBnClickedChartEditor) //ON_BN_CLICKED(IDC_RADIO_DRAW, &CTIandSIDlg::OnClickedRadioDraw) ON_BN_CLICKED(IDRESULT, &CTIandSIDlg::OnBnClickedResult) ON_WM_DROPFILES() ON_BN_CLICKED(IDPAUSE, &CTIandSIDlg::OnBnClickedPause) ON_BN_CLICKED(IDSTOP, &CTIandSIDlg::OnBnClickedStop) ON_COMMAND(IDWEBSITE, &CTIandSIDlg::OnWebsite) ON_COMMAND(ID_URLLIST_QUICKADD, &CTIandSIDlg::OnUrllistQuickadd) END_MESSAGE_MAP() // CTIandSIDlg 消息处理程序 BOOL CTIandSIDlg::OnInitDialog() { CDialogEx::OnInitDialog(); // 将“关于...”菜单项添加到系统菜单中。 // IDM_ABOUTBOX 必须在系统命令范围内。 ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX); ASSERT(IDM_ABOUTBOX < 0xF000); CMenu* pSysMenu = GetSystemMenu(FALSE); if (pSysMenu != NULL) { BOOL bNameValid; CString strAboutMenu; bNameValid = strAboutMenu.LoadString(IDS_ABOUTBOX); ASSERT(bNameValid); if (!strAboutMenu.IsEmpty()) { pSysMenu->AppendMenu(MF_SEPARATOR); pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu); } } // 设置此对话框的图标。当应用程序主窗口不是对话框时，框架将自动 // 执行此操作 SetIcon(m_hIcon, TRUE); // 设置大图标 SetIcon(m_hIcon, FALSE); // 设置小图标 //--------------- m_output.SetCheck(TRUE); UpdateData(FALSE); //初始化 m_progressalltext.SetWindowText(L"0%"); m_progresscurtext.SetWindowText(L"0%"); m_progressall.SetRange(0,100); m_progresscur.SetRange(0,100); //SDL========================== sdlparam.graphically==true; //SDL_putenv()放在前面 char variable[256]; CWnd* pWnd = GetDlgItem(IDC_SCREEN); //获取图片控件的窗口指针 sprintf(variable,"SDL_WINDOWID=0x%1x",pWnd->GetSafeHwnd()); // 格式化字符串 SDL_putenv(variable); if(SDL_Init(SDL_INIT_VIDEO)) { AfxMessageBox(L"Could not initialize SDL"); return 0; } CRect screenrect; GetDlgItem(IDC_SCREEN)->GetWindowRect(screenrect); sdlparam.screen = SDL_SetVideoMode(screenrect.Width(), screenrect.Height(), 0, 0); if(!sdlparam.screen) { AfxMessageBox(L"SDL: could not set video mode"); return 0; } sdlparam.rect.x=0; sdlparam.rect.y=0; sdlparam.rect.w=screenrect.Width(); sdlparam.rect.h=screenrect.Height(); //------------- resultdlg=new ResultDlg; resultdlg->Create(IDD_RESULT_DIALOG); m_draw.InsertString(0,L"SI"); m_draw.InsertString(1,L"TI"); m_draw.InsertString(2,L"不显示"); m_draw.SetCurSel(0); SetState(SYSTEM_PREPARE); //============================= return TRUE; // 除非将焦点设置到控件，否则返回 TRUE } void CTIandSIDlg::OnSysCommand(UINT nID, LPARAM lParam) { if ((nID & 0xFFF0) == IDM_ABOUTBOX) { CAboutDlg dlgAbout; dlgAbout.DoModal(); } else { CDialogEx::OnSysCommand(nID, lParam); } } // 如果向对话框添加最小化按钮，则需要下面的代码 // 来绘制该图标。对于使用文档/视图模型的 MFC 应用程序， // 这将由框架自动完成。 void CTIandSIDlg::OnPaint() { if (IsIconic()) { CPaintDC dc(this); // 用于绘制的设备上下文 SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0); // 使图标在工作区矩形中居中 int cxIcon = GetSystemMetrics(SM_CXICON); int cyIcon = GetSystemMetrics(SM_CYICON); CRect rect; GetClientRect(&rect); int x = (rect.Width() - cxIcon + 1) / 2; int y = (rect.Height() - cyIcon + 1) / 2; // 绘制图标 dc.DrawIcon(x, y, m_hIcon); } else { CDialogEx::OnPaint(); } } //当用户拖动最小化窗口时系统调用此函数取得光标 //显示。 HCURSOR CTIandSIDlg::OnQueryDragIcon() { return static_cast<HCURSOR>(m_hIcon); } //解码视频并且计算 int CTIandSIDlg::TIandSI(CString url,LPVOID lparam) { CTIandSIDlg *dlg=(CTIandSIDlg *)lparam;; //---------------- AVFormatContext *pFormatCtx; int i, video_stream,audio_stream; AVCodecContext *pCodecCtx,*pCodecCtx_au; AVCodec *pCodec,*pCodec_au; char in_url[500]={0}; char out_url[500]={0}; USES_CONVERSION; sprintf(in_url,"%s",W2A(url)); char *suffix=(char *)strrchr(in_url, '.');// *suffix='\0'; strcpy(out_url,in_url); *suffix='.'; sprintf(out_url,"%s_tisi.csv",out_url); av_register_all(); pFormatCtx = avformat_alloc_context(); if(avformat_open_input(&pFormatCtx,in_url,NULL,NULL)!=0){ AfxMessageBox(L"Couldn't open file.\n"); return FALSE; } if(av_find_stream_info(pFormatCtx)<0) { AfxMessageBox(L"Couldn't find stream information.\n"); return FALSE; } video_stream=-1; audio_stream=-1; for(i=0; i<pFormatCtx->nb_streams; i++) { if(pFormatCtx->streams[i]->codec->codec_type==AVMEDIA_TYPE_VIDEO){ video_stream=i; }if(pFormatCtx->streams[i]->codec->codec_type==AVMEDIA_TYPE_AUDIO){ audio_stream=i; } } if(video_stream==-1) { AfxMessageBox(L"Didn't find a video stream.\n"); return FALSE; } if(video_stream!=-1){ pCodecCtx=pFormatCtx->streams[video_stream]->codec; pCodec=avcodec_find_decoder(pCodecCtx->codec_id); if(pCodec==NULL) { AfxMessageBox(L"Codec not found.\n"); return FALSE; } if(avcodec_open2(pCodecCtx, pCodec,NULL)<0) { AfxMessageBox(L"Could not open codec.\n"); return FALSE; } //vtype=VE_TIL_SIL; std::vector<float> silist; std::vector<float> tilist; AVFrame *pFrame,*pFrameYUV; pFrame=avcodec_alloc_frame(); pFrameYUV=avcodec_alloc_frame(); uint8_t *out_buffer; out_buffer=(uint8_t *)av_malloc(avpicture_get_size(PIX_FMT_YUV420P, pCodecCtx->width, pCodecCtx->height)); avpicture_fill((AVPicture *)pFrameYUV, out_buffer, PIX_FMT_YUV420P, pCodecCtx->width, pCodecCtx->h