data_feeder.hpp

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 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.
 */
 
#ifndef NVIDIA_DEEPSTREAM_DATA_FEEDER
#define NVIDIA_DEEPSTREAM_DATA_FEEDER
 
#include <thread>
#include <mutex>
#include <condition_variable>
 
#include "signal_handler.hpp"
#include "buffer.hpp"
 
namespace deepstream {
 
class DataFeeder : public SignalHandler {
 public:
  class IDataProvider {
   public:
    virtual Buffer read(DataFeeder& feeder, unsigned int size, bool& eos) = 0;
    virtual ~IDataProvider() {}
  };
 
  DataFeeder(const std::string &name, IDataProvider* provider);
 
  DataFeeder(const std::string &name, const char* factory, IDataProvider* provider);
 
  virtual ~DataFeeder();
 
  void startFeed(void* appsrc, unsigned int size);
  void stopFeed();
  bool doFeed(void* appsrc, unsigned int size);
 
 protected:
  std::unique_ptr<IDataProvider> data_provider_;
  std::thread worker_;
  std::mutex mutex_;
  std::condition_variable cv_data_;
  unsigned int data_size_;
  bool data_ready_;
  void* appsrc_;
  bool eos_;
 
};
}
 
#endif