Script JSON 通信

View as Markdown

1#!/usr/bin/env python3
2import depthai as dai
3import json
4
5pipeline = dai.Pipeline()
6
7xin = pipeline.create(dai.node.XLinkIn)
8xin.setStreamName('in')
9
10script = pipeline.create(dai.node.Script)
11xin.out.link(script.inputs['in'])
12script.setScript("""
13    import json
14
15    # Receive bytes from the host
16    data = node.io['in'].get().getData()
17    jsonStr = str(data, 'utf-8')
18    dict = json.loads(jsonStr)
19
20    # Change initial dictionary a bit
21    dict['one'] += 1
22    dict['foo'] = "baz"
23
24    b = Buffer(30)
25    b.setData(json.dumps(dict).encode('utf-8'))
26    node.io['out'].send(b)
27""")
28
29xout = pipeline.create(dai.node.XLinkOut)
30xout.setStreamName('out')
31script.outputs['out'].link(xout.input)
32
33# Connect to device with pipeline
34with dai.Device(pipeline) as device:
35    # This dict will be serialized (JSON), sent to device (Script node),
36    # edited a bit and sent back to the host
37    dict = {'one':1, 'foo': 'bar'}
38    print('dict', dict)
39    data = json.dumps(dict).encode('utf-8')
40    buffer = dai.Buffer()
41    buffer.setData(list(data))
42    device.getInputQueue("in").send(buffer)
43
44    # Wait for the script to send the changed dictionary back
45    jsonData = device.getOutputQueue("out").get()
46    jsonText = str(jsonData.getData(), 'utf-8')
47    changedDict = json.loads(jsonText)
48    print('changedDict', changedDict)

1#include <chrono>
2#include <iostream>
3#include <thread>
4
5// Includes common necessary includes for development using depthai library
6#include "depthai/depthai.hpp"
7
8// Include nlohmann json
9#include "nlohmann/json.hpp"
10
11int main() {
12    using namespace std;
13
14    dai::Pipeline pipeline;
15
16    auto xin = pipeline.create<dai::node::XLinkIn>();
17    xin->setStreamName("in");
18
19    auto script = pipeline.create<dai::node::Script>();
20    xin->out.link(script->inputs["in"]);
21    script->setScript(R"(
22        import json
23
24        # Receive bytes from the host
25        data = node.io['in'].get().getData()
26        jsonStr = str(data, 'utf-8')
27        dict = json.loads(jsonStr)
28
29        # Change initial dictionary a bit
30        dict['one'] += 1
31        dict['foo'] = "baz"
32
33        b = Buffer(30)
34        b.setData(json.dumps(dict).encode('utf-8'))
35        node.io['out'].send(b)
36    )");
37
38    auto xout = pipeline.create<dai::node::XLinkOut>();
39    xout->setStreamName("out");
40    script->outputs["out"].link(xout->input);
41
42    // Connect to device with pipeline
43    dai::Device device(pipeline);
44
45    // This dict will be serialized (JSON), sent to device (Script node),
46    // edited a bit and sent back to the host
47    nlohmann::json dict{{"one", 1}, {"foo", "bar"}};
48    cout << "dict: " << dict << "\n";
49    auto buffer = dai::Buffer();
50    auto data = dict.dump();
51    buffer.setData({data.begin(), data.end()});
52    device.getInputQueue("in")->send(buffer);
53
54    // Wait for the script to send the changed dictionary back
55    auto jsonData = device.getOutputQueue("out")->get<dai::Buffer>();
56    auto changedDict = nlohmann::json::parse(jsonData->getData());
57    cout << "changedDict: " << changedDict << "\n";
58    const nlohmann::json expectedDict{{"one", 2}, {"foo", "baz"}};
59    if(expectedDict != changedDict) return 1;
60    return 0;
61}