Skip to content

Commit 00a4cef

Browse files
committed
Add function stubs. Going to redo file system.
1 parent b8999e5 commit 00a4cef

5 files changed

Lines changed: 25 additions & 1 deletion

File tree

bridge/include/bridge.h

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,8 @@ bridge_tensor_t resize(bridge_tensor_t input,int height,int width);
3737
bridge_tensor_t imagenet_normalize(bridge_tensor_t input);
3838

3939

40+
bridge_tensor_t capture_webcam_bridge(int cam_index);
41+
4042
int baz(void);
4143

4244
void wrHello(void);

bridge/lib/bridge.cpp

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,8 @@
1414
#include <vector>
1515
#include <cstdint>
1616

17+
#include <cvtool.hpp>
18+
1719

1820

1921

@@ -262,6 +264,11 @@ extern "C" bridge_tensor_t imagenet_normalize(bridge_tensor_t input) {
262264
return torch_to_bridge(output);
263265
}
264266

267+
extern "C" bridge_tensor_t capture_webcam_bridge(int cam_index) {
268+
torch::Tensor image = capture_webcam(cam_index);
269+
return torch_to_bridge(image);
270+
}
271+
265272

266273

267274
// extern "C"
Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
1+
use Tensor;
2+
3+
4+
proc main(args: [] string) {
5+
const a = Bridge.captureWebcam(0);
6+
7+
writeln(a.type:string);
8+
const b = Bridge.bridgeTensorToArray(3,a);
9+
writeln(b.shape);
10+
writeln(b);
11+
12+
}

lib/Bridge.chpl

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,9 @@ module Bridge {
8181

8282
extern "imagenet_normalize" proc imageNetNormalize(
8383
in input: bridge_tensor_t): bridge_tensor_t;
84-
84+
85+
extern "capture_webcam_bridge" proc captureWebcam(
86+
in cam_index: int(32)): bridge_tensor_t;
8587

8688

8789
proc getSizeArray(const ref arr: [] ?eltType): [] int(32) {

lib/Tensor.chpl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,4 +9,5 @@ module Tensor {
99
public import Utilities as util;
1010
public import Utilities.Standard;
1111
public import Testing;
12+
public import Bridge;
1213
}

0 commit comments

Comments
 (0)