react-native-yolo 0.0.5 → 0.0.8

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,10 +14,8 @@ 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);
-      prototype.registerHybridMethod("detect", &HybridYoloSpec::detect);
     });
   }
 

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

@@ -13,16 +13,15 @@
 #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; }
-// Forward declaration of `Detection` to properly resolve imports.
-namespace margelo::nitro::yolo { struct Detection; }
 
-#include <string>
 #include <memory>
+#include "HybridYoloModelSpec.hpp"
+#include <string>
 #include <VisionCamera/HybridFrameSpec.hpp>
-#include "Detection.hpp"
-#include <vector>
 
 namespace margelo::nitro::yolo {
 
@@ -55,10 +54,8 @@ 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;
-      virtual std::vector<Detection> detect(const std::shared_ptr<margelo::nitro::camera::HybridFrameSpec>& frame) = 0;
 
     protected:
       // Hybrid Setup

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "react-native-yolo",
-  "version": "0.0.5",
+  "version": "0.0.8",
   "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

@@ -13,12 +13,21 @@ export type BoundingBox = {
   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'
 }> {
-  sum(num1: number, num2: number): number
-  loadModel(modelPath: string): void
+  loadModel(modelPath: string): YoloModel
   frameToBase64(frame: Frame): string
-  detect(frame: Frame): Detection[]
 }
+
+