Aegisub/vsfilter/subpic/DX9SubPic.cpp

362 lines
8.9 KiB
C++

/*
* Copyright (C) 2003-2006 Gabest
* http://www.gabest.org
*
* This Program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2, or (at your option)
* any later version.
*
* This Program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with GNU Make; see the file COPYING. If not, write to
* the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
* http://www.gnu.org/copyleft/gpl.html
*
*/
#include "stdafx.h"
#include <d3d9.h>
#include <Vmr9.h>
#include "DX9SubPic.h"
//
// CDX9SubPic
//
CDX9SubPic::CDX9SubPic(IDirect3DSurface9* pSurface)
: m_pSurface(pSurface)
{
D3DSURFACE_DESC d3dsd;
ZeroMemory(&d3dsd, sizeof(d3dsd));
if(SUCCEEDED(m_pSurface->GetDesc(&d3dsd)))
{
m_maxsize.SetSize(d3dsd.Width, d3dsd.Height);
m_rcDirty.SetRect(0, 0, d3dsd.Width, d3dsd.Height);
}
}
// ISubPic
STDMETHODIMP_(void*) CDX9SubPic::GetObject()
{
CComPtr<IDirect3DTexture9> pTexture;
if(SUCCEEDED(m_pSurface->GetContainer(IID_IDirect3DTexture9, (void**)&pTexture)))
return (void*)(IDirect3DTexture9*)pTexture;
return NULL;
}
STDMETHODIMP CDX9SubPic::GetDesc(SubPicDesc& spd)
{
D3DSURFACE_DESC d3dsd;
ZeroMemory(&d3dsd, sizeof(d3dsd));
if(FAILED(m_pSurface->GetDesc(&d3dsd)))
return E_FAIL;
spd.type = 0;
spd.w = m_size.cx;
spd.h = m_size.cy;
spd.bpp =
d3dsd.Format == D3DFMT_A8R8G8B8 ? 32 :
d3dsd.Format == D3DFMT_A4R4G4B4 ? 16 : 0;
spd.pitch = 0;
spd.bits = NULL;
spd.vidrect = m_vidrect;
return S_OK;
}
STDMETHODIMP CDX9SubPic::CopyTo(ISubPic* pSubPic)
{
HRESULT hr;
if(FAILED(hr = __super::CopyTo(pSubPic)))
return hr;
if(m_rcDirty.IsRectEmpty())
return S_FALSE;
CComPtr<IDirect3DDevice9> pD3DDev;
if(!m_pSurface || FAILED(m_pSurface->GetDevice(&pD3DDev)) || !pD3DDev)
return E_FAIL;
hr = pD3DDev->UpdateTexture((IDirect3DTexture9*)GetObject(), (IDirect3DTexture9*)pSubPic->GetObject());
return SUCCEEDED(hr) ? S_OK : E_FAIL;
}
STDMETHODIMP CDX9SubPic::ClearDirtyRect(DWORD color)
{
if(m_rcDirty.IsRectEmpty())
return S_FALSE;
CComPtr<IDirect3DDevice9> pD3DDev;
if(!m_pSurface || FAILED(m_pSurface->GetDevice(&pD3DDev)) || !pD3DDev)
return E_FAIL;
SubPicDesc spd;
if(SUCCEEDED(Lock(spd)))
{
int h = m_rcDirty.Height();
BYTE* ptr = (BYTE*)spd.bits + spd.pitch*m_rcDirty.top + (m_rcDirty.left*spd.bpp>>3);
if(spd.bpp == 16)
{
while(h-- > 0)
{
WORD* start = (WORD*)ptr;
WORD* end = start + m_rcDirty.Width();
while(start < end) *start++ = (WORD)color;
ptr += spd.pitch;
}
}
else if(spd.bpp == 32)
{
while(h-- > 0)
{
DWORD* start = (DWORD*)ptr;
DWORD* end = start + m_rcDirty.Width();
while(start < end) *start++ = color;
ptr += spd.pitch;
}
}
/*
DWORD* ptr = (DWORD*)bm.bits;
DWORD* end = ptr + bm.h*bm.wBytes/4;
while(ptr < end) *ptr++ = color;
*/
Unlock(NULL);
}
// HRESULT hr = pD3DDev->ColorFill(m_pSurface, m_rcDirty, color);
m_rcDirty.SetRectEmpty();
return S_OK;
}
STDMETHODIMP CDX9SubPic::Lock(SubPicDesc& spd)
{
D3DSURFACE_DESC d3dsd;
ZeroMemory(&d3dsd, sizeof(d3dsd));
if(FAILED(m_pSurface->GetDesc(&d3dsd)))
return E_FAIL;
D3DLOCKED_RECT LockedRect;
ZeroMemory(&LockedRect, sizeof(LockedRect));
if(FAILED(m_pSurface->LockRect(&LockedRect, NULL, 0)))
return E_FAIL;
spd.type = 0;
spd.w = m_size.cx;
spd.h = m_size.cy;
spd.bpp =
d3dsd.Format == D3DFMT_A8R8G8B8 ? 32 :
d3dsd.Format == D3DFMT_A4R4G4B4 ? 16 : 0;
spd.pitch = LockedRect.Pitch;
spd.bits = LockedRect.pBits;
spd.vidrect = m_vidrect;
return S_OK;
}
STDMETHODIMP CDX9SubPic::Unlock(RECT* pDirtyRect)
{
m_pSurface->UnlockRect();
if(pDirtyRect)
{
m_rcDirty = *pDirtyRect;
m_rcDirty.InflateRect(1, 1);
m_rcDirty.left &= ~127;
m_rcDirty.top &= ~63;
m_rcDirty.right = (m_rcDirty.right + 127) & ~127;
m_rcDirty.bottom = (m_rcDirty.bottom + 63) & ~63;
m_rcDirty &= CRect(CPoint(0, 0), m_size);
}
else
{
m_rcDirty = CRect(CPoint(0, 0), m_size);
}
return S_OK;
}
STDMETHODIMP CDX9SubPic::AlphaBlt(RECT* pSrc, RECT* pDst, SubPicDesc* pTarget)
{
ASSERT(pTarget == NULL);
if(!pSrc || !pDst)
return E_POINTER;
CRect src(*pSrc), dst(*pDst);
CComPtr<IDirect3DDevice9> pD3DDev;
CComPtr<IDirect3DTexture9> pTexture = (IDirect3DTexture9*)GetObject();
if(!pTexture || FAILED(pTexture->GetDevice(&pD3DDev)) || !pD3DDev)
return E_NOINTERFACE;
HRESULT hr;
do
{
D3DSURFACE_DESC d3dsd;
ZeroMemory(&d3dsd, sizeof(d3dsd));
if(FAILED(pTexture->GetLevelDesc(0, &d3dsd)) /*|| d3dsd.Type != D3DRTYPE_TEXTURE*/)
break;
float w = (float)d3dsd.Width;
float h = (float)d3dsd.Height;
struct
{
float x, y, z, rhw;
float tu, tv;
}
pVertices[] =
{
{(float)dst.left, (float)dst.top, 0.5f, 2.0f, (float)src.left / w, (float)src.top / h},
{(float)dst.right, (float)dst.top, 0.5f, 2.0f, (float)src.right / w, (float)src.top / h},
{(float)dst.left, (float)dst.bottom, 0.5f, 2.0f, (float)src.left / w, (float)src.bottom / h},
{(float)dst.right, (float)dst.bottom, 0.5f, 2.0f, (float)src.right / w, (float)src.bottom / h},
};
/*
for(int i = 0; i < countof(pVertices); i++)
{
pVertices[i].x -= 0.5;
pVertices[i].y -= 0.5;
}
*/
hr = pD3DDev->SetTexture(0, pTexture);
DWORD abe, sb, db;
hr = pD3DDev->GetRenderState(D3DRS_ALPHABLENDENABLE, &abe);
hr = pD3DDev->GetRenderState(D3DRS_SRCBLEND, &sb);
hr = pD3DDev->GetRenderState(D3DRS_DESTBLEND, &db);
hr = pD3DDev->SetRenderState(D3DRS_CULLMODE, D3DCULL_NONE);
hr = pD3DDev->SetRenderState(D3DRS_LIGHTING, FALSE);
hr = pD3DDev->SetRenderState(D3DRS_ZENABLE, FALSE);
hr = pD3DDev->SetRenderState(D3DRS_ALPHABLENDENABLE, TRUE);
hr = pD3DDev->SetRenderState(D3DRS_SRCBLEND, D3DBLEND_ONE); // pre-multiplied src and ...
hr = pD3DDev->SetRenderState(D3DRS_DESTBLEND, D3DBLEND_SRCALPHA); // ... inverse alpha channel for dst
hr = pD3DDev->SetTextureStageState(0, D3DTSS_COLOROP, D3DTOP_SELECTARG1);
hr = pD3DDev->SetTextureStageState(0, D3DTSS_COLORARG1, D3DTA_TEXTURE);
hr = pD3DDev->SetTextureStageState(0, D3DTSS_ALPHAARG1, D3DTA_TEXTURE);
hr = pD3DDev->SetSamplerState(0, D3DSAMP_MAGFILTER, D3DTEXF_LINEAR);
hr = pD3DDev->SetSamplerState(0, D3DSAMP_MINFILTER, D3DTEXF_LINEAR);
hr = pD3DDev->SetSamplerState(0, D3DSAMP_MIPFILTER, D3DTEXF_LINEAR);
hr = pD3DDev->SetSamplerState(0, D3DSAMP_ADDRESSU, D3DTADDRESS_CLAMP);
hr = pD3DDev->SetSamplerState(0, D3DSAMP_ADDRESSV, D3DTADDRESS_CLAMP);
/*//
D3DCAPS9 d3dcaps9;
hr = pD3DDev->GetDeviceCaps(&d3dcaps9);
if(d3dcaps9.AlphaCmpCaps & D3DPCMPCAPS_LESS)
{
hr = pD3DDev->SetRenderState(D3DRS_ALPHAREF, (DWORD)0x000000FE);
hr = pD3DDev->SetRenderState(D3DRS_ALPHATESTENABLE, TRUE);
hr = pD3DDev->SetRenderState(D3DRS_ALPHAFUNC, D3DPCMPCAPS_LESS);
}
*///
hr = pD3DDev->SetPixelShader(NULL);
if(FAILED(hr = pD3DDev->BeginScene()))
break;
hr = pD3DDev->SetFVF(D3DFVF_XYZRHW | D3DFVF_TEX1);
hr = pD3DDev->DrawPrimitiveUP(D3DPT_TRIANGLESTRIP, 2, pVertices, sizeof(pVertices[0]));
hr = pD3DDev->EndScene();
//
pD3DDev->SetTexture(0, NULL);
pD3DDev->SetRenderState(D3DRS_ALPHABLENDENABLE, abe);
pD3DDev->SetRenderState(D3DRS_SRCBLEND, sb);
pD3DDev->SetRenderState(D3DRS_DESTBLEND, db);
return S_OK;
}
while(0);
return E_FAIL;
}
//
// CDX9SubPicAllocator
//
CDX9SubPicAllocator::CDX9SubPicAllocator(IDirect3DDevice9* pD3DDev, SIZE maxsize, bool fPow2Textures)
: ISubPicAllocatorImpl(maxsize, true, fPow2Textures)
, m_pD3DDev(pD3DDev)
, m_maxsize(maxsize)
{
m_pD3DDev = pD3DDev;
m_maxsize = maxsize;
}
// ISubPicAllocator
STDMETHODIMP CDX9SubPicAllocator::ChangeDevice(IUnknown* pDev)
{
CComQIPtr<IDirect3DDevice9> pD3DDev = pDev;
if(!pD3DDev) return E_NOINTERFACE;
CAutoLock cAutoLock(this);
m_pD3DDev = pD3DDev;
return __super::ChangeDevice(pDev);
}
// ISubPicAllocatorImpl
bool CDX9SubPicAllocator::Alloc(bool fStatic, ISubPic** ppSubPic)
{
if(!ppSubPic)
return(false);
CAutoLock cAutoLock(this);
*ppSubPic = NULL;
CComPtr<IDirect3DSurface9> pSurface;
int Width = m_maxsize.cx;
int Height = m_maxsize.cy;
if(m_fPow2Textures)
{
Width = Height = 1;
while(Width < m_maxsize.cx) Width <<= 1;
while(Height < m_maxsize.cy) Height <<= 1;
}
CComPtr<IDirect3DTexture9> pTexture;
if(FAILED(m_pD3DDev->CreateTexture(Width, Height, 1, 0, D3DFMT_A8R8G8B8, fStatic?D3DPOOL_SYSTEMMEM:D3DPOOL_DEFAULT, &pTexture, NULL)))
return(false);
if(FAILED(pTexture->GetSurfaceLevel(0, &pSurface)))
return(false);
if(!(*ppSubPic = new CDX9SubPic(pSurface)))
return(false);
(*ppSubPic)->AddRef();
return(true);
}