mirror of
https://github.com/opencv/opencv.git
synced 2024-11-30 06:10:02 +08:00
115 lines
4.5 KiB
Python
115 lines
4.5 KiB
Python
#/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)
|
|
with open(output_dir+'/cv.m', 'wb') as f:
|
|
f.write(populated)
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
# add the hdr_parser to the path
|
|
import sys, re, os, time
|
|
sys.path.append(sys.argv[1])
|
|
from string import Template
|
|
from hdr_parser import CppHeaderParser
|
|
from parse_tree import ParseTree, todict, constants
|
|
from filters import *
|
|
from jinja2 import Environment, FileSystemLoader
|
|
|
|
# get the IO from the command line arguments
|
|
input_files = sys.argv[2:-1]
|
|
output_dir = sys.argv[-1]
|
|
|
|
# create the generator
|
|
mwg = MatlabWrapperGenerator()
|
|
mwg.gen(input_files, output_dir)
|