ni_p2p_test.c

Go to the documentation of this file.

/*******************************************************************************
 *
 * Copyright (C) 2022 NETINT Technologies
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 *
 ******************************************************************************/
 
/*!*****************************************************************************
 *  \file   ni_p2p_test.c
 *
 *  \brief  Application for performing video processing using libxcoder API and
 *          P2P DMA. Its code provides examples on how to programatically use
 *          libxcoder API in conjunction with P2P DMA.
 ******************************************************************************/
 
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <stddef.h>
#include <stdint.h>
 
#define _POSIX_C_SOURCE 200809L // NOLINT(bugprone-reserved-identifier,cert-dcl37-c,cert-dcl51-cpp)
#include <getopt.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <sys/time.h>
#include <time.h>
 
#include "ni_device_api.h"
#include "ni_util.h"
 
// max YUV frame size
#define MAX_YUV_FRAME_SIZE (7680 * 4320 * 3 / 2)
#define MAX_ABGR_FRAME_SIZE (7680 * 4320 * 4)
#define POOL_SIZE 2
#define FILE_NAME_LEN 256
 
int send_fin_flag = 0;
int receive_fin_flag = 0;
int enc_eos_sent = 0;
 
uint32_t number_of_frames = 0;
uint32_t number_of_packets = 0;
uint64_t data_left_size = 0;
int g_repeat = 1;
 
struct timeval start_time;
struct timeval previous_time;
struct timeval current_time;
 
time_t start_timestamp = 0;
time_t previous_timestamp = 0;
time_t current_timestamp = 0;
 
unsigned long total_file_size = 0;
 
uint8_t *g_curr_cache_pos = NULL;
uint8_t *g_yuv_frame[POOL_SIZE] = {NULL, NULL};
uint8_t *g_rgba_frame[POOL_SIZE] = {NULL, NULL};
 
uint8_t g_rgb2yuv_csc = 0;
 
/*!****************************************************************************
 *  \brief  Exit on argument error
 *
 *  \param[in]  arg_name    pointer to argument name
 *        [in]  param       pointer to provided parameter
 *
 *  \return     None        program exit
 ******************************************************************************/
void arg_error_exit(char *arg_name, char *param)
{
    (void)fprintf(stderr, "Error: unrecognized argument for %s, \"%s\"\n", arg_name,
            param);
    exit(-1); // NOLINT(concurrency-mt-unsafe) - single-threaded CLI tool
}
 
/*!****************************************************************************
 *  \brief  Read the next frame
 *
 *  \param[in]  fd          file descriptor of input file
 *  \param[out] p_dst       pointer to place the frame
 *  \param[in]  to_read     number of bytes to copy to the pointer
 *
 *  \return     bytes copied
 ******************************************************************************/
int read_next_chunk_from_file(int fd, uint8_t *p_dst, uint32_t to_read)
{
    uint8_t *tmp_dst = p_dst;
    ni_log(NI_LOG_DEBUG,
        "read_next_chunk_from_file:p_dst %p len %u totalSize %llu left %llu\n",
        tmp_dst, to_read, (unsigned long long)total_file_size, (unsigned long long)data_left_size);
    int to_copy = (int)to_read;
    unsigned long tmpFileSize = to_read;
    if (data_left_size == 0)
    {
        if (g_repeat > 1)
        {
            data_left_size = total_file_size;
            g_repeat--;
            ni_log(NI_LOG_DEBUG, "input processed %d left\n", g_repeat);
            lseek(fd, 0, SEEK_SET);   //back to beginning
        } else
        {
            return 0;
        }
    } else if (data_left_size < to_read)
    {
        tmpFileSize = data_left_size;
        to_copy = (int)data_left_size;
    }
 
    int one_read_size = (int)read(fd, tmp_dst, to_copy);
    if (one_read_size == -1)
    {
        (void)fprintf(stderr, "Error: reading file, quit! left-to-read %lu\n",
                tmpFileSize);
        (void)fprintf(stderr, "Error: input file read error\n");
        return -1;
    }
    data_left_size -= one_read_size;
 
    return to_copy;
}
 
/*!****************************************************************************
 *  \brief  Load the input file into memory
 *
 *  \param [in] filename        name of input file
 *         [out] bytes_read     number of bytes read from file
 *
 *  \return     0 on success
 *              < 0 on error
 ******************************************************************************/
int load_input_file(const char *filename, unsigned long *bytes_read)
{
    struct stat info;
 
    /* Get information on the file */
    if (stat(filename, &info) < 0)
    {
        (void)fprintf(stderr, "Can't stat %s\n", filename);
        return -1;
    }
 
    /* Check the file size */
    if (info.st_size <= 0)
    {
        (void)fprintf(stderr, "File %s is empty\n", filename);
        return -1;
    }
 
    *bytes_read = info.st_size;
 
    return 0;
}
 
/*!****************************************************************************
*  \brief  Recycle hw frames back to Quadra
*
*  \param [in] p2p_frame - array of hw frames to recycle
*
*  \return  Returns the number of hw frames that have been recycled
*******************************************************************************/
int recycle_frames(ni_frame_t p2p_frame[])
{
    int i;
    int cnt = 0;
    ni_retcode_t rc;
 
    for (i = 0; i < POOL_SIZE; i++)
    {
        rc = ni_hwframe_p2p_buffer_recycle(&p2p_frame[i]);
 
        if (rc != NI_RETCODE_SUCCESS)
        {
            (void)fprintf(stderr, "Recycle failed\n");
        }
 
        cnt += (rc == NI_RETCODE_SUCCESS) ? 1 : 0;
    }
 
    return cnt;
}
 
