winamp/Src/Plugins/Input/in_dshow/dshowrender.cpp

584 lines
16 KiB
C++

//
// This file contains code that supports an alternate method playing dshow data...
// This will have dshow take control of rendering the data (audio or video).
// Used for mms streams
//
#ifdef WINAMPX
#include <windows.h>
#include <math.h>
#include "message.h"
#include "../jnetlib/jnetlib.h"
#include "in2.h"
#include <AtlBase.h>
#include <streams.h>
#include <qedit.h>
#include <qnetwork.h>
#ifdef DEFGUID
#include <initguid.h> // declares DEFINE_GUID to declare an EXTERN_C const.
#endif
#define IPC_GET_IVIDEOOUTPUT 500
// Externs
extern HRESULT AddToRot(IUnknown *pUnkGraph, DWORD *pdwRegister);
extern void RemoveFromRot(DWORD pdwRegister);
extern bool ReportMissingCodec(char *fn);
extern In_Module mod; // the output module (filled in near the bottom of this file)
extern char lastfn[MAX_PATH]; // currently playing file (used for getting info on the current file)
extern char lastfn_status[256];
extern int file_length; // file length, in bytes
extern int decode_pos_ms; // current decoding position, in milliseconds.
extern int paused; // are we paused?
extern volatile int seek_needed; // if != -1, it is the point that the decode thread should seek to, in ms.
extern int m_length;
extern int paused; // are we paused?
extern bool s_using_dsr;
// Static Vars and Defines
class IVideoOutput
{
public:
virtual ~IVideoOutput() { }
virtual int open(int w, int h, int vflip, double aspectratio, unsigned int fmt)=0;
virtual void setcallback(LRESULT (*msgcallback)(void *token, HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam), void *token) { }
virtual void close()=0;
virtual void draw(void *frame)=0;
virtual void drawSubtitle(/*SubsItem **/ void *item) { }
virtual void showStatusMsg(const char *text) { }
virtual int get_latency() { return 0; }
virtual void notifyBufferState(int bufferstate) { } /* 0-255*/
virtual int extended(int param1, int param2, int param3) { return 0; } // Dispatchable, eat this!
};
static IVideoOutput * m_video_output;
static CComPtr<IGraphBuilder> s_pGraphBuilder;
static CComPtr<IMediaEventEx> s_pMediaEventEx;
static CComPtr<IMediaControl> s_pMediaControl;
static CComPtr<IVideoWindow> s_pVideoWindow;
static CComPtr<IBasicAudio> s_pBasicAudio;
static CComPtr<IBasicVideo> s_pBasicVideo;
static CComPtr<IMediaSeeking> s_pMediaSeeking;
static HWND s_dsr_notif_hwnd = NULL;
static HWND s_hVideoWnd = NULL;
static WNDPROC s_OriginalVideoWndProc = NULL;
static RECT s_parentRect;
static DWORD GraphEdit_dwRegister = 0;
static int s_buffering = 0;
static int s_bufferstat = 0;
static BOOL s_bAudioOnly;
static int s_setVolumeOnStart = -1;
// Forward Declarations
LRESULT CALLBACK dsr_TimerWndProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam);
LRESULT CALLBACK dsr_SubclassParentWndProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam);
void dsr_setvolume(int volume);
void dsr_handleNotifyEvents();
void dsr_stop();
// Macros
#define BeginEnumFilters(pFilterGraph, pEnumFilters, pBaseFilter) \
{CComPtr<IEnumFilters> pEnumFilters; \
if(pFilterGraph && SUCCEEDED(pFilterGraph->EnumFilters(&pEnumFilters))) \
{ \
for(CComPtr<IBaseFilter> pBaseFilter; S_OK == pEnumFilters->Next(1, &pBaseFilter, 0); pBaseFilter = NULL) \
{ \
#define EndEnumFilters }}}
static void SendStatus(int status, int arg ) {
mod.fire_winampstatus( status, arg );
}
//------------------------------------------------------------------------------
// Name: CheckVisibility()
// Desc: Set global values for presence of video window.
//------------------------------------------------------------------------------
bool CheckVisibility(void) // returns false for failure
{
HRESULT hr;
if ((!s_pVideoWindow) || (!s_pBasicVideo))
{
s_bAudioOnly = TRUE; // Audio-only files have no video interfaces.
return TRUE;
}
s_bAudioOnly = FALSE;
long lVisible;
hr = s_pVideoWindow->get_Visible(&lVisible); // If this is an audio-only clip, get_Visible() won't work.
if ((FAILED(hr)) && (hr == E_NOINTERFACE))
{
s_bAudioOnly = TRUE;
return TRUE;
}
return !FAILED(hr);
}
static RECT fitMediaToWindow(RECT rectWindow, int mediaWidth, int mediaHeight)
{
RECT retval;
int windowWidth = rectWindow.right - rectWindow.left;
int windowHeight = rectWindow.bottom - rectWindow.top;
if (mediaHeight*windowWidth > mediaWidth*windowHeight)
{
// Gap is on left&right sides
int nOutWidth = windowHeight* mediaWidth / mediaHeight;
int nGap = (windowWidth - nOutWidth)/2;
retval.top = rectWindow.top;
retval.bottom = retval.top + windowHeight;
retval.left = rectWindow.left + nGap;
retval.right = retval.left + nOutWidth;
}
else
{
// Gap is on the top/bottom sides
int nOutHeight = windowWidth* mediaHeight / mediaWidth;
int nGap = (windowHeight - nOutHeight)/2;
retval.left = rectWindow.left;
retval.right = retval.left + windowWidth;
retval.top = rectWindow.top + nGap;
retval.bottom = retval.top + nOutHeight;
}
return retval;
}
void dsr_releaseObjects()
{
if(s_dsr_notif_hwnd)
{
KillTimer(s_dsr_notif_hwnd,112);
DestroyWindow(s_dsr_notif_hwnd);
}
s_dsr_notif_hwnd=NULL;
if ((s_OriginalVideoWndProc) && (s_hVideoWnd))
{
SetWindowLong(s_hVideoWnd, GWL_WNDPROC, (LONG_PTR)s_OriginalVideoWndProc);
s_OriginalVideoWndProc = NULL;
s_hVideoWnd = NULL;
}
s_pMediaEventEx = NULL;
s_pGraphBuilder = NULL;
s_pMediaControl = NULL;
s_pVideoWindow = NULL;
s_pBasicAudio = NULL;
s_pBasicVideo = NULL;
s_pMediaSeeking = NULL;
s_using_dsr = false;
}
// Prefix used for functions here are dsd_ (
int dsr_play(char *fn)
{
HRESULT hr;
hr = s_pGraphBuilder.CoCreateInstance(CLSID_FilterGraph, NULL, CLSCTX_INPROC_SERVER);
if (FAILED(hr)) return 1;
#ifdef _DEBUG
AddToRot(s_pGraphBuilder, &GraphEdit_dwRegister);
#endif
s_pGraphBuilder->QueryInterface(IID_IMediaEvent, (void **)&s_pMediaEventEx);
if(!s_pMediaEventEx)
{
dsr_releaseObjects();
return 1;
}
m_video_output=(IVideoOutput *)SendMessage(mod.hMainWindow,WM_USER,0,IPC_GET_IVIDEOOUTPUT);
if(!m_video_output)
{
dsr_releaseObjects();
return -200; // Can't play file
}
// Create window that will receive filter notifications
static int classReg=0;
if(!classReg)
{
WNDCLASS wc={0,};
wc.style = CS_DBLCLKS;
wc.lpfnWndProc = dsr_TimerWndProc;
wc.hInstance = mod.hDllInstance;
wc.hIcon = NULL;
wc.hCursor = NULL;
wc.lpszClassName = "in_dshowClass2";
if (!RegisterClass(&wc))
{
dsr_releaseObjects();
return 1;
}
classReg=1;
}
s_dsr_notif_hwnd=CreateWindow("in_dshowClass2","dshow_notif2",NULL,0,0,1,1,NULL,NULL,mod.hDllInstance,NULL);
SetTimer(s_dsr_notif_hwnd,112,500,0);
// Build a normal graph (start rendering)
WCHAR f[4096];
MultiByteToWideChar(CP_ACP,0,fn,lstrlen(fn)+1,f,4096);
hr = s_pGraphBuilder->RenderFile(f,NULL);
if (FAILED(hr)) {
dsr_handleNotifyEvents();
dsr_releaseObjects();
if ((hr == CLASS_E_CLASSNOTAVAILABLE) || (hr == VFW_E_UNSUPPORTED_VIDEO) || (hr == VFW_E_NO_DECOMPRESSOR))
{
if (ReportMissingCodec(fn)) // returns true if we sent a message
return -500; // Unsupported format
return -200; // Can't play file
}
return 1;
}
// Check if it's a partial playing of the file (likely video codec missing)
if ((hr == VFW_S_PARTIAL_RENDER) || (hr == VFW_S_VIDEO_NOT_RENDERED))
{
if (!ReportMissingCodec(fn)) // Report the missing codec if we can determine it
mod.fire_winampstatus(WINAMPX_STATUS_MISSING_AVI_CODEC, 0); // If we can't report a null codec missing
}
if (FAILED(s_pGraphBuilder->QueryInterface(IID_IMediaControl, (void **)&s_pMediaControl)) ||
FAILED(s_pGraphBuilder->QueryInterface(IID_IMediaSeeking, (void **)&s_pMediaSeeking)) ||
FAILED(s_pMediaSeeking->SetTimeFormat(&TIME_FORMAT_MEDIA_TIME)))
{
dsr_releaseObjects();
return 1;
}
// Get length of file
int mylength = -1;
LONGLONG length;
if (SUCCEEDED(s_pMediaSeeking->GetDuration(&length)))
{
mylength=(int)(length/10000);
}
m_length = mylength;
// Query for video interfaces, which may not be relevant for audio files
s_pGraphBuilder->QueryInterface(IID_IVideoWindow, (void **)&s_pVideoWindow);
s_pGraphBuilder->QueryInterface(IID_IBasicVideo, (void **)&s_pBasicVideo);
// Query for audio interfaces, which may not be relevant for video-only files
s_pGraphBuilder->QueryInterface(IID_IBasicAudio, (void **)&s_pBasicAudio);
if (s_setVolumeOnStart != -1)
{
dsr_setvolume(s_setVolumeOnStart);
s_setVolumeOnStart = -1;
}
// Is this an audio-only file (no video component)?
CheckVisibility();
if (!s_bAudioOnly)
{
m_video_output->open(1, 1, 0, 1.0f, VIDEO_MAKETYPE('N','O','N','E')); // Dummy Size of 1x1
s_hVideoWnd = (HWND)m_video_output->extended(VIDUSER_GET_VIDEOHWND, 0, 0);
InvalidateRect(s_hVideoWnd, NULL, TRUE);
// Setup the video window
s_pVideoWindow->put_Owner((OAHWND) s_hVideoWnd);
s_pVideoWindow->put_WindowStyle(WS_CHILD | WS_CLIPSIBLINGS);
RECT grc;
GetClientRect(s_hVideoWnd, &grc);
s_parentRect = grc;
if ((s_pBasicVideo) && (s_pVideoWindow))
{
long videoWidth, videoHeight;
if (SUCCEEDED(s_pBasicVideo->GetVideoSize(&videoWidth, &videoHeight)))
{
RECT r = fitMediaToWindow(grc, videoWidth, videoHeight);
s_pVideoWindow->SetWindowPosition(r.left, r.top, r.right-r.left, r.bottom-r.top);
}
}
// Intercept resize messages
s_OriginalVideoWndProc = (WNDPROC) SetWindowLong(s_hVideoWnd, GWL_WNDPROC, (LONG_PTR)dsr_SubclassParentWndProc);
}
// Run the graph to play the media file
hr = s_pMediaControl->Run();
if (FAILED(hr))
{
dsr_stop();
return 1;
}
return 0;
}
// stop playing.
void dsr_stop()
{
if (s_pMediaControl)
s_pMediaControl->Stop();
if (s_pVideoWindow)
{
s_pVideoWindow->put_Visible(OAFALSE);
s_pVideoWindow->put_Owner(NULL);
}
if (m_video_output)
m_video_output->close();
mod.outMod->Close();
mod.SAVSADeInit();
dsr_releaseObjects();
m_length=-1;
lastfn[0]=0;
}
// Standard Pause Routines
void dsr_pause() {
paused=1;
if (s_pMediaControl)
{
s_pMediaControl->Pause();
}
}
void dsr_unpause() {
paused=0;
if (s_pMediaControl)
{
s_pMediaControl->Run();
}
}
int dsr_getoutputtime() {
if (s_bufferstat)
return s_bufferstat;
if (s_pMediaSeeking)
{
LONGLONG pos;
if (SUCCEEDED(s_pMediaSeeking->SetTimeFormat(&TIME_FORMAT_MEDIA_TIME)) &&
SUCCEEDED(s_pMediaSeeking->GetCurrentPosition(&pos)))
{
return (int)(pos/10000);
}
}
return 0;
}
void dsr_setoutputtime(int time_in_ms) {
if(s_pMediaSeeking) {
DWORD dwCaps = AM_SEEKING_CanSeekAbsolute;
if (s_pMediaSeeking->CheckCapabilities(&dwCaps) == S_OK)
{
int oldpause=paused;
if(oldpause) dsr_unpause();
LONGLONG l=((LONGLONG)time_in_ms)*10000;
s_pMediaSeeking->SetTimeFormat(&TIME_FORMAT_MEDIA_TIME);
s_pMediaSeeking->SetPositions(&l,AM_SEEKING_AbsolutePositioning,NULL,AM_SEEKING_NoPositioning);
mod.outMod->Flush(time_in_ms);
if(oldpause) dsr_pause();
}
}
}
void dsr_setvolume(int volume)
{
if (s_pBasicAudio)
{
volume = min(volume, 255);
volume = max(volume, 0);
s_pBasicAudio->put_Volume(log(volume+1)/log(256)*10000-10000); // Map (0,255) to (-10000,0) log scale (as ActiveX is in DB's)
}
else
{
s_setVolumeOnStart = volume;
}
}
LRESULT CALLBACK dsr_TimerWndProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
if(uMsg==WM_TIMER && wParam==112 && s_dsr_notif_hwnd)
{
dsr_handleNotifyEvents();
if(s_buffering)
{
BeginEnumFilters(s_pGraphBuilder, pEF, pBF)
{
if(CComQIPtr<IAMNetworkStatus, &IID_IAMNetworkStatus> pAMNS = pBF)
{
long BufferingProgress = 0;
if(SUCCEEDED(pAMNS->get_BufferingProgress(&BufferingProgress)) && BufferingProgress > 0)
{
wsprintf(lastfn_status,"Buffering (%ld%%)",BufferingProgress);
if(m_video_output) m_video_output->extended(VIDUSER_SET_INFOSTRING,(int)&lastfn_status,0);
int bpos=BufferingProgress;
int csa = mod.SAGetMode();
char tempdata[75*2]={0,};
int x;
if (csa&1)
{
for (x = 0; x < bpos*75/100; x ++)
{
tempdata[x]=x*16/75;
}
}
if (csa&2)
{
int offs=(csa&1) ? 75 : 0;
x=0;
while (x < bpos*75/100)
{
tempdata[offs + x++]=-6+x*14/75;
}
while (x < 75)
{
tempdata[offs + x++]=0;
}
}
if (csa==4)
{
tempdata[0]=tempdata[1]=(bpos*127/100);
}
if (csa) mod.SAAdd(tempdata,++s_bufferstat,(csa==3)?0x80000003:csa);
PostMessage(mod.hMainWindow,WM_USER,0,243);
SendStatus(WINAMPX_STATUS_PREBUFFERING_PCT, 255*BufferingProgress/100);
break;
}
}
}
EndEnumFilters
}
}
return (DefWindowProc(hwnd, uMsg, wParam, lParam));
}
void dsr_handleNotifyEvents()
{
// char s[256];
for (;;) {
long evCode, param1, param2;
HRESULT h = s_pMediaEventEx->GetEvent(&evCode, &param1, &param2, 0);
if (FAILED(h)) break;
switch(evCode) {
case EC_BUFFERING_DATA:
{
// sprintf(s, "Handling Event: EC_BUFFERING_DATA: %d\n", param1);
// OutputDebugString(s);
s_buffering=param1;
if(!s_buffering)
{
lastfn_status[0]=0;
s_bufferstat=0;
PostMessage(mod.hMainWindow,WM_USER,0,243);
break;
}
}
break;
case EC_COMPLETE:
{
// OutputDebugString("Handling Event: EC_COMPLETE\n");
PostMessage(mod.hMainWindow,WM_WA_MPEG_EOF,0,0);
}
default:
{
// sprintf(s, "Handling Event: 0x%x : param1=0x%x; param2 = 0x%x\n", evCode, param1, param2);
// OutputDebugString(s);
}
}
s_pMediaEventEx->FreeEventParams(evCode, param1, param2);
}
}
// Subclass of the Window containing the code. (Used to check if the parent got resized)
LRESULT CALLBACK dsr_SubclassParentWndProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
LRESULT retval = CallWindowProc(s_OriginalVideoWndProc, hwnd, uMsg, wParam, lParam);
if ((uMsg == WM_SIZE) && (s_hVideoWnd))
{
RECT grc;
GetClientRect(s_hVideoWnd, &grc);
if (!EqualRect(&grc, &s_parentRect))
{
s_parentRect = grc;
if ((s_pBasicVideo) && (s_pVideoWindow))
{
long videoWidth, videoHeight;
if (SUCCEEDED(s_pBasicVideo->GetVideoSize(&videoWidth, &videoHeight)))
{
RECT r = fitMediaToWindow(grc, videoWidth, videoHeight);
s_pVideoWindow->SetWindowPosition(r.left, r.top, r.right-r.left, r.bottom-r.top);
}
}
}
}
if ((uMsg == WM_PARENTNOTIFY) && (LOWORD(wParam)==WM_RBUTTONDOWN))
{
SendStatus(WINAMPX_STATUS_VIDEO_RIGHT_CLICK, lParam);
}
if ((uMsg == WM_PARENTNOTIFY) && (LOWORD(wParam)==WM_LBUTTONDOWN))
{
static DWORD dwLastTime = 0;
int dwTimeNow = GetTickCount();
if (((dwTimeNow - dwLastTime) < GetDoubleClickTime()) && (dwLastTime != 0))
{
PostMessage(hwnd, WM_USER_DSR_LBUTTONDBLCLK, 0, 0); // Notify the video window that we double clicked
dwLastTime = 0;
}
else
dwLastTime = dwTimeNow;
}
if ((uMsg == WM_USER_DSR_FULLSCREEN) && (s_pVideoWindow))
{
s_pVideoWindow->HideCursor(wParam ? OATRUE : OAFALSE);
}
return retval;
}
#endif // WINAMPX