v4lrecorder.cpp

Go to the documentation of this file.

// -*- Mode: c++ -*-

#ifdef USING_V4L1
#include <linux/videodev.h> // for vbi_format
#endif // USING_V4L1

#ifdef USING_V4L2
#include <linux/videodev2.h>
#endif // USING_V4L2

#include <sys/ioctl.h>      // for ioctl
#include <sys/time.h>       // for gettimeofday
#include <unistd.h>         // for IO_NONBLOCK
#include <fcntl.h>          // for IO_NONBLOCK

#include "vbi608extractor.h"
#include "mythcontext.h"
#include "mythlogging.h"
#include "v4lrecorder.h"
#include "vbitext/vbi.h"
#include "tv_rec.h"

#define TVREC_CARDNUM \
        ((m_tvrec != nullptr) ? QString::number(m_tvrec->GetInputId()) : "NULL")

#define LOC QString("V4LRec[%1](%2): ") \
            .arg(TVREC_CARDNUM).arg(m_videodevice)

V4LRecorder::~V4LRecorder()
{
    {
        QMutexLocker locker(&m_pauseLock);
        m_request_helper = false;
        m_unpauseWait.wakeAll();
    }

    if (m_vbi_thread)
    {
        m_vbi_thread->wait();
        delete m_vbi_thread;
        m_vbi_thread = nullptr;
        CloseVBIDevice();
    }
}

void V4LRecorder::StopRecording(void)
{
    DTVRecorder::StopRecording();
    while (m_vbi_thread && m_vbi_thread->isRunning())
        m_vbi_thread->wait();
}

bool V4LRecorder::IsHelperRequested(void) const
{
    QMutexLocker locker(&m_pauseLock);
    return m_request_helper && m_request_recording;
}

void V4LRecorder::SetOption(const QString &name, const QString &value)
{
    if (name == "audiodevice")
        m_audiodevice = value;
    else if (name == "vbidevice")
        m_vbidevice = value;
    else if (name == "vbiformat")
        m_vbimode = VBIMode::Parse(value);
    else
        DTVRecorder::SetOption(name, value);
}

static void vbi_event(struct VBIData *data, struct vt_event *ev)
{
    switch (ev->type)
    {
       case EV_PAGE:
       {
            auto *vtp = (struct vt_page *) ev->p1;
            if (vtp->flags & PG_SUBTITLE)
            {
#if 0
                LOG(VB_GENERAL, LOG_DEBUG, QString("subtitle page %1.%2")
                      .arg(vtp->pgno, 0, 16) .arg(vtp->subno, 0, 16));
#endif
                data->foundteletextpage = true;
                memcpy(&(data->teletextpage), vtp, sizeof(vt_page));
            }
       }
       break;

       case EV_HEADER:
       case EV_XPACKET:
           break;
    }
}

int V4LRecorder::OpenVBIDevice(void)
{
    int fd = -1;
    if (m_vbi_fd >= 0)
        return m_vbi_fd;

    struct VBIData *vbi_cb = nullptr;
    struct vbi     *pal_tt = nullptr;
    uint width = 0, start_line = 0, line_count = 0;

    QByteArray vbidev = m_vbidevice.toLatin1();
    if (VBIMode::PAL_TT == m_vbimode)
    {
        pal_tt = vbi_open(vbidev.constData(), nullptr, 99, -1);
        if (pal_tt)
        {
            fd = pal_tt->fd;
            vbi_cb = new VBIData;
            memset(vbi_cb, 0, sizeof(VBIData));
            vbi_cb->nvr = this;
            vbi_add_handler(pal_tt, (void*) vbi_event, vbi_cb);
        }
    }
    else if (VBIMode::NTSC_CC == m_vbimode)
    {
        fd = open(vbidev.constData(), O_RDONLY/*|O_NONBLOCK*/);
    }
    else
    {
        LOG(VB_GENERAL, LOG_ERR, LOC + "Invalid CC/Teletext mode");
        return -1;
    }

    if (fd < 0)
    {
        LOG(VB_GENERAL, LOG_ERR, LOC +
            QString("Can't open vbi device: '%1'").arg(m_vbidevice));
        return -1;
    }

    if (VBIMode::NTSC_CC == m_vbimode)
    {
#ifdef USING_V4L2
        struct v4l2_format fmt {};
        fmt.type = V4L2_BUF_TYPE_VBI_CAPTURE;
        if (0 != ioctl(fd, VIDIOC_G_FMT, &fmt))
        {
#ifdef USING_V4L1
            LOG(VB_RECORD, LOG_INFO, "V4L2 VBI setup failed, trying v1 ioctl");
            // Try V4L v1 VBI ioctls, iff V4L v2 fails
            struct vbi_format old_fmt;
            memset(&old_fmt, 0, sizeof(vbi_format));
            if (ioctl(fd, VIDIOCGVBIFMT, &old_fmt) < 0)
            {
                LOG(VB_GENERAL, LOG_ERR, LOC +
                    "Failed to query vbi capabilities (V4L1)");
                close(fd);
                return -1;
            }
            fmt.fmt.vbi.sampling_rate    = old_fmt.sampling_rate;
            fmt.fmt.vbi.offset           = 0;
            fmt.fmt.vbi.samples_per_line = old_fmt.samples_per_line;
            fmt.fmt.vbi.start[0]         = old_fmt.start[0];
            fmt.fmt.vbi.start[1]         = old_fmt.start[1];
            fmt.fmt.vbi.count[0]         = old_fmt.count[0];
            fmt.fmt.vbi.count[1]         = old_fmt.count[1];
            fmt.fmt.vbi.flags            = old_fmt.flags;
#else // if !USING_V4L1
            LOG(VB_RECORD, LOG_ERR, "V4L2 VBI setup failed");
            close(fd);
            return -1;
#endif // !USING_V4L1
        }
        LOG(VB_RECORD, LOG_INFO, LOC +
            QString("vbi_format  rate: %1"
                    "\n\t\t\t          offset: %2"
                    "\n\t\t\tsamples_per_line: %3"
                    "\n\t\t\t          starts: %4, %5"
                    "\n\t\t\t          counts: %6, %7"
                    "\n\t\t\t           flags: 0x%8")
                .arg(fmt.fmt.vbi.sampling_rate)
                .arg(fmt.fmt.vbi.offset)
                .arg(fmt.fmt.vbi.samples_per_line)
                .arg(fmt.fmt.vbi.start[0])
                .arg(fmt.fmt.vbi.start[1])
                .arg(fmt.fmt.vbi.count[0])
                .arg(fmt.fmt.vbi.count[1])
                .arg(fmt.fmt.vbi.flags,0,16));

        width      = fmt.fmt.vbi.samples_per_line;
        start_line = fmt.fmt.vbi.start[0];
        line_count = fmt.fmt.vbi.count[0];
        if (line_count != fmt.fmt.vbi.count[1])
        {
            LOG(VB_GENERAL, LOG_ERR, LOC +
                    "VBI must have the same number of "
                    "odd and even fields for our decoder");
            close(fd);
            return -1;
        }
        if (start_line > 21 || start_line + line_count < 22)
        {
            LOG(VB_GENERAL, LOG_ERR, LOC + "VBI does not include line 21");
            // TODO We could try to set the VBI format ourselves..
            close(fd);
            return -1;
        }
#endif // USING_V4L2
    }

    if (VBIMode::PAL_TT == m_vbimode)
    {
        m_pal_vbi_cb = vbi_cb;
        m_pal_vbi_tt = pal_tt;
    }
    else if (VBIMode::NTSC_CC == m_vbimode)
    {
        m_ntsc_vbi_width      = width;
        m_ntsc_vbi_start_line = start_line;
        m_ntsc_vbi_line_count = line_count;
        m_vbi608 = new VBI608Extractor();
    }

    m_vbi_fd = fd;

    return fd;
}

