forked from mia/Aegisub
362 lines
8.9 KiB
C++
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);
|
|
}
|