hbristow
12 years ago
parent
ef2c1e1a24
commit
52dc51a62c
9 changed files with 396 additions and 21 deletions
@ -0,0 +1,58 @@ |
||||
#!/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 --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('--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() |
||||
|
||||
from filters import * |
||||
from jinja2 import Environment, FileSystemLoader |
||||
|
||||
# populate the mex base template |
||||
substitute(cv, cv.outdir) |
||||
@ -0,0 +1,46 @@ |
||||
function mex(varargin) |
||||
%CV.MEX compile MEX-function with OpenCV linkages |
||||
% |
||||
% Usage: |
||||
% CV.MEX [options ...] file [file file ...] |
||||
% |
||||
% Description: |
||||
% CV.MEX compiles one or more C/C++ source files into a shared-library |
||||
% called a mex-file. This function is equivalent to the builtin MEX |
||||
% routine, with the notable exception that it automatically resolves |
||||
% OpenCV includes, and links in the OpenCV libraries where appropriate. |
||||
% It also forwards the flags used to build OpenCV, so architecture- |
||||
% specific optimizations can be used. |
||||
% |
||||
% CV.MEX is designed to be used in situations where the source(s) you |
||||
% are compiling contain OpenCV definitions. In such cases, it streamlines |
||||
% the finding and including of appropriate OpenCV libraries. |
||||
% |
||||
% See also: mex |
||||
% |
||||
% Copyright {{ time.strftime("%Y", time.localtime()) }} The OpenCV Foundation |
||||
% |
||||
|
||||
% forward the OpenCV build flags (C++ only) |
||||
EXTRA_FLAGS = ['"CXXFLAGS="\$CXXFLAGS '... |
||||
'{{ cv.flags | trim | wordwrap(60, false, '\'...\n \'') }}""']; |
||||
|
||||
% add the OpenCV include dirs |
||||
INCLUDE_DIRS = {{ cv.include_dirs | split | cellarray | wordwrap(60, false, '...\n ') }}; |
||||
|
||||
% add the lib dir (singular in both build tree and install tree) |
||||
LIB_DIR = '{{ cv.lib_dir }}'; |
||||
|
||||
% add the OpenCV libs. Only the used libs will actually be linked |
||||
LIBS = {{ cv.libs | split | cellarray | wordwrap(60, false, '...\n ') }}; |
||||
|
||||
% add the mex opts (usually at least -largeArrayDims) |
||||
OPTS = {{ cv.opts | split | cellarray | wordwrap(60, false, '...\n ') }}; |
||||
|
||||
% merge all of the default options (EXTRA_FLAGS, LIBS, etc) and the options |
||||
% and files passed by the user (varargin) into a single cell array |
||||
merged = [ {EXTRA_FLAGS}, INCLUDE_DIRS, {LIB_DIR}, LIBS, OPTS, varargin ]; |
||||
|
||||
% expand the merged argument list into the builtin mex utility |
||||
mex(merged{:}); |
||||
end |
||||
Loading…
Reference in new issue