/*!****************************************************************************
 *  \brief  Reads YUV data from input file then calls a special libxcoder API
 *          function to transfer the YUV data into the hardware frame on
 *          the Quadra device.
 *
 *  \param  [in]    p_upl_ctx           pointer to upload session context
 *          [in]    fd                  file descriptor of input file
 *          [in]    p_yuv420p_frame     address of pointer to YUV data
 *          [in]    p_in_frame          pointer to hardware frame
 *          [in]    input_video_width   video width
 *          [in]    input_video_height  video height
 *          [out]   bytes_sent          updated byte count of total data read
 *          [out]   input_exhausted     set to 1 when we reach end-of-file
 *
 *  \return  0 on success
 *          -1 on error
 ******************************************************************************/
int p2p_upload_send_data(ni_session_context_t *p_upl_ctx, int fd,
                         uint8_t **p_yuv420p_frame, ni_frame_t *p_in_frame,
                         int input_video_width, int input_video_height,
                         unsigned long *bytes_sent, int *input_exhausted)
{
    static uint8_t tmp_buf[MAX_YUV_FRAME_SIZE];
    void *p_buffer;
    uint8_t *p_src[NI_MAX_NUM_DATA_POINTERS];
    uint8_t *p_dst[NI_MAX_NUM_DATA_POINTERS];
    int src_stride[NI_MAX_NUM_DATA_POINTERS];
    int src_height[NI_MAX_NUM_DATA_POINTERS];
    int dst_stride[NI_MAX_NUM_DATA_POINTERS] = {0, 0, 0, 0};
    int dst_height[NI_MAX_NUM_DATA_POINTERS] = {0, 0, 0, 0};
    int frame_size;
    int chunk_size;
    int alignedh;
    int Ysize;
    int Usize;
    int Vsize;
    int total_size;
 
    ni_log2(p_upl_ctx, NI_LOG_DEBUG,  "===> p2p upload_send_data <===\n");
 
    /* An 8-bit YUV420 planar frame occupies [(width x height x 3)/2] bytes */
    frame_size = input_video_height * input_video_width * 3 / 2;
 
    chunk_size = read_next_chunk_from_file(fd, tmp_buf, frame_size);
 
    if (chunk_size == 0)
    {
        ni_log2(p_upl_ctx, NI_LOG_DEBUG,  "p2p_upload_send_data: read chunk size 0, eos!\n");
        *input_exhausted = 1;
    }
 
    p_in_frame->video_width = input_video_width;
    p_in_frame->video_height = input_video_height;
    p_in_frame->extra_data_len = 0;
 
    ni_get_hw_yuv420p_dim(input_video_width, input_video_height,
                          p_upl_ctx->bit_depth_factor, 0, dst_stride,
                          dst_height);
 
    ni_log2(p_upl_ctx, NI_LOG_DEBUG,  "p_dst alloc linesize = %d/%d/%d  src height=%d  "
                   "dst height aligned = %d/%d/%d  \n",
                   dst_stride[0], dst_stride[1], dst_stride[2],
                   input_video_height, dst_height[0], dst_height[1],
                   dst_height[2]);
 
    src_stride[0] = input_video_width * p_upl_ctx->bit_depth_factor;
    src_stride[1] = src_stride[2] = src_stride[0] / 2;
 
    src_height[0] = input_video_height;
    src_height[1] = src_height[0] / 2;
    src_height[2] = src_height[1];
 
    p_src[0] = tmp_buf;
    p_src[1] = tmp_buf + (ptrdiff_t)src_stride[0] * src_height[0];
    p_src[2] = p_src[1] + (ptrdiff_t)src_stride[1] * src_height[1];
 
    alignedh = (input_video_height + 1) & ~1;
 
    Ysize = dst_stride[0] * alignedh;
    Usize = dst_stride[1] * alignedh / 2;
    Vsize = dst_stride[2] * alignedh / 2;
 
    total_size = Ysize + Usize + Vsize;
    total_size = ((total_size + 4095) & ~4095) + 4096;
 
    if (*p_yuv420p_frame == NULL)
    {
        if (ni_posix_memalign(&p_buffer, sysconf(_SC_PAGESIZE), total_size))
        {
            (void)fprintf(stderr, "Can't alloc memory\n");
            return -1;
        }
 
        *p_yuv420p_frame = p_buffer;
    }
 
    p_dst[0] = *p_yuv420p_frame;
    p_dst[1] = *p_yuv420p_frame + Ysize;
    p_dst[2] = *p_yuv420p_frame + Ysize + Usize;
    p_dst[3] = NULL;
 
    ni_copy_hw_yuv420p(p_dst, p_src, input_video_width, input_video_height, 1,
                       0, 0, dst_stride, dst_height, src_stride, src_height);
#ifndef _WIN32
    if (ni_uploader_p2p_test_send(p_upl_ctx, *p_yuv420p_frame, total_size,
                                  p_in_frame))
    {
        (void)fprintf(stderr, "Error: failed ni_uploader_p2p_test_send()\n");
        return -1;
    } else
#endif
    {
        *bytes_sent = total_size;
    }
 
    return 0;
}
 
/*!****************************************************************************
 *  \brief  Reads RGBA data from input file then calls a special libxcoder API
 *          function to transfer the RGBA data into the hardware frame on
 *          the Quadra device.
 *
 *  \param  [in]    p_upl_ctx           pointer to upload session context
 *          [in]    fd                  file descriptor of input file
 *          [in]    p_rgba_frame        address of pointer to RGBA data
 *          [in]    p_in_frame          pointer to hardware frame
 *          [in]    input_video_width   video width
 *          [in]    input_video_height  video height
 *          [out]   bytes_sent          updated byte count of total data read
 *          [out]   input_exhausted     set to 1 when we reach end-of-file
 *
 *  \return  0 on success
 *          -1 on error
 ******************************************************************************/
