#ifndef _smartai_shm_struct_parser_h
|
#define _smartai_shm_struct_parser_h
|
|
#include <stdint.h>
|
#include "std_target.h"
|
|
// struct image
|
struct stimg{
|
uint64_t id;
|
unsigned char* data;
|
uint64_t data_size;
|
uint32_t width;
|
uint32_t height;
|
char* timestamp;
|
uint32_t timestamp_size;
|
char* camera_id;
|
uint32_t camera_id_size;
|
char* camera_name;
|
uint32_t camera_name_size;
|
};
|
|
//////////////////////////////////////////////
|
// sdk result
|
|
// struct rect
|
struct strc{
|
int32_t left;
|
int32_t top;
|
int32_t right;
|
int32_t bottom;
|
};
|
|
// struct target from analysis
|
struct sttgt{
|
uint64_t id;
|
char* type;
|
uint32_t type_size;
|
int32_t confidence;
|
struct strc rect;
|
unsigned char* feature;
|
uint32_t feature_size;
|
unsigned char* attribute;
|
uint32_t attribute_size;
|
};
|
|
// struct sdk analysis run
|
struct stsdk{
|
char* sdktype;
|
uint32_t sdktype_size;
|
char* sdkid;
|
uint32_t sdkid_size;
|
char* sdkname;
|
uint32_t sdkname_size;
|
char* timestamp;
|
uint32_t timestamp_size;
|
|
uint32_t tgt_count;
|
// targets [struct sttgt]
|
struct sttgt* tgt;
|
};
|
|
// struct rule message for rule server
|
struct strule{
|
char* handletrack;
|
uint32_t handletrack_size;
|
char* datatype;
|
uint32_t datatype_size;
|
|
uint32_t sdk_count;
|
// struct stsdk
|
struct stsdk* sdk;
|
};
|
|
#ifdef __cplusplus
|
extern "C"{
|
#endif
|
|
/*
|
以下api提供将image放入shm和从shm取出的功能
|
make_image_ref 将 image 信息组装成 stimg 结构，不发生内存拷贝，参数生命周期由外部管理保证
|
image2shm 将 stimg 放入共享内存 shm 中，由外部保证 shm 足够大
|
也可以不使用 make_image_ref 自己组装 struct stimg
|
shm2image 将共享内存 shm 的内容组装成 stimg 结构，不产生内存拷贝，仅移动指针
|
返回一个 struct stimg 结构的指针，需要调用 free_stimg 释放
|
free_stimg 释放 shm2image 返回的指针
|
*/
|
struct stimg* make_image_ref(const uint64_t id, const unsigned char* data, const uint32_t size,
|
const uint32_t width, const uint32_t height,
|
const char* timestamp, const uint32_t ts_size,
|
const char* camera_id, const uint32_t cid_size,
|
const char* camera_name,const uint32_t cname_size);
|
void image2shm(void* shm, struct stimg* img);
|
struct stimg* shm2image(void* shm);
|
void free_stimg(struct stimg* img);
|
|
/*
|
rule2shm 在 decoder 中调用，初始化一个 rulemsg 给后续 analysis 使用
|
add_result2rule_in_shm 将 tresult 加入到 shm 中保存的 rule 中使用的结构体中
|
shm2rule 获取 shm 中保存的 rule 结构体，不产生内存拷贝
|
返回一个 struct strule 结构的指针，需要调用 free_strule 释放
|
free_strule 释放 shm2rule 返回的指针
|
*/
|
void rule2shm(void* shm, const char* handletrack, const uint32_t ht_size);
|
void add_result2rule_in_shm(void* shm, const TResult* res,
|
const char* sdktype, const uint32_t st_size,
|
const char* timestamp, const uint32_t ts_size);
|
struct strule* shm2rule(void* shm);
|
void free_strule(struct strule* rule);
|
|
///////////////////////////////////////////////////////////////
|
// go need function
|
// go image
|
void goImage2Shm(void* shm, void* img);
|
void goImageGet(void* img, void** data, uint64_t* size,
|
void** ts, uint32_t* tss,
|
void** cid, uint32_t* cids,
|
void** cname, uint32_t* cnames,
|
uint64_t* id, uint32_t* w, uint32_t* h);
|
|
// go rule/sdk/result
|
void goTargetsDel(void* tgts);
|
void* goTargetsNew(const uint32_t size);
|
void goTargetAdd(void* tgts, int idx, uint64_t id, char* type, uint32_t tsize, int32_t confidence,
|
int left, int top, int right, int bottom,
|
unsigned char* feat, uint32_t featsize,
|
unsigned char* attr, uint32_t attrsize);
|
void goSdkPut(void* shm, void* tgts, uint32_t tgtcnt,
|
char* type, uint32_t tsize,
|
char* timestamp, uint32_t tmsize);
|
|
// unserialize
|
void goHandletrackGet(void* rule, void** data, uint32_t* size);
|
void goDatatypeGet(void* rule, void** data, uint32_t* size);
|
|
void goSdkArrayGet(void* rule, void** data, uint32_t* size);
|
void goSdkGet(void* sdk, int idx, void** t, uint32_t* ts,
|
void** i, uint32_t* is, void** n, uint32_t* ns,
|
void** tm, uint32_t* tms, void** data, uint32_t* size);
|
void goTargetGet(void* tgt, int idx, uint64_t* id, int32_t* c,
|
int32_t* l,int32_t* t,int32_t* r,int32_t* b,
|
void** tp, uint32_t* tps,
|
void** feat, uint32_t* fs, void** attr, uint32_t* attrs);
|
|
|
#ifdef __cplusplus
|
}
|
#endif
|
#endif
|
