sdk: add ObjectDetectionGenerator

koush · Sep 10, 2023 · 988f297 · 988f297
1 parent 6e109d8
commit 988f297
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 5 deletions.
diff --git a/sdk/types/scrypted_python/scrypted_sdk/types.py b/sdk/types/scrypted_python/scrypted_sdk/types.py
@@ -146,6 +146,7 @@ class ScryptedInterface(str, Enum):
     Notifier = "Notifier"
     OauthClient = "OauthClient"
     ObjectDetection = "ObjectDetection"
+    ObjectDetectionGenerator = "ObjectDetectionGenerator"
     ObjectDetector = "ObjectDetector"
     ObjectTracker = "ObjectTracker"
     OccupancySensor = "OccupancySensor"
@@ -1109,6 +1110,9 @@ async def getDetectionModel(self, settings: Any = None) -> ObjectDetectionModel:
         pass
 
 
+class ObjectDetectionGenerator:
+
+
 class ObjectDetector:
     """ObjectDetector is found on Cameras that have smart detection capabilities."""
 
@@ -2913,6 +2917,11 @@ def applicationInfo(self, value: LauncherApplicationInfo):
     ],
     "properties": []
   },
+  "ObjectDetectionGenerator": {
+    "name": "ObjectDetectionGenerator",
+    "methods": [],
+    "properties": []
+  },
   "HumiditySetting": {
     "name": "HumiditySetting",
     "methods": [

diff --git a/sdk/types/src/build.ts b/sdk/types/src/build.ts
@@ -21,15 +21,16 @@ function toTypescriptType(type: any): string {
 
 for (const name of Object.values(ScryptedInterface)) {
     const td = schema.children.find((child: any) => child.name === name);
-    const properties = td.children.filter((child: any) => child.kindString === 'Property').map((child: any) => child.name);
-    const methods = td.children.filter((child: any) => child.kindString === 'Method').map((child: any) => child.name);
+    const children = td.children || [];
+    const properties = children.filter((child: any) => child.kindString === 'Property').map((child: any) => child.name);
+    const methods = children.filter((child: any) => child.kindString === 'Method').map((child: any) => child.name);
     ScryptedInterfaceDescriptors[name] = {
         name,
         methods,
         properties,
     };
 
-    for (const p of td.children.filter((child: any) => child.kindString === 'Property')) {
+    for (const p of children.filter((child: any) => child.kindString === 'Property')) {
         allProperties[p.name] = p.type;
     }
 }
@@ -196,8 +197,9 @@ class ${td.name}:
 ${toDocstring(td)}
 `;
 
-    const properties = td.children.filter((child: any) => child.kindString === 'Property');
-    const methods = td.children.filter((child: any) => child.kindString === 'Method');
+    const children = td.children || [];
+    const properties = children.filter((child: any) => child.kindString === 'Property');
+    const methods = children.filter((child: any) => child.kindString === 'Method');
     for (const property of properties) {
         python += `    ${property.name}: ${toPythonType(property.type)}${toComment(property)}
 `

diff --git a/sdk/types/src/types.input.ts b/sdk/types/src/types.input.ts
@@ -1383,6 +1383,9 @@ export interface ObjectDetection {
   detectObjects(mediaObject: MediaObject, session?: ObjectDetectionSession): Promise<ObjectsDetected>;
   getDetectionModel(settings?: { [key: string]: any }): Promise<ObjectDetectionModel>;
 }
+
+export interface ObjectDetectionGenerator {
+}
 export type ImageFormat = 'gray' | 'rgba' | 'rgb' | 'jpg';
 export interface ImageOptions {
   crop?: {
@@ -2006,6 +2009,7 @@ export enum ScryptedInterface {
   ObjectTracker = "ObjectTracker",
   ObjectDetector = "ObjectDetector",
   ObjectDetection = "ObjectDetection",
+  ObjectDetectionGenerator = "ObjectDetectionGenerator",
   HumiditySetting = "HumiditySetting",
   Fan = "Fan",
   RTCSignalingChannel = "RTCSignalingChannel",