/*
 * SPDX-FileCopyrightText: Copyright (c) 2021-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 * SPDX-License-Identifier: LicenseRef-NvidiaProprietary
 *
 * NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
 * property and proprietary rights in and to this material, related
 * documentation and any modifications thereto. Any use, reproduction,
 * disclosure or distribution of this material and related documentation
 * without an express license agreement from NVIDIA CORPORATION or
 * its affiliates is strictly prohibited.
 */

/*
 * NvSciStream Event Loop Driven Sample App - ipc blocks
 */

#include <stdlib.h>
#include <stdio.h>
#include "nvsciipc.h"
#include "nvscistream.h"
#include "block_info.h"
#include "event_loop.h"

/*
 * Ipc blocks do not require any block-specific interactions so we
 *   use the set of common functions to handle its events. However
 *   they do have an additional data field which needs to be cleaned
 *   up when the block is destroyed, so we use more than the common
 *   data structure and delete function.
 */

/* Create and register a new ipcsrc block */
int32_t createIpcSrc(
    NvSciStreamBlock* ipcsrc,
    const char*       channel,
    bool              useExtEventService)
{
    NvSciError err;
    uint32_t i;
    uint32_t slot = 0;

    /* Create a data structure to track the block's status */
    BlockData* blockData = createCommon("IpcSrc", 0);
    if (NULL == blockData) {
        return 0;
    }

    for (i=0; i< MAX_CONSUMERS; i++) {
        if (!strcmp(ipcEP[i].ipcChannel, channel)) {
            slot = i;
            break;
        }
    }

    /* Open the named channel */
    err = NvSciIpcOpenEndpoint(channel, &ipcEP[slot].ipcEndpoint);
    if (NvSciError_Success != err) {
        printf("Failed (%x) to open channel (%s) for IpcSrc\n",
               err, channel);
        deleteCommon(blockData);
        return 0;
    }

    err = NvSciIpcResetEndpointSafe(ipcEP[slot].ipcEndpoint);
    if (NvSciError_Success != err) {
        printf("Failed (%x) to reset IPC endpoint", err);
    }

    /* Create a ipcsrc block */
    err = NvSciStreamIpcSrcCreate(ipcEP[slot].ipcEndpoint,
                                  sciSyncModule,
                                  sciBufModule,
                                  &blockData->block);
    if (NvSciError_Success != err) {
        printf("Failed (%x) to create IpcSrc block\n", err);
        deleteCommon(blockData);
        return 0;
    }

    /* Register block with event handling mechanism */
    if (!eventFuncs->reg(blockData->block, blockData, handleCommon)) {
        deleteCommon(blockData);
        return 0;
    }

    /* Configuate block to use the external event service for
     *  internal I/O messages */
    if (useExtEventService && !eventFuncs->regInt(blockData->block)) {
        deleteCommon(blockData);
        return 0;
    }

    *ipcsrc = blockData->block;
    ipcEP[slot].ipcBlock = blockData->block;
    ipcEP[slot].ipcOpened = true;
    ipcEP[slot].ipcConnected = true;

    return 1;
}


/* Create and register a new ipcsrc block */
int32_t createIpcSrc2(
    NvSciStreamBlock* ipcsrc,
    NvSciIpcEndpoint  endpoint,
    bool              useExtEventService)
{
    NvSciError err;

    /* Create a data structure to track the block's status */
    BlockData* blockData = createCommon("IpcSrc", 0);
    if (NULL == blockData) {
        return 0;
    }

    /* Create a ipcsrc block */
    err = NvSciStreamIpcSrcCreate(endpoint,
                                  sciSyncModule,
                                  sciBufModule,
                                  &blockData->block);
    if (NvSciError_Success != err) {
        printf("Failed (%x) to create IpcSrc block\n", err);
        deleteCommon(blockData);
        return 0;
    }

    /* Register block with event handling mechanism */
    if (!eventFuncs->reg(blockData->block, blockData, handleCommon)) {
        deleteCommon(blockData);
        return 0;
    }

    /* Configuate block to use the external event service for
     *  internal I/O messages */
    if (useExtEventService && !eventFuncs->regInt(blockData->block)) {
        deleteCommon(blockData);
        return 0;
    }

    *ipcsrc = blockData->block;

    return 1;
}

/* Create and register a new ipcdst block */
int32_t createIpcDst(
    NvSciStreamBlock* ipcdst,
    const char*       channel,
    bool              useExtEventService)
{
    NvSciError err;

    /* Create a data structure to track the block's status */
    BlockData* blockData = createCommon("IpcDst", 0);
    if (NULL == blockData) {
        return 0;
    }

    /* Open the named channel */
    err = NvSciIpcOpenEndpoint(channel, &ipcEP[0].ipcEndpoint);
    if (NvSciError_Success != err) {
        printf("Failed (%x) to open channel (%s) for IpcDst\n",
               err, channel);
        deleteCommon(blockData);
        return 0;
    }

    err = NvSciIpcResetEndpointSafe(ipcEP[0].ipcEndpoint);
    if (NvSciError_Success != err) {
        printf("Failed (%x) to reset IPC endpoint", err);
    }

    /* Create a ipcdst block */
    err = NvSciStreamIpcDstCreate(ipcEP[0].ipcEndpoint,
                                  sciSyncModule,
                                  sciBufModule,
                                  &blockData->block);
    if (NvSciError_Success != err) {
        printf("Failed (%x) to create IpcDst block\n", err);
        deleteCommon(blockData);
        return 0;
    }

    /* Register block with event handling mechanism */
    if (!eventFuncs->reg(blockData->block, blockData, handleCommon)) {
        deleteCommon(blockData);
        return 0;
    }

    /* Configuate block to use the external event service for
     *  internal I/O messages */
    if (useExtEventService && !eventFuncs->regInt(blockData->block)) {
        deleteCommon(blockData);
        return 0;
    }

    *ipcdst = blockData->block;
    return 1;
}