main.cpp

Go to the documentation of this file.

/* SPDX-License-Identifier: MIT */
//  Includes
#include "ntv2utils.h"
#include "ajatypes.h"
#include "ntv2m31publicinterface.h"
#include "ajabase/common/options_popt.h"
#include "ajabase/system/systemtime.h"
#include "ntv2encodehevcvif.h"
#include <signal.h>
#include <iostream>
#include <iomanip>

using namespace std;


//  Globals
const M31VideoPreset kCodecPreset[] = {
    M31_VIF_3840X2160_420_8_50p,
    M31_VIF_3840X2160_420_8_5994p,
    M31_VIF_3840X2160_420_8_60p,

    M31_VIF_3840X2160_420_10_50p,
    M31_VIF_3840X2160_420_10_5994p,
    M31_VIF_3840X2160_420_10_60p,

    M31_VIF_3840X2160_422_10_50p,
    M31_VIF_3840X2160_422_10_5994p,
    M31_VIF_3840X2160_422_10_60p,
};

const NTV2FrameBufferFormat kCodecFormat[] = {
    NTV2_FBF_8BIT_YCBCR_420PL2,
    NTV2_FBF_10BIT_YCBCR_420PL2,
    NTV2_FBF_8BIT_YCBCR_422PL2,
    NTV2_FBF_10BIT_YCBCR_422PL2
};

static const size_t gNumCodecPresets    (sizeof(kCodecPreset)/sizeof(M31VideoPreset));
static const size_t gNumCodecFormats    (sizeof(kCodecFormat)/sizeof(NTV2FrameBufferFormat));
static const int    kNoSelection        (1000000000);
static bool         gGlobalQuit         (false);    //  Set this "true" to exit gracefully


static void SignalHandler (int inSignal)
{   (void) inSignal;
    gGlobalQuit = true;
}

static int printPresets (void)
{
    cout << "M31 Presets" << endl;
    for (size_t ndx(0);  ndx < gNumCodecPresets;  ndx++)
    {   const string presetStr (::NTV2M31VideoPresetToString (kCodecPreset[ndx], true));
        if (!presetStr.empty())
            cout << ndx << ":  " << presetStr << endl;
    }
    return 0;
}

static int printFormats (void)
{
    cout << "M31 Formats" << endl;
    for (size_t ndx(0);  ndx < gNumCodecFormats;  ndx++)
    {   const string pixFormatStr (::NTV2FrameBufferFormatToString(kCodecFormat[ndx], true));
        if (!pixFormatStr.empty())
            cout << ndx << ":  " << pixFormatStr << endl;
    }
    return 0;
}


