DX9SubPic.cpp

/* 
 *      Copyright (C) 2003-2005 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 &= 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);

        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);
                }

                *///

                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);

                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);
}

---