metadata.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_METADATA
#define NVIDIA_DEEPSTREAM_METADATA
 
#include <functional>
#include <memory>
#include <string>
 
#include "nvll_osd_struct.h"
 
namespace deepstream {
 
#define USER_METADATA_START = (4096 + 4096 + 1)
 
class Metadata {
 public:
  Metadata(void* data);
 
  virtual ~Metadata();
 
  virtual operator bool() { return data_ != nullptr; }
 
 protected:
  void* data_; //> opaque data pointer
};
 
template <typename T>
class AbstractIterator {
 public:
  AbstractIterator(const T& data) : data_(data) {}
 
  virtual ~AbstractIterator() {}
 
  T& operator*() { return data_; }
 
  T* operator->() { return &data_; }
 
  T& get() { return data_; }
 
  virtual AbstractIterator<T>& next() = 0;
 
  virtual bool done() = 0;
 
  virtual bool operator==(const AbstractIterator<T>& other) const = 0;
 
  bool operator!=(const AbstractIterator<T>& other) const {
    return !(*this == other);
  }
 
 protected:
  T data_;
};
 
class UserMetadata : public Metadata {
 public:
  typedef std::unique_ptr<AbstractIterator<UserMetadata>> Iterator;
 
  UserMetadata(void*);
  virtual ~UserMetadata();
 
  friend class BatchMetadata;
  friend class FrameMetadata;
 
 protected:
  void get_(void*&);
};
 
class ClassifierMetadata : public Metadata {
public:
  typedef std::unique_ptr<AbstractIterator<ClassifierMetadata>> Iterator;
 
  ClassifierMetadata(void* data = nullptr);
  virtual ~ClassifierMetadata();
 
  unsigned int nLabels() const;
  unsigned int uniqueComponentId() const;
  std::string getLabel(unsigned int nth) const;
 
};
 
class ObjectMetadata : public Metadata {
 public:
  typedef std::unique_ptr<AbstractIterator<ObjectMetadata>> Iterator;
 
  ObjectMetadata(void* data = nullptr);
  virtual ~ObjectMetadata();
 
  unsigned int uniqueComponentId() const;
  void setUniqueComponentId(unsigned int);
  unsigned int classId() const;
  void setClassId(unsigned int);
  unsigned long  int objectId() const;
  void setObjectId(unsigned long);
  float confidence() const;
  void setConfidence(float);
  float trackerConfidence() const;
  void setTrackerConfidence(float);
  NvOSD_RectParams& rectParams() const;
  void setRectParams(const NvOSD_RectParams&);
  NvOSD_MaskParams& maskParams() const;
  void setMaskParams(const NvOSD_MaskParams&);
  NvOSD_TextParams& textParams() const;
  void setTextParams(const NvOSD_TextParams&);
  std::string label() const;
  void setLabel(std::string);
 
  unsigned int iterate(const std::function<void(const ClassifierMetadata&)>& func) const;
  void initiateIterator(ClassifierMetadata::Iterator&) const;
 
  unsigned int iterate(const std::function<void(const UserMetadata&)>& func, int meta_type) const;
  void initiateIterator(UserMetadata::Iterator&, int meta_type) const;
 
 friend class FrameMetadata;
};
 
 
class FrameMetadata;
class EventMessageUserMetadata : public UserMetadata {
 public:
  EventMessageUserMetadata(void* data=nullptr);
  virtual ~EventMessageUserMetadata();
 
  void generate(
    const ObjectMetadata&,
    const FrameMetadata&,
    const std::string sensor="N/A",
    const std::string uri="N/A",
    const std::vector<std::string>labels=std::vector<std::string>());
};
 
class Tensor;
class TensorOutputUserMetadata : public UserMetadata {
 public:
  TensorOutputUserMetadata(void* data=nullptr);
  TensorOutputUserMetadata(const UserMetadata&);
  virtual ~TensorOutputUserMetadata();
 
