6 Commits
1.0.2 ... main

Author SHA1 Message Date
f343d28540 完善预编译库下载工具 2025-03-21 12:34:44 +08:00
049a70afc2 额,忘了处理缓存了 2025-03-17 16:15:10 +08:00
f7d8926443 下载时增加版本判断 2025-03-17 16:12:49 +08:00
b86366e244 清除多余日志 2025-03-17 15:58:11 +08:00
0518c9a02f 修复CVVideoCapture::Set方法 2025-03-17 15:56:50 +08:00
2a980fbdf8 增加VideoCapture 2025-03-17 15:55:43 +08:00
20 changed files with 919 additions and 477 deletions

View File

@ -37,6 +37,19 @@ if(CMAKE_JS_RESULT EQUAL 0)
OUTPUT_STRIP_TRAILING_WHITESPACE
)
endif()
# ReleaseVersion
if(CMAKE_JS_RESULT EQUAL 0)
execute_process(
COMMAND node ${CMAKE_SOURCE_DIR}/thirdpart/cmake-js-util.js --release
RESULT_VARIABLE CMAKE_JS_RESULT
OUTPUT_VARIABLE RELEASE_VERSION
OUTPUT_STRIP_TRAILING_WHITESPACE
)
if(CMAKE_JS_RESULT EQUAL 0)
message(STATUS "RELEASE_VERSION: ${RELEASE_VERSION}")
add_compile_definitions(RELEASE_VERSION="${RELEASE_VERSION}")
endif()
endif()
# NAPI
if(CMAKE_JS_RESULT EQUAL 0)
@ -77,6 +90,7 @@ endif()
add_library(cv SHARED ${CMAKE_JS_SRC}
cxx/node.cc
cxx/mat.cc
cxx/videocap.cc
cxx/util.cc
)
target_link_libraries(cv ${CMAKE_JS_LIB} ${OpenCV_LIBS})

View File

