git.maemo.org Git - opencv/blob - utils/extract_macros.py

   1 #! /usr/bin/env python
   2 """
   3 This script extracts macros functions from the OpenCV headers and attempts to generate
   4 standard C function prototypes.  Type information is missing in the macros, so SWIG
   5 cannot generate C code for them unless we provide this.
   6 """
   7
   8 import sys, re
   9
  10 ARG_MAP = {
  11     "mat":"CvMat*",
  12     "type":"int",
  13     "flags":"int",
  14     "img":"CvArr *",
  15     "image":"IplImage *",
  16     "mat1":"CvMat*",
  17     "mat2":"CvMat*",
  18     "seq":"CvSeq*",
  19     "elem_ptr":"void *",
  20     "elem":"CvPoint",
  21     "elem_type":"ignore",
  22     "elemtype":"ignore",
  23     "elem_size":"int",
  24     "edge":"CvGraphEdge *",
  25     "vertex":"CvGraphVtx *",
  26     "vtx":"CvGraphVtx *",
  27     "reader":"CvSeqReader",
  28     "writer":"CvSeqWriter",
  29     "hist":"CvHistogram *",
  30     "ptr":"void *",
  31     "arr":"CvArr *",
  32     "header":"CvMat *",
  33     "src":"CvArr *",
  34     "src1":"CvArr *",
  35     "src2":"CvArr *",
  36     "src3":"CvArr *",
  37     "dst":"CvArr *",
  38     "pt1":"CvPoint",
  39     "pt2":"CvPoint",
  40     "_pt":"CvPoint",
  41     "index":"int",
  42     "idx":"int",
  43     "set":"CvSet *",
  44     "n":"int",
  45     "a":"int",
  46     "b":"int",
  47     "t":"int",
  48     "value":"double",
  49     "row":"int",
  50     "col":"int",
  51     "cn":"int",
  52     "new_cn":"int",
  53     "pix_size":"int",
  54     "depth":"int",
  55     "node":"CvSparseNode *",
  56     "storage":"CvMemStorage *",
  57     "new_dims": "int",
  58     "new_sizes": "int *",
  59     "A":"CvArr *",
  60     "B":"CvArr *",
  61     "C":"CvArr *",
  62     "real_scalar":"double",
  63     "graph":"CvGraph *",
  64     "r":"double",
  65     "g":"double",
  66 #    "b":"double",
  67     "line_iterator":"CvLineIterator",
  68     "deltas":"double *",
  69     "step":"int",
  70     "haar":"void *",
  71 #    "contour":"const void*",  # handled as a special case in cvshadow
  72     "range":"CvSize",
  73     "nch":"int",
  74     "method":"int",
  75     "factor":"double"
  76 }
  77 RET_MAP = {
  78     "CV_RGB":"CvScalar",
  79     "CV_NEXT_GRAPH_EDGE":"CvGraphEdge *",
  80     "CV_IMIN":"int",
  81     "CV_IMAX":"int",
  82     "CV_IABS":"int",
  83     "CV_MAT_CN":"int",
  84     "CV_MAT_DEPTH":"int",
  85     "CV_NEXT_LINE_POINT":"void",
  86 }
  87
  88 # special cases
  89 MACROS = {
  90     #"CV_MAKETYPE":"",  # SWIG 1.3.29 doesn't like this one for some indeterminant reason
  91     "CV_TURN_ON_IPL_COMPATIBILITY":"",
  92     "CV_MAT_ELEM_PTR_FAST":"void * CV_MAT_ELEM_PTR_FAST(CvMat mat,int row,int col,int pix_size);",
  93     "CV_MAT_ELEM_PTR":"void * CV_MAT_ELEM_PTR(CvMat mat,int row,int col);",
  94     "CV_NODE_VAL":"void * CV_NODE_VAL(CvSparseMat* mat,CvSparseNode * node);",
  95     "CV_NODE_IDX":"int * CV_NODE_IDX(CvSparseMat* mat,CvSparseNode * node);",
  96     "CV_READ_CHAIN_POINT":"void CV_READ_CHAIN_POINT(CvPoint _pt, CvChainPtReader reader);",
  97     "CV_SUBDIV2D_NEXT_EDGE":"CvQuadEdge2D* CV_SUBDIV2D_NEXT_EDGE(CvSubdiv2DEdge edge);",
  98     "cvFree":"void cvFree(void ** ptr);",
  99
 100
 101 }
 102
 103 print """
 104 /*//////////////////////////////////////////////////////////////////////////////////////////////////
 105 // This file is automatically generated from the extract_macros.py script found in the 'utils'
 106 // subdirectory of the OpenCV distribution.  If the generated function prototypes are missing or
 107 // incorrect, it is likely that a name->type mapping will have to be added to the script
 108 /////////////////////////////////////////////////////////////////////////////////////////////////M*/
 109 """
 110
 111 def determine_return_type(name, arguments):
 112     if RET_MAP.has_key( name ):
 113         return RET_MAP[name]
 114     if name.find("_IS_")>=0 or \
 115        name.find("_HAS_")>=0 or \
 116        name.find("_KIND")>=0 or \
 117        name.find("_ARE_")>=0 or \
 118        name.find("_SIZE")>=0 or \
 119        name.find("Idx")>=0 or \
 120        name.find("Count")>=0 or \
 121        (name.find("TYPE")>=0 and not name.find("TYPES")>=0):
 122         return "int"
 123     if re.match( r"CV_(?:8|16|32|64)(?:U|S|F)C", name ):
 124         return "int"
 125     if len(arguments) is 1 and arguments[0].startswith("double"):
 126         return "double"
 127     if name.find("_PTR")>=0:
 128         return "void *"
 129     if name.endswith("POINT"):
 130         return "CvPoint"
 131     return "void"
 132
 133 for m in MACROS:
 134     print MACROS[m]
 135
 136 for fn in sys.argv[1:]:
 137     f = open( fn, "r" )
 138     in_define = False
 139     fstr=""
 140     macro_name=""
 141     for l in f.xreadlines():
 142         m = re.match( r"^#define\s+((?:CV_|IPL_|cv)\w+)\s*\(([_, a-zA-Z0-9]*)\)\s*(.*)", l )
 143
 144         if m and not m.group(1).endswith("FIELDS") and not MACROS.has_key(m.group(1)):
 145             macro_name = m.group(1)
 146             args = m.group(2).strip().split(",")
 147
 148             # assign return type
 149             ret=determine_return_type( macro_name, args )
 150
 151             # assign type to each argument
 152             no_args = len(args) is 0
 153             typed_args = []
 154             ignore = False
 155             for arg in args:
 156                 arg = arg.strip()
 157                 if len(arg) is 0:
 158                     no_args = True
 159                     break
 160                 if ARG_MAP.has_key( arg ):
 161                     if ARG_MAP[arg] is "ignore":
 162                         ignore=True
 163                         break
 164                     typed_args.append( "%s %s"%( ARG_MAP[arg], arg ) )
 165                 else:
 166                     sys.stderr.write( "\"%s\":\"?\", in macro '%s'\n" % (arg, macro_name) )
 167                     typed_args = []
 168                     break
 169             if not ignore and (no_args or len(typed_args)>0):
 170                 decl = "%s %s(%s);" % (ret, macro_name, ",".join( typed_args) )
 171                 MACROS[ macro_name ] = decl
 172                 print decl
 173
 174     f.close()