owl_patch 0.8.0

Rust SDK for Rebel Technology Owl2/3 devices
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import sys

import argparse
import json
from collections import OrderedDict as odict

import os
dir_path = os.path.dirname(os.path.realpath(__file__))
sys.path.append('%s/hvcc' % dir_path)
import core.hv2ir.HeavyLangObject as HeavyLangObject

heavy_hash = HeavyLangObject.HeavyLangObject.get_hash

import jinja2

OWL_BUTTONS = ['Push', 'B1', 'B2', 'B3', 'B4', 'B5', 'B6', 'B7', 'B8']
               
def get_template(path):
    templateLoader = jinja2.FileSystemLoader(searchpath=dir_path)
    templateEnv = jinja2.Environment(loader=templateLoader)
    template = templateEnv.get_template(path)
    return template

def main():
    parser = argparse.ArgumentParser(
        description="Generates the constants file for send_receive puredata objects.")

    parser.add_argument(
        "infilename",
        help="Input ir file",
    )

    parser.add_argument(
        "outfilename",
        help="Output file",
        default="Heavy_owl_constants.h"
    )

    args = parser.parse_args()
    jdata = list()

    # Read receive and send object data
    with open(args.infilename, mode="r") as f:
        ir = json.load(f)

        for name, v in ir['control']['receivers'].iteritems():
            # skip __hv_init and similar
            if name.startswith("__"):
                continue

            # If a name has been specified
            if 'owl' in v['attributes'] and v['attributes']['owl'] is not None:
                key = v['attributes']['owl']
                jdata.append((key, name, 'RECV', "0x%x" % heavy_hash(name),
                              v['attributes']['min'],
                              v['attributes']['max'],
                              v['attributes']['default'],
                              key in OWL_BUTTONS))

            elif name.startswith('Channel-'):
                key = name.split('Channel-', 1)[1]
                jdata.append((key, name, 'RECV', "0x%x" % heavy_hash(name),
                              0, 1, None, key in OWL_BUTTONS))

        for k, v in ir['objects'].iteritems():
            try:
                if v['type'] == '__send':
                    name = v['args']['name']
                    if 'owl' in v['args']['attributes'] and v['args']['attributes']['owl'] is not None:
                        key = v['args']['attributes']['owl']
                        jdata.append((key, name+'>', 'SEND', "0x%x" % heavy_hash(name),
                                      v['args']['attributes']['min'],
                                      v['args']['attributes']['max'],
                                      v['args']['attributes']['default'],
                                      key in OWL_BUTTONS))
                    elif name.startswith('Channel-'):
                        key = name.split('Channel-', 1)[1]
                        jdata.append((key, name+'>', 'SEND', "0x%x" % heavy_hash(name),
                                      0, 1, None, key in OWL_BUTTONS))
            except:
                pass

    # TODO, check that there is not channel defined both as input and output
                                                              
    # Write to files
    with open(args.outfilename, mode="w") as f:
        template = get_template("Heavy_owl_constants.tpl.h")
        outputText = template.render(jdata=jdata)  # this is where to put args to the template renderer
        f.write(outputText)

    HeavyPatchpath = "/".join(args.outfilename.split("/")[:-1]) + '/HeavyPatch.hpp'
    with open(HeavyPatchpath, mode="w") as f:
        template = get_template("HeavyPatch.tpl.hpp")
        outputText = template.render(jdata=jdata)  # this is where to put args to the template renderer
        f.write(outputText)

if __name__ == '__main__':
    main()