nvdsmeta_schema.h

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 NVDSMETA_H_
#define NVDSMETA_H_
 
#include <glib.h>
 
#ifdef __cplusplus
extern "C"
{
#endif
 
typedef enum NvDsEventType {
  NVDS_EVENT_ENTRY,
  NVDS_EVENT_EXIT,
  NVDS_EVENT_MOVING,
  NVDS_EVENT_STOPPED,
  NVDS_EVENT_EMPTY,
  NVDS_EVENT_PARKED,
  NVDS_EVENT_RESET,
 
  NVDS_EVENT_RESERVED = 0x100,
  NVDS_EVENT_CUSTOM = 0x101,
  NVDS_EVENT_FORCE32 = 0x7FFFFFFF
} NvDsEventType;
 
typedef enum NvDsObjectType {
  NVDS_OBJECT_TYPE_VEHICLE,
  NVDS_OBJECT_TYPE_PERSON,
  NVDS_OBJECT_TYPE_FACE,
  NVDS_OBJECT_TYPE_BAG,
  NVDS_OBJECT_TYPE_BICYCLE,
  NVDS_OBJECT_TYPE_ROADSIGN,
  NVDS_OBJECT_TYPE_VEHICLE_EXT,
  NVDS_OBJECT_TYPE_PERSON_EXT,
  NVDS_OBJECT_TYPE_FACE_EXT,
  NVDS_OBJECT_TYPE_3D,
  NVDS_OBJECT_TYPE_PRODUCT, 
  NVDS_OBJECT_TYPE_PRODUCT_EXT, 
  NVDS_OBJECT_TYPE_RESERVED = 0x100,
  NVDS_OBJECT_TYPE_CUSTOM = 0x101,
  NVDS_OBJECT_TYPE_UNKNOWN = 0x102,
  NVDS_OBEJCT_TYPE_FORCE32 = 0x7FFFFFFF
} NvDsObjectType;
 
typedef enum NvDsPayloadType {
  NVDS_PAYLOAD_DEEPSTREAM,
  NVDS_PAYLOAD_DEEPSTREAM_MINIMAL,
  NVDS_PAYLOAD_DEEPSTREAM_PROTOBUF,
  NVDS_PAYLOAD_RESERVED = 0x100,
  NVDS_PAYLOAD_CUSTOM = 0x101,
  NVDS_PAYLOAD_FORCE32 = 0x7FFFFFFF
} NvDsPayloadType;
 
typedef struct NvDsRect {
  float top;     
  float left;    
  float width;   
  float height;  
} NvDsRect;
 
typedef struct NvDsGeoLocation {
  gdouble lat;      
  gdouble lon;      
  gdouble alt;      
} NvDsGeoLocation;
 
typedef struct NvDsCoordinate {
  gdouble x;        
  gdouble y;        
  gdouble z;        
} NvDsCoordinate;
 
typedef struct NvDsObjectSignature {
  gdouble *signature;
  guint size;
} NvDsObjectSignature;
 
typedef struct NvDsProductObjectExt {
  gchar *brand; 
  gchar *type; 
  gchar *shape; 
  GList *mask;
} NvDsProductObjectExt;
 
typedef struct NvDsVehicleObject {
  gchar *type;      
  gchar *make;      
  gchar *model;     
  gchar *color;     
  gchar *region;    
  gchar *license;   
} NvDsVehicleObject;
 
typedef struct NvDsPersonObject {
  gchar *gender;    
  gchar *hair;      
  gchar *cap;       
  gchar *apparel;   
  guint age;        
} NvDsPersonObject;
 
typedef struct NvDsFaceObject {
  gchar *gender;    
  gchar *hair;      
  gchar *cap;       
  gchar *glasses;   
  gchar *facialhair;
  gchar *name;      
  gchar *eyecolor;  
  guint age;        
} NvDsFaceObject;
 
typedef struct NvDsVehicleObjectExt {
  gchar *type;      
  gchar *make;      
  gchar *model;     
  gchar *color;     
  gchar *region;    
  gchar *license;   
  GList *mask;      
} NvDsVehicleObjectExt;
 
typedef struct NvDsPersonObjectExt {
  gchar *gender;    
  gchar *hair;      
  gchar *cap;       
  gchar *apparel;   
  guint age;        
  GList *mask;      
} NvDsPersonObjectExt;
 
typedef struct NvDsFaceObjectWithExt {
  gchar *gender;    
  gchar *hair;      
  gchar *cap;       
  gchar *glasses;   
  gchar *facialhair;
  gchar *name;      
  gchar *eyecolor;  
  guint age;        
  GList *mask;      
} NvDsFaceObjectExt;
 
typedef struct NvDsJoint {
  float x;     
  float y;     
  float z;     
  float confidence;   
} NvDsJoint;
 
typedef struct NvDsJoints {
  NvDsJoint *joints;   
  int num_joints;      
  int pose_type;       
} NvDsJoints;
 
typedef struct NvDsEmbedding {
  float *embedding_vector; 
  guint embedding_length; 
} NvDsEmbedding;
 
typedef struct NvDsProductObject {
  gchar *brand; 
  gchar *type; 
  gchar *shape; 
} NvDsProductObject;
 
typedef struct NvDsConvexHull {
  gint* points;
  guint numFilled;
} NvDsConvexHull;
 
typedef struct NvDs3DTracking {
  gfloat ptWorldFeet[2];
  gfloat ptImgFeet[2];
  gfloat visibility;
  NvDsConvexHull convexHull;
} NvDs3DTracking;
 
typedef struct _NvDsBbox3dObject {
    float boxes_3d[10];
    float scores_3d;
    guint64 labels_3d;
    float cls_scores; // confidence
    guint instance_ids;
    NvDsEmbedding *instance_feats;
} NvDsBbox3dObject;
 
#define MAX_NUM_OBJECTS 300  // Define a maximum number of objects
typedef struct _NvDsBbox3dObjectList {
    NvDsBbox3dObject objects[MAX_NUM_OBJECTS];
    guint num_objects;
    guint64 timestamp;
    gint frame_num;
    guint camera_id;
    gchar *sensorStr;
} NvDsBbox3dObjectList;
 
typedef struct {
    gchar *id;
    gchar *query;
    gchar *response;
} NvDsConversationObject;
 
typedef struct {
    GHashTable *info;
    NvDsConversationObject *conversations;
    guint num_conversations;
    NvDsEmbedding *embeddings;
    guint num_embeddings;
} NvDsLLMObject;
 
typedef struct {
    gchar *id;
    gchar **objectIds;
    guint num_objectIds;
    NvDsCoordinate *coordinates;
    guint num_coordinates;
    gchar *description;
    GHashTable *info;
    gint64 timestamp;  // New field added
} NvDsInteractionObject;
 
typedef struct {
    gchar *id;
    gchar **objectIds;
    guint num_objectIds;
    float amount;
    GHashTable *info;
    gint64 timestamp;  // New field added
} NvDsCongestionObject;
 
typedef struct {
    gchar *id;
    gchar *type;
    gint32 count;
    NvDsCoordinate *coordinates;
    guint num_coordinates;
    gchar **objectIds;
    guint num_objectIds;
    GHashTable *info;
} NvDsTypeMetricsObject;
 
typedef struct {
    gint32 *mask;
    guint mask_size;
    GHashTable *info;
} NvDsSegmentationObject;
 
typedef struct NvDsEventMsgMeta {
  NvDsEventType type;
  NvDsObjectType objType;
  NvDsRect bbox;
  NvDsGeoLocation location;
  NvDsCoordinate coordinate;
  NvDsObjectSignature objSignature;
  gint objClassId;
  gint sensorId;
  gint moduleId;
  gint placeId;
  gint componentId;
  gint frameId;
  gdouble confidence;
  guint64 trackingId;
  gchar *ts;
  gchar *objectId;
  gchar *sensorStr;
  NvDsBbox3dObject *bbox3d;
  NvDsLLMObject *llm;
  gint num_interactions;
  NvDsInteractionObject *interactions;
  gint num_congestions;
  NvDsCongestionObject *congestions;
  gint num_fov;
  NvDsTypeMetricsObject *fov;
  NvDsSegmentationObject *segmentation;
 
  gchar *otherAttrs;
  gchar *videoPath;
  gpointer extMsg;
  guint extMsgSize;
  NvDsJoints pose;
  NvDsEmbedding embedding;
  bool has3DTracking;
  NvDs3DTracking singleView3DTracking;
} NvDsEventMsgMeta;
 
typedef struct _NvDsEvent {
  NvDsEventType eventType;
  NvDsEventMsgMeta *metadata;
} NvDsEvent;
 
typedef struct _NvDsCustomMsgInfo {
  void *message;
  guint size;
}NvDsCustomMsgInfo;
 
typedef struct NvDsPayload {
  gpointer payload;
  guint payloadSize;
  guint componentId;
} NvDsPayload;
 
#ifdef __cplusplus
}
#endif
#endif /* NVDSMETA_H_ */