IChannelPacket.hpp

Go to the documentation of this file.

 
//
// Notice
// ALL NVIDIA DESIGN SPECIFICATIONS AND CODE ("MATERIALS") ARE PROVIDED "AS IS" NVIDIA MAKES
// NO REPRESENTATIONS, WARRANTIES, EXPRESSED, IMPLIED, STATUTORY, OR OTHERWISE WITH RESPECT TO
// THE MATERIALS, AND EXPRESSLY DISCLAIMS ANY IMPLIED WARRANTIES OF NONINFRINGEMENT,
// MERCHANTABILITY, OR FITNESS FOR A PARTICULAR PURPOSE.
//
// NVIDIA CORPORATION & AFFILIATES assumes no responsibility for the consequences of use of such
// information or for any infringement of patents or other rights of third parties that may
// result from its use. No license is granted by implication or otherwise under any patent
// or patent rights of NVIDIA CORPORATION & AFFILIATES. No third party distribution is allowed unless
// expressly authorized by NVIDIA. Details are subject to change without notice.
// This code supersedes and replaces all information previously supplied.
// NVIDIA CORPORATION & AFFILIATES products are not authorized for use as critical
// components in life support devices or systems without express written approval of
// NVIDIA CORPORATION & AFFILIATES.
//
// SPDX-FileCopyrightText: Copyright (c) 2020-2022 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.
//
 
#ifndef DW_FRAMEWORK_ICHANNEL_PACKET_HPP_
#define DW_FRAMEWORK_ICHANNEL_PACKET_HPP_
 
#include <typeinfo>
#include <cstddef>
#include <dwcgf/channel/ChannelPacketTypes.hpp>
#include <dwcgf/channel/ChannelParameters.hpp>
#include <nvscibuf.h>
#include <nvscisync.h>
#include <dw/core/language/Function.hpp>
 
namespace dw
{
namespace framework
{
 
class IChannelPacket
{
public:
    // Enable ownership via this interface
    virtual ~IChannelPacket() = default;
 
    virtual GenericData getGenericData() = 0;
 
    class SocketCallbacks
    {
    public:
        virtual uint8_t* getBuffer() = 0;
        virtual size_t getBufferSize() = 0;
        virtual size_t serialize() = 0;
        virtual void deserialize(size_t) = 0;
    };
 
    class NvSciCallbacks
    {
    public:
        virtual uint32_t getNumBuffers() const = 0;
        virtual void fillNvSciBufAttributes(uint32_t bufferIndex, NvSciBufAttrList& attrList) const = 0;
        virtual void initializeFromNvSciBufObjs(dw::core::span<NvSciBufObj> bufs) = 0;
 
        virtual void pack() = 0;
        virtual void unpack() = 0;
    };
 
    SocketCallbacks& getSocketCallbacks();
    NvSciCallbacks& getNvSciCallbacks();
};
 
// TODO(chale): this should be made private to the framework once external entities no longer create
// Channel packets
class ChannelPacketDefaultBase : public IChannelPacket
{
public:
    ChannelPacketDefaultBase(size_t typeSize);
 
    GenericData getGenericData() final;
 
protected:
    size_t m_typeSize;
    std::unique_ptr<uint8_t[]> m_buffer;
    GenericData m_data;
};
 
class ChannelPacketDefault : public ChannelPacketDefaultBase, public IChannelPacket::SocketCallbacks
{
public:
    ChannelPacketDefault(size_t typeSize);
 
    uint8_t* getBuffer() final;
 
    size_t getBufferSize() final;
 
    size_t serialize() final;
 
    void deserialize(size_t) final;
};
 
class ChannelNvSciPacketDefault : public IChannelPacket, public IChannelPacket::NvSciCallbacks
{
    static constexpr char LOG_TAG[] = "ChannelNvSciPacketDefault";
 
public:
    ChannelNvSciPacketDefault(size_t typeSize);
 
    uint32_t getNumBuffers() const final;
    void fillNvSciBufAttributes(uint32_t bufferIndex, NvSciBufAttrList& attrList) const final;
 
    void initializeFromNvSciBufObjs(dw::core::span<NvSciBufObj> bufs);
 
    void pack() final;
 
    void unpack() final;
 
    GenericData getGenericData() final;
 
private:
    void fillCpuPacketDataAttributes(NvSciBufAttrList& output) const;
 
    size_t m_typeSize{};
    void* m_data{};
};
 
class IChannelPacketFactory
{
public:
    virtual std::unique_ptr<IChannelPacket> makePacket(GenericDataReference ref, ChannelType channelType) = 0;
};
using ChannelPacketFactoryPtr = std::shared_ptr<IChannelPacketFactory>;
 
} // namespace framework
} // namespace dw
 
#endif // DW_FRAMEWORK_ICHANNEL_PACKET_HPP_