int p2p_upload_rgba_send_data(ni_session_context_t *p_upl_ctx, int fd,
                         uint8_t **p_rgba_frame, ni_frame_t *p_in_frame,
                         int input_video_width, int input_video_height,
                         unsigned long *bytes_sent, int *input_exhausted)
{
    static uint8_t tmp_buf[MAX_ABGR_FRAME_SIZE];
    void *p_buffer;
    uint8_t *p_src,*p_dst;
    int linewidth;
    int frame_size;
    int chunk_size;
    int total_size;
    int row;
 
    ni_log2(p_upl_ctx, NI_LOG_DEBUG,  "===> p2p upload_rgba_send_data <===\n");
 
    /* An 8-bit RGBA frame is in a packed raster (or linear) format */
    /* and occupies width * height * 4 bytes.                       */
    frame_size = input_video_width * input_video_height * 4;
 
    chunk_size = read_next_chunk_from_file(fd, tmp_buf, frame_size);
 
    if (chunk_size == 0)
    {
        ni_log2(p_upl_ctx, NI_LOG_DEBUG,  "p2p_upload_rgba_send_data: eos!\n");
        *input_exhausted = 1;
    }
 
    p_in_frame->video_width = input_video_width;
    p_in_frame->video_height = input_video_height;
    p_in_frame->extra_data_len = 0;
 
    linewidth = input_video_width * 4;
 
    // Round up to 4K
    total_size = NI_VPU_ALIGN4096(frame_size);
 
    if (*p_rgba_frame == NULL)
    {
        if (ni_posix_memalign(&p_buffer, sysconf(_SC_PAGESIZE), total_size))
        {
            (void)fprintf(stderr, "Can't alloc memory\n");
            return -1;
        }
 
        *p_rgba_frame = p_buffer;
    }
 
    p_src = tmp_buf;
    p_dst = *p_rgba_frame;
 
    for (row = 0; row < input_video_height; row++)
    {
        memcpy(p_dst, p_src, linewidth);
        p_src += linewidth;
        p_dst += linewidth;
    }
 
    if (ni_uploader_p2p_test_send(p_upl_ctx, *p_rgba_frame, total_size,
                                  p_in_frame))
    {
        (void)fprintf(stderr, "Error: failed ni_uploader_p2p_test_send()\n");
        return -1;
    } else
    {
        *bytes_sent = total_size;
    }
 
    return 0;
}
 
/*!****************************************************************************
 *  \brief  Prepare frames to simulate P2P transfers
 *
 *  \param [in] p_upl_ctx           pointer to caller allocated uploader
 *                                  session context
 *         [in] input_video_width   video width
 *         [in] input_video_height  video height
 *         [out] p2p_frame          array of hw frames
 *
 *  \return  0  on success
 *          -1  on error
 ******************************************************************************/
int p2p_prepare_frames(ni_session_context_t *p_upl_ctx, int input_video_width,
                       int input_video_height, ni_frame_t p2p_frame[])
{
    int i;
    int ret = 0;
    ni_frame_t *p_in_frame;
 
    // Allocate memory for two hardware frames
    for (i = 0; i < POOL_SIZE; i++)
    {
        p_in_frame = &p2p_frame[i];
 
        p_in_frame->start_of_stream = 0;
        p_in_frame->end_of_stream = 0;
        p_in_frame->force_key_frame = 0;
        p_in_frame->extra_data_len = 0;
 
        // Allocate a hardware ni_frame structure for the encoder
        if (ni_frame_buffer_alloc_hwenc(
                p_in_frame, input_video_width, input_video_height,
                (int)p_in_frame->extra_data_len) != NI_RETCODE_SUCCESS)
        {
            (void)fprintf(stderr, "Error: could not allocate hw frame buffer!");
            ret = -1;
            goto fail_out;
        }
 
#ifndef _WIN32
        // Acquire a hw frame from the upload session. This obtains a handle
        // to Quadra memory from the previously created frame pool.
        if (ni_device_session_acquire(p_upl_ctx, p_in_frame))
        {
            (void)fprintf(stderr, "Error: failed ni_device_session_acquire()\n");
            ret = -1;
            goto fail_out;
        }
#endif
    }
 
    return ret;
 
fail_out:
    for (i = 0; i < POOL_SIZE; i++)
    {
        ni_frame_buffer_free(&(p2p_frame[i]));
    }
 
    return ret;
}
 
/*!****************************************************************************
 *  \brief  Send the Quadra encoder a hardware frame which triggers
 *          Quadra to encode the frame
 *
 *  \param  [in] p_enc_ctx              pointer to encoder context
 *          [in] p_in_frame             pointer to hw frame
 *          [in] input_exhausted        flag indicating this is the last frame
 *          [in/out] need_to_resend     flag indicating need to re-send
 *
 *  \return  0 on success
 *          -1 on failure
 ******************************************************************************/
