first commit

project/fm_viewer/fav/hw_decoder/VideoDecoderD3D.h

@@ -0,0 +1,145 @@
+/******************************************************************************
+    QtAV:  Media play library based on Qt and FFmpeg
+    Copyright (C) 2012-2016 Wang Bin <wbsecg1@gmail.com>
+
+*   This file is part of QtAV (from 2016)
+
+    This library is free software; you can redistribute it and/or
+    modify it under the terms of the GNU Lesser General Public
+    License as published by the Free Software Foundation; either
+    version 2.1 of the License, or (at your option) any later version.
+
+    This library 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
+    Lesser General Public License for more details.
+
+    You should have received a copy of the GNU Lesser General Public
+    License along with this library; if not, write to the Free Software
+    Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
+******************************************************************************/
+
+#ifndef QTAV_VIDEODECODERD3D_H
+#define QTAV_VIDEODECODERD3D_H
+#ifdef RM_USE_HW_DECODER
+
+#include <unknwn.h>
+#include <inttypes.h>
+#include "VideoDecoderFFmpegHW.h"
+#include "VideoDecoderFFmpegHW_p.h"
+
+namespace FAV {
+
+struct d3d_format_t {
+    const char *name;
+    int fourcc;
+    VideoFormat::PixelFormat pixfmt;
+};
+bool isIntelClearVideo(const GUID *guid);
+bool isNoEncrypt(const GUID* guid);
+int getSupportedFourcc(int *formats, UINT nb_formats);
+VideoFormat::PixelFormat pixelFormatFromFourcc(int format);
+
+class VideoDecoderD3DPrivate;
+class VideoDecoderD3D : public VideoDecoderFFmpegHW
+{
+    DPTR_DECLARE_PRIVATE(VideoDecoderD3D)
+    Q_OBJECT
+    Q_PROPERTY(int surfaces READ surfaces WRITE setSurfaces NOTIFY surfacesChanged)
+public:
+    // properties
+    void setSurfaces(int num);
+    int surfaces() const;
+Q_SIGNALS:
+    void surfacesChanged();
+protected:
+    VideoDecoderD3D(VideoDecoderD3DPrivate& d);
+};
+
+
+struct va_surface_t {
+    va_surface_t() : ref(0), order(0) {}
+    virtual ~va_surface_t() {}
+    virtual void setSurface(IUnknown* s) = 0;
+    virtual IUnknown* getSurface() const = 0;
+
+    int ref;
+    int order;
+};
+
+class VideoDecoderD3DPrivate : public VideoDecoderFFmpegHWPrivate
+{
+public:
+    VideoDecoderD3DPrivate();
+
+    bool open() Q_DECL_OVERRIDE;
+    void close() Q_DECL_OVERRIDE;
+    void* setup(AVCodecContext *avctx) Q_DECL_OVERRIDE;
+    bool getBuffer(void **opaque, uint8_t **data) Q_DECL_OVERRIDE;
+    void releaseBuffer(void *opaque, uint8_t *data) Q_DECL_OVERRIDE;
+
+    int aligned(int x);
+private:
+    virtual bool setupSurfaceInterop() {return true;}
+    virtual bool createDevice() = 0; //d3d device, video context etc.
+    virtual void destroyDevice() = 0;
+    virtual bool checkDevice() {return true;}
+    virtual QVector<GUID> getSupportedCodecs() const = 0;
+
+    virtual void* setupAVVAContext() = 0;
+    /// create surfaces and decoder. width and height are coded value, maybe not aligned for d3d surface
+    /// surfaces count is given, but not allocated
+    virtual bool createDecoder(AVCodecID codec, int width, int height, QVector<va_surface_t*>& surf) = 0;
+    virtual void destroyDecoder() = 0;
+    virtual int fourccFor(const GUID *guid) const = 0;
+    const d3d_format_t *getFormat(const AVCodecContext* avctx, const QVector<GUID>& guids, GUID *selected) const;
+public:
+    // set by user. don't reset in when call destroy
+    bool surface_auto;
+    int surface_count;
+    QVector<IUnknown*> hw_surfaces;
+    int format_fcc;
+    GUID codec_guid;
+private:
+    int surface_width;
+    int surface_height;
+    unsigned surface_order;
+    QVector<va_surface_t*> surfaces; //TODO: delete on close()
+};
+
+template<typename T> int SelectConfig(AVCodecID codec_id, const T* cfgs, int nb_cfgs, T* cfg)
+{
+    qDebug("Decoder configurations: %d", nb_cfgs);
+    /* Select the best decoder configuration */
+    int cfg_score = 0;
+    for (int i = 0; i < nb_cfgs; i++) {
+        const T &c = cfgs[i];
+        qDebug("configuration[%d] ConfigBitstreamRaw %d", i, c.ConfigBitstreamRaw);
+        /* */
+        int score = 0;
+        if (c.ConfigBitstreamRaw == 1)
+            score = 1;
+        else if (codec_id == QTAV_CODEC_ID(H264) && c.ConfigBitstreamRaw == 2)
+            score = 2;
+        else
+            continue;
+        if (isNoEncrypt(&c.guidConfigBitstreamEncryption))
+            score += 16;
+        if (cfg_score < score) {
+            *cfg = c;
+            cfg_score = score;
+        }
+    }
+    if (cfg_score <= 0)
+        qWarning("Failed to find a supported decoder configuration");
+    return cfg_score;
+}
+
+#ifndef MAKEFOURCC //winrt
+#define MAKEFOURCC(ch0, ch1, ch2, ch3) \
+  ((DWORD)(BYTE)(ch0)|((DWORD)(BYTE)(ch1)<<8)|((DWORD)(BYTE)(ch2)<<16)|((DWORD)(BYTE)(ch3)<<24))
+#endif //MAKEFOURCC
+} //namespace QtAV
+
+#endif // RM_USE_HW_DECODER
+#endif //QTAV_VIDEODECODERD3D_H