react-native-yolo 0.0.4 → 0.0.7

package/nitrogen/generated/shared/c++/HybridYoloModelSpec.cpp

@@ -0,0 +1,22 @@
+///
+/// HybridYoloModelSpec.cpp
+/// This file was generated by nitrogen. DO NOT MODIFY THIS FILE.
+/// https://github.com/mrousavy/nitro
+/// Copyright © Marc Rousavy @ Margelo
+///
+
+#include "HybridYoloModelSpec.hpp"
+
+namespace margelo::nitro::yolo {
+
+  void HybridYoloModelSpec::loadHybridMethods() {
+    // load base methods/properties
+    HybridObject::loadHybridMethods();
+    // load custom methods/properties
+    registerHybrids(this, [](Prototype& prototype) {
+      prototype.registerHybridMethod("detect", &HybridYoloModelSpec::detect);
+      prototype.registerHybridMethod("close", &HybridYoloModelSpec::close);
+    });
+  }
+
+} // namespace margelo::nitro::yolo

package/nitrogen/generated/shared/c++/HybridYoloModelSpec.hpp

@@ -0,0 +1,69 @@
+///
+/// HybridYoloModelSpec.hpp
+/// This file was generated by nitrogen. DO NOT MODIFY THIS FILE.
+/// https://github.com/mrousavy/nitro
+/// Copyright © Marc Rousavy @ Margelo
+///
+
+#pragma once
+
+#if __has_include(<NitroModules/HybridObject.hpp>)
+#include <NitroModules/HybridObject.hpp>
+#else
+#error NitroModules cannot be found! Are you sure you installed NitroModules properly?
+#endif
+
+// Forward declaration of `Detection` to properly resolve imports.
+namespace margelo::nitro::yolo { struct Detection; }
+// Forward declaration of `HybridFrameSpec` to properly resolve imports.
+namespace margelo::nitro::camera { class HybridFrameSpec; }
+
+#include "Detection.hpp"
+#include <vector>
+#include <memory>
+#include <VisionCamera/HybridFrameSpec.hpp>
+
+namespace margelo::nitro::yolo {
+
+  using namespace margelo::nitro;
+
+  /**
+   * An abstract base class for `YoloModel`
+   * Inherit this class to create instances of `HybridYoloModelSpec` in C++.
+   * You must explicitly call `HybridObject`'s constructor yourself, because it is virtual.
+   * @example
+   * ```cpp
+   * class HybridYoloModel: public HybridYoloModelSpec {
+   * public:
+   *   HybridYoloModel(...): HybridObject(TAG) { ... }
+   *   // ...
+   * };
+   * ```
+   */
+  class HybridYoloModelSpec: public virtual HybridObject {
+    public:
+      // Constructor
+      explicit HybridYoloModelSpec(): HybridObject(TAG) { }
+
+      // Destructor
+      ~HybridYoloModelSpec() override = default;
+
+    public:
+      // Properties
+      
+
+    public:
+      // Methods
+      virtual std::vector<Detection> detect(const std::shared_ptr<margelo::nitro::camera::HybridFrameSpec>& frame) = 0;
+      virtual void close() = 0;
+
+    protected:
+      // Hybrid Setup
+      void loadHybridMethods() override;
+
+    protected:
+      // Tag for logging
+      static constexpr auto TAG = "YoloModel";
+  };
+
+} // namespace margelo::nitro::yolo

package/nitrogen/generated/shared/c++/HybridYoloSpec.cpp

@@ -14,7 +14,6 @@ namespace margelo::nitro::yolo {
     HybridObject::loadHybridMethods();
     // load custom methods/properties
     registerHybrids(this, [](Prototype& prototype) {
-      prototype.registerHybridMethod("sum", &HybridYoloSpec::sum);
       prototype.registerHybridMethod("loadModel", &HybridYoloSpec::loadModel);
       prototype.registerHybridMethod("frameToBase64", &HybridYoloSpec::frameToBase64);
     });

package/nitrogen/generated/shared/c++/HybridYoloSpec.hpp

@@ -13,11 +13,14 @@
 #error NitroModules cannot be found! Are you sure you installed NitroModules properly?
 #endif
 
+// Forward declaration of `HybridYoloModelSpec` to properly resolve imports.
+namespace margelo::nitro::yolo { class HybridYoloModelSpec; }
 // Forward declaration of `HybridFrameSpec` to properly resolve imports.
 namespace margelo::nitro::camera { class HybridFrameSpec; }
 
-#include <string>
 #include <memory>
+#include "HybridYoloModelSpec.hpp"
+#include <string>
 #include <VisionCamera/HybridFrameSpec.hpp>
 
 namespace margelo::nitro::yolo {
@@ -51,8 +54,7 @@ namespace margelo::nitro::yolo {
 
     public:
       // Methods
-      virtual double sum(double num1, double num2) = 0;
-      virtual void loadModel(const std::string& modelPath) = 0;
+      virtual std::shared_ptr<HybridYoloModelSpec> loadModel(const std::string& modelPath) = 0;
       virtual std::string frameToBase64(const std::shared_ptr<margelo::nitro::camera::HybridFrameSpec>& frame) = 0;
 
     protected:

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "react-native-yolo",
-  "version": "0.0.4",
+  "version": "0.0.7",
   "description": "react-native-yolo is a react native package built with Nitro",
   "main": "./lib/commonjs/index.js",
   "module": "./lib/module/index.js",

package/src/index.ts

@@ -1,12 +1,22 @@
 import { NitroModules } from 'react-native-nitro-modules'
 import type { Yolo as YoloSpec } from './specs/yolo.nitro'
 import { Image } from 'react-native'
+import type { Frame } from 'react-native-vision-camera';
 
 const NativeYolo = NitroModules.createHybridObject<YoloSpec>('Yolo')
 
-export const Yolo = Object.assign(NativeYolo, {
-  loadModelTest(modelAssetId: number){
-    const { uri } = Image.resolveAssetSource(modelAssetId);
+export const Yolo = {
+  loadModel(modelAssetId: number) {
+    const { uri } = Image.resolveAssetSource(modelAssetId)
     return NativeYolo.loadModel(uri)
   },
-})
+
+  frameToBase64(frame: Frame) {
+    return NativeYolo.frameToBase64(frame)
+  },
+}
+export type {
+  Detection,
+  BoundingBox,
+  YoloModel,
+} from './specs/yolo.nitro'

package/src/specs/yolo.nitro.ts

@@ -1,8 +1,33 @@
 import type { HybridObject } from 'react-native-nitro-modules'
 import type { Frame } from 'react-native-vision-camera'
 
-export interface Yolo extends HybridObject<{ ios: 'swift', android: 'kotlin' }> {
-  sum(num1: number, num2: number): number
-  loadModel(modelPath: string): void
+export type Detection = {
+  classId: number
+  score: number
+  boundingBox: BoundingBox
+}
+export type BoundingBox = {
+  x1: number
+  y1: number
+  x2: number
+  y2: number
+}
+
+export interface YoloModel extends HybridObject<{
+  ios: 'swift'
+  android: 'kotlin'
+}> {
+  detect(frame: Frame): Detection[]
+  close(): void
+
+}
+
+export interface Yolo extends HybridObject<{
+  ios: 'swift'
+  android: 'kotlin'
+}> {
+  loadModel(modelPath: string): YoloModel
   frameToBase64(frame: Frame): string
-}
+}
+
+