int encoder_encode_frame(ni_session_context_t *p_enc_ctx,
                         ni_frame_t *p_in_frame, int input_exhausted,
                         int *need_to_resend)
{
    static int started = 0;
    int oneSent;
    ni_session_data_io_t in_data;
 
    ni_log2(p_enc_ctx, NI_LOG_DEBUG,  "===> encoder_encode_frame <===\n");
 
    if (enc_eos_sent == 1)
    {
        ni_log2(p_enc_ctx, NI_LOG_DEBUG,  "encoder_encode_frame: ALL data (incl. eos) sent "
                       "already!\n");
        return 0;
    }
 
    if (*need_to_resend)
    {
        goto send_frame;
    }
 
    p_in_frame->start_of_stream = 0;
 
    // If this is the first frame, mark the frame as start-of-stream
    if (!started)
    {
        started = 1;
        p_in_frame->start_of_stream = 1;
    }
 
    // If this is the last frame, mark the frame as end-of-stream
    p_in_frame->end_of_stream = input_exhausted ? 1 : 0;
    p_in_frame->force_key_frame = 0;
 
send_frame:
 
    in_data.data.frame = *p_in_frame;
    oneSent =
        ni_device_session_write(p_enc_ctx, &in_data, NI_DEVICE_TYPE_ENCODER);
 
    if (oneSent < 0)
    {
        (void)fprintf(stderr,
                "Error: failed ni_device_session_write() for encoder\n");
        *need_to_resend = 1;
        return -1;
    } else if (oneSent == 0 && !p_enc_ctx->ready_to_close)
    {
        *need_to_resend = 1;
        ni_log2(p_enc_ctx, NI_LOG_DEBUG,  "NEEDED TO RESEND");
    } else
    {
        *need_to_resend = 0;
 
        ni_log2(p_enc_ctx, NI_LOG_DEBUG,  "encoder_encode_frame: total sent data size=%u\n",
                       p_in_frame->data_len[3]);
 
        ni_log2(p_enc_ctx, NI_LOG_DEBUG,  "encoder_encode_frame: success\n");
 
        if (p_enc_ctx->ready_to_close)
        {
            enc_eos_sent = 1;
        }
    }
 
    return 0;
}
 
/*!****************************************************************************
 *  \brief  Receive output packet data from the Quadra encoder
 *
 *  \param  [in] p_enc_ctx              pointer to encoder session context
 *          [in] p_out_data             pointer to output data session
 *          [in] p_file                 pointer to file to write the packet
 *          [out] total_bytes_received  running counter of bytes read
 *          [in] print_time             1 = print the time
 *
 *  \return 0 - success got packet
 *          1 - received eos
 *          2 - got nothing, need retry
 *         -1 - failure
 ******************************************************************************/
int encoder_receive_data(ni_session_context_t *p_enc_ctx,
                         ni_session_data_io_t *p_out_data, FILE *p_file,
                         unsigned long long *total_bytes_received,
                         int print_time)
{
    int packet_size = NI_MAX_TX_SZ;
    int rc = 0;
    int end_flag = 0;
    int rx_size = 0;
    int meta_size = (int)p_enc_ctx->meta_size;
    ni_packet_t *p_out_pkt = &(p_out_data->data.packet);
    static int received_stream_header = 0;
 
    ni_log2(p_enc_ctx, NI_LOG_DEBUG,  "===> encoder_receive_data <===\n");
 
    if (NI_INVALID_SESSION_ID == p_enc_ctx->session_id ||
        NI_INVALID_DEVICE_HANDLE == p_enc_ctx->blk_io_handle)
    {
        ni_log2(p_enc_ctx, NI_LOG_DEBUG,  "encode session not opened yet, return\n");
        return 0;
    }
 
    if (p_file == NULL)
    {
        ni_log2(p_enc_ctx, NI_LOG_ERROR,  "Bad file pointer, return\n");
        return -1;
    }
 
    rc = ni_packet_buffer_alloc(p_out_pkt, packet_size);
    if (rc != NI_RETCODE_SUCCESS)
    {
        (void)fprintf(stderr, "Error: malloc packet failed, ret = %d!\n", rc);
        return -1;
    }
 
    /*
     * The first data read from the encoder session context
     * is a stream header read.
     */
    if (!received_stream_header)
    {
        /* Read the encoded stream header */
        rc = ni_encoder_session_read_stream_header(p_enc_ctx, p_out_data);
 
        if (rc > 0)
        {
            /* Write out the stream header */
            if (fwrite((uint8_t *)p_out_pkt->p_data + meta_size,
                       p_out_pkt->data_len - meta_size, 1, p_file) != 1)
            {
                (void)fprintf(stderr, "Error: writing data %u bytes error!\n",
                        p_out_pkt->data_len - meta_size);
                (void)fprintf(stderr, "Error: ferror rc = %d\n", ferror(p_file));
            }
 
            rx_size = rc;
            *total_bytes_received += rx_size;
            number_of_packets++;
            received_stream_header = 1;
        } else if (rc != 0)
        {
            (void)fprintf(stderr, "Error: reading header %d\n", rc);
            return -1;
        }
 
        if (print_time)
        {
            int timeDiff = (int)(current_time.tv_sec - start_time.tv_sec);
            if (timeDiff == 0)
            {
                timeDiff = 1;
            }
            printf("[R] Got:%d   Packets= %u fps=%u  Total bytes %llu\n",
                   rx_size, number_of_packets, number_of_packets / timeDiff,
                   *total_bytes_received);
        }
 
        /* This shouldn't happen */
        if (p_out_pkt->end_of_stream)
        {
            return 1;
        } else if (rc == 0)
        {
            return 2;
        }
    }
 
receive_data:
    rc = ni_device_session_read(p_enc_ctx, p_out_data, NI_DEVICE_TYPE_ENCODER);
 
    end_flag = (int)p_out_pkt->end_of_stream;
    rx_size = rc;
 
    ni_log2(p_enc_ctx, NI_LOG_DEBUG,  "encoder_receive_data: received data size=%d\n", rx_size);
 
    if (rx_size > meta_size)
    {
        if (fwrite((uint8_t *)p_out_pkt->p_data + meta_size,
                   p_out_pkt->data_len - meta_size, 1, p_file) != 1)
        {
            (void)fprintf(stderr, "Error: writing data %u bytes error!\n",
                    p_out_pkt->data_len - meta_size);
            (void)fprintf(stderr, "Error: ferror rc = %d\n", ferror(p_file));
        }
 
        *total_bytes_received += rx_size - meta_size;
        number_of_packets++;
 
        ni_log2(p_enc_ctx, NI_LOG_DEBUG,  "Got:   Packets= %u\n", number_of_packets);
    } else if (rx_size != 0)
    {
        (void)fprintf(stderr, "Error: received %d bytes, <= metadata size %d!\n",
                rx_size, meta_size);
        return -1;
    } else if (!end_flag &&
               (((ni_xcoder_params_t *)(p_enc_ctx->p_session_config))
                   ->low_delay_mode))
    {
        ni_log2(p_enc_ctx, NI_LOG_DEBUG,  "low delay mode and NO pkt, keep reading...\n");
        goto receive_data;
    }
 
    if (print_time)
    {
        int timeDiff = (int)(current_time.tv_sec - start_time.tv_sec);
        if (timeDiff == 0)
        {
            timeDiff = 1;
        }
        printf("[R] Got:%d   Packets= %u fps=%u  Total bytes %llu\n", rx_size,
               number_of_packets, number_of_packets / timeDiff,
               *total_bytes_received);
    }
 
    if (end_flag)
    {
        printf("Encoder Receiving done\n");
        return 1;
    } else if (0 == rx_size)
    {
        return 2;
    }
 
    ni_log2(p_enc_ctx, NI_LOG_DEBUG,  "encoder_receive_data: success\n");
 
    return 0;
}
 
