videoout_d3d.cpp

Go to the documentation of this file.

// -*- Mode: c++ -*-
 
#include <windows.h>
#include <mmsystem.h>
 
#include <map>
#include <iostream>
#include <algorithm>
 
#include "libmythbase/mythlogging.h"
#include "libmythui/mythmainwindow.h"
 
#include "fourcc.h"
#include "mythavutil.h"
#include "mythplayer.h"
#include "mythvideoprofile.h"
#include "osd.h"
#include "tv.h"
#include "videoout_d3d.h"
 
extern "C" {
#include "libavutil/imgutils.h"
}
 
#undef UNICODE
 
const int kNumBuffers = 31;
const int kNeedFreeFrames = 1;
const int kPrebufferFramesNormal = 10;
const int kPrebufferFramesSmall = 4;
 
#define LOC      QString("VideoOutputD3D: ")
 
void VideoOutputD3D::GetRenderOptions(RenderOptions &Options)
{
    Options.renderers->append("direct3d");
    (*Options.safe_renderers)["dummy"].append("direct3d");
    (*Options.safe_renderers)["nuppel"].append("direct3d");
    if (Options.decoders->contains("ffmpeg"))
        (*Options.safe_renderers)["ffmpeg"].append("direct3d");
    Options.priorities->insert("direct3d", 70);
 
#if CONFIG_DXVA2
    if (Options.decoders->contains("dxva2"))
        (*Options.safe_renderers)["dxva2"].append("direct3d");
#endif
}
 
VideoOutputD3D::VideoOutputD3D(MythMainWindow* MainWindow, MythRenderD3D9* Render,
                   MythD3D9Painter* Painter, MythDisplay* Display,
                   const MythVideoProfilePtr& VideoProfile, QString& Profile)
  : MythVideoOutputGPU(MainWindow, Render, Painter, Display, VideoProfile, Profile)
{
    m_pauseFrame.m_buffer = nullptr;
}
 
VideoOutputD3D::~VideoOutputD3D()
{
    TearDown();
}
 
void VideoOutputD3D::TearDown(void)
{
    QMutexLocker locker(&m_lock);
    m_videoBuffers.DiscardFrames(true);
    m_videoBuffers.Reset();
    m_videoBuffers.DeleteBuffers();
    if (m_pauseFrame.m_buffer)
    {
        delete [] m_pauseFrame.m_buffer;
        m_pauseFrame.m_buffer = nullptr;
    }
 
    DeleteDecoder();
    DestroyContext();
}
 
void VideoOutputD3D::DestroyContext(void)
{
    QMutexLocker locker(&m_lock);
    m_renderValid = false;
    m_renderReset = false;
 
    if (m_video)
    {
        delete m_video;
        m_video = nullptr;
    }
 
    if (m_render)
    {
        m_render->DecrRef();
        m_render = nullptr;
    }
}
 
bool VideoOutputD3D::InputChanged(QSize        video_dim_buf,
                                  QSize        video_dim_disp,
                                  float        video_aspect,
                                  MythCodecID  av_codec_id,
                                  bool        &aspect_only,
                                  int          [[maybe_unused]] reference_frames,
                                  bool         force_change)
{
    QMutexLocker locker(&m_lock);
 
    QSize cursize = GetVideoDim();
 
    LOG(VB_PLAYBACK, LOG_INFO, LOC +
        QString("InputChanged from %1: %2x%3 aspect %4 to %5: %6x%7 aspect %9")
            .arg(toString(m_videoCodecID)).arg(cursize.width())
            .arg(cursize.height()).arg(GetVideoAspect())
            .arg(toString(av_codec_id)).arg(video_dim_disp.width())
            .arg(video_dim_disp.height()).arg(video_aspect));
 
 
    bool cid_changed = (m_videoCodecID != av_codec_id);
    bool res_changed = video_dim_disp != cursize;
    bool asp_changed = video_aspect   != GetVideoAspect();
 
    if (!res_changed && !cid_changed && !force_change)
    {
        if (asp_changed)
        {
            aspect_only = true;
            VideoAspectRatioChanged(video_aspect);
            MoveResize();
        }
        return true;
    }
 
    TearDown();
    QRect disp = GetDisplayVisibleRect();
    if (Init(video_dim_buf, video_dim_disp,
             video_aspect, disp, av_codec_id))
    {
        return true;
    }
 
    LOG(VB_GENERAL, LOG_ERR, LOC + "Failed to re-initialise video output.");
    m_errorState = kError_Unknown;
 
    return false;
}
 
bool VideoOutputD3D::SetupContext()
{
    QMutexLocker locker(&m_lock);
    DestroyContext();
    QSize size = GetVideoDim();
    m_render = new MythRenderD3D9();
    if (!(m_render && m_render->Create(GetDisplayVisibleRect().size(),
                                 m_hWnd)))
        return false;
 
    m_video = new D3D9Image(m_render, size, true);
    if (!(m_video && m_video->IsValid()))
        return false;
 
    LOG(VB_PLAYBACK, LOG_INFO, LOC +
        "Direct3D device successfully initialized.");
    m_renderValid = true;
    return true;
}
 
