38 lines
1.7 KiB
C++
38 lines
1.7 KiB
C++
// scripts/node-addon-api/src/macros.h
|
|
//
|
|
// Copyright (c) 2024 Xiaomi Corporation
|
|
#ifndef SCRIPTS_NODE_ADDON_API_SRC_MACROS_H_
|
|
#define SCRIPTS_NODE_ADDON_API_SRC_MACROS_H_
|
|
|
|
#include <algorithm>
|
|
#include <string>
|
|
|
|
#define SHERPA_ONNX_ASSIGN_ATTR_STR(c_name, js_name) \
|
|
do { \
|
|
if (o.Has(#js_name) && o.Get(#js_name).IsString()) { \
|
|
Napi::String _str = o.Get(#js_name).As<Napi::String>(); \
|
|
std::string s = _str.Utf8Value(); \
|
|
char *p = new char[s.size() + 1]; \
|
|
std::copy(s.begin(), s.end(), p); \
|
|
p[s.size()] = 0; \
|
|
\
|
|
c.c_name = p; \
|
|
} \
|
|
} while (0)
|
|
|
|
#define SHERPA_ONNX_ASSIGN_ATTR_INT32(c_name, js_name) \
|
|
do { \
|
|
if (o.Has(#js_name) && o.Get(#js_name).IsNumber()) { \
|
|
c.c_name = o.Get(#js_name).As<Napi::Number>().Int32Value(); \
|
|
} \
|
|
} while (0)
|
|
|
|
#define SHERPA_ONNX_ASSIGN_ATTR_FLOAT(c_name, js_name) \
|
|
do { \
|
|
if (o.Has(#js_name) && o.Get(#js_name).IsNumber()) { \
|
|
c.c_name = o.Get(#js_name).As<Napi::Number>().FloatValue(); \
|
|
} \
|
|
} while (0)
|
|
|
|
#endif // SCRIPTS_NODE_ADDON_API_SRC_MACROS_H_
|