/*!****************************************************************************
 *  \brief  Open an encoder session to Quadra
 *
 *  \param  [out] p_enc_ctx         pointer to an encoder session context
 *          [in]  dst_codec_format  AVC or HEVC
 *          [in]  iXcoderGUID       id to identify the Quadra device
 *          [in]  p_enc_params      sets the encoder parameters
 *          [in]  width             width of frames to encode
 *          [in]  height            height of frames to encode
 *
 *  \return 0 if successful, < 0 otherwise
 ******************************************************************************/
int encoder_open_session(ni_session_context_t *p_enc_ctx, int dst_codec_format,
                         int iXcoderGUID, ni_xcoder_params_t *p_enc_params,
                         int width, int height, ni_frame_t *p_frame)
{
    int ret = 0;
 
    // Enable hardware frame encoding
    p_enc_ctx->hw_action = NI_CODEC_HW_ENABLE;
    p_enc_params->hwframes = 1;
 
    // Provide the first frame to the Quadra encoder
    p_enc_params->p_first_frame = p_frame;
 
    // Specify codec, AVC vs HEVC
    p_enc_ctx->codec_format = dst_codec_format;
 
    p_enc_ctx->p_session_config = p_enc_params;
    p_enc_ctx->session_id = NI_INVALID_SESSION_ID;
 
    // Assign the card GUID in the encoder context to open a session
    // to that specific Quadra device
    p_enc_ctx->device_handle = NI_INVALID_DEVICE_HANDLE;
    p_enc_ctx->blk_io_handle = NI_INVALID_DEVICE_HANDLE;
    p_enc_ctx->hw_id = iXcoderGUID;
 
    if (g_rgb2yuv_csc)
        p_enc_ctx->pixel_format = NI_PIX_FMT_ABGR;
 
    ni_encoder_set_input_frame_format(p_enc_ctx, p_enc_params, width, height, 8,
                                      NI_FRAME_LITTLE_ENDIAN, 1);
 
    // Encoder will operate in P2P mode
    ret = ni_device_session_open(p_enc_ctx, NI_DEVICE_TYPE_ENCODER);
    if (ret != NI_RETCODE_SUCCESS)
    {
        (void)fprintf(stderr, "Error: encoder open session failure\n");
    } else
    {
        printf("Encoder device %d session open successful\n", iXcoderGUID);
    }
 
    return ret;
}
 
/*!****************************************************************************
 *  \brief  Open an upload session to Quadra
 *
 *  \param  [out] p_upl_ctx   pointer to an upload context of the open session
 *          [in]  iXcoderGUID pointer to  Quadra card hw id
 *          [in]  width       width of the frames
 *          [in]  height      height of the frames
 *
 *  \return 0 if successful, < 0 otherwise
 ******************************************************************************/
int uploader_open_session(ni_session_context_t *p_upl_ctx, int *iXcoderGUID,
                          int width, int height)
{
    int ret = 0;
    ni_pix_fmt_t frame_format;
 
    p_upl_ctx->session_id = NI_INVALID_SESSION_ID;
 
    // Assign the card GUID in the encoder context
    p_upl_ctx->device_handle = NI_INVALID_DEVICE_HANDLE;
    p_upl_ctx->blk_io_handle = NI_INVALID_DEVICE_HANDLE;
 
    // Assign the card id to specify the specific Quadra device
    p_upl_ctx->hw_id = *iXcoderGUID;
 
    // Assign the pixel format we want to use
    frame_format = g_rgb2yuv_csc ? NI_PIX_FMT_ABGR : NI_PIX_FMT_YUV420P;
 
    // Set the input frame format of the upload session
    ni_uploader_set_frame_format(p_upl_ctx, width, height, frame_format,
                                 1);
 
    ret = ni_device_session_open(p_upl_ctx, NI_DEVICE_TYPE_UPLOAD);
    if (ret != NI_RETCODE_SUCCESS)
    {
        (void)fprintf(stderr, "Error: uploader_open_session failure!\n");
        return ret;
    } else
    {
        printf("Uploader device %d session opened successfully\n",
               *iXcoderGUID);
        *iXcoderGUID = p_upl_ctx->hw_id;
    }
 
    // Create a P2P frame pool for the uploader sesson of pool size 2
    ret = ni_device_session_init_framepool(p_upl_ctx, POOL_SIZE, 1);
    if (ret < 0)
    {
        (void)fprintf(stderr, "Error: Can't create frame pool\n");
        ni_device_session_close(p_upl_ctx, 1, NI_DEVICE_TYPE_UPLOAD);
    } else
    {
        printf("Uploader device %d configured successfully\n", *iXcoderGUID);
    }
 
    return ret;
}
 