@ -1,459 +1,488 @@
#include "node.h"
#include "mat.h"
using namespace Napi;
#define MAT_INSTANCE_METHOD(method) InstanceMethod<&CVMat::method>(#method, static_cast<napi_property_attributes>(napi_writable | napi_configurable))
#define MAT_STATIC_METHOD(method) StaticMethod<&CVMat::method>(#method, static_cast<napi_property_attributes>(napi_writable | napi_configurable))
#define NODE_FUNCTION_IMPL(name) Napi::Value CVMat::name(const Napi::CallbackInfo &info)
static FunctionReference *constructor = nullptr;
FunctionReference *CVMat::constructor = nullptr;
class CVMat : public ObjectWrap<CVMat> {
public:
static Napi::Object Init(Napi::Env env, Napi::Object exports)
{
Function func = DefineClass(env, "Mat", {
MAT_STATIC_METHOD(ImRead),
MAT_STATIC_METHOD(ImDecode),
MAT_STATIC_METHOD(ImWrite),
MAT_STATIC_METHOD(ImEncode),
Napi::Object CVMat::Init(Napi::Env env, Napi::Object exports)
{
Function func = DefineClass(env, "Mat", {
MAT_STATIC_METHOD(ImRead),
MAT_STATIC_METHOD(ImDecode),
MAT_STATIC_METHOD(ImWrite),
MAT_STATIC_METHOD(ImEncode),
MAT_STATIC_METHOD(Crop),
MAT_STATIC_METHOD(Resize),
MAT_STATIC_METHOD(WarpAffine),
MAT_STATIC_METHOD(Blur),
MAT_STATIC_METHOD(CopyMakeBorder),
MAT_STATIC_METHOD(Flip),
MAT_STATIC_METHOD(GetRotationMatrix2D),
MAT_STATIC_METHOD(Crop),
MAT_STATIC_METHOD(Resize),
MAT_STATIC_METHOD(WarpAffine),
MAT_STATIC_METHOD(Blur),
MAT_STATIC_METHOD(CopyMakeBorder),
MAT_STATIC_METHOD(Flip),
MAT_STATIC_METHOD(GetRotationMatrix2D),
MAT_STATIC_METHOD(Rectangle),
MAT_STATIC_METHOD(Circle),
MAT_STATIC_METHOD(Line),
MAT_STATIC_METHOD(Ellipse),
MAT_STATIC_METHOD(Polylines),
MAT_STATIC_METHOD(FillPoly),
MAT_STATIC_METHOD(FillConvexPoly),
MAT_STATIC_METHOD(DrawMarker),
MAT_STATIC_METHOD(PutText),
MAT_STATIC_METHOD(Rectangle),
MAT_STATIC_METHOD(Circle),
MAT_STATIC_METHOD(Line),
MAT_STATIC_METHOD(Ellipse),
MAT_STATIC_METHOD(Polylines),
MAT_STATIC_METHOD(FillPoly),
MAT_STATIC_METHOD(FillConvexPoly),
MAT_STATIC_METHOD(DrawMarker),
MAT_STATIC_METHOD(PutText),
MAT_INSTANCE_METHOD(IsEmpty),
MAT_INSTANCE_METHOD(GetCols),
MAT_INSTANCE_METHOD(GetRows),
MAT_INSTANCE_METHOD(GetType),
MAT_INSTANCE_METHOD(GetDepth),
MAT_INSTANCE_METHOD(GetChannels),
MAT_INSTANCE_METHOD(GetFlags),
MAT_INSTANCE_METHOD(GetDims),
MAT_INSTANCE_METHOD(GetIsContinuous),
MAT_INSTANCE_METHOD(GetIsSubmatrix),
MAT_INSTANCE_METHOD(GetElemSize),
MAT_INSTANCE_METHOD(GetElemSize1),
MAT_INSTANCE_METHOD(GetTotal),
MAT_INSTANCE_METHOD(GetTotalWithDim),
MAT_INSTANCE_METHOD(GetSize),
MAT_INSTANCE_METHOD(GetData),
MAT_INSTANCE_METHOD(IsEmpty),
MAT_INSTANCE_METHOD(GetCols),
MAT_INSTANCE_METHOD(GetRows),
MAT_INSTANCE_METHOD(GetType),
MAT_INSTANCE_METHOD(GetDepth),
MAT_INSTANCE_METHOD(GetChannels),
MAT_INSTANCE_METHOD(GetFlags),
MAT_INSTANCE_METHOD(GetDims),
MAT_INSTANCE_METHOD(GetIsContinuous),
MAT_INSTANCE_METHOD(GetIsSubmatrix),
MAT_INSTANCE_METHOD(GetElemSize),
MAT_INSTANCE_METHOD(GetElemSize1),
MAT_INSTANCE_METHOD(GetTotal),
MAT_INSTANCE_METHOD(GetTotalWithDim),
MAT_INSTANCE_METHOD(GetSize),
MAT_INSTANCE_METHOD(GetData),
MAT_INSTANCE_METHOD(Col),
MAT_INSTANCE_METHOD(ColRange),
MAT_INSTANCE_METHOD(Row),
MAT_INSTANCE_METHOD(RowRange),
MAT_INSTANCE_METHOD(Diag),
MAT_INSTANCE_METHOD(Col),
MAT_INSTANCE_METHOD(ColRange),
MAT_INSTANCE_METHOD(Row),
MAT_INSTANCE_METHOD(RowRange),
MAT_INSTANCE_METHOD(Diag),
MAT_INSTANCE_METHOD(Clone),
MAT_INSTANCE_METHOD(CopyTo),
MAT_INSTANCE_METHOD(Clone),
MAT_INSTANCE_METHOD(CopyTo),
});
constructor = new FunctionReference();
*constructor = Napi::Persistent(func);
exports.Set("Mat", func);
env.SetInstanceData<FunctionReference>(constructor);
return exports;
}
});
constructor = new FunctionReference();
*constructor = Napi::Persistent(func);
exports.Set("Mat", func);
env.SetInstanceData<FunctionReference>(constructor);
return exports;
}
CVMat(const CallbackInfo &info)
: ObjectWrap<CVMat>(info)
{
if (info.Length() >= 2) {
auto sizesArray = info[0].As<Array>();
std::vector<int> sizes(sizesArray.Length());
for (auto i = 0; i < sizes.size(); ++i) sizes[i] = sizesArray.Get(i).As<Number>().Int32Value();
auto type = info[1].As<Number>().Int32Value();
// 使用sizes和type初始化
if (info.Length() < 3) mat_ = cv::Mat(sizes, type);
// 使用数据初始化
else if (info[2].IsTypedArray()) {
auto dataArray = info[2].As<TypedArray>();
auto data = static_cast<uint8_t *>(dataArray.ArrayBuffer().Data()) + dataArray.ByteOffset();
mat_ = cv::Mat(sizes, type, data);
}
// 其他 TODO
else mat_ = cv::Mat(sizes, type);
CVMat::CVMat(const CallbackInfo &info)
: ObjectWrap<CVMat>(info)
{
if (info.Length() >= 2) {
auto sizesArray = info[0].As<Array>();
std::vector<int> sizes(sizesArray.Length());
for (auto i = 0; i < sizes.size(); ++i) sizes[i] = sizesArray.Get(i).As<Number>().Int32Value();
auto type = info[1].As<Number>().Int32Value();
// 使用sizes和type初始化
if (info.Length() < 3) mat_ = cv::Mat(sizes, type);
// 使用数据初始化
else if (info[2].IsTypedArray()) {
auto dataArray = info[2].As<TypedArray>();
auto data = static_cast<uint8_t *>(dataArray.ArrayBuffer().Data()) + dataArray.ByteOffset();
mat_ = cv::Mat(sizes, type, data);
}
// 其他 TODO
else mat_ = cv::Mat(sizes, type);
}
}
static Napi::Value ImRead(const Napi::CallbackInfo &info)
{
auto filename = info[0].As<String>().Utf8Value();
int flag = (info.Length() >= 2) ? info[1].As<Number>().Int32Value() : cv::IMREAD_COLOR_BGR;
return CreateMat(info.Env(), [&](CVMat &mat) { mat.mat_ = cv::imread(filename, flag); });
NODE_FUNCTION_IMPL(ImRead)
{
auto filename = info[0].As<String>().Utf8Value();
int flag = (info.Length() >= 2) ? info[1].As<Number>().Int32Value() : cv::IMREAD_COLOR_BGR;
return CreateMat(info.Env(), [&](CVMat &mat) { mat.mat_ = cv::imread(filename, flag); });
}
NODE_FUNCTION_IMPL(ImDecode)
{
auto arr = info[0].As<TypedArray>();
auto data = static_cast<uint8_t *>(arr.ArrayBuffer().Data()) + arr.ByteOffset();
std::vector<uint8_t> buffer(data, data + arr.ByteLength());
int flag = (info.Length() >= 2) ? info[1].As<Number>().Int32Value() : cv::IMREAD_COLOR_BGR;
return CreateMat(info.Env(), [&](CVMat &mat) { mat.mat_ = cv::imdecode(buffer, flag); });
}
NODE_FUNCTION_IMPL(ImWrite)
{
auto &mat = GetMat(info[0].As<Object>());
auto filename = info[1].As<String>().Utf8Value();
std::vector<int> params;
if (info[2].IsArray()) {
auto paramArray = info[2].As<Array>();
params.resize(paramArray.Length());
for (auto i = 0; i < params.size(); ++i) params[i] = paramArray.Get(i).As<Number>().Int32Value();
}
static Napi::Value ImDecode(const Napi::CallbackInfo &info)
{
auto arr = info[0].As<TypedArray>();
auto data = static_cast<uint8_t *>(arr.ArrayBuffer().Data()) + arr.ByteOffset();
std::vector<uint8_t> buffer(data, data + arr.ByteLength());
int flag = (info.Length() >= 2) ? info[1].As<Number>().Int32Value() : cv::IMREAD_COLOR_BGR;
return CreateMat(info.Env(), [&](CVMat &mat) { mat.mat_ = cv::imdecode(buffer, flag); });
auto res = cv::imwrite(filename, mat.mat_, params);
return Boolean::New(info.Env(), res);
}
NODE_FUNCTION_IMPL(ImEncode)
{
auto &mat = GetMat(info[0].As<Object>());
auto extname = info[1].As<String>().Utf8Value();
std::vector<int> params;
if (info[2].IsArray()) {
auto paramArray = info[2].As<Array>();
params.resize(paramArray.Length());
for (auto i = 0; i < params.size(); ++i) params[i] = paramArray.Get(i).As<Number>().Int32Value();
}
static Napi::Value ImWrite(const Napi::CallbackInfo &info)
{
auto &mat = GetMat(info[0].As<Object>());
auto filename = info[1].As<String>().Utf8Value();
std::vector<int> params;
if (info[2].IsArray()) {
auto paramArray = info[2].As<Array>();
params.resize(paramArray.Length());
for (auto i = 0; i < params.size(); ++i) params[i] = paramArray.Get(i).As<Number>().Int32Value();
}
auto res = cv::imwrite(filename, mat.mat_, params);
return Boolean::New(info.Env(), res);
}
static Napi::Value ImEncode(const Napi::CallbackInfo &info)
{
auto &mat = GetMat(info[0].As<Object>());
auto extname = info[1].As<String>().Utf8Value();
std::vector<int> params;
if (info[2].IsArray()) {
auto paramArray = info[2].As<Array>();
params.resize(paramArray.Length());
for (auto i = 0; i < params.size(); ++i) params[i] = paramArray.Get(i).As<Number>().Int32Value();
}
std::vector<uchar> buf;
if (!cv::imencode(extname, mat.mat_, buf, params)) return info.Env().Undefined();
auto arrayBuffer = ArrayBuffer::New(info.Env(), buf.size());
auto bufferPtr = static_cast<uchar *>(arrayBuffer.Data());
for (auto ch : buf) {
*bufferPtr = ch;
bufferPtr++;
}
return arrayBuffer;
std::vector<uchar> buf;
if (!cv::imencode(extname, mat.mat_, buf, params)) return info.Env().Undefined();
auto arrayBuffer = ArrayBuffer::New(info.Env(), buf.size());
auto bufferPtr = static_cast<uchar *>(arrayBuffer.Data());
for (auto ch : buf) {
*bufferPtr = ch;
bufferPtr++;
}
return arrayBuffer;
}
Napi::Value IsEmpty(const Napi::CallbackInfo &info) { return Boolean::New(info.Env(), mat_.empty()); }
Napi::Value GetCols(const Napi::CallbackInfo &info) { return Number::New(info.Env(), mat_.cols); }
Napi::Value GetRows(const Napi::CallbackInfo &info) { return Number::New(info.Env(), mat_.rows); }
Napi::Value GetType(const Napi::CallbackInfo &info) { return Number::New(info.Env(), mat_.type()); }
Napi::Value GetDepth(const Napi::CallbackInfo &info) { return Number::New(info.Env(), mat_.depth()); }
Napi::Value GetChannels(const Napi::CallbackInfo &info) { return Number::New(info.Env(), mat_.channels()); }
Napi::Value GetFlags(const Napi::CallbackInfo &info) { return Number::New(info.Env(), mat_.flags); }
Napi::Value GetDims(const Napi::CallbackInfo &info) { return Number::New(info.Env(), mat_.dims); }
Napi::Value GetIsContinuous(const Napi::CallbackInfo &info) { return Boolean::New(info.Env(), mat_.isContinuous()); }
Napi::Value GetIsSubmatrix(const Napi::CallbackInfo &info) { return Boolean::New(info.Env(), mat_.isSubmatrix()); }
Napi::Value GetElemSize(const Napi::CallbackInfo &info) { return Number::New(info.Env(), static_cast<double>(mat_.elemSize())); }
Napi::Value GetElemSize1(const Napi::CallbackInfo &info) { return Number::New(info.Env(), static_cast<double>(mat_.elemSize1())); }
Napi::Value GetTotal(const Napi::CallbackInfo &info) { return Number::New(info.Env(), static_cast<double>(mat_.total())); }
Napi::Value GetTotalWithDim(const Napi::CallbackInfo &info) { return Number::New(info.Env(), static_cast<double>(mat_.total(info[0].As<Number>().Int32Value(), info[1].As<Number>().Int32Value()))); }
Napi::Value GetSize(const Napi::CallbackInfo &info)
{
auto ret = Array::New(info.Env(), mat_.dims);
auto &size = mat_.size;
for (int i = 0; i < mat_.dims; ++i) ret.Set(i, size[i]);
return ret;
}
Napi::Value GetData(const Napi::CallbackInfo &info)
{
auto ptr = mat_.ptr();
auto bytes = mat_.elemSize() * mat_.total();
return ArrayBuffer::New(info.Env(), ptr, bytes);
}
NODE_FUNCTION_IMPL(IsEmpty)
{
return Boolean::New(info.Env(), mat_.empty());
}
NODE_FUNCTION_IMPL(GetCols)
{
return Number::New(info.Env(), mat_.cols);
}
NODE_FUNCTION_IMPL(GetRows)
{
return Number::New(info.Env(), mat_.rows);
}
NODE_FUNCTION_IMPL(GetType)
{
return Number::New(info.Env(), mat_.type());
}
NODE_FUNCTION_IMPL(GetDepth)
{
return Number::New(info.Env(), mat_.depth());
}
NODE_FUNCTION_IMPL(GetChannels)
{
return Number::New(info.Env(), mat_.channels());
}
NODE_FUNCTION_IMPL(GetFlags)
{
return Number::New(info.Env(), mat_.flags);
}
NODE_FUNCTION_IMPL(GetDims)
{
return Number::New(info.Env(), mat_.dims);
}
NODE_FUNCTION_IMPL(GetIsContinuous)
{
return Boolean::New(info.Env(), mat_.isContinuous());
}
NODE_FUNCTION_IMPL(GetIsSubmatrix)
{
return Boolean::New(info.Env(), mat_.isSubmatrix());
}
NODE_FUNCTION_IMPL(GetElemSize)
{
return Number::New(info.Env(), static_cast<double>(mat_.elemSize()));
}
NODE_FUNCTION_IMPL(GetElemSize1)
{
return Number::New(info.Env(), static_cast<double>(mat_.elemSize1()));
}
NODE_FUNCTION_IMPL(GetTotal)
{
return Number::New(info.Env(), static_cast<double>(mat_.total()));
}
NODE_FUNCTION_IMPL(GetTotalWithDim)
{
return Number::New(info.Env(), static_cast<double>(mat_.total(info[0].As<Number>().Int32Value(), info[1].As<Number>().Int32Value())));
}
NODE_FUNCTION_IMPL(GetSize)
{
auto ret = Array::New(info.Env(), mat_.dims);
auto &size = mat_.size;
for (int i = 0; i < mat_.dims; ++i) ret.Set(i, size[i]);
return ret;
}
NODE_FUNCTION_IMPL(GetData)
{
auto ptr = mat_.ptr();
auto bytes = mat_.elemSize() * mat_.total();
return ArrayBuffer::New(info.Env(), ptr, bytes);
}
Napi::Value Col(const Napi::CallbackInfo &info)
{
auto col = info[0].As<Number>().Int32Value();
return CreateMat(info.Env(), [&](CVMat &mat) { mat.mat_ = mat_.col(col); });
}
Napi::Value ColRange(const Napi::CallbackInfo &info)
{
auto start = info[0].As<Number>().Int32Value();
auto end = info[1].As<Number>().Int32Value();
return CreateMat(info.Env(), [&](CVMat &mat) { mat.mat_ = mat_.colRange(start, end); });
}
Napi::Value Row(const Napi::CallbackInfo &info)
{
auto row = info[0].As<Number>().Int32Value();
return CreateMat(info.Env(), [&](CVMat &mat) { mat.mat_ = mat_.row(row); });
}
Napi::Value RowRange(const Napi::CallbackInfo &info)
{
auto start = info[0].As<Number>().Int32Value();
auto end = info[1].As<Number>().Int32Value();
return CreateMat(info.Env(), [&](CVMat &mat) { mat.mat_ = mat_.rowRange(start, end); });
}
Napi::Value Diag(const Napi::CallbackInfo &info)
{
auto d = info[0].As<Number>().Int32Value();
return CreateMat(info.Env(), [&](CVMat &mat) { mat.mat_ = mat_.diag(d); });
}
NODE_FUNCTION_IMPL(Col)
{
auto col = info[0].As<Number>().Int32Value();
return CreateMat(info.Env(), [&](CVMat &mat) { mat.mat_ = mat_.col(col); });
}
NODE_FUNCTION_IMPL(ColRange)
{
auto start = info[0].As<Number>().Int32Value();
auto end = info[1].As<Number>().Int32Value();
return CreateMat(info.Env(), [&](CVMat &mat) { mat.mat_ = mat_.colRange(start, end); });
}
NODE_FUNCTION_IMPL(Row)
{
auto row = info[0].As<Number>().Int32Value();
return CreateMat(info.Env(), [&](CVMat &mat) { mat.mat_ = mat_.row(row); });
}
NODE_FUNCTION_IMPL(RowRange)
{
auto start = info[0].As<Number>().Int32Value();
auto end = info[1].As<Number>().Int32Value();
return CreateMat(info.Env(), [&](CVMat &mat) { mat.mat_ = mat_.rowRange(start, end); });
}
NODE_FUNCTION_IMPL(Diag)
{
auto d = info[0].As<Number>().Int32Value();
return CreateMat(info.Env(), [&](CVMat &mat) { mat.mat_ = mat_.diag(d); });
}
Napi::Value Clone(const Napi::CallbackInfo &info)
{
return CreateMat(info.Env(), [&](CVMat &mat) { mat.mat_ = mat_.clone(); });
}
Napi::Value CopyTo(const Napi::CallbackInfo &info) {
auto &target = GetMat(info[0].As<Object>());
mat_.copyTo(target.mat_);
return info.Env().Undefined();
}
NODE_FUNCTION_IMPL(Clone)
{
return CreateMat(info.Env(), [&](CVMat &mat) { mat.mat_ = mat_.clone(); });
}
NODE_FUNCTION_IMPL(CopyTo)
{
auto &target = GetMat(info[0].As<Object>());
mat_.copyTo(target.mat_);
return info.Env().Undefined();
}
static Napi::Value Crop(const Napi::CallbackInfo &info)
{
auto &src = GetMat(info[0].As<Object>());
auto &dst = GetMat(info[1].As<Object>());
auto rangeArray = info[2].As<Array>();
std::vector<cv::Range> ranges(rangeArray.Length());
for (int i = 0; i < ranges.size(); ++i) {
auto item = rangeArray.Get(i).As<Object>();
auto start = item.Get("start").As<Number>().Int32Value();
auto end = item.Get("end").As<Number>().Int32Value();
ranges[i] = cv::Range(start, end);
}
dst.mat_ = src.mat_(ranges);
return info[1];
}
static Napi::Value Resize(const Napi::CallbackInfo &info)
{
auto &src = GetMat(info[0].As<Object>());
auto &dst = GetMat(info[1].As<Object>());
auto width = info[2].As<Number>().Int32Value();
auto height = info[3].As<Number>().Int32Value();
auto fx = info[4].As<Number>().DoubleValue();
auto fy = info[5].As<Number>().DoubleValue();
auto interpolation = info[6].As<Number>().Int32Value();
cv::resize(src.mat_, dst.mat_, cv::Size(width, height), fx, fy, interpolation);
return info[1];
}
static Napi::Value WarpAffine(const Napi::CallbackInfo &info)
{
auto &src = GetMat(info[0].As<Object>());
auto &dst = GetMat(info[1].As<Object>());
auto &transformMat = GetMat(info[2].As<Object>());
auto dwidth = info[3].As<Number>().Int32Value();
auto dheight = info[4].As<Number>().Int32Value();
auto flags = info[5].As<Number>().Int32Value();
auto borderMode = info[6].As<Number>().Int32Value();
cv::warpAffine(src.mat_, dst.mat_, transformMat.mat_, cv::Size(dwidth, dheight), flags, borderMode);
return info[1];
}
static Napi::Value Blur(const Napi::CallbackInfo &info)
{
auto &src = GetMat(info[0].As<Object>());
auto &dst = GetMat(info[1].As<Object>());
auto kwidth = info[2].As<Number>().Int32Value();
auto kheight = info[3].As<Number>().Int32Value();
auto anchorX = info[4].As<Number>().Int32Value();
auto anchorY = info[5].As<Number>().Int32Value();
auto borderType = info[6].As<Number>().Int32Value();
cv::blur(src.mat_, dst.mat_, cv::Size(kwidth, kheight), cv::Point(anchorX, anchorY), borderType);
return info[1];
}
static Napi::Value CopyMakeBorder(const Napi::CallbackInfo &info)
{
auto &src = GetMat(info[0].As<Object>());
auto &dst = GetMat(info[1].As<Object>());
auto top = info[2].As<Number>().Int32Value();
auto bottom = info[3].As<Number>().Int32Value();
auto left = info[4].As<Number>().Int32Value();
auto right = info[5].As<Number>().Int32Value();
auto borderType = info[6].As<Number>().Int32Value();
cv::copyMakeBorder(src.mat_, dst.mat_, top, bottom, left, right, borderType);
return info[1];
}
static Napi::Value Flip(const Napi::CallbackInfo &info)
{
auto &src = GetMat(info[0].As<Object>());
auto &dst = GetMat(info[1].As<Object>());
auto flipCode = info[2].As<Number>().Int32Value();
cv::flip(src.mat_, dst.mat_, flipCode);
return info[1];
}
static Napi::Value GetRotationMatrix2D(const Napi::CallbackInfo &info)
{
auto centerX = info[0].As<Number>().FloatValue();
auto centerY = info[1].As<Number>().FloatValue();
auto angle = info[2].As<Number>().DoubleValue();
auto scale = info[3].As<Number>().DoubleValue();
return CreateMat(info.Env(), [&](CVMat &mat) { mat.mat_ = cv::getRotationMatrix2D(cv::Point2f(centerX, centerY), angle, scale); });
NODE_FUNCTION_IMPL(Crop)
{
auto &src = GetMat(info[0].As<Object>());
auto &dst = GetMat(info[1].As<Object>());
auto rangeArray = info[2].As<Array>();
std::vector<cv::Range> ranges(rangeArray.Length());
for (int i = 0; i < ranges.size(); ++i) {
auto item = rangeArray.Get(i).As<Object>();
auto start = item.Get("start").As<Number>().Int32Value();
auto end = item.Get("end").As<Number>().Int32Value();
ranges[i] = cv::Range(start, end);
}
dst.mat_ = src.mat_(ranges);
return info[1];
}
NODE_FUNCTION_IMPL(Resize)
{
auto &src = GetMat(info[0].As<Object>());
auto &dst = GetMat(info[1].As<Object>());
auto width = info[2].As<Number>().Int32Value();
auto height = info[3].As<Number>().Int32Value();
auto fx = info[4].As<Number>().DoubleValue();
auto fy = info[5].As<Number>().DoubleValue();
auto interpolation = info[6].As<Number>().Int32Value();
cv::resize(src.mat_, dst.mat_, cv::Size(width, height), fx, fy, interpolation);
return info[1];
}
NODE_FUNCTION_IMPL(WarpAffine)
{
auto &src = GetMat(info[0].As<Object>());
auto &dst = GetMat(info[1].As<Object>());
auto &transformMat = GetMat(info[2].As<Object>());
auto dwidth = info[3].As<Number>().Int32Value();
auto dheight = info[4].As<Number>().Int32Value();
auto flags = info[5].As<Number>().Int32Value();
auto borderMode = info[6].As<Number>().Int32Value();
cv::warpAffine(src.mat_, dst.mat_, transformMat.mat_, cv::Size(dwidth, dheight), flags, borderMode);
return info[1];
}
NODE_FUNCTION_IMPL(Blur)
{
auto &src = GetMat(info[0].As<Object>());
auto &dst = GetMat(info[1].As<Object>());
auto kwidth = info[2].As<Number>().Int32Value();
auto kheight = info[3].As<Number>().Int32Value();
auto anchorX = info[4].As<Number>().Int32Value();
auto anchorY = info[5].As<Number>().Int32Value();
auto borderType = info[6].As<Number>().Int32Value();
cv::blur(src.mat_, dst.mat_, cv::Size(kwidth, kheight), cv::Point(anchorX, anchorY), borderType);
return info[1];
}
NODE_FUNCTION_IMPL(CopyMakeBorder)
{
auto &src = GetMat(info[0].As<Object>());
auto &dst = GetMat(info[1].As<Object>());
auto top = info[2].As<Number>().Int32Value();
auto bottom = info[3].As<Number>().Int32Value();
auto left = info[4].As<Number>().Int32Value();
auto right = info[5].As<Number>().Int32Value();
auto borderType = info[6].As<Number>().Int32Value();
cv::copyMakeBorder(src.mat_, dst.mat_, top, bottom, left, right, borderType);
return info[1];
}
NODE_FUNCTION_IMPL(Flip)
{
auto &src = GetMat(info[0].As<Object>());
auto &dst = GetMat(info[1].As<Object>());
auto flipCode = info[2].As<Number>().Int32Value();
cv::flip(src.mat_, dst.mat_, flipCode);
return info[1];
}
NODE_FUNCTION_IMPL(GetRotationMatrix2D)
{
auto centerX = info[0].As<Number>().FloatValue();
auto centerY = info[1].As<Number>().FloatValue();
auto angle = info[2].As<Number>().DoubleValue();
auto scale = info[3].As<Number>().DoubleValue();
return CreateMat(info.Env(), [&](CVMat &mat) { mat.mat_ = cv::getRotationMatrix2D(cv::Point2f(centerX, centerY), angle, scale); });
}
static Napi::Value Rectangle(const Napi::CallbackInfo &info)
{
auto &img = GetMat(info[0].As<Object>());
auto x1 = info[1].As<Number>().Int32Value();
auto y1 = info[2].As<Number>().Int32Value();
auto x2 = info[3].As<Number>().Int32Value();
auto y2 = info[4].As<Number>().Int32Value();
auto b = info[5].As<Number>().DoubleValue();
auto g = info[6].As<Number>().DoubleValue();
auto r = info[7].As<Number>().DoubleValue();
auto thickness = info[8].As<Number>().Int32Value();
auto lineType = info[9].As<Number>().Int32Value();
auto shift = info[10].As<Number>().Int32Value();
cv::rectangle(img.mat_, cv::Point(x1, y1), cv::Point(x2, y2), cv::Scalar(b, g, r), thickness, lineType, shift);
return info.Env().Undefined();
NODE_FUNCTION_IMPL(Rectangle)
{
auto &img = GetMat(info[0].As<Object>());
auto x1 = info[1].As<Number>().Int32Value();
auto y1 = info[2].As<Number>().Int32Value();
auto x2 = info[3].As<Number>().Int32Value();
auto y2 = info[4].As<Number>().Int32Value();
auto b = info[5].As<Number>().DoubleValue();
auto g = info[6].As<Number>().DoubleValue();
auto r = info[7].As<Number>().DoubleValue();
auto thickness = info[8].As<Number>().Int32Value();
auto lineType = info[9].As<Number>().Int32Value();
auto shift = info[10].As<Number>().Int32Value();
cv::rectangle(img.mat_, cv::Point(x1, y1), cv::Point(x2, y2), cv::Scalar(b, g, r), thickness, lineType, shift);
return info.Env().Undefined();
}
NODE_FUNCTION_IMPL(Circle)
{
auto &img = GetMat(info[0].As<Object>());
auto x = info[1].As<Number>().Int32Value();
auto y = info[2].As<Number>().Int32Value();
auto radius = info[3].As<Number>().Int32Value();
auto b = info[4].As<Number>().DoubleValue();
auto g = info[5].As<Number>().DoubleValue();
auto r = info[6].As<Number>().DoubleValue();
auto thickness = info[7].As<Number>().Int32Value();
auto lineType = info[8].As<Number>().Int32Value();
auto shift = info[9].As<Number>().Int32Value();
cv::circle(img.mat_, cv::Point(x, y), radius, cv::Scalar(b, g, r), thickness, lineType, shift);
return info.Env().Undefined();
}
NODE_FUNCTION_IMPL(Line)
{
auto &img = GetMat(info[0].As<Object>());
auto x1 = info[1].As<Number>().Int32Value();
auto y1 = info[2].As<Number>().Int32Value();
auto x2 = info[3].As<Number>().Int32Value();
auto y2 = info[4].As<Number>().Int32Value();
auto b = info[5].As<Number>().DoubleValue();
auto g = info[6].As<Number>().DoubleValue();
auto r = info[7].As<Number>().DoubleValue();
auto thickness = info[8].As<Number>().Int32Value();
auto lineType = info[9].As<Number>().Int32Value();
auto shift = info[10].As<Number>().Int32Value();
cv::line(img.mat_, cv::Point(x1, y1), cv::Point(x2, y2), cv::Scalar(b, g, r), thickness, lineType, shift);
return info.Env().Undefined();
}
NODE_FUNCTION_IMPL(Ellipse)
{
auto &img = GetMat(info[0].As<Object>());
auto x = info[1].As<Number>().Int32Value();
auto y = info[2].As<Number>().Int32Value();
auto width = info[3].As<Number>().Int32Value();
auto height = info[4].As<Number>().Int32Value();
auto angle = info[5].As<Number>().DoubleValue();
auto startAngle = info[6].As<Number>().DoubleValue();
auto endAngle = info[7].As<Number>().DoubleValue();
auto b = info[8].As<Number>().DoubleValue();
auto g = info[9].As<Number>().DoubleValue();
auto r = info[10].As<Number>().DoubleValue();
auto thickness = info[11].As<Number>().Int32Value();
auto lineType = info[12].As<Number>().Int32Value();
auto shift = info[13].As<Number>().Int32Value();
cv::ellipse(img.mat_, cv::Point(x, y), cv::Size(width, height), angle, startAngle, endAngle, cv::Scalar(b, g, r), thickness, lineType, shift);
return info.Env().Undefined();
}
NODE_FUNCTION_IMPL(Polylines)
{
auto &img = GetMat(info[0].As<Object>());
auto points_ = info[1].As<Array>();
std::vector<cv::Point> points(points_.Length());
for (uint32_t i = 0; i < points_.Length(); i++) {
auto pt = points_.Get(i).As<Array>();
points[i] = cv::Point(pt.Get(0U).As<Number>().Int32Value(), pt.Get(1U).As<Number>().Int32Value());
}
static Napi::Value Circle(const Napi::CallbackInfo &info)
{
auto &img = GetMat(info[0].As<Object>());
auto x = info[1].As<Number>().Int32Value();
auto y = info[2].As<Number>().Int32Value();
auto radius = info[3].As<Number>().Int32Value();
auto b = info[4].As<Number>().DoubleValue();
auto g = info[5].As<Number>().DoubleValue();
auto r = info[6].As<Number>().DoubleValue();
auto thickness = info[7].As<Number>().Int32Value();
auto lineType = info[8].As<Number>().Int32Value();
auto shift = info[9].As<Number>().Int32Value();
cv::circle(img.mat_, cv::Point(x, y), radius, cv::Scalar(b, g, r), thickness, lineType, shift);
return info.Env().Undefined();
auto isClosed = info[2].As<Boolean>().Value();
auto b = info[3].As<Number>().DoubleValue();
auto g = info[4].As<Number>().DoubleValue();
auto r = info[5].As<Number>().DoubleValue();
auto thickness = info[6].As<Number>().Int32Value();
auto lineType = info[7].As<Number>().Int32Value();
auto shift = info[8].As<Number>().Int32Value();
cv::polylines(img.mat_, points, isClosed, cv::Scalar(b, g, r), thickness, lineType, shift);
return info.Env().Undefined();
}
NODE_FUNCTION_IMPL(FillPoly)
{
auto &img = GetMat(info[0].As<Object>());
auto points_ = info[1].As<Array>();
std::vector<cv::Point> points(points_.Length());
for (uint32_t i = 0; i < points_.Length(); i++) {
auto pt = points_.Get(i).As<Array>();
points[i] = cv::Point(pt.Get(0U).As<Number>().Int32Value(), pt.Get(1U).As<Number>().Int32Value());
}
static Napi::Value Line(const Napi::CallbackInfo &info)
{
auto &img = GetMat(info[0].As<Object>());
auto x1 = info[1].As<Number>().Int32Value();
auto y1 = info[2].As<Number>().Int32Value();
auto x2 = info[3].As<Number>().Int32Value();
auto y2 = info[4].As<Number>().Int32Value();
auto b = info[5].As<Number>().DoubleValue();
auto g = info[6].As<Number>().DoubleValue();
auto r = info[7].As<Number>().DoubleValue();
auto thickness = info[8].As<Number>().Int32Value();
auto lineType = info[9].As<Number>().Int32Value();
auto shift = info[10].As<Number>().Int32Value();
cv::line(img.mat_, cv::Point(x1, y1), cv::Point(x2, y2), cv::Scalar(b, g, r), thickness, lineType, shift);
return info.Env().Undefined();
auto b = info[2].As<Number>().DoubleValue();
auto g = info[3].As<Number>().DoubleValue();
auto r = info[4].As<Number>().DoubleValue();
auto lineType = info[5].As<Number>().Int32Value();
auto shift = info[6].As<Number>().Int32Value();
cv::fillPoly(img.mat_, points, cv::Scalar(b, g, r), lineType, shift);
return info.Env().Undefined();
}
NODE_FUNCTION_IMPL(FillConvexPoly)
{
auto &img = GetMat(info[0].As<Object>());
auto points_ = info[1].As<Array>();
std::vector<cv::Point> points(points_.Length());
for (uint32_t i = 0; i < points_.Length(); i++) {
auto pt = points_.Get(i).As<Array>();
points[i] = cv::Point(pt.Get(0U).As<Number>().Int32Value(), pt.Get(1U).As<Number>().Int32Value());
}
static Napi::Value Ellipse(const Napi::CallbackInfo &info)
{
auto &img = GetMat(info[0].As<Object>());
auto x = info[1].As<Number>().Int32Value();
auto y = info[2].As<Number>().Int32Value();
auto width = info[3].As<Number>().Int32Value();
auto height = info[4].As<Number>().Int32Value();
auto angle = info[5].As<Number>().DoubleValue();
auto startAngle = info[6].As<Number>().DoubleValue();
auto endAngle = info[7].As<Number>().DoubleValue();
auto b = info[8].As<Number>().DoubleValue();
auto g = info[9].As<Number>().DoubleValue();
auto r = info[10].As<Number>().DoubleValue();
auto thickness = info[11].As<Number>().Int32Value();
auto lineType = info[12].As<Number>().Int32Value();
auto shift = info[13].As<Number>().Int32Value();
cv::ellipse(img.mat_, cv::Point(x, y), cv::Size(width, height), angle, startAngle, endAngle, cv::Scalar(b, g, r), thickness, lineType, shift);
return info.Env().Undefined();
}
static Napi::Value Polylines(const Napi::CallbackInfo &info)
{
auto &img = GetMat(info[0].As<Object>());
auto points_ = info[1].As<Array>();
std::vector<cv::Point> points(points_.Length());
for (uint32_t i = 0; i < points_.Length(); i++) {
auto pt = points_.Get(i).As<Array>();
points[i] = cv::Point(pt.Get(0U).As<Number>().Int32Value(), pt.Get(1U).As<Number>().Int32Value());
}
auto isClosed = info[2].As<Boolean>().Value();
auto b = info[3].As<Number>().DoubleValue();
auto g = info[4].As<Number>().DoubleValue();
auto r = info[5].As<Number>().DoubleValue();
auto thickness = info[6].As<Number>().Int32Value();
auto lineType = info[7].As<Number>().Int32Value();
auto shift = info[8].As<Number>().Int32Value();
cv::polylines(img.mat_, points, isClosed, cv::Scalar(b, g, r), thickness, lineType, shift);
return info.Env().Undefined();
}
static Napi::Value FillPoly(const Napi::CallbackInfo &info)
{
auto &img = GetMat(info[0].As<Object>());
auto points_ = info[1].As<Array>();
std::vector<cv::Point> points(points_.Length());
for (uint32_t i = 0; i < points_.Length(); i++) {
auto pt = points_.Get(i).As<Array>();
points[i] = cv::Point(pt.Get(0U).As<Number>().Int32Value(), pt.Get(1U).As<Number>().Int32Value());
}
auto b = info[2].As<Number>().DoubleValue();
auto g = info[3].As<Number>().DoubleValue();
auto r = info[4].As<Number>().DoubleValue();
auto lineType = info[5].As<Number>().Int32Value();
auto shift = info[6].As<Number>().Int32Value();
cv::fillPoly(img.mat_, points, cv::Scalar(b, g, r), lineType, shift);
return info.Env().Undefined();
}
static Napi::Value FillConvexPoly(const Napi::CallbackInfo &info)
{
auto &img = GetMat(info[0].As<Object>());
auto points_ = info[1].As<Array>();
std::vector<cv::Point> points(points_.Length());
for (uint32_t i = 0; i < points_.Length(); i++) {
auto pt = points_.Get(i).As<Array>();
points[i] = cv::Point(pt.Get(0U).As<Number>().Int32Value(), pt.Get(1U).As<Number>().Int32Value());
}
auto b = info[2].As<Number>().DoubleValue();
auto g = info[3].As<Number>().DoubleValue();
auto r = info[4].As<Number>().DoubleValue();
auto lineType = info[5].As<Number>().Int32Value();
auto shift = info[6].As<Number>().Int32Value();
cv::fillConvexPoly(img.mat_, points, cv::Scalar(b, g, r), lineType, shift);
return info.Env().Undefined();
}
static Napi::Value DrawMarker(const Napi::CallbackInfo &info)
{
auto &img = GetMat(info[0].As<Object>());
auto x = info[1].As<Number>().Int32Value();
auto y = info[2].As<Number>().Int32Value();
auto b = info[3].As<Number>().DoubleValue();
auto g = info[4].As<Number>().DoubleValue();
auto r = info[5].As<Number>().DoubleValue();
auto markerType = info[6].As<Number>().Int32Value();
auto markerSize = info[7].As<Number>().Int32Value();
auto lineType = info[8].As<Number>().Int32Value();
auto shift = info[9].As<Number>().Int32Value();
cv::drawMarker(img.mat_, cv::Point(x, y), cv::Scalar(b, g, r), markerType, markerSize, lineType, shift);
return info.Env().Undefined();
}
static Napi::Value PutText(const Napi::CallbackInfo &info)
{
auto &img = GetMat(info[0].As<Object>());
auto text = info[1].As<String>().Utf8Value();
auto x = info[2].As<Number>().Int32Value();
auto y = info[3].As<Number>().Int32Value();
auto fontFace = info[4].As<Number>().Int32Value();
auto fontScale = info[5].As<Number>().DoubleValue();
auto b = info[6].As<Number>().DoubleValue();
auto g = info[7].As<Number>().DoubleValue();
auto r = info[8].As<Number>().DoubleValue();
auto thickness = info[9].As<Number>().Int32Value();
auto lineType = info[10].As<Number>().Int32Value();
auto shift = info[11].As<Number>().Int32Value();
auto bottomLeftOrigin = info[12].As<Boolean>().Value();
cv::putText(img.mat_, text, cv::Point(x, y), fontFace, fontScale, cv::Scalar(b, g, r), thickness, lineType, bottomLeftOrigin);
return info.Env().Undefined();
}
private:
inline static Napi::Object EmptyMat(Napi::Env env) { return constructor->New({}).As<Object>(); }
inline static CVMat &GetMat(Napi::Object obj) { return *ObjectWrap<CVMat>::Unwrap(obj); }
inline static Napi::Object CreateMat(Napi::Env env, std::function<void(CVMat &mat)> callback)
{
auto obj = EmptyMat(env);
callback(GetMat(obj));
return obj;
}
private:
cv::Mat mat_;
};
auto b = info[2].As<Number>().DoubleValue();
auto g = info[3].As<Number>().DoubleValue();
auto r = info[4].As<Number>().DoubleValue();
auto lineType = info[5].As<Number>().Int32Value();
auto shift = info[6].As<Number>().Int32Value();
cv::fillConvexPoly(img.mat_, points, cv::Scalar(b, g, r), lineType, shift);
return info.Env().Undefined();
}
NODE_FUNCTION_IMPL(DrawMarker)
{
auto &img = GetMat(info[0].As<Object>());
auto x = info[1].As<Number>().Int32Value();
auto y = info[2].As<Number>().Int32Value();
auto b = info[3].As<Number>().DoubleValue();
auto g = info[4].As<Number>().DoubleValue();
auto r = info[5].As<Number>().DoubleValue();
auto markerType = info[6].As<Number>().Int32Value();
auto markerSize = info[7].As<Number>().Int32Value();
auto lineType = info[8].As<Number>().Int32Value();
auto shift = info[9].As<Number>().Int32Value();
cv::drawMarker(img.mat_, cv::Point(x, y), cv::Scalar(b, g, r), markerType, markerSize, lineType, shift);
return info.Env().Undefined();
}
NODE_FUNCTION_IMPL(PutText)
{
auto &img = GetMat(info[0].As<Object>());
auto text = info[1].As<String>().Utf8Value();
auto x = info[2].As<Number>().Int32Value();
auto y = info[3].As<Number>().Int32Value();
auto fontFace = info[4].As<Number>().Int32Value();
auto fontScale = info[5].As<Number>().DoubleValue();
auto b = info[6].As<Number>().DoubleValue();
auto g = info[7].As<Number>().DoubleValue();
auto r = info[8].As<Number>().DoubleValue();
auto thickness = info[9].As<Number>().Int32Value();
auto lineType = info[10].As<Number>().Int32Value();
auto shift = info[11].As<Number>().Int32Value();
auto bottomLeftOrigin = info[12].As<Boolean>().Value();
cv::putText(img.mat_, text, cv::Point(x, y), fontFace, fontScale, cv::Scalar(b, g, r), thickness, lineType, bottomLeftOrigin);
return info.Env().Undefined();
}
void InitMatAPI(Napi::Env env, Napi::Object exports)
{

79
cxx/mat.h Normal file
View File

@ -0,0 +1,79 @@
#ifndef __CV_MAT_H__
#define __CV_MAT_H__
#include "node.h"
class CVMat : public Napi::ObjectWrap<CVMat> {
public:
friend class CVVideoCapture;
public:
static Napi::Object Init(Napi::Env env, Napi::Object exports);
CVMat(const Napi::CallbackInfo &info);
static Napi::Value ImRead(const Napi::CallbackInfo &info);
static Napi::Value ImDecode(const Napi::CallbackInfo &info);
static Napi::Value ImWrite(const Napi::CallbackInfo &info);
static Napi::Value ImEncode(const Napi::CallbackInfo &info);
Napi::Value IsEmpty(const Napi::CallbackInfo &info);
Napi::Value GetCols(const Napi::CallbackInfo &info);
Napi::Value GetRows(const Napi::CallbackInfo &info);
Napi::Value GetType(const Napi::CallbackInfo &info);
Napi::Value GetDepth(const Napi::CallbackInfo &info);
Napi::Value GetChannels(const Napi::CallbackInfo &info);
Napi::Value GetFlags(const Napi::CallbackInfo &info);
Napi::Value GetDims(const Napi::CallbackInfo &info);
Napi::Value GetIsContinuous(const Napi::CallbackInfo &info);
Napi::Value GetIsSubmatrix(const Napi::CallbackInfo &info);
Napi::Value GetElemSize(const Napi::CallbackInfo &info);
Napi::Value GetElemSize1(const Napi::CallbackInfo &info);
Napi::Value GetTotal(const Napi::CallbackInfo &info);
Napi::Value GetTotalWithDim(const Napi::CallbackInfo &info);
Napi::Value GetSize(const Napi::CallbackInfo &info);
Napi::Value GetData(const Napi::CallbackInfo &info);
Napi::Value Col(const Napi::CallbackInfo &info);
Napi::Value ColRange(const Napi::CallbackInfo &info);
Napi::Value Row(const Napi::CallbackInfo &info);
Napi::Value RowRange(const Napi::CallbackInfo &info);
Napi::Value Diag(const Napi::CallbackInfo &info);
Napi::Value Clone(const Napi::CallbackInfo &info);
Napi::Value CopyTo(const Napi::CallbackInfo &info);
static Napi::Value Crop(const Napi::CallbackInfo &info);
static Napi::Value Resize(const Napi::CallbackInfo &info);
static Napi::Value WarpAffine(const Napi::CallbackInfo &info);
static Napi::Value Blur(const Napi::CallbackInfo &info);
static Napi::Value CopyMakeBorder(const Napi::CallbackInfo &info);
static Napi::Value Flip(const Napi::CallbackInfo &info);
static Napi::Value GetRotationMatrix2D(const Napi::CallbackInfo &info);
static Napi::Value Rectangle(const Napi::CallbackInfo &info);
static Napi::Value Circle(const Napi::CallbackInfo &info);
static Napi::Value Line(const Napi::CallbackInfo &info);
static Napi::Value Ellipse(const Napi::CallbackInfo &info);
static Napi::Value Polylines(const Napi::CallbackInfo &info);
static Napi::Value FillPoly(const Napi::CallbackInfo &info);
static Napi::Value FillConvexPoly(const Napi::CallbackInfo &info);
static Napi::Value DrawMarker(const Napi::CallbackInfo &info);
static Napi::Value PutText(const Napi::CallbackInfo &info);
private:
inline static Napi::Object EmptyMat(Napi::Env env) { return constructor->New({}).As<Napi::Object>(); }
inline static CVMat &GetMat(Napi::Object obj) { return *ObjectWrap<CVMat>::Unwrap(obj); }
inline static Napi::Object CreateMat(Napi::Env env, std::function<void(CVMat &mat)> callback)
{
auto obj = EmptyMat(env);
callback(GetMat(obj));
return obj;
}
private:
static Napi::FunctionReference *constructor;
cv::Mat mat_;
};
#endif

View File

@ -4,8 +4,13 @@
static Napi::Object Init(Napi::Env env, Napi::Object exports)
{
#ifdef RELEASE_VERSION
exports.Set("__release__", Napi::String::New(env, RELEASE_VERSION));
#endif
InitMatAPI(env, exports);
InitProcAPI(env, exports);
InitVideoCaptureAPI(env, exports);
InitUtilAPI(env, exports);
return exports;
}
NODE_API_MODULE(addon, Init);

View File

@ -4,7 +4,11 @@
#include <napi.h>
#include <opencv2/opencv.hpp>
#define NODE_FUNCTION(name) Napi::Value name(const CallbackInfo &info)
#define NODE_FUNCTION_STATIC(name) static Napi::Value name(const CallbackInfo &info)
void InitMatAPI(Napi::Env env, Napi::Object exports);
void InitProcAPI(Napi::Env env, Napi::Object exports);
void InitVideoCaptureAPI(Napi::Env env, Napi::Object exports);
void InitUtilAPI(Napi::Env env, Napi::Object exports);
#endif

View File

@ -27,7 +27,7 @@ DEFINE_FUNCTION(GetFontScaleFromHeight)
return Number::New(info.Env(), cv::getFontScaleFromHeight(fontFace, pixelHeight, thickness));
}
void InitProcAPI(Napi::Env env, Napi::Object exports)
void InitUtilAPI(Napi::Env env, Napi::Object exports)
{
auto obj = Object::New(env);
ADD_PROC_FUNCTION(GetTextSize);

114
cxx/videocap.cc Normal file
View File

@ -0,0 +1,114 @@
#include <iostream>
#include <vector>
#include "node.h"
#include "mat.h"
#include "videocap.h"
using namespace Napi;
#define MAT_INSTANCE_METHOD(method) InstanceMethod<&CVVideoCapture::method>(#method, static_cast<napi_property_attributes>(napi_writable | napi_configurable))
#define MAT_STATIC_METHOD(method) StaticMethod<&CVVideoCapture::method>(#method, static_cast<napi_property_attributes>(napi_writable | napi_configurable))
FunctionReference *CVVideoCapture::constructor = nullptr;
Napi::Object CVVideoCapture::Init(Napi::Env env, Napi::Object exports)
{
Function func = DefineClass(env, "VideoCapture", {
MAT_INSTANCE_METHOD(Open),
MAT_INSTANCE_METHOD(IsOpened),
MAT_INSTANCE_METHOD(Grab),
MAT_INSTANCE_METHOD(Retrieve),
MAT_INSTANCE_METHOD(Read),
MAT_INSTANCE_METHOD(Set),
MAT_INSTANCE_METHOD(Get),
MAT_INSTANCE_METHOD(GetBackendName),
});
constructor = new FunctionReference();
*constructor = Napi::Persistent(func);
exports.Set("VideoCapture", func);
env.SetInstanceData<FunctionReference>(constructor);
return exports;
}
CVVideoCapture::CVVideoCapture(const Napi::CallbackInfo &info)
: ObjectWrap<CVVideoCapture>(info)
{
if (info[0].IsString()) {
auto apiPreference = (info.Length() > 1) ? info[1].As<Number>().Int32Value() : cv::CAP_ANY;
std::vector<int> params;
if (info.Length() > 2) {
auto paramsArray = info[2].As<Array>();
params.resize(paramsArray.Length());
for (auto i = 0; i < params.size(); ++i) params[i] = paramsArray.Get(i).As<Number>().Int32Value();
}
capture_ = cv::VideoCapture(info[0].As<String>().Utf8Value(), apiPreference, params);
}
else if (info[0].IsNumber()) {
auto index = info[0].As<Number>().Int32Value();
auto apiPreference = (info.Length() > 1) ? info[1].As<Number>().Int32Value() : cv::CAP_ANY;
std::vector<int> params;
if (info.Length() > 2) {
auto paramsArray = info[2].As<Array>();
params.resize(paramsArray.Length());
for (auto i = 0; i < params.size(); ++i) params[i] = paramsArray.Get(i).As<Number>().Int32Value();
}
capture_ = cv::VideoCapture(index, apiPreference, params);
}
else capture_ = cv::VideoCapture();
}
Napi::Value CVVideoCapture::Open(const Napi::CallbackInfo &info)
{
bool res = false;
auto apiPreference = (info.Length() > 1) ? info[1].As<Number>().Int32Value() : cv::CAP_ANY;
std::vector<int> params;
if (info.Length() > 2) {
auto paramsArray = info[2].As<Array>();
params.resize(paramsArray.Length());
for (auto i = 0; i < params.size(); ++i) params[i] = paramsArray.Get(i).As<Number>().Int32Value();
}
if (info[0].IsString()) res = capture_.open(info[0].As<String>().Utf8Value(), apiPreference, params);
else if (info[0].IsNumber()) res = capture_.open(info[0].As<Number>().Int32Value(), apiPreference, params);
return Boolean::New(info.Env(), res);
}
Napi::Value CVVideoCapture::IsOpened(const Napi::CallbackInfo &info)
{
return Boolean::New(info.Env(), capture_.isOpened());
}
Napi::Value CVVideoCapture::Grab(const Napi::CallbackInfo &info)
{
return Boolean::New(info.Env(), capture_.grab());
}
Napi::Value CVVideoCapture::Retrieve(const Napi::CallbackInfo &info)
{
auto flag = info[0].As<Number>().Int32Value();
return CVMat::CreateMat(info.Env(), [&](CVMat &mat) {
capture_.retrieve(mat.mat_, flag);
});
}
Napi::Value CVVideoCapture::Read(const Napi::CallbackInfo &info)
{
return CVMat::CreateMat(info.Env(), [&](CVMat &mat) {
capture_.read(mat.mat_);
});
}
Napi::Value CVVideoCapture::Set(const Napi::CallbackInfo &info)
{
capture_.set(info[0].As<Number>().Int32Value(), info[1].As<Number>().DoubleValue());
return info.Env().Undefined();
}
Napi::Value CVVideoCapture::Get(const Napi::CallbackInfo &info)
{
return Number::New(info.Env(), capture_.get(info[0].As<Number>().Int32Value()));
}
Napi::Value CVVideoCapture::GetBackendName(const Napi::CallbackInfo &info)
{
return String::New(info.Env(), capture_.getBackendName());
}
void InitVideoCaptureAPI(Napi::Env env, Napi::Object exports)
{
CVVideoCapture::Init(env, exports);
}

26
cxx/videocap.h Normal file
View File

@ -0,0 +1,26 @@
#ifndef __CV_VIDEO_CAP_H__
#define __CV_VIDEO_CAP_H__
#include "node.h"
class CVVideoCapture : public Napi::ObjectWrap<CVVideoCapture> {
public:
static Napi::Object Init(Napi::Env env, Napi::Object exports);
CVVideoCapture(const Napi::CallbackInfo &info);
Napi::Value Open(const Napi::CallbackInfo &info);
Napi::Value IsOpened(const Napi::CallbackInfo &info);
Napi::Value Grab(const Napi::CallbackInfo &info);
Napi::Value Retrieve(const Napi::CallbackInfo &info);
Napi::Value Read(const Napi::CallbackInfo &info);
Napi::Value Set(const Napi::CallbackInfo &info);
Napi::Value Get(const Napi::CallbackInfo &info);
Napi::Value GetBackendName(const Napi::CallbackInfo &info);
private:
static Napi::FunctionReference *constructor;
cv::VideoCapture capture_;
};
#endif

View File

@ -1,7 +1,7 @@
{
"name": "@yizhi/cv",
"version": "1.0.2",
"releaseVersion": "1.0.2",
"version": "1.0.4",
"releaseVersion": "1.0.3",
"main": "dist/index.js",
"types": "typing/index.d.ts",
"scripts": {

View File

@ -18,5 +18,5 @@ export function getConfig<N extends keyof AddonConfig>(name: N): AddonConfig[N]
export function setConfig<N extends keyof AddonConfig>(name: N, value: AddonConfig[N]) { config[name] = value; }
export function CVMat() { return getAddon().Mat; }
export function CVVideoCapture() { return getAddon().VideoCapture; }
export function CVUtil() { return getAddon().util; }

View File

@ -1,10 +1,14 @@
export { getConfig as C } from "./addon";
import { Mat } from "./mat";
import { VideoCapture } from "./videocapture";
export type TypedArray = Uint8Array | Int8Array | Uint16Array | Int16Array | Uint32Array | Int32Array | BigUint64Array | BigInt64Array | Float32Array | Float64Array
export function FromCV(mat: any): Mat { return new (Mat as any)(mat); }
export function __Mat(mat: any): Mat { return new (Mat as any)(mat); }
export function __VideoCapture(cap: any): VideoCapture { return new (VideoCapture as any)(cap); }
export function M(mat: Mat) { return (mat as any).__mat__; }
export function VC(cap: VideoCapture) { return (cap as any).__cap__; }
export function resolveArgs<R extends Array<any>>(args: any[], checker: boolean | (() => boolean)): R {
if (typeof checker === "function") checker = checker();

View File

@ -164,3 +164,114 @@ export const MARKER_DIAMOND = 3;
export const MARKER_SQUARE = 4;
export const MARKER_TRIANGLE_UP = 5;
export const MARKER_TRIANGLE_DOWN = 6;
export const CAP_ANY = 0;
export const CAP_VFW = 200;
export const CAP_V4L = 200;
export const CAP_V4L2 = CAP_V4L;
export const CAP_FIREWIRE = 300;
export const CAP_FIREWARE = CAP_FIREWIRE;
export const CAP_IEEE1394 = CAP_FIREWIRE;
export const CAP_DC1394 = CAP_FIREWIRE;
export const CAP_CMU1394 = CAP_FIREWIRE;
export const CAP_QT = 500;
export const CAP_UNICAP = 600;
export const CAP_DSHOW = 700;
export const CAP_PVAPI = 800;
export const CAP_OPENNI = 900;
export const CAP_OPENNI_ASUS = 910;
export const CAP_ANDROID = 1000;
export const CAP_XIAPI = 1100;
export const CAP_AVFOUNDATION = 1200;
export const CAP_GIGANETIX = 1300;
export const CAP_MSMF = 1400;
export const CAP_WINRT = 1410;
export const CAP_INTELPERC = 1500;
export const CAP_REALSENSE = 1500;
export const CAP_OPENNI2 = 1600;
export const CAP_OPENNI2_ASUS = 1610;
export const CAP_OPENNI2_ASTRA = 1620;
export const CAP_GPHOTO2 = 1700;
export const CAP_GSTREAMER = 1800;
export const CAP_FFMPEG = 1900;
export const CAP_IMAGES = 2000;
export const CAP_ARAVIS = 2100;
export const CAP_OPENCV_MJPEG = 2200;
export const CAP_INTEL_MFX = 2300;
export const CAP_XINE = 2400;
export const CAP_UEYE = 2500;
export const CAP_OBSENSOR = 2600;
export const CAP_PROP_POS_MSEC = 0;
export const CAP_PROP_POS_FRAMES = 1;
export const CAP_PROP_POS_AVI_RATIO = 2;
export const CAP_PROP_FRAME_WIDTH = 3;
export const CAP_PROP_FRAME_HEIGHT = 4;
export const CAP_PROP_FPS = 5;
export const CAP_PROP_FOURCC = 6;
export const CAP_PROP_FRAME_COUNT = 7;
export const CAP_PROP_FORMAT = 8;
export const CAP_PROP_MODE = 9;
export const CAP_PROP_BRIGHTNESS = 10;
export const CAP_PROP_CONTRAST = 11;
export const CAP_PROP_SATURATION = 12;
export const CAP_PROP_HUE = 13;
export const CAP_PROP_GAIN = 14;
export const CAP_PROP_EXPOSURE = 15;
export const CAP_PROP_CONVERT_RGB = 16;
export const CAP_PROP_WHITE_BALANCE_BLUE_U = 17;
export const CAP_PROP_RECTIFICATION = 18;
export const CAP_PROP_MONOCHROME = 1;
export const CAP_PROP_SHARPNESS = 2;
export const CAP_PROP_AUTO_EXPOSURE = 21;
export const CAP_PROP_GAMMA = 2;
export const CAP_PROP_TEMPERATURE = 2;
export const CAP_PROP_TRIGGER = 2;
export const CAP_PROP_TRIGGER_DELAY = 2;
export const CAP_PROP_WHITE_BALANCE_RED_V = 2;
export const CAP_PROP_ZOOM = 2;
export const CAP_PROP_FOCUS = 2;
export const CAP_PROP_GUID = 2;
export const CAP_PROP_ISO_SPEED = 3;
export const CAP_PROP_BACKLIGHT = 3;
export const CAP_PROP_PAN = 3;
export const CAP_PROP_TILT = 3;
export const CAP_PROP_ROLL = 3;
export const CAP_PROP_IRIS = 3;
export const CAP_PROP_SETTINGS = 37;
export const CAP_PROP_BUFFERSIZE = 3;
export const CAP_PROP_AUTOFOCUS = 3;
export const CAP_PROP_SAR_NUM = 40;
export const CAP_PROP_SAR_DEN = 41;
export const CAP_PROP_BACKEND = 42;
export const CAP_PROP_CHANNEL = 43;
export const CAP_PROP_AUTO_WB = 44;
export const CAP_PROP_WB_TEMPERATURE = 45;
export const CAP_PROP_CODEC_PIXEL_FORMAT = 46;
export const CAP_PROP_BITRATE = 47;
export const CAP_PROP_ORIENTATION_META = 48;
export const CAP_PROP_ORIENTATION_AUTO = 49;
export const CAP_PROP_HW_ACCELERATION = 50;
export const CAP_PROP_HW_DEVICE = 51;
export const CAP_PROP_HW_ACCELERATION_USE_OPENCL = 52;
export const CAP_PROP_OPEN_TIMEOUT_MSEC = 53;
export const CAP_PROP_READ_TIMEOUT_MSEC = 54;
export const CAP_PROP_STREAM_OPEN_TIME_USEC = 55;
export const CAP_PROP_VIDEO_TOTAL_CHANNELS = 5;
export const CAP_PROP_VIDEO_STREAM = 5;
export const CAP_PROP_AUDIO_STREAM = 5;
export const CAP_PROP_AUDIO_POS = 5;
export const CAP_PROP_AUDIO_SHIFT_NSEC = 6;
export const CAP_PROP_AUDIO_DATA_DEPTH = 6;
export const CAP_PROP_AUDIO_SAMPLES_PER_SECOND = 6;
export const CAP_PROP_AUDIO_BASE_INDEX = 6;
export const CAP_PROP_AUDIO_TOTAL_CHANNELS = 6;
export const CAP_PROP_AUDIO_TOTAL_STREAMS = 6;
export const CAP_PROP_AUDIO_SYNCHRONIZE = 6;
export const CAP_PROP_LRF_HAS_KEY_FRAME = 6;
export const CAP_PROP_CODEC_EXTRADATA_INDEX = 6;
export const CAP_PROP_FRAME_TYPE = 6;
export const CAP_PROP_N_THREADS = 7;
export const CAP_PROP_PTS = 7;
export const CAP_PROP_DTS_DELAY = 7;

View File

@ -3,34 +3,34 @@ import fs from "fs";
import path from "path";
import { C } from "./common";
const URLS = {
GITHUB: `https://github.com/kangkang520/node-addons/releases/download/cv{{version}}/cv_{{platform}}_{{arch}}.node`,
URNAS: `http://git.urnas.cn:5200/yizhi-js-lib/opencv/releases/download/{{version}}/cv_{{platform}}_{{arch}}.node`,
GITHUB: `https://github.com/kangkang520/node-addons/releases/download/cv{{version}}/{{filename}}`,
URNAS: `http://git.urnas.cn:5200/yizhi-js-lib/opencv/releases/download/{{version}}/{{filename}}`,
}
function getURL(template: string) {
const version = require("../../package.json").releaseVersion;
let platform = "";
let arch = "";
switch (os.platform()) {
case "win32":
platform = "windows";
break;
case "linux":
platform = "linux";
break;
default:
throw new Error(`Unsupported platform: ${os.platform()}, Please compile the addon yourself.`);
}
switch (os.arch()) {
case "x64":
arch = "x64";
break;
default:
throw new Error(`Unsupported architecture: ${os.arch()}, Please compile the addon yourself.`);
}
function releaseVersion() { return require("../../package.json").releaseVersion }
return template.replaceAll("{{version}}", version).replaceAll("{{platform}}", platform).replaceAll("{{arch}}", arch);
function getURL(template: string) {
const URL_CONFIG: Record<string, Record<string, string>> = {
"win32": {
"x64": "cv_windows_x64.node",
},
"linux": {
"x64": "cv_linux_x64.node",
"arm64": "cv_linux_arm64.node",
},
"darwin": {
"arm64": "cv_macos_arm64.node"
}
};
const archConfig = URL_CONFIG[os.platform()];
if (!archConfig) throw new Error(`Unsupported platform: ${os.platform()}`);
const downloadName = archConfig[os.arch()];
if (!downloadName) throw new Error(`Unsupported architecture: ${os.arch()}`);
return template.replaceAll("{{version}}", releaseVersion()).replaceAll("{{filename}}", downloadName);
}
async function getStream() {
@ -48,7 +48,14 @@ async function getStream() {
export async function downloadAddon(savename?: string) {
const defaultAddon = path.resolve(process.cwd(), C("ADDON_PATH"));
const saveName = savename ? path.resolve(path.dirname(defaultAddon), savename) : defaultAddon;
if (fs.existsSync(saveName)) return saveName;
if (fs.existsSync(saveName)) {
try {
const addon = require(saveName);
if (addon.__release__ === releaseVersion()) return saveName;
//清除缓存
delete require.cache[saveName];
} catch (err) { }
}
await fs.promises.mkdir(path.dirname(saveName), { recursive: true });

View File

@ -1,6 +1,7 @@
export { setConfig as config } from "./addon";
export { downloadAddon } from "./download";
export * from "./mat";
export * from "./videocapture";
export * from "./consts";
export * from "./proc";
export * from "./draw";

View File

@ -1,5 +1,5 @@
import { CVMat } from "./addon";
import { FromCV, M, TypedArray } from "./common";
import { __Mat, M, TypedArray } from "./common";
import { IMREAD_COLOR_BGR } from "./consts";
@ -40,18 +40,18 @@ export class Mat {
else return this.copyTo(this.clone()).data;
}
public clone() { return FromCV(this.#mat.Clone()); }
public clone() { return __Mat(this.#mat.Clone()); }
public copyTo(mat: Mat) { return this.#mat.CopyTo(M(mat)), mat; }
public row(y: number) { return FromCV(this.#mat.Row(y)); }
public col(x: number) { return FromCV(this.#mat.Col(x)); }
public rowRange(start: number, end: number) { return FromCV(this.#mat.RowRange(start, end)); }
public colRange(start: number, end: number) { return FromCV(this.#mat.ColRange(start, end)); }
public diag(d: number) { return FromCV(this.#mat.Diag(d)); }
public row(y: number) { return __Mat(this.#mat.Row(y)); }
public col(x: number) { return __Mat(this.#mat.Col(x)); }
public rowRange(start: number, end: number) { return __Mat(this.#mat.RowRange(start, end)); }
public colRange(start: number, end: number) { return __Mat(this.#mat.ColRange(start, end)); }
public diag(d: number) { return __Mat(this.#mat.Diag(d)); }
};
export function imread(filename: string, flag?: number) { return FromCV(CVMat().ImRead(filename, flag ?? IMREAD_COLOR_BGR)); }
export function imdecode(data: Uint8Array, flag?: number) { return FromCV(CVMat().ImDecode(data, flag ?? IMREAD_COLOR_BGR)); }
export function imread(filename: string, flag?: number) { return __Mat(CVMat().ImRead(filename, flag ?? IMREAD_COLOR_BGR)); }
export function imdecode(data: Uint8Array, flag?: number) { return __Mat(CVMat().ImDecode(data, flag ?? IMREAD_COLOR_BGR)); }
export function imwrite(filename: string, mat: Mat, params?: number[]): boolean { return CVMat().ImWrite(M(mat), filename, params); }
export function imencode(ext: string, mat: Mat, params?: number[]): Uint8Array | null {
const res = CVMat().ImEncode(M(mat), ext, params);

View File

@ -1,5 +1,5 @@
import { Mat } from "./mat";
import { FromCV, M, resolveArgs } from "./common";
import { __Mat, M, resolveArgs } from "./common";
import { BORDER_CONSTANT, BORDER_DEFAULT, INTER_LINEAR } from "./consts";
import { CVMat } from "./addon";
@ -9,14 +9,14 @@ export function crop(src: Mat, dst: Mat, range: CropRange): Mat;
export function crop(...args: any[]) {
let [src, dst, range] = resolveArgs<[src: Mat, dst: Mat, range: CropRange]>(args, args[1] instanceof Mat);
if (!(range instanceof Array)) range = [{ start: range.y, end: range.y + range.height }, { start: range.x, end: range.x + range.width }];
return FromCV(CVMat().Crop(M(src), M(dst), range));
return __Mat(CVMat().Crop(M(src), M(dst), range));
}
export function resize(src: Mat, width: number, height: number, fx?: number, fy?: number, interpolation?: number): Mat;
export function resize(src: Mat, dst: Mat, width: number, height: number, fx?: number, fy?: number, interpolation?: number): Mat;
export function resize(...args: any[]) {
const [src, dst, width, height, fx, fy, interpolation] = resolveArgs<[src: Mat, dst: Mat, width: number, height: number, fx?: number, fy?: number, interpolation?: number]>(args, args[1] instanceof Mat);
return FromCV(CVMat().Resize(M(src), M(dst), width, height, fx ?? 0, fy ?? 0, interpolation ?? INTER_LINEAR));
return __Mat(CVMat().Resize(M(src), M(dst), width, height, fx ?? 0, fy ?? 0, interpolation ?? INTER_LINEAR));
}
export function blur(src: Mat, kernelWidth: number, kernelHeight: number, anchorX?: number, anchorY?: number, borderType?: number): Mat
@ -24,14 +24,14 @@ export function blur(src: Mat, dst: Mat, kernelWidth: number, kernelHeight: numb
export function blur(...args: any[]) {
let [src, dst, kernelWidth, kernelHeight, anchorX = -1, anchorY = -1, borderType = BORDER_DEFAULT] = resolveArgs<[src: Mat, dst: Mat, kernelWidth: number, kernelHeight: number, anchorX?: number, anchorY?: number, borderType?: number]>(args, args[1] instanceof Mat);
if (anchorX == -1 || anchorY == -1) anchorX = anchorY = -1;
return FromCV(CVMat().Blur(M(src), M(dst), kernelWidth, kernelHeight, anchorX, anchorY, borderType));
return __Mat(CVMat().Blur(M(src), M(dst), kernelWidth, kernelHeight, anchorX, anchorY, borderType));
}
export function copyMakeBorder(src: Mat, top: number, bottom: number, left: number, right: number, borderType: number): Mat
export function copyMakeBorder(src: Mat, dst: Mat, top: number, bottom: number, left: number, right: number, borderType: number): Mat
export function copyMakeBorder(...args: any[]) {
const [src, dst, top, bottom, left, right, borderType] = resolveArgs<[src: Mat, dst: Mat, top: number, bottom: number, left: number, right: number, borderType: number]>(args, args[1] instanceof Mat);
return FromCV(CVMat().CopyMakeBorder(M(src), M(dst), top, bottom, left, right, borderType));
return __Mat(CVMat().CopyMakeBorder(M(src), M(dst), top, bottom, left, right, borderType));
}
type FlipCode = 0 | 1 | -1 | "x" | "y" | "both";
@ -40,14 +40,14 @@ export function flip(src: Mat, dst: Mat, flipCode: FlipCode): Mat
export function flip(...args: any[]) {
let [src, dst, flipCode] = resolveArgs<[src: Mat, dst: Mat, flipCode: FlipCode]>(args, args[1] instanceof Mat);
if (typeof flipCode == "string") flipCode = ({ x: 0, y: 1, both: -1 } as Record<"x" | "y" | "both", 0 | 1 | -1>)[flipCode];
return FromCV(CVMat().Flip(M(src), M(dst), flipCode));
return __Mat(CVMat().Flip(M(src), M(dst), flipCode));
}
export function warpAffine(src: Mat, transfromMat: Mat, dwidth: number, dheight: number, flags?: number, borderMode?: number): Mat;
export function warpAffine(src: Mat, dst: Mat, transfromMat: Mat, dwidth: number, dheight: number, flags?: number, borderMode?: number): Mat;
export function warpAffine(...args: any[]) {
const [src, dst, transfromMat, dwidth, dheight, flags, borderMode] = resolveArgs<[src: Mat, dst: Mat, transfromMat: Mat, dwidth: number, dheight: number, flags?: number, borderMode?: number]>(args, () => args[2] instanceof Mat);
return FromCV(CVMat().WarpAffine(M(src), M(dst), M(transfromMat), dwidth, dheight, flags ?? INTER_LINEAR, borderMode ?? BORDER_CONSTANT));
return __Mat(CVMat().WarpAffine(M(src), M(dst), M(transfromMat), dwidth, dheight, flags ?? INTER_LINEAR, borderMode ?? BORDER_CONSTANT));
}
export function getRotationMatrix2D(centerX: number, centerY: number, angle: number, scale: number) { return FromCV(CVMat().GetRotationMatrix2D(centerX, centerY, angle, scale)); }
export function getRotationMatrix2D(centerX: number, centerY: number, angle: number, scale: number) { return __Mat(CVMat().GetRotationMatrix2D(centerX, centerY, angle, scale)); }

33
src/cv/videocapture.ts Normal file
View File

@ -0,0 +1,33 @@
import { CVVideoCapture } from "./addon";
import { __Mat } from "./common";
import { CAP_ANY } from "./consts";
export class VideoCapture {
#cap: any
constructor(source: string | number, apiPreference?: number, params?: number[]) {
const CVCap = CVVideoCapture();
if ((source as any) instanceof CVCap) this.#cap = source;
else this.#cap = new CVCap(source, apiPreference ?? CAP_ANY, params ?? []);
}
private get __cap__() { return this.#cap; }
public open(source: string | number, apiPreference?: number, params?: number[]): boolean { return this.#cap.Open(source, apiPreference ?? CAP_ANY, params ?? []); }
public get isOpened(): boolean { return this.#cap.IsOpened(); }
public grab(): boolean { return this.#cap.Grab(); }
public retrieve(flag?: number) { return __Mat(this.#cap.Retrieve(flag ?? 0)); }
public read() { return __Mat(this.#cap.Read()); }
public get backendName(): string { return this.#cap.GetBackendName(); }
public prop(propId: number, value: number): void
public prop(propId: number): number
public prop(propId: number, value?: number) {
if (typeof value === "undefined") return this.#cap.Get(propId);
else this.#cap.Set(propId, value);
}
}

View File

@ -1,7 +1,7 @@
import cv from ".";
async function test() {
async function testImage() {
await cv.downloadAddon();
const fs = await import("fs");
// const buffer = fs.readFileSync("data/im1.jpeg")
@ -32,5 +32,15 @@ async function test() {
cv.imwrite("test_data/draw.jpg", res);
}
async function testMovie() {
await cv.downloadAddon();
const cap = new cv.VideoCapture("test_data/movie.mp4");
if (cap.grab()) {
const im = cap.retrieve();
debugger;
}
const im = cap.read();
debugger
}
test();
testMovie();

View File

@ -15,4 +15,5 @@ function runCmakeJS(args) {
if (args.includes("--include")) runCmakeJS(["print-cmakejs-include"]);
else if (args.includes("--src")) runCmakeJS(["print-cmakejs-src"]);
else if (args.includes("--lib")) runCmakeJS(["print-cmakejs-lib"]);
else if (args.includes("--napi")) console.log(require("node-addon-api").include.replace(/^"/, "").replace(/"$/, ""));
else if (args.includes("--napi")) console.log(require("node-addon-api").include.replace(/^"/, "").replace(/"$/, ""));
else if (args.includes("--release")) console.log(require("../package.json").releaseVersion);

View File

@ -140,7 +140,6 @@ async function downloadFromURL(name, url, resolver) {
fs.rmSync(outputDir, { recursive: true, force: true });
}
async function main() {
//OpenCV
if (buildOptions.withOpenCV) cmakeBuildFromSource("OpenCV", "https://github.com/opencv/opencv.git", "4.11.0", null, [
@ -155,7 +154,12 @@ async function main() {
"-DBUILD_FAT_JAVA_LIB=OFF",
"-DBUILD_ANDROID_SERVICE=OFF",
"-DBUILD_JAVA=OFF",
"-DBUILD_PERF_TESTS=OFF"
"-DBUILD_PERF_TESTS=OFF",
"-DBUILD_TIFF=ON",
"-DBUILD_OPENJPEG=ON",
"-DBUILD_JPEG=ON",
"-DBUILD_PNG=ON",
"-DBUILD_WEBP=ON",
], (root) => [
`set(OpenCV_STATIC ON)`,
os.platform() == "win32" ?