9.1/service-maker/includes/data_feeder.hpp

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2024-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 * SPDX-License-Identifier: Apache-2.0
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
 
#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