/*!****************************************************************************
 *  \brief    Print usage information
 *
 *  \param    none
 *
 *  \return   none
 ******************************************************************************/
void print_usage(void)
{
    printf("Video encoder/P2P application directly using Netint "
           "Libxcoder release v%s\n"
           "Usage: xcoderp2p [options]\n"
           "\n"
           "options:\n"
           "--------------------------------------------------------------------------------\n"
           "  -h | --help        Show help.\n"
           "  -v | --version     Print version info.\n"
           "  -l | --loglevel    Set loglevel of libxcoder API.\n"
           "                     [none, fatal, error, info, debug, trace]\n"
           "                     Default: info\n"
           "  -c | --card        Set card index to use.\n"
           "                     See `ni_rsrc_mon` for cards on system.\n"
           "                     (Default: 0)\n"
           "  -i | --input       Input file path.\n"
           "  -r | --repeat      (Positive integer) to Repeat input X times "
           "for performance \n"
           "                     test. (Default: 1)\n"
           "  -s | --size        Resolution of input file in format "
           "WIDTHxHEIGHT.\n"
           "                     (eg. '1920x1080')\n"
           "  -m | --mode        Input to output codec processing mode in "
           "format:\n"
           "                     INTYPE2OUTTYPE. [p2a, p2h, r2a, r2h]\n"
           "                     Type notation: p=P2P, a=AVC, h=HEVC, r=ABGR\n"
           "  -o | --output      Output file path.\n",
           NI_XCODER_REVISION);
}
 
/*!****************************************************************************
 *  \brief  Parse user command line arguments
 *
 *  \param [in] argc                argument count
 *         [in] argv                argument vector
 *         [out] input_filename     input filename
 *         [out] output_filename    output filename
 *         [out] iXcoderGUID        Quadra device
 *         [out] arg_width          resolution width
 *         [out] arg_height         resolution height
 *         [out] dst_codec_format   codec (AVC vs HEVC)
 *
 *  \return nothing                 program exit on error
 ******************************************************************************/
void parse_arguments(int argc, char *argv[], char *input_filename,
                     char *output_filename, int *iXcoderGUID, int *arg_width,
                     int *arg_height, int *dst_codec_format)
{
    char xcoderGUID[32];
    char mode_description[128];
    char *n;   // used for parsing width and height from --size
    size_t i;
    int opt;
    int opt_index;
    ni_log_level_t log_level;
 
    static const char *opt_string = "hvl:c:i:s:m:o:r:";
    static const struct option long_options[] = {
        {"help", no_argument, NULL, 'h'},
        {"version", no_argument, NULL, 'v'},
        {"loglevel", no_argument, NULL, 'l'},
        {"card", required_argument, NULL, 'c'},
        {"input", required_argument, NULL, 'i'},
        {"size", required_argument, NULL, 's'},
        {"mode", required_argument, NULL, 'm'},
        {"output", required_argument, NULL, 'o'},
        {"repeat", required_argument, NULL, 'r'},
        {NULL, 0, NULL, 0},
    };
 
    while ((opt = getopt_long(argc, argv, opt_string, long_options, // NOLINT(concurrency-mt-unsafe) - single-threaded CLI tool
                              &opt_index)) != -1)
    {
        switch (opt)
        {
            case 'h':
                print_usage();
                exit(0); // NOLINT(concurrency-mt-unsafe) - single-threaded CLI tool
            case 'v':
                printf("Release ver: %s\n"
                       "API ver:     %s\n"
                       "Date:        %s\n"
                       "ID:          %s\n",
                       NI_XCODER_REVISION, LIBXCODER_API_VERSION,
                       NI_SW_RELEASE_TIME, NI_SW_RELEASE_ID);
                exit(0); // NOLINT(concurrency-mt-unsafe) - single-threaded CLI tool
            case 'l':
                log_level = arg_to_ni_log_level(optarg);
                if (log_level != NI_LOG_INVALID)
                {
                    ni_log_set_level(log_level);
                } else {
                    arg_error_exit("-l | --loglevel", optarg);
                }
                break;
            case 'c':
                ni_strcpy(xcoderGUID, sizeof(xcoderGUID), optarg);
                *iXcoderGUID = (int)strtol(optarg, &n, 10);
                // No numeric characters found in left side of optarg
                if (n == xcoderGUID)
                    arg_error_exit("-c | --card", optarg);
                break;
            case 'i':
                ni_strcpy(input_filename, FILE_NAME_LEN, optarg);
                break;
            case 's':
                *arg_width = (int)strtol(optarg, &n, 10);
                {
                    int32_t tmp_height;
                    if (ni_strtoi(n + 1, &tmp_height) != NI_RETCODE_SUCCESS)
                        arg_error_exit("-s | --size", optarg);
                    *arg_height = (int)tmp_height;
                }
                if ((*n != 'x') || (!*arg_width || !*arg_height))
                    arg_error_exit("-s | --size", optarg);
                break;
            case 'm':
                if (!(strlen(optarg) == 3))
                    arg_error_exit("-m | --mode", optarg);
 
                // convert to lower case for processing
                for (i = 0; i < strlen(optarg); i++)
                    optarg[i] = (char)tolower((unsigned char)optarg[i]);
 
                if (strcmp(optarg, "p2a") != 0 && strcmp(optarg, "p2h") != 0 &&
                    strcmp(optarg, "r2a") != 0 && strcmp(optarg, "r2h") != 0)
                    arg_error_exit("-, | --mode", optarg);
 
                // determine codec
                ni_sprintf(mode_description, 128, "P2P + Encoding");
 
                g_rgb2yuv_csc = (optarg[0] == 'r') ? 1 : 0;
 
                if (optarg[2] == 'a')
                {
                    *dst_codec_format = NI_CODEC_FORMAT_H264;
                    ni_strcat(mode_description, 128, " to AVC");
                }
 
                if (optarg[2] == 'h')
                {
                    *dst_codec_format = NI_CODEC_FORMAT_H265;
                    ni_strcat(mode_description, 128, " to HEVC");
                }
                printf("%s...\n", mode_description);
 
                break;
            case 'o':
                ni_strcpy(output_filename, FILE_NAME_LEN, optarg);
                break;
            case 'r':
            {
                int32_t tmp_repeat;
                if (ni_strtoi(optarg, &tmp_repeat) != NI_RETCODE_SUCCESS)
                    arg_error_exit("-r | --repeat", optarg);
                if (!(tmp_repeat >= 1))
                    arg_error_exit("-r | --repeat", optarg);
                g_repeat = (int)tmp_repeat;
                break;
            }
            default:
                print_usage();
                exit(1); // NOLINT(concurrency-mt-unsafe) - single-threaded CLI tool
        }
    }
 
    // Check required args are present
    if (!input_filename[0])
    {
        printf("Error: missing argument for -i | --input\n");
        exit(-1); // NOLINT(concurrency-mt-unsafe) - single-threaded CLI tool
    }
 
    if (!output_filename[0])
    {
        printf("Error: missing argument for -o | --output\n");
        exit(-1); // NOLINT(concurrency-mt-unsafe) - single-threaded CLI tool
    }
}
 