  unsigned int uniqueId() const;
  std::unordered_map<std::string, Tensor*> getLayers();
};
 
 
template <class C, int N>
class UserMetadataTemplate : public UserMetadata {
 public:
  UserMetadataTemplate(void* data=nullptr) : UserMetadata(data) {}
  UserMetadataTemplate(const UserMetadata& other): UserMetadata(other) {}
  virtual ~UserMetadataTemplate() {}
 
  C& get() {
    void* p = nullptr;
    this->get_(p);
    return *((C*)p);
  }
 
  static void* copy(void* data, void* user_data) {
    void* p = nullptr;
    UserMetadataTemplate<C, N>(data).get_(p);
    C*& src = (C*&) p;
    C* dst = new C(*src);
    return dst;
  }
 
  static void free(void* data, void* user_data) {
    void* p = nullptr;
    UserMetadataTemplate<C, N>(data).get_(p);
    C*& src = (C*&) p;
    delete src;
    p = nullptr;
  }
};
 
class DisplayMetadata : public Metadata {
 public:
  typedef std::unique_ptr<AbstractIterator<DisplayMetadata>> Iterator;
 
  DisplayMetadata(void* data = nullptr);
  virtual ~DisplayMetadata();
 
  unsigned int nRects() const;
  unsigned int nLabels() const;
  unsigned int nLines() const;
  unsigned int nArrows() const;
  unsigned int nCircles() const;
 
  bool add(NvOSD_TextParams&);
  bool add(NvOSD_RectParams&);
  bool add(NvOSD_LineParams&);
  bool add(NvOSD_ArrowParams&);
  bool add(NvOSD_CircleParams&);
 
  friend class FrameMetadata;
};
 
class FrameMetadata : public Metadata {
 public:
  typedef std::unique_ptr<AbstractIterator<FrameMetadata>> Iterator;
 
  FrameMetadata(void* data=nullptr);
  virtual ~FrameMetadata();
 
  unsigned int iterate(const std::function<void(const ObjectMetadata&)>& func) const;
  void initiateIterator(ObjectMetadata::Iterator&) const;
 
  unsigned int iterate(const std::function<void(const DisplayMetadata&)>& func) const;
  void initiateIterator(DisplayMetadata::Iterator&) const;
 
  unsigned int iterate(const std::function<void(const UserMetadata&)>& func, int meta_type) const;
  void initiateIterator(UserMetadata::Iterator&, int meta_type) const;
 
  unsigned int padIndex() const;
  unsigned int batchId() const;
  int frameNum() const;
  unsigned int sourceId() const;
  unsigned int sourceWidth() const;
  unsigned int sourceHeight() const;
 
  unsigned int pipelineWidth() const;
  unsigned int pipelineHeight() const;
 
  uint64_t bufferPTS() const;
  uint64_t ntpTimestamp() const;
 
  void append(const DisplayMetadata&);
  void append(const UserMetadata&);
  void append(const ObjectMetadata&);
 
  friend class BatchMetadata;
};
 
class BatchMetadata : public Metadata {
 public:
  BatchMetadata(void *data = nullptr);
 
  virtual ~BatchMetadata();
 
  unsigned int iterate(const std::function<void(const FrameMetadata&)>& func) const;
  void initiateIterator(FrameMetadata::Iterator&) const;
 
  unsigned int iterate(const std::function<void(const UserMetadata&)>& func, int meta_type) const;
  void initiateIterator(UserMetadata::Iterator&, int meta_type) const;
 
  bool acquire(DisplayMetadata&);
  bool acquire(FrameMetadata&);
  bool acquire(ObjectMetadata&);
  bool acquire(EventMessageUserMetadata&);
  template<class C, int N>
  bool acquire(UserMetadataTemplate<C, N>& user_meta) {
    user_meta = acquireUserMetadata_(
        new C, N, UserMetadataTemplate<C, N>::copy, UserMetadataTemplate<C, N>::free
    );
    return user_meta ? true : false;
  }
 
  void append(const FrameMetadata&);
  void append(const UserMetadata&);
 
  unsigned int nFrames() const;
 
 protected:
  UserMetadata acquireUserMetadata_(
    void* data, unsigned int type, void*(*copy)(void*, void*), void(*free)(void*, void*)
  );
};
 
}
 
#endif