99 lines
3.7 KiB
Plaintext
99 lines
3.7 KiB
Plaintext
//
|
|
// FrameHostObject.m
|
|
// VisionCamera
|
|
//
|
|
// Created by Marc Rousavy on 22.03.21.
|
|
// Copyright © 2021 mrousavy. All rights reserved.
|
|
//
|
|
|
|
#import "FrameHostObject.h"
|
|
#import <Foundation/Foundation.h>
|
|
#import <jsi/jsi.h>
|
|
|
|
std::vector<jsi::PropNameID> FrameHostObject::getPropertyNames(jsi::Runtime& rt) {
|
|
std::vector<jsi::PropNameID> result;
|
|
result.push_back(jsi::PropNameID::forUtf8(rt, std::string("toString")));
|
|
result.push_back(jsi::PropNameID::forUtf8(rt, std::string("isValid")));
|
|
result.push_back(jsi::PropNameID::forUtf8(rt, std::string("width")));
|
|
result.push_back(jsi::PropNameID::forUtf8(rt, std::string("height")));
|
|
result.push_back(jsi::PropNameID::forUtf8(rt, std::string("bytesPerRow")));
|
|
result.push_back(jsi::PropNameID::forUtf8(rt, std::string("planesCount")));
|
|
result.push_back(jsi::PropNameID::forUtf8(rt, std::string("close")));
|
|
return result;
|
|
}
|
|
|
|
jsi::Value FrameHostObject::get(jsi::Runtime& runtime, const jsi::PropNameID& propName) {
|
|
auto name = propName.utf8(runtime);
|
|
|
|
if (name == "toString") {
|
|
auto toString = [this] (jsi::Runtime& runtime, const jsi::Value&, const jsi::Value*, size_t) -> jsi::Value {
|
|
if (this->frame == nil) {
|
|
return jsi::String::createFromUtf8(runtime, "[closed frame]");
|
|
}
|
|
auto imageBuffer = CMSampleBufferGetImageBuffer(frame.buffer);
|
|
auto width = CVPixelBufferGetWidth(imageBuffer);
|
|
auto height = CVPixelBufferGetHeight(imageBuffer);
|
|
|
|
NSMutableString* string = [NSMutableString stringWithFormat:@"%lu x %lu Frame", width, height];
|
|
return jsi::String::createFromUtf8(runtime, string.UTF8String);
|
|
};
|
|
return jsi::Function::createFromHostFunction(runtime, jsi::PropNameID::forUtf8(runtime, "toString"), 0, toString);
|
|
}
|
|
if (name == "close") {
|
|
auto close = [this] (jsi::Runtime& runtime, const jsi::Value&, const jsi::Value*, size_t) -> jsi::Value {
|
|
if (this->frame == nil) {
|
|
throw jsi::JSError(runtime, "Trying to close an already closed frame! Did you call frame.close() twice?");
|
|
}
|
|
this->close();
|
|
return jsi::Value::undefined();
|
|
};
|
|
return jsi::Function::createFromHostFunction(runtime, jsi::PropNameID::forUtf8(runtime, "close"), 0, close);
|
|
}
|
|
|
|
if (name == "isValid") {
|
|
auto isValid = frame != nil && CMSampleBufferIsValid(frame.buffer);
|
|
return jsi::Value(isValid);
|
|
}
|
|
if (name == "width") {
|
|
this->assertIsFrameStrong(runtime, name);
|
|
auto imageBuffer = CMSampleBufferGetImageBuffer(frame.buffer);
|
|
auto width = CVPixelBufferGetWidth(imageBuffer);
|
|
return jsi::Value((double) width);
|
|
}
|
|
if (name == "height") {
|
|
this->assertIsFrameStrong(runtime, name);
|
|
auto imageBuffer = CMSampleBufferGetImageBuffer(frame.buffer);
|
|
auto height = CVPixelBufferGetHeight(imageBuffer);
|
|
return jsi::Value((double) height);
|
|
}
|
|
if (name == "bytesPerRow") {
|
|
this->assertIsFrameStrong(runtime, name);
|
|
auto imageBuffer = CMSampleBufferGetImageBuffer(frame.buffer);
|
|
auto bytesPerRow = CVPixelBufferGetBytesPerRow(imageBuffer);
|
|
return jsi::Value((double) bytesPerRow);
|
|
}
|
|
if (name == "planesCount") {
|
|
this->assertIsFrameStrong(runtime, name);
|
|
auto imageBuffer = CMSampleBufferGetImageBuffer(frame.buffer);
|
|
auto planesCount = CVPixelBufferGetPlaneCount(imageBuffer);
|
|
return jsi::Value((double) planesCount);
|
|
}
|
|
|
|
return jsi::Value::undefined();
|
|
}
|
|
|
|
void FrameHostObject::assertIsFrameStrong(jsi::Runtime &runtime, const std::string &accessedPropName) {
|
|
if (frame == nil) {
|
|
auto message = "Cannot get `" + accessedPropName + "`, frame is already closed!";
|
|
throw jsi::JSError(runtime, message.c_str());
|
|
}
|
|
}
|
|
|
|
void FrameHostObject::close() {
|
|
if (frame != nil) {
|
|
CMSampleBufferInvalidate(frame.buffer);
|
|
// ARC will hopefully delete it lol
|
|
this->frame = nil;
|
|
}
|
|
}
|