int main(int argc, char *argv[])
{
    static char input_filename[FILE_NAME_LEN];
    static char output_filename[FILE_NAME_LEN];
    unsigned long total_bytes_sent;
    unsigned long long total_bytes_received;
    int input_video_width;
    int input_video_height;
    int iXcoderGUID = 0;
    int arg_width = 0;
    int arg_height = 0;
    int input_exhausted = 0;
    int num_post_recycled = 0;
    int dst_codec_format = 0;
    int ret;
    int timeDiff;
    int print_time;
    int need_to_resend = 0;
    int render_index = 0;
    int encode_index = -1;
    FILE *p_file = NULL;
    ni_xcoder_params_t api_param;
    ni_session_context_t enc_ctx = {0};
    ni_session_context_t upl_ctx = {0};
    ni_frame_t p2p_frame[POOL_SIZE];
    ni_session_data_io_t out_packet = {0};
    int input_file_fd = -1;
 
    parse_arguments(argc, argv, input_filename, output_filename, &iXcoderGUID,
                    &arg_width, &arg_height, &dst_codec_format);
 
    // Load input file into memory
    if (load_input_file(input_filename, &total_file_size) < 0)
    {
        exit(-1); // NOLINT(concurrency-mt-unsafe) - single-threaded CLI tool
    }
 
    data_left_size = total_file_size;
 
    // Create output file
    if (strcmp(output_filename, "null") != 0)
    {
        ni_fopen(&p_file, output_filename, "wb");
        if (p_file == NULL)
        {
            (void)fprintf(stderr, "Error: cannot open %s\n", output_filename);
            goto end;
        }
    }
 
    printf("SUCCESS: Opened output file: %s\n", output_filename);
 
    if (ni_device_session_context_init(&enc_ctx) < 0)
    {
        (void)fprintf(stderr, "Error: init encoder context error\n");
        return -1;
    }
 
    if (ni_device_session_context_init(&upl_ctx) < 0)
    {
        (void)fprintf(stderr, "Error: init uploader context error\n");
        return -1;
    }
 
    total_bytes_received = 0;
    total_bytes_sent = 0;
 
    send_fin_flag = 0;
    receive_fin_flag = 0;
 
    printf("User video resolution: %dx%d\n", arg_width, arg_height);
 
    if (arg_width == 0 || arg_height == 0)
    {
        input_video_width = 1280;
        input_video_height = 720;
    } else
    {
        input_video_width = arg_width;
        input_video_height = arg_height;
    }
 
    ni_gettimeofday(&start_time, NULL);
    ni_gettimeofday(&previous_time, NULL);
    ni_gettimeofday(&current_time, NULL);
    start_timestamp = previous_timestamp = current_timestamp = time(NULL);
 
    printf("P2P Encoding resolution: %dx%d\n", input_video_width,
           input_video_height);
 
    // Open an uploader session to Quadra
    if (uploader_open_session(&upl_ctx, &iXcoderGUID, arg_width, arg_height))
    {
        goto end;
    }
 
    // Configure the encoder parameter structure. We'll use some basic
    // defaults: 30 fps, 200000 bps CBR encoding, AVC or HEVC encoding
    if (ni_encoder_init_default_params(&api_param, 30, 1, 200000, arg_width,
                                       arg_height, enc_ctx.codec_format) < 0)
    {
        (void)fprintf(stderr, "Error: encoder init default set up error\n");
        ni_device_session_close(&upl_ctx, 1, NI_DEVICE_TYPE_UPLOAD);
        return -1;
    }
 
    // For P2P demo, change some of the encoding parameters from
    // the default. Enable low delay encoding.
    ret = ni_encoder_params_set_value(&api_param, "lowDelay", "1");
    if (ret != NI_RETCODE_SUCCESS)
    {
        (void)fprintf(stderr, "Error: can't set low delay mode %d\n", ret);
        ni_device_session_close(&upl_ctx, 1, NI_DEVICE_TYPE_UPLOAD);
        return -1;
    }
 
    // Use a GOP preset of 9 which represents a GOP pattern of
    // IPPPPPPP....This will be low latency.
    ret = ni_encoder_params_set_value(&api_param, "gopPresetIdx", "9");
    if (ret != NI_RETCODE_SUCCESS)
    {
        (void)fprintf(stderr, "Error: can't set gop preset %d\n", ret);
        ni_device_session_close(&upl_ctx, 1, NI_DEVICE_TYPE_UPLOAD);
        return -1;
    }
 
    // Prepare two frames for double buffering
    ret = p2p_prepare_frames(&upl_ctx, input_video_width, input_video_height,
                             p2p_frame);
 
    if (ret < 0)
    {
        ni_device_session_close(&upl_ctx, 1, NI_DEVICE_TYPE_UPLOAD);
        goto end;
    }
 
    // Open the encoder session with given parameters
    ret = encoder_open_session(&enc_ctx, dst_codec_format, iXcoderGUID,
                               &api_param, arg_width, arg_height,
                               &p2p_frame[render_index]);
    if (ret < 0)
    {
        ni_device_session_close(&upl_ctx, 1, NI_DEVICE_TYPE_UPLOAD);
        goto end;
    }
 
#ifdef _WIN32
    input_file_fd = open(input_filename, O_RDONLY | O_BINARY);
#else
    input_file_fd = open(input_filename, O_RDONLY);
#endif
 
    if (input_file_fd < 0)
    {
        (void)fprintf(stderr, "Error: can not open input file %s\n", input_filename);
        goto end;
    }
 
    if (g_rgb2yuv_csc)
    {
        // upload an rgba frame to quadra
        if (p2p_upload_rgba_send_data(
                &upl_ctx, input_file_fd, &g_rgba_frame[render_index],
                &p2p_frame[render_index], input_video_width, input_video_height,
                &total_bytes_sent, &input_exhausted))
        {
            (void)fprintf(stderr, "Error: upload frame error\n");
            ni_device_session_close(&upl_ctx, 1, NI_DEVICE_TYPE_UPLOAD);
            goto end;
        }
    } else
    {
        /* send out a frame to do rendering */
        if (p2p_upload_send_data(
                &upl_ctx, input_file_fd, &g_yuv_frame[render_index],
                &p2p_frame[render_index], input_video_width, input_video_height,
                &total_bytes_sent, &input_exhausted))
        {
            (void)fprintf(stderr, "Error: upload frame error\n");
            close(input_file_fd);
            return -1;
        }
    }
 
    while (send_fin_flag == 0 || receive_fin_flag == 0)
    {
        ni_gettimeofday(&current_time, NULL);
 
        // Print the time if >= 1 second has passed
        print_time = ((current_time.tv_sec - previous_time.tv_sec) > 1);
        encode_index = render_index;
 
        // Encode the frame
        send_fin_flag = encoder_encode_frame(&enc_ctx, &p2p_frame[encode_index],
                                             input_exhausted, &need_to_resend);
 
        // Error, exit
        if (send_fin_flag == 2)
        {
            break;
        }
 
        // Switch to the other hw frame buffer
        render_index = !render_index;
 
        // Fill the frame buffer with YUV data while the previous frame is being encoded
        if (!input_exhausted && need_to_resend == 0)
        {
            if (g_rgb2yuv_csc)
            {
                if (p2p_upload_rgba_send_data(
                        &upl_ctx, input_file_fd, &g_rgba_frame[render_index],
                        &p2p_frame[render_index], input_video_width,
                        input_video_height, &total_bytes_sent, &input_exhausted))
                {
                    (void)fprintf(stderr, "Error: upload frame error\n");
                    close(input_file_fd);
                    return -1;
                }
            }
            else
            {
                if (p2p_upload_send_data(
                        &upl_ctx, input_file_fd, &g_yuv_frame[render_index],
                        &p2p_frame[render_index], input_video_width,
                        input_video_height, &total_bytes_sent, &input_exhausted))
                {
                    (void)fprintf(stderr, "Error: upload frame error\n");
                    close(input_file_fd);
                    return -1;
                }
            }
        }
 
        // Receive encoded packet data from the encoder
        receive_fin_flag = encoder_receive_data(
            &enc_ctx, &out_packet, p_file, &total_bytes_received, print_time);
 
        if (print_time)
        {
            previous_time = current_time;
        }
 
        // Error or eos
        if (receive_fin_flag < 0 || out_packet.data.packet.end_of_stream)
        {
            break;
        }
    }
 
    timeDiff = (int)(current_time.tv_sec - start_time.tv_sec);
    timeDiff = (timeDiff > 0) ? timeDiff : 1;   // avoid division by zero
 
    printf("[R] Got:  Packets= %u fps=%u  Total bytes %llu\n",
           number_of_packets, number_of_packets / timeDiff,
           total_bytes_received);
 
    // Recycle the hardware frames back to the pool prior
    // to closing the uploader session.
    num_post_recycled = recycle_frames(p2p_frame);
 
    ni_log(NI_LOG_DEBUG, "Cleanup recycled %d internal buffers\n", num_post_recycled);
 
    ni_device_session_close(&enc_ctx, 1, NI_DEVICE_TYPE_ENCODER);
    ni_device_session_close(&upl_ctx, 1, NI_DEVICE_TYPE_UPLOAD);
 
    ni_device_session_context_clear(&enc_ctx);
    ni_device_session_context_clear(&upl_ctx);
 
    for (int i = 0; i < POOL_SIZE; i++)
    {
        ni_frame_buffer_free(&(p2p_frame[i]));
    }
 
    ni_packet_buffer_free(&(out_packet.data.packet));
 
end:
    close(input_file_fd);
 
    if (p_file)
    {
        (void)fclose(p_file);
    }
 
    printf("All done\n");
 
    return 0;
}