bool VideoOutputD3D::Init(QSize video_dim_buf,
                          QSize video_dim_disp,
                          float video_aspect,
                          QRect win_rect,MythCodecID codec_id)
{
    MythPainter *painter = GetMythPainter();
    if (painter)
        painter->FreeResources();
 
    QMutexLocker locker(&m_lock);
    m_hWnd      = (HWND)m_display->m_widget->winId();
 
    MythVideoOutput::Init(video_dim_buf, video_dim_disp,
                          video_aspect, win_rect, codec_id);
 
    LOG(VB_PLAYBACK, LOG_INFO, LOC + QString("Init with codec: %1")
                         .arg(toString(codec_id)));
    SetProfile();
 
    bool success = true;
    success &= SetupContext();
 
    if (codec_is_dxva2(m_videoCodecID))
    {
        if (!CreateDecoder())
            return false;
    }
 
    success &= CreateBuffers();
    success &= InitBuffers();
 
    MoveResize();
 
    if (!success)
        TearDown();
 
    return success;
}
 
void VideoOutputD3D::SetProfile(void)
{
    if (m_videoProfile)
        m_videoProfile->SetVideoRenderer("direct3d");
}
 
bool VideoOutputD3D::CreateBuffers(void)
{
    if (codec_is_dxva2(m_videoCodecID))
    {
        m_videoBuffers.Init(VideoBuffers::GetNumBuffers(FMT_DXVA2), 2, 1, 4);
        LOG(VB_PLAYBACK, LOG_INFO, LOC + QString("Created %1 empty DXVA2 buffers.")
            .arg(VideoBuffers::GetNumBuffers(FMT_DXVA2)));
        return true;
    }
 
    m_videoBuffers.Init(VideoBuffers::GetNumBuffers(FMT_YV12), kNeedFreeFrames,
                  kPrebufferFramesNormal, kPrebufferFramesSmall);
    return true;
 
}
 
bool VideoOutputD3D::InitBuffers(void)
{
#if CONFIG_DXVA2
    if ((codec_is_dxva2(m_videoCodecID)) && m_decoder)
    {
        QMutexLocker locker(&m_lock);
        const QSize video_dim = GetVideoDim();
        bool ok = true;
        for (int i = 0; i < NUM_DXVA2_BUFS; i++)
        {
            ok &= vbuffers.CreateBuffer(video_dim.width(),
                                        video_dim.height(), i,
                                        m_decoder->GetSurface(i), FMT_DXVA2);
        }
        if (ok)
            LOG(VB_PLAYBACK, LOG_INFO, LOC + "Initialised DXVA2 buffers.");
        return ok;
    }
#endif
    return m_videoBuffers.CreateBuffers(FMT_YV12,
                                  GetVideoDim().width(),
                                  GetVideoDim().height());
}
 
void VideoOutputD3D::RenderFrame(MythVideoFrame *buffer,
                                 FrameScanType t)
{
    if (IsErrored())
    {
        LOG(VB_GENERAL, LOG_ERR, LOC +
            "RenderFrame() called while IsErrored is true.");
        return;
    }
 
    if (!buffer && codec_is_std(m_videoCodecID))
        buffer = m_videoBuffers.GetScratchFrame();
 
    bool dummy = false;
    if (buffer)
    {
        dummy = buffer->m_dummy;
        m_framesPlayed = buffer->m_frameNumber + 1;
    }
 
    if (!m_render || !m_video)
        return;
 
    m_renderValid = m_render->Test(m_renderReset);
    if (m_renderValid)
    {
        QRect dvr = GetDisplayVideoRect();
        bool ok = m_render->ClearBuffer();
        if (ok && !dummy)
            ok = m_video->UpdateVertices(dvr, GetVideoRect(),
                                         255, true);
 
        if (ok)
        {
            ok = m_render->Begin();
            if (ok)
            {
                if (!dummy)
                    m_video->Draw();
            }
        }
 
    }
}
 
void VideoOutputD3D::RenderEnd()
{
    if (!m_renderValid)
        return;
    m_render->End();
}
 
void VideoOutputD3D::EndFrame()
{
    if (IsErrored())
    {
        LOG(VB_GENERAL, LOG_ERR, LOC +
            "Show() called while IsErrored is true.");
        return;
    }
 
    if (!m_render)
        return;
 
    m_renderValid = m_render->Test(m_renderReset);
    if (m_renderValid)
        m_render->Present(IsEmbedding() ? m_hEmbedWnd : nullptr);
}
 
