mirror of
https://github.com/opencv/opencv.git
synced 2024-12-26 18:58:16 +08:00
64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
#!/usr/bin/env python
|
|
|
|
def substitute(cv, output_dir):
|
|
|
|
# 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 filters
|
|
jtemplate.filters['cellarray'] = cellarray
|
|
jtemplate.filters['split'] = split
|
|
jtemplate.filters['csv'] = csv
|
|
|
|
# load the template
|
|
template = jtemplate.get_template('template_cvmex_base.m')
|
|
|
|
# create the build directory
|
|
output_dir = output_dir+'/+cv'
|
|
if not os.path.isdir(output_dir):
|
|
os.mkdir(output_dir)
|
|
|
|
# populate template
|
|
populated = template.render(cv=cv, time=time)
|
|
with open(os.path.join(output_dir, 'mex.m'), 'wb') as f:
|
|
f.write(populated)
|
|
|
|
if __name__ == "__main__":
|
|
"""
|
|
Usage: python cvmex.py --jinja2 /path/to/jinja2/engine
|
|
--opts [-list -of -opts]
|
|
--include_dirs [-list -of -opencv_include_directories]
|
|
--lib_dir opencv_lib_directory
|
|
--libs [-lopencv_core -lopencv_imgproc ...]
|
|
--flags [-Wall -opencv_build_flags ...]
|
|
--outdir /path/to/generated/output
|
|
|
|
cvmex.py generates a custom mex compiler that automatically links OpenCV
|
|
libraries to built sources where appropriate. The calling syntax is the
|
|
same as the builtin mex compiler, with added cv qualification:
|
|
>> cv.mex(..., ...);
|
|
"""
|
|
|
|
# parse the input options
|
|
import sys, re, os, time
|
|
from argparse import ArgumentParser
|
|
parser = ArgumentParser()
|
|
parser.add_argument('--jinja2')
|
|
parser.add_argument('--opts')
|
|
parser.add_argument('--include_dirs')
|
|
parser.add_argument('--lib_dir')
|
|
parser.add_argument('--libs')
|
|
parser.add_argument('--flags')
|
|
parser.add_argument('--outdir')
|
|
cv = parser.parse_args()
|
|
|
|
# add jinja to the path
|
|
sys.path.append(cv.jinja2)
|
|
|
|
from filters import *
|
|
from jinja2 import Environment, FileSystemLoader
|
|
|
|
# populate the mex base template
|
|
substitute(cv, cv.outdir)
|