Open Source Computer Vision Library
https://opencv.org/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
121 lines
4.7 KiB
121 lines
4.7 KiB
#/usr/bin/env python |
|
|
|
class MatlabWrapperGenerator(object): |
|
|
|
def gen(self, input_files, output_dir): |
|
# parse each of the files and store in a dictionary |
|
# as a separate "namespace" |
|
parser = CppHeaderParser() |
|
ns = {} |
|
for file in input_files: |
|
# get the file name |
|
# TODO: Is there a cleaner way to do this? |
|
try: |
|
name = re.findall('include/opencv2/([^./]+)', file)[0] |
|
except: |
|
name = os.path.splitext(os.path.basename(file))[0] |
|
|
|
# add the file to the namespace |
|
try: |
|
ns[name] = ns[name] + parser.parse(file) |
|
except KeyError: |
|
ns[name] = parser.parse(file) |
|
|
|
# cleanify the parser output |
|
parse_tree = ParseTree() |
|
parse_tree.build(ns) |
|
|
|
# setup the template engine |
|
template_dir = os.path.join(os.path.dirname(__file__), 'templates') |
|
jtemplate = Environment(loader=FileSystemLoader(template_dir), trim_blocks=True, lstrip_blocks=True) |
|
|
|
# add the custom filters |
|
jtemplate.filters['formatMatlabConstant'] = formatMatlabConstant |
|
jtemplate.filters['convertibleToInt'] = convertibleToInt |
|
jtemplate.filters['toUpperCamelCase'] = toUpperCamelCase |
|
jtemplate.filters['toLowerCamelCase'] = toLowerCamelCase |
|
jtemplate.filters['toUnderCase'] = toUnderCase |
|
jtemplate.filters['comment'] = comment |
|
jtemplate.filters['inputs'] = inputs |
|
jtemplate.filters['ninputs'] = ninputs |
|
jtemplate.filters['outputs'] = outputs |
|
jtemplate.filters['noutputs'] = noutputs |
|
jtemplate.filters['only'] = only |
|
jtemplate.filters['void'] = void |
|
jtemplate.filters['not'] = flip |
|
|
|
# load the templates |
|
tfunction = jtemplate.get_template('template_function_base.cpp') |
|
tclassm = jtemplate.get_template('template_class_base.m') |
|
tclassc = jtemplate.get_template('template_class_base.cpp') |
|
tdoc = jtemplate.get_template('template_doc_base.m') |
|
tconst = jtemplate.get_template('template_map_base.m') |
|
|
|
# create the build directory |
|
output_source_dir = output_dir+'/src' |
|
output_private_dir = output_source_dir+'/private' |
|
output_class_dir = output_dir+'/+cv' |
|
output_map_dir = output_dir+'/map' |
|
if not os.path.isdir(output_source_dir): |
|
os.mkdir(output_source_dir) |
|
if not os.path.isdir(output_private_dir): |
|
os.mkdir(output_private_dir) |
|
if not os.path.isdir(output_class_dir): |
|
os.mkdir(output_class_dir) |
|
if not os.path.isdir(output_map_dir): |
|
os.mkdir(output_map_dir) |
|
|
|
# populate templates |
|
for namespace in parse_tree.namespaces: |
|
# functions |
|
for method in namespace.methods: |
|
populated = tfunction.render(fun=method, time=time, includes=namespace.name) |
|
with open(output_source_dir+'/'+method.name+'.cpp', 'wb') as f: |
|
f.write(populated) |
|
populated = tdoc.render(fun=method, time=time) |
|
with open(output_class_dir+'/'+method.name+'.m', 'wb') as f: |
|
f.write(populated) |
|
# classes |
|
for clss in namespace.classes: |
|
# cpp converter |
|
populated = tclassc.render(clss=clss, time=time) |
|
with open(output_private_dir+'/'+clss.name+'Bridge.cpp', 'wb') as f: |
|
f.write(populated) |
|
# matlab classdef |
|
populated = tclassm.render(clss=clss, time=time) |
|
with open(output_class_dir+'/'+clss.name+'.m', 'wb') as f: |
|
f.write(populated) |
|
|
|
# create a global constants lookup table |
|
const = dict(constants(todict(parse_tree.namespaces))) |
|
populated = tconst.render(constants=const, time=time) |
|
with open(output_dir+'/cv.m', 'wb') as f: |
|
f.write(populated) |
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
|
|
# parse the input options |
|
import sys, re, os, time |
|
from argparse import ArgumentParser |
|
parser = ArgumentParser() |
|
parser.add_argument('--hdrparser') |
|
parser.add_argument('--rstparser') |
|
parser.add_argument('--headers', nargs='+') |
|
parser.add_argument('--outdir') |
|
args = parser.parse_args() |
|
|
|
# add the hdr_parser and rst_parser modules to the path |
|
sys.path.append(args.hdrparser) |
|
sys.path.append(args.rstparser) |
|
|
|
from string import Template |
|
from hdr_parser import CppHeaderParser |
|
from parse_tree import ParseTree, todict, constants |
|
from filters import * |
|
from jinja2 import Environment, FileSystemLoader |
|
|
|
# create the generator |
|
mwg = MatlabWrapperGenerator() |
|
mwg.gen(args.headers, args.outdir)
|
|
|