void VideoOutputD3D::UpdatePauseFrame(std::chrono::milliseconds &disp_timecode, FrameScanType)
{
    QMutexLocker locker(&m_lock);
    MythVideoFrame *used_frame = m_videoBuffers.Head(kVideoBuffer_used);
 
    if (codec_is_std(m_videoCodecID))
    {
        if (!used_frame)
            used_frame = m_videoBuffers.GetScratchFrame();
        CopyFrame(&m_pauseFrame, used_frame);
        disp_timecode = m_pauseFrame.m_displayTimecode;
    }
    else if (codec_is_dxva2(m_videoCodecID))
    {
        if (used_frame)
        {
            m_pauseSurface = used_frame->m_buffer;
            disp_timecode = used_frame->m_displayTimecode;
        }
        else
            LOG(VB_PLAYBACK, LOG_WARNING, LOC + "Failed to update pause frame");
    }
}
 
void VideoOutputD3D::UpdateFrame(MythVideoFrame *frame, D3D9Image *img)
{
    if (codec_is_dxva2(m_videoCodecID))
        return;
 
    // TODO - add a size check
    bool hardware_conv = false;
    uint pitch = 0;
    uint8_t *buf = img->GetBuffer(hardware_conv, pitch);
    if (buf && hardware_conv)
    {
        copybuffer(buf, frame, pitch);
    }
    else if (buf && !hardware_conv)
    {
        AVFrame image_out;
        av_image_fill_arrays(image_out.data, image_out.linesize,
            (uint8_t*)buf,
            AV_PIX_FMT_RGB32, frame->m_width, frame->m_height, IMAGE_ALIGN);
        image_out.linesize[0] = pitch;
        m_copyFrame.Copy(&image_out, frame,(uint8_t*)buf, AV_PIX_FMT_RGB32);
    }
    img->ReleaseBuffer();
}
 
void VideoOutputD3D::PrepareFrame(MythVideoFrame *frame, FrameScanType scan)
{
    if (!m_render || !m_video)
        return;
 
    QMutexLocker locker(&m_lock);
    if (IsErrored())
    {
        LOG(VB_GENERAL, LOG_ERR, LOC +
            "ProcessFrame() called while IsErrored is true.");
        return;
    }
 
    bool gpu = codec_is_dxva2(m_videoCodecID);
 
    if (gpu && frame && frame->m_type != FMT_DXVA2)
    {
        LOG(VB_GENERAL, LOG_ERR, LOC + "Wrong frame format");
        return;
    }
 
    bool dummy = false;
    bool pauseframe = false;
    if (!frame)
    {
        if (!gpu)
        {
            frame = m_videoBuffers.GetScratchFrame();
            CopyFrame(m_videoBuffers.GetScratchFrame(), &m_pauseFrame);
        }
        pauseframe = true;
    }
 
    if (frame)
        dummy = frame->m_dummy;
 
    // Test the device
    m_renderValid |= m_render->Test(m_renderReset);
    if (m_renderReset)
        SetupContext();
 
    // Update a software decoded frame
    if (m_renderValid && !gpu && !dummy)
        UpdateFrame(frame, m_video);
 
    // Update a GPU decoded frame
    if (m_renderValid && gpu && !dummy)
    {
        m_renderValid = m_render->Test(m_renderReset);
        if (m_renderReset)
            CreateDecoder();
 
        if (m_renderValid && frame)
        {
            m_render->CopyFrame(frame->m_buffer, m_video);
        }
        else if (m_renderValid && pauseframe)
        {
            m_render->CopyFrame(m_pauseSurface, m_video);
        }
    }
}
 
QStringList VideoOutputD3D::GetAllowedRenderers(
    MythCodecID myth_codec_id, const QSize &video_dim)
{
    QStringList list;
    if (codec_is_std(myth_codec_id) || (codec_is_dxva2_hw(myth_codec_id) &&
        !getenv("NO_DXVA2")))
    {
        list += "direct3d";
    }
    return list;
}
 
MythCodecID VideoOutputD3D::GetSupportedCodec(
    AVCodecContext **Context, const AVCodec ** Codec,
    const QString &decoder, uint stream_type)
{
#if CONFIG_DXVA2
    MythCodecID test_cid = (MythCodecID)(kCodec_MPEG1_DXVA2 + (stream_type - 1));
    bool use_cpu = !codec_is_dxva2_hw(test_cid);
    if ((decoder == "dxva2") && !getenv("NO_DXVA2") && !use_cpu)
        return test_cid;
#endif
    return (MythCodecID)(kCodec_MPEG1 + (stream_type - 1));
}
 
bool VideoOutputD3D::CreateDecoder(void)
{
#if CONFIG_DXVA2
    QMutexLocker locker(&m_lock);
    if (m_decoder)
        DeleteDecoder();
    QSize video_dim = GetVideoDim();
    m_decoder = new DXVA2Decoder(NUM_DXVA2_BUFS, m_videoCodecID,
                                 video_dim.width(), video_dim.height());
    return (m_decoder && m_decoder->Init(m_render));
#else
    return false;
#endif
}
 
void VideoOutputD3D::DeleteDecoder(void)
{
#if CONFIG_DXVA2
    QMutexLocker locker(&m_lock);
    delete m_decoder;
    m_decoder = nullptr;
#endif
}
 
/* vim: set expandtab tabstop=4 shiftwidth=4: */