src/subpic/MemSubPic.h

   1 /*
   2  *      Copyright (C) 2003-2006 Gabest
   3  *      http://www.gabest.org
   4  *
   5  *  This Program is free software; you can redistribute it and/or modify
   6  *  it under the terms of the GNU General Public License as published by
   7  *  the Free Software Foundation; either version 2, or (at your option)
   8  *  any later version.
   9  *
  10  *  This Program is distributed in the hope that it will be useful,
  11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13  *  GNU General Public License for more details.
  14  *
  15  *  You should have received a copy of the GNU General Public License
  16  *  along with GNU Make; see the file COPYING.  If not, write to
  17  *  the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
  18  *  http://www.gnu.org/copyleft/gpl.html
  19  *
  20  */
  21
  22 #pragma once
  23
  24 #include "SubPicImpl.h"
  25
  26 // CMemSubPic
  27
  28 typedef const UINT8 CUINT8, *PCUINT8;
  29
  30 class CMemSubPic : public CSubPicExImpl
  31 {
  32 public:
  33     static void AlphaBltYv12Luma(byte* dst, int dst_pitch, int w, int h, const byte* sub, const byte* alpha, int sub_pitch);
  34     static void AlphaBltYv12LumaC(byte* dst, int dst_pitch, int w, int h, const byte* sub, const byte* alpha, int sub_pitch);
  35
  36     static void AlphaBltYv12Chroma(byte* dst, int dst_pitch, int w, int chroma_h, const byte* sub_chroma,
  37         const byte* alpha, int sub_pitch);
  38     static void AlphaBltYv12ChromaC(byte* dst, int dst_pitch, int w, int chroma_h, const byte* sub_chroma,
  39         const byte* alpha, int sub_pitch);
  40
  41     static HRESULT AlphaBltAnv12_P010(const BYTE* src_a, const BYTE* src_y, const BYTE* src_uv, int src_pitch,
  42         BYTE* dst_y, BYTE* dst_uv, int dst_pitch,
  43         int w, int h);
  44     static HRESULT AlphaBltAnv12_P010_C(const BYTE* src_a, const BYTE* src_y, const BYTE* src_uv, int src_pitch,
  45         BYTE* dst_y, BYTE* dst_uv, int dst_pitch,
  46         int w, int h);
  47     static HRESULT AlphaBltAnv12_Nv12(const BYTE* src_a, const BYTE* src_y, const BYTE* src_uv, int src_pitch,
  48         BYTE* dst_y, BYTE* dst_uv, int dst_pitch,
  49         int w, int h);
  50     static HRESULT AlphaBltAnv12_Nv12_C(const BYTE* src_a, const BYTE* src_y, const BYTE* src_uv, int src_pitch,
  51         BYTE* dst_y, BYTE* dst_uv, int dst_pitch,
  52         int w, int h);
  53
  54     static void AlphaBlt_YUY2(int w, int h, BYTE* d, int dstpitch, PCUINT8 s, int srcpitch);
  55
  56     static void SubsampleAndInterlace(BYTE* dst, const BYTE* u, const BYTE* v, int h, int w, int pitch);
  57     static void SubsampleAndInterlaceC(BYTE* dst, const BYTE* u, const BYTE* v, int h, int w, int pitch);
  58 public:
  59     CMemSubPic(SubPicDesc& spd, int alpha_blt_dst_type);
  60     virtual ~CMemSubPic();
  61
  62     // ISubPic
  63     STDMETHODIMP GetDesc(SubPicDesc& spd) const;
  64     STDMETHODIMP ClearDirtyRect(DWORD color);
  65     STDMETHODIMP Lock(SubPicDesc& spd);
  66     STDMETHODIMP AlphaBlt(const RECT* pSrc, const RECT* pDst, SubPicDesc* pTarget);
  67
  68     // ISubPicEx
  69     STDMETHODIMP CopyTo(ISubPicEx* pSubPic);
  70     STDMETHODIMP Unlock(CAtlList<CRect>* dirtyRectList);
  71     STDMETHODIMP SetDirtyRectEx(CAtlList<CRect>* dirtyRectList);
  72
  73 protected:
  74     int m_type;
  75     bool converted;
  76
  77     int m_alpha_blt_dst_type;
  78
  79         SubPicDesc m_spd;
  80
  81 protected:
  82         STDMETHODIMP_(void*) GetObject() const; // returns SubPicDesc*
  83
  84     HRESULT AlphaBltAxyuAxyv_P010(const RECT* pSrc, const RECT* pDst, SubPicDesc* pTarget);
  85     HRESULT AlphaBltAxyuAxyv_Yv12(const RECT* pSrc, const RECT* pDst, SubPicDesc* pTarget);
  86     HRESULT AlphaBltAxyuAxyv_Nv12(const RECT* pSrc, const RECT* pDst, SubPicDesc* pTarget);
  87     HRESULT AlphaBltAnv12_P010(const RECT* pSrc, const RECT* pDst, SubPicDesc* pTarget);
  88     HRESULT AlphaBltAnv12_Nv12(const RECT* pSrc, const RECT* pDst, SubPicDesc* pTarget);
  89     HRESULT AlphaBltOther(const RECT* pSrc, const RECT* pDst, SubPicDesc* pTarget);
  90
  91     HRESULT UnlockRGBA_YUV(CAtlList<CRect>* dirtyRectList);
  92     HRESULT UnlockOther(CAtlList<CRect>* dirtyRectList);
  93
  94     void SubsampleAndInterlace( const CRect& cRect, bool u_first );
  95 };
  96
  97 // CMemSubPicAllocator
  98
  99 class CMemSubPicAllocator : public CSubPicExAllocatorImpl
 100 {
 101         int m_type;
 102     int m_alpha_blt_dst_type;
 103         CSize m_maxsize;
 104
 105         bool AllocEx(bool fStatic, ISubPicEx** ppSubPic);
 106
 107 public:
 108         CMemSubPicAllocator(int alpha_blt_dst_type, SIZE maxsize, int type=-1);
 109 };
 110
 111