void V4LRecorder::CloseVBIDevice(void)
{
    if (m_vbi_fd < 0)
        return;

    if (m_pal_vbi_tt)
    {
        vbi_del_handler(m_pal_vbi_tt, (void*) vbi_event, m_pal_vbi_cb);
        vbi_close(m_pal_vbi_tt);
        delete m_pal_vbi_cb;
        m_pal_vbi_cb = nullptr;
    }
    else
    {
        delete m_vbi608; m_vbi608 = nullptr;
        close(m_vbi_fd);
    }

    m_vbi_fd = -1;
}

void V4LRecorder::RunVBIDevice(void)
{
    if (m_vbi_fd < 0)
        return;

    unsigned char *buf = nullptr, *ptr = nullptr, *ptr_end = nullptr;
    if (m_ntsc_vbi_width)
    {
        uint sz   = m_ntsc_vbi_width * m_ntsc_vbi_line_count * 2;
        buf = ptr = new unsigned char[sz];
        ptr_end   = buf + sz;
    }

    while (IsHelperRequested() && !IsErrored())
    {
        if (PauseAndWait())
            continue;

        if (!IsHelperRequested() || IsErrored())
            break;

        struct timeval tv {0, 5000};
        fd_set rdset;

        FD_ZERO(&rdset);
        FD_SET(m_vbi_fd, &rdset);

        int nr = select(m_vbi_fd + 1, &rdset, nullptr, nullptr, &tv);
        if (nr < 0)
            LOG(VB_GENERAL, LOG_ERR, LOC + "vbi select failed" + ENO);

        if (nr <= 0)
        {
            if (nr==0)
                LOG(VB_GENERAL, LOG_DEBUG, LOC + "vbi select timed out");
            continue; // either failed or timed out..
        }
        if (VBIMode::PAL_TT == m_vbimode)
        {
            m_pal_vbi_cb->foundteletextpage = false;
            vbi_handler(m_pal_vbi_tt, m_pal_vbi_tt->fd);
            if (m_pal_vbi_cb->foundteletextpage)
            {
                // decode VBI as teletext subtitles
                FormatTT(m_pal_vbi_cb);
            }
        }
        else if (VBIMode::NTSC_CC == m_vbimode)
        {
            int ret = read(m_vbi_fd, ptr, ptr_end - ptr);
            ptr = (ret > 0) ? ptr + ret : ptr;
            if ((ptr_end - ptr) == 0)
            {
                unsigned char *line21_field1 =
                    buf + ((21 - m_ntsc_vbi_start_line) * m_ntsc_vbi_width);
                unsigned char *line21_field2 =
                    buf + ((m_ntsc_vbi_line_count + 21 - m_ntsc_vbi_start_line)
                           * m_ntsc_vbi_width);
                bool cc1 = m_vbi608->ExtractCC12(line21_field1, m_ntsc_vbi_width);
                bool cc2 = m_vbi608->ExtractCC34(line21_field2, m_ntsc_vbi_width);
                if (cc1 || cc2)
                {
                    int code1 = m_vbi608->GetCode1();
                    int code2 = m_vbi608->GetCode2();
                    code1 = (0xFFFF==code1) ? -1 : code1;
                    code2 = (0xFFFF==code2) ? -1 : code2;
                    FormatCC(code1, code2);
                }
                ptr = buf;
            }
            else if (ret < 0)
            {
                LOG(VB_GENERAL, LOG_ERR, LOC + "Reading VBI data" + ENO);
            }
        }
    }

    delete [] buf;
}

/* vim: set expandtab tabstop=4 shiftwidth=4: */