int main (int argc, const char ** argv)
{
    M31VideoPreset          m31Preset       (M31_NUMVIDEOPRESETS);              //  Codec preset
    NTV2FrameBufferFormat   pixelFormat     (NTV2_FBF_NUMFRAMEBUFFERFORMATS);   //  Codec frame format
    char *                  pWhatToList     (AJA_NULL);                         //  Value of --list argument
    char *                  pDeviceSpec     (AJA_NULL);                         //  Device specifier string, if any
    int                     codecFormat     (kNoSelection);                     //  Codec format index
    int                     codecPreset     (kNoSelection);                     //  Codec preset index
    int                     infoData        (0);                                //  Picture and encoded information
    int                     maxFrames       (kNoSelection);                     //  Maximum number of catpured frames
    int                     audioChannels   (0);                                //  Number of audio channels
    poptContext             optionsContext;                                     //  Context for parsing command line arguments

    AJADebug::Open();

    //  Command line option descriptions:
    const struct poptOption userOptionsTable [] =
    {
        {"audio",       'a',    POPT_ARG_INT,           &audioChannels, 0,  "number of audio channels", "1-16"},
        {"device",      'd',    POPT_ARG_STRING,        &pDeviceSpec,   0,  "device to use",            "index#, serial#, or model"},
        {"format",      'f',    POPT_ARG_INT,           &codecFormat,   0,  "format to use",            "use '-lf' for list"},
        {"preset",      'p',    POPT_ARG_INT,           &codecPreset,   0,  "codec preset to use",      "use '-lp' for list"},
        {"list",        'l',    POPT_ARG_STRING,        &pWhatToList,   0,  "list options",             "p or f (presets or formats)"},
        {"info",        'i',    POPT_ARG_NONE,          &infoData,      0,  "write encoded info file",  ""},
        {"maxframes",   'x',    POPT_ARG_INT |
                                POPT_ARGFLAG_OPTIONAL,  &maxFrames,     0,  "limit output file frames", ""},
        POPT_AUTOHELP
        POPT_TABLEEND
    };

    //  Read command line arguments...
    optionsContext = ::poptGetContext (AJA_NULL, argc, argv, userOptionsTable, 0);
    ::poptGetNextOpt(optionsContext);
    optionsContext = ::poptFreeContext(optionsContext);

    if (pWhatToList && string(pWhatToList) == "p")
        return printPresets();
    else if (pWhatToList && string(pWhatToList) == "f")
        return printFormats();
    else if (pWhatToList)
        {cerr << "## ERROR:  Invalid argument to --list option, expected 'p' or 'f'" << endl;  return 2;}

    if ((codecFormat != kNoSelection) && (codecFormat >= int(gNumCodecFormats)))
        {cerr << "## ERROR:  Invalid M31 format " << codecFormat << " -- expected 0 thru " << (gNumCodecFormats) << endl;  return 2;}

    if (codecFormat < int(gNumCodecFormats))
        pixelFormat = kCodecFormat[codecFormat];

    if ((codecPreset != kNoSelection) && (codecPreset >= int(gNumCodecPresets)))
        {cerr << "## ERROR:  Invalid M31 preset " << codecPreset << " -- expected 0 thru " << (gNumCodecPresets) << endl;  return 2;}

    if (codecPreset < int(gNumCodecPresets))
        m31Preset = kCodecPreset[codecPreset];

    ::signal (SIGINT, SignalHandler);
    #if defined(AJAMac)
        ::signal (SIGHUP, SignalHandler);
        ::signal (SIGQUIT, SignalHandler);
    #endif

    //  Instantiate the NTV2EncodeHEVCVif object, using the specified AJA device
    const bool  infoMode (infoData ? true : false); //  Picture and encoded information mode
    const uint32_t hevcMaxFrames (maxFrames >= kNoSelection ? 0xffffffff : uint32_t(maxFrames));
    NTV2EncodeHEVCVif hevcVifEncoder (pDeviceSpec ? string (pDeviceSpec) : "0",
                                        m31Preset, pixelFormat, UWord(audioChannels),
                                        infoMode, hevcMaxFrames);

    //  Initialize the NTV2EncodeHEVCVif instance
    if (AJA_FAILURE(hevcVifEncoder.Init()))
        {cerr << "## ERROR:  Initialization failed" << endl;  return 1;}

    //  Get the current codec preset
    m31Preset = hevcVifEncoder.GetCodecPreset();

    cout    << endl
            << "Capture: " << ::NTV2M31VideoPresetToString (m31Preset, false) << endl
            << endl;

    //  Run the capturer...
    hevcVifEncoder.Run();

    cout << "           Capture  Capture" << endl
         << "   Frames   Frames   Buffer" << endl
         << "Processed  Dropped    Level" << endl;

    //  Loop til told to stop...
    do
    {
        AVHevcStatus info;
        hevcVifEncoder.GetStatus(&info);
        cout << setw(9) << info.framesProcessed
             << setw(9) << info.framesDropped
             << setw(9) << info.bufferLevel
             << "\r" << flush;
        AJATime::Sleep(1000);
    } while (!gGlobalQuit);
    cout << endl;

    //  Quit the capturer
    hevcVifEncoder.Quit();

    return 0;

}   //  main