ntv2outputtestpattern.cpp

Go to the documentation of this file.

/* SPDX-License-Identifier: MIT */
//  Includes
#include "ntv2outputtestpattern.h"
#include "ntv2devicescanner.h"
#include "ntv2testpatterngen.h"
#include "ajabase/system/process.h"
 
#define AsULWordPtr(_p_)    reinterpret_cast<const ULWord*> (_p_)
 
 
using namespace std;
 
 
const uint32_t  kAppSignature   (NTV2_FOURCC('T','e','s','t'));
 
 
AJALabelValuePairs TestPatConfig::Get (const bool inCompact) const
{
    AJALabelValuePairs result (PlayerConfig::Get (inCompact));
    AJASystemInfo::append(result, "Background Pattern", fTestPatternName);
    return result;
}
 
std::ostream & operator << (std::ostream & ioStrm,  const TestPatConfig & inObj)
{
    return ioStrm << AJASystemInfo::ToString(inObj.Get());
}
 
 
NTV2OutputTestPattern::NTV2OutputTestPattern (const TestPatConfig & inConfig)
    :   mConfig             (inConfig),
        mDeviceID           (DEVICE_ID_NOTFOUND),
        mSavedTaskMode      (NTV2_TASK_MODE_INVALID),
        mSavedConnections   ()
{
}   //  constructor
 
 
NTV2OutputTestPattern::~NTV2OutputTestPattern ()
{
    //  Restore prior widget routing...
    mDevice.ApplySignalRoute(mSavedConnections, /*replace?*/true);
 
    //  Restore the prior service level, and release the device...
    mDevice.SetEveryFrameServices(mSavedTaskMode);
    mDevice.ReleaseStreamForApplication (kAppSignature, static_cast<int32_t>(AJAProcess::GetPid()));
 
}   //  destructor
 
 
AJAStatus NTV2OutputTestPattern::Init (void)
{
    //  Open the device...
    if (!CNTV2DeviceScanner::GetFirstDeviceFromArgument (mConfig.fDeviceSpec, mDevice))
        {cerr << "## ERROR:  Device '" << mConfig.fDeviceSpec << "' not found" << endl;  return AJA_STATUS_OPEN;}
    mDeviceID = mDevice.GetDeviceID();  //  Keep this ID handy -- it's used frequently
 
    if (!mDevice.IsDeviceReady(false))
        {cerr << "## ERROR:  Device '" << mDevice.GetDisplayName() << "' not ready" << endl;  return AJA_STATUS_INITIALIZE;}
    if (!::NTV2DeviceCanDoPlayback(mDeviceID))
        {cerr << "## ERROR:  '" << mDevice.GetDisplayName() << "' is capture-only" << endl;  return AJA_STATUS_FEATURE;}
 
    const UWord maxNumChannels (::NTV2DeviceGetNumFrameStores(mDeviceID));
 
    if ((mConfig.fOutputChannel == NTV2_CHANNEL1) && (!::NTV2DeviceCanDoFrameStore1Display(mDeviceID)))
    {   //  Some older devices (e.g. Corvid1) can only output from FrameStore 2...
        mConfig.fOutputChannel = NTV2_CHANNEL2;
        cerr << "## WARNING:  '" << mDevice.GetDisplayName() << "' switched to Ch2 (Ch1 is input-only)" << endl;
    }
    if (UWord(mConfig.fOutputChannel) >= maxNumChannels)
    {
        cerr    << "## ERROR:  '" << mDevice.GetDisplayName() << "' can't use Ch" << DEC(mConfig.fOutputChannel+1)
                << " -- only supports Ch1" << (maxNumChannels > 1  ?  string("-Ch") + string(1, char(maxNumChannels+'0'))  :  "") << endl;
        return AJA_STATUS_UNSUPPORTED;
    }
 
    if (!mConfig.fDoMultiFormat)
    {
        mDevice.GetConnections(mSavedConnections);      //  Save current routing, so it can be restored later
        mDevice.GetEveryFrameServices(mSavedTaskMode);  //  Save the current task mode
        if (!mDevice.AcquireStreamForApplication (kDemoAppSignature, int32_t(AJAProcess::GetPid())))
            return AJA_STATUS_BUSY;     //  Device is in use by another app -- fail
    }
    mDevice.SetEveryFrameServices(NTV2_OEM_TASKS);          //  Set OEM service level
 
    if (::NTV2DeviceCanDoMultiFormat(mDeviceID))
        mDevice.SetMultiFormatMode(mConfig.fDoMultiFormat);
    else
        mConfig.fDoMultiFormat = false;
 
    //  Set up the desired video configuration...
    AJAStatus status (SetUpVideo());
    if (AJA_FAILURE(status))
        return status;
 
    #if defined(_DEBUG)
        cerr << mConfig << endl;
    #endif  //  defined(_DEBUG)
    return AJA_STATUS_SUCCESS;
 
}   //  Init
 
 
AJAStatus NTV2OutputTestPattern::SetUpVideo (void)
{
    if (mConfig.fVideoFormat == NTV2_FORMAT_UNKNOWN)
    {
        //  User didn't specify a video format.
        //  Get the device's current video format and use it to create the test pattern...
        bool enabled(false);
        mDevice.IsChannelEnabled(mConfig.fOutputChannel, enabled);
        if (!enabled)
        if (!mDevice.GetVideoFormat (mConfig.fVideoFormat, mConfig.fOutputChannel))
            return AJA_STATUS_FAIL;
 
        //  Read the current VANC mode, as this can affect the NTV2FormatDescriptor and host frame buffer size...
        if (!mDevice.GetVANCMode (mConfig.fVancMode, mConfig.fOutputChannel))
            return AJA_STATUS_FAIL;
    }
 
    if (mConfig.fVideoFormat != NTV2_FORMAT_UNKNOWN)
    {
        //  User specified a video format -- is it legal for this device?
        if (!::NTV2DeviceCanDoVideoFormat(mDeviceID, mConfig.fVideoFormat))
        {   cerr << "## ERROR: '" << mDevice.GetDisplayName() << "' cannot do " << ::NTV2VideoFormatToString(mConfig.fVideoFormat) << endl;
            return AJA_STATUS_UNSUPPORTED;
        }
 
        //  Set the video format -- is it legal for this device?
        if (!mDevice.SetVideoFormat (mConfig.fVideoFormat, /*retail?*/false, /*keepVANC*/false, mConfig.fOutputChannel))
        {   cerr << "## ERROR: SetVideoFormat '" << ::NTV2VideoFormatToString(mConfig.fVideoFormat) << "' failed" << endl;
            return AJA_STATUS_FAIL;
        }
 
        //  Set the VANC mode
        if (!mDevice.SetEnableVANCData (NTV2_IS_VANCMODE_ON(mConfig.fVancMode), NTV2_IS_VANCMODE_TALLER(mConfig.fVancMode), mConfig.fOutputChannel))
        {   cerr << "## ERROR: SetEnableVANCData '" << ::NTV2VANCModeToString(mConfig.fVancMode,true) << "' failed" << endl;
            return AJA_STATUS_FAIL;
        }
    }
 
    //  SD/HD/2K only -- no 4K or 8K...
    if (NTV2_IS_4K_VIDEO_FORMAT(mConfig.fVideoFormat) ||  NTV2_IS_8K_VIDEO_FORMAT(mConfig.fVideoFormat))
    {   cerr << "## ERROR: This demo only supports SD/HD/2K1080, not '" << ::NTV2VideoFormatToString(mConfig.fVideoFormat) << "'" << endl;
        return AJA_STATUS_UNSUPPORTED;
    }
 
    //  This is a "playback" application, so set the board reference to free run...
    if (!mDevice.SetReference(NTV2_REFERENCE_FREERUN))
        return AJA_STATUS_FAIL;
 
    //  Set the FrameStore's pixel format...
    if (!mDevice.SetFrameBufferFormat (mConfig.fOutputChannel, mConfig.fPixelFormat))
        return AJA_STATUS_FAIL;
 
    //  Enable the FrameStore (if currently disabled)...
    mDevice.EnableChannel(mConfig.fOutputChannel);
 
    //  Set the FrameStore mode to "playout" (not capture)...
    if (!mDevice.SetMode (mConfig.fOutputChannel, NTV2_MODE_DISPLAY))
        return AJA_STATUS_FAIL;
 
    //  Enable SDI output from the channel being used, but only if the device supports bi-directional SDI...
    if (::NTV2DeviceHasBiDirectionalSDI(mDeviceID))
        mDevice.SetSDITransmitEnable (mConfig.fOutputChannel, true);
 
    return AJA_STATUS_SUCCESS;
 
}   //  SetUpVideo
 
 
void NTV2OutputTestPattern::RouteOutputSignal (void)
{
    NTV2XptConnections  connections;
 
    //  Build a set of crosspoint connections (input-to-output)
    //  between the relevant signal processing widgets on the device.
    //  By default, the main output crosspoint that feeds the SDI output widgets is the FrameStore's video output:
    NTV2OutputXptID outputXpt (::GetFrameBufferOutputXptFromChannel(mConfig.fOutputChannel, ::IsRGBFormat(mConfig.fPixelFormat)));
    if (::IsRGBFormat(mConfig.fPixelFormat))
    {   //  Even though this demo defaults to using an 8-bit YUV frame buffer,
        //  this code block allows it to work with RGB frame buffers, which
        //  necessitate inserting a CSC between the FrameStore and the SDI output(s)...
        connections.insert(NTV2Connection(::GetCSCInputXptFromChannel(mConfig.fOutputChannel), outputXpt)); //  CSC video input to FrameStore output
        outputXpt = ::GetCSCOutputXptFromChannel(mConfig.fOutputChannel);   //  The CSC output feeds all SDIOut inputs
    }
 
    //  Route all SDI outputs to the outputXpt...
    const NTV2ChannelSet    sdiOutputs  (::NTV2MakeChannelSet(NTV2_CHANNEL1, ::NTV2DeviceGetNumVideoOutputs(mDeviceID)));
    const NTV2Standard      videoStd    (::GetNTV2StandardFromVideoFormat(mConfig.fVideoFormat));
    const NTV2ChannelList   sdiOuts     (::NTV2MakeChannelList(sdiOutputs));
 
    //  Some devices have bi-directional SDI connectors...
    if (::NTV2DeviceHasBiDirectionalSDI(mDeviceID))
        mDevice.SetSDITransmitEnable (sdiOutputs, true);    //  Set to "transmit"
 
    //  For every SDI output, set video standard, and disable level A/B conversion...
    mDevice.SetSDIOutputStandard (sdiOutputs, videoStd);
    mDevice.SetSDIOutLevelAtoLevelBConversion(sdiOutputs, false);
    mDevice.SetSDIOutRGBLevelAConversion(sdiOutputs, false);
 
    //  Connect each SDI output to the main output crosspoint...
    for (size_t ndx(0);  ndx < sdiOutputs.size();  ndx++)
    {
        const NTV2Channel sdiOut(sdiOuts.at(ndx));
        connections.insert(NTV2Connection(::GetSDIOutputInputXpt(sdiOut), outputXpt));
    }
 
    //  And connect analog video output, if the device has one...
    if (::NTV2DeviceGetNumAnalogVideoOutputs(mDeviceID))
        connections.insert(NTV2Connection(::GetOutputDestInputXpt(NTV2_OUTPUTDESTINATION_ANALOG), outputXpt));
 
    //  And connect HDMI video output, if the device has one...
    if (::NTV2DeviceGetNumHDMIVideoOutputs(mDeviceID))
        connections.insert(NTV2Connection(::GetOutputDestInputXpt(NTV2_OUTPUTDESTINATION_HDMI), outputXpt));
 
    //  Apply all the accumulated connections...
    mDevice.ApplySignalRoute(connections, /*replaceExistingRoutes*/true);
 
}   //  RouteOutputSignal
 
 
AJAStatus NTV2OutputTestPattern::EmitPattern (void)
{
    //  Connect the FrameStore to the video output...
    RouteOutputSignal();
 
    //  Allocate a host video buffer that will hold our test pattern raster...
    NTV2FormatDescriptor fd (mConfig.fVideoFormat, mConfig.fPixelFormat, mConfig.fVancMode);
    NTV2Buffer hostBuffer (fd.GetTotalBytes());
    if (hostBuffer.IsNULL())
        return AJA_STATUS_MEMORY;
 
    //  Write the requested test pattern into host buffer...
    NTV2TestPatternGen  testPatternGen;
    testPatternGen.setVANCToLegalBlack(fd.IsVANC());
    if (!testPatternGen.DrawTestPattern (mConfig.fTestPatternName,  fd, hostBuffer))
        return AJA_STATUS_FAIL;
 
    //  Find out which frame is currently being output from the frame store...
    uint32_t currentOutputFrame(0);
    if (!mDevice.GetOutputFrame (mConfig.fOutputChannel, currentOutputFrame))
        return AJA_STATUS_FAIL; //  ReadRegister failure?
 
    //  Now simply transfer the contents of the host buffer to the device's current output frame...
    if (!mDevice.DMAWriteFrame (currentOutputFrame,             //  Device frame number
                                AsULWordPtr(fd.GetRowAddress(hostBuffer.GetHostPointer(), 0)),  //  Host buffer address
                                hostBuffer.GetByteCount()))     //  # bytes to xfer
        return AJA_STATUS_FAIL;
 
    return AJA_STATUS_SUCCESS;
 
}   //  EmitPattern