modules/video_chroma/dxgi_fmt.c

   1 /*****************************************************************************
   2  * dxgi_fmt.c : DXGI GPU surface conversion module for vlc
   3  *****************************************************************************
   4  * Copyright © 2015 VLC authors, VideoLAN and VideoLabs
   5  *
   6  * Authors: Steve Lhomme <robux4@gmail.com>
   7  *
   8  * This program is free software; you can redistribute it and/or modify it
   9  * under the terms of the GNU Lesser General Public License as published by
  10  * the Free Software Foundation; either version 2.1 of the License, or
  11  * (at your option) any later version.
  12  *
  13  * This program is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  16  * GNU Lesser General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU Lesser General Public License
  19  * along with this program; if not, write to the Free Software Foundation,
  20  * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  21  *****************************************************************************/
  22
  23 #ifdef HAVE_CONFIG_H
  24 # include "config.h"
  25 #endif
  26
  27 #include <vlc_es.h>
  28
  29 #include "dxgi_fmt.h"
  30
  31 #include <assert.h>
  32
  33 typedef struct
  34 {
  35     const char   *name;
  36     DXGI_FORMAT  format;
  37     vlc_fourcc_t vlc_format;
  38 } dxgi_format_t;
  39
  40 static const dxgi_format_t dxgi_formats[] = {
  41     { "NV12",        DXGI_FORMAT_NV12,                VLC_CODEC_NV12     },
  42     { "I420_OPAQUE", DXGI_FORMAT_420_OPAQUE,          0                  },
  43     { "RGBA",        DXGI_FORMAT_R8G8B8A8_UNORM,      VLC_CODEC_RGBA     },
  44     { "RGBA_SRGB",   DXGI_FORMAT_R8G8B8A8_UNORM_SRGB, VLC_CODEC_RGBA     },
  45     { "BGRX",        DXGI_FORMAT_B8G8R8X8_UNORM,      VLC_CODEC_RGB32    },
  46     { "BGRA",        DXGI_FORMAT_B8G8R8A8_UNORM,      VLC_CODEC_BGRA     },
  47     { "BGRA_SRGB",   DXGI_FORMAT_B8G8R8A8_UNORM_SRGB, VLC_CODEC_BGRA     },
  48     { "AYUV",        DXGI_FORMAT_AYUV,                VLC_CODEC_VUYA     },
  49     { "YUY2",        DXGI_FORMAT_YUY2,                VLC_CODEC_YUYV     },
  50     { "AI44",        DXGI_FORMAT_AI44,                0                  },
  51     { "P8",          DXGI_FORMAT_P8,                  0                  },
  52     { "A8P8",        DXGI_FORMAT_A8P8,                0                  },
  53     { "B5G6R5",      DXGI_FORMAT_B5G6R5_UNORM,        VLC_CODEC_RGB16    },
  54     { "Y416",        DXGI_FORMAT_Y416,                0                  },
  55     { "P010",        DXGI_FORMAT_P010,                VLC_CODEC_P010     },
  56     { "P016",        DXGI_FORMAT_P016,                VLC_CODEC_P016     },
  57     { "Y210",        DXGI_FORMAT_Y210,                VLC_CODEC_Y210     },
  58     { "Y410",        DXGI_FORMAT_Y410,                VLC_CODEC_Y410     },
  59     { "NV11",        DXGI_FORMAT_NV11,                0                  },
  60     { "RGB10A2",     DXGI_FORMAT_R10G10B10A2_UNORM,   VLC_CODEC_RGBA10   },
  61     { "RGB16",       DXGI_FORMAT_R16G16B16A16_UNORM,  VLC_CODEC_RGBA64   },
  62     { "RGB16_FLOAT", DXGI_FORMAT_R16G16B16A16_FLOAT,  0                  },
  63     { "UNKNOWN",     DXGI_FORMAT_UNKNOWN,             0                  },
  64
  65     { NULL, 0, 0}
  66 };
  67
  68 static const d3d_format_t d3d_formats[] = {
  69     { "NV12",     DXGI_FORMAT_NV12,           VLC_CODEC_NV12,              8, 2, 2, { DXGI_FORMAT_R8_UNORM,       DXGI_FORMAT_R8G8_UNORM } },
  70     { "VA_NV12",  DXGI_FORMAT_NV12,           VLC_CODEC_D3D11_OPAQUE,      8, 2, 2, { DXGI_FORMAT_R8_UNORM,       DXGI_FORMAT_R8G8_UNORM } },
  71     { "P010",     DXGI_FORMAT_P010,           VLC_CODEC_P010,             10, 2, 2, { DXGI_FORMAT_R16_UNORM,      DXGI_FORMAT_R16G16_UNORM } },
  72     { "VA_P010",  DXGI_FORMAT_P010,           VLC_CODEC_D3D11_OPAQUE_10B, 10, 2, 2, { DXGI_FORMAT_R16_UNORM,      DXGI_FORMAT_R16G16_UNORM } },
  73     { "VA_AYUV",  DXGI_FORMAT_AYUV,           VLC_CODEC_D3D11_OPAQUE,      8, 1, 1, { DXGI_FORMAT_R8G8B8A8_UNORM } },
  74     { "YUY2",     DXGI_FORMAT_YUY2,           VLC_CODEC_YUYV,              8, 1, 2, { DXGI_FORMAT_R8G8B8A8_UNORM } },
  75     { "VA_YUY2",  DXGI_FORMAT_YUY2,           VLC_CODEC_D3D11_OPAQUE,      8, 1, 2, { DXGI_FORMAT_R8G8B8A8_UNORM } },
  76 #ifdef BROKEN_PIXEL
  77     { "Y416",     DXGI_FORMAT_Y416,           VLC_CODEC_I444_16L,     16, 1, 1, { DXGI_FORMAT_R16G16B16A16_UINT } },
  78 #endif
  79     { "VA_Y210",  DXGI_FORMAT_Y210,           VLC_CODEC_D3D11_OPAQUE_10B, 10, 1, 2, { DXGI_FORMAT_R16G16B16A16_UNORM } },
  80     { "VA_Y410",  DXGI_FORMAT_Y410,           VLC_CODEC_D3D11_OPAQUE_10B, 10, 1, 1, { DXGI_FORMAT_R10G10B10A2_UNORM } },
  81 #ifdef UNTESTED
  82     { "Y210",     DXGI_FORMAT_Y210,           VLC_CODEC_I422_10L,     10, 1, 2, { DXGI_FORMAT_R16G16B16A16_UNORM } },
  83     { "Y410",     DXGI_FORMAT_Y410,           VLC_CODEC_I444,         10, 1, 1, { DXGI_FORMAT_R10G10B10A2_UNORM } },
  84     { "NV11",     DXGI_FORMAT_NV11,           VLC_CODEC_I411,          8, 4, 1, { DXGI_FORMAT_R8_UNORM,           DXGI_FORMAT_R8G8_UNORM} },
  85 #endif
  86     { "I420",     DXGI_FORMAT_UNKNOWN,        VLC_CODEC_I420,          8, 2, 2, { DXGI_FORMAT_R8_UNORM,      DXGI_FORMAT_R8_UNORM, DXGI_FORMAT_R8_UNORM } },
  87     { "I420_10",  DXGI_FORMAT_UNKNOWN,        VLC_CODEC_I420_10L,     10, 2, 2, { DXGI_FORMAT_R16_UNORM,     DXGI_FORMAT_R16_UNORM, DXGI_FORMAT_R16_UNORM } },
  88     { "YUVA",     DXGI_FORMAT_UNKNOWN,        VLC_CODEC_YUVA,          8, 1, 1, { DXGI_FORMAT_R8_UNORM,      DXGI_FORMAT_R8_UNORM, DXGI_FORMAT_R8_UNORM, DXGI_FORMAT_R8_UNORM } },
  89     { "I444_16",  DXGI_FORMAT_UNKNOWN,        VLC_CODEC_I444_16L,     16, 1, 1, { DXGI_FORMAT_R16_UNORM,     DXGI_FORMAT_R16_UNORM, DXGI_FORMAT_R16_UNORM } },
  90     { "B8G8R8A8", DXGI_FORMAT_B8G8R8A8_UNORM, VLC_CODEC_BGRA,          8, 1, 1, { DXGI_FORMAT_B8G8R8A8_UNORM } },
  91     { "VA_BGRA",  DXGI_FORMAT_B8G8R8A8_UNORM, VLC_CODEC_D3D11_OPAQUE_BGRA,  8, 1, 1, { DXGI_FORMAT_B8G8R8A8_UNORM } },
  92     { "R8G8B8A8", DXGI_FORMAT_R8G8B8A8_UNORM, VLC_CODEC_RGBA,          8, 1, 1, { DXGI_FORMAT_R8G8B8A8_UNORM } },
  93     { "VA_RGBA",  DXGI_FORMAT_R8G8B8A8_UNORM, VLC_CODEC_D3D11_OPAQUE_RGBA,  8, 1, 1, { DXGI_FORMAT_R8G8B8A8_UNORM } },
  94     { "R8G8B8X8", DXGI_FORMAT_B8G8R8X8_UNORM, VLC_CODEC_RGB32,         8, 1, 1, { DXGI_FORMAT_B8G8R8X8_UNORM } },
  95     { "RGBA64",   DXGI_FORMAT_R16G16B16A16_UNORM, VLC_CODEC_RGBA64,   16, 1, 1, { DXGI_FORMAT_R16G16B16A16_UNORM } },
  96     { "RGB10A2",  DXGI_FORMAT_R10G10B10A2_UNORM, VLC_CODEC_RGBA10,    10, 1, 1, { DXGI_FORMAT_R10G10B10A2_UNORM } },
  97     { "VA_RGB10", DXGI_FORMAT_R10G10B10A2_UNORM, VLC_CODEC_D3D11_OPAQUE_RGBA, 10, 1, 1, { DXGI_FORMAT_R10G10B10A2_UNORM } },
  98     { "AYUV",     DXGI_FORMAT_AYUV,           VLC_CODEC_VUYA,          8, 1, 1, { DXGI_FORMAT_R8G8B8A8_UNORM } },
  99     { "B5G6R5",   DXGI_FORMAT_B5G6R5_UNORM,   VLC_CODEC_RGB16,         5, 1, 1, { DXGI_FORMAT_B5G6R5_UNORM } },
 100     { "I420_OPAQUE", DXGI_FORMAT_420_OPAQUE,  VLC_CODEC_D3D11_OPAQUE,  8, 2, 2, { DXGI_FORMAT_UNKNOWN } },
 101
 102     { NULL, 0, 0, 0, 0, 0, {} }
 103 };
 104
 105 const char *DxgiFormatToStr(DXGI_FORMAT format)
 106 {
 107     for (const dxgi_format_t *f = dxgi_formats; f->name != NULL; ++f)
 108     {
 109         if (f->format == format)
 110             return f->name;
 111     }
 112     return NULL;
 113 }
 114
 115 vlc_fourcc_t DxgiFormatFourcc(DXGI_FORMAT format)
 116 {
 117     for (const dxgi_format_t *f = dxgi_formats; f->name != NULL; ++f)
 118     {
 119         if (f->format == format)
 120             return f->vlc_format;
 121     }
 122     return 0;
 123 }
 124
 125 DXGI_FORMAT DxgiFourccFormat(vlc_fourcc_t fcc)
 126 {
 127     for (const dxgi_format_t *f = dxgi_formats; f->name != NULL; ++f)
 128     {
 129         if (f->vlc_format == fcc)
 130             return f->format;
 131     }
 132     return DXGI_FORMAT_UNKNOWN;
 133 }
 134
 135 const d3d_format_t *DxgiGetRenderFormatList(void)
 136 {
 137     return d3d_formats;
 138 }
 139
 140 void DxgiFormatMask(DXGI_FORMAT format, video_format_t *fmt)
 141 {
 142     if (format == DXGI_FORMAT_B8G8R8X8_UNORM || format == DXGI_FORMAT_B8G8R8A8_UNORM)
 143     {
 144         fmt->i_rmask = 0x0000ff00;
 145         fmt->i_gmask = 0x00ff0000;
 146         fmt->i_bmask = 0xff000000;
 147     }
 148 }
 149
 150 const char *DxgiVendorStr(unsigned int gpu_vendor)
 151 {
 152     static const struct {
 153         unsigned   id;
 154         const char name[32];
 155     } vendors [] = {
 156         { GPU_MANUFACTURER_AMD,      "ATI"         },
 157         { GPU_MANUFACTURER_NVIDIA,   "NVIDIA"      },
 158         { GPU_MANUFACTURER_VIA,      "VIA"         },
 159         { GPU_MANUFACTURER_INTEL,    "Intel"       },
 160         { GPU_MANUFACTURER_S3,       "S3 Graphics" },
 161         { GPU_MANUFACTURER_QUALCOMM, "Qualcomm"    },
 162         { 0,                         "Unknown" }
 163     };
 164
 165     int i = 0;
 166     for (i = 0; vendors[i].id != 0; i++) {
 167         if (vendors[i].id == gpu_vendor)
 168             break;
 169     }
 170     return vendors[i].name;
 171 }
 172
 173 UINT DxgiResourceCount(const d3d_format_t *d3d_fmt)
 174 {
 175     for (UINT count=0; count<DXGI_MAX_SHADER_VIEW; count++)
 176     {
 177         if (d3d_fmt->resourceFormat[count] == DXGI_FORMAT_UNKNOWN)
 178             return count;
 179     }
 180     return DXGI_MAX_SHADER_VIEW;
 181 }
 182
 183 bool DxgiIsRGBFormat(const d3d_format_t *cfg)
 184 {
 185     return cfg->resourceFormat[0] != DXGI_FORMAT_R8_UNORM &&
 186            cfg->resourceFormat[0] != DXGI_FORMAT_R16_UNORM &&
 187            cfg->formatTexture != DXGI_FORMAT_YUY2 &&
 188            cfg->formatTexture != DXGI_FORMAT_AYUV &&
 189            cfg->formatTexture != DXGI_FORMAT_Y210 &&
 190            cfg->formatTexture != DXGI_FORMAT_Y410 &&
 191            cfg->formatTexture != DXGI_FORMAT_420_OPAQUE;
 192 }
 193
 194 void DXGI_GetBlackColor( const d3d_format_t *pixelFormat,
 195                          union DXGI_Color black[DXGI_MAX_RENDER_TARGET],
 196                          size_t colors[DXGI_MAX_RENDER_TARGET] )
 197 {
 198     static const union DXGI_Color blackY    = { .y = 0.0f };
 199     static const union DXGI_Color blackUV   = { .u = 0.5f, .v = 0.5f };
 200     static const union DXGI_Color blackRGBA = { .r = 0.0f, .g = 0.0f, .b = 0.0f, .a = 1.0f };
 201     static const union DXGI_Color blackYUY2 = { .r = 0.0f, .g = 0.5f, .b = 0.0f, .a = 0.5f };
 202     static const union DXGI_Color blackVUYA = { .r = 0.5f, .g = 0.5f, .b = 0.0f, .a = 1.0f };
 203     static const union DXGI_Color blackY210 = { .r = 0.0f, .g = 0.5f, .b = 0.5f, .a = 0.0f };
 204
 205     static_assert(DXGI_MAX_RENDER_TARGET >= 2, "we need at least 2 RenderTargetView for NV12/P010");
 206
 207     switch (pixelFormat->formatTexture)
 208     {
 209     case DXGI_FORMAT_NV12:
 210     case DXGI_FORMAT_P010:
 211         colors[0] = 1; black[0] = blackY;
 212         colors[1] = 2; black[1] = blackUV;
 213         break;
 214     case DXGI_FORMAT_R8G8B8A8_UNORM:
 215     case DXGI_FORMAT_B8G8R8A8_UNORM:
 216     case DXGI_FORMAT_B8G8R8X8_UNORM:
 217     case DXGI_FORMAT_R10G10B10A2_UNORM:
 218     case DXGI_FORMAT_B5G6R5_UNORM:
 219         colors[0] = 4; black[0] = blackRGBA;
 220         colors[1] = 0;
 221         break;
 222     case DXGI_FORMAT_YUY2:
 223         colors[0] = 4; black[0] = blackYUY2;
 224         colors[1] = 0;
 225         break;
 226     case DXGI_FORMAT_Y410:
 227         colors[0] = 4; black[0] = blackVUYA;
 228         colors[1] = 0;
 229         break;
 230     case DXGI_FORMAT_Y210:
 231         colors[0] = 4; black[0] = blackY210;
 232         colors[1] = 0;
 233         break;
 234     case DXGI_FORMAT_AYUV:
 235         colors[0] = 4; black[0] = blackVUYA;
 236         colors[1] = 0;
 237         break;
 238     default:
 239         vlc_assert_unreachable();
 240     }
 241 }