1200 lines
48 KiB
Python
1200 lines
48 KiB
Python
import sys, re, os.path
|
|
from string import Template
|
|
|
|
try:
|
|
from cStringIO import StringIO
|
|
except:
|
|
from StringIO import StringIO
|
|
|
|
class_ignore_list = (
|
|
#core
|
|
"FileNode", "FileStorage",
|
|
#highgui
|
|
"VideoWriter", "VideoCapture",
|
|
#features2d
|
|
"KeyPoint",
|
|
)
|
|
|
|
const_ignore_list = (
|
|
"CV_CAP_OPENNI",
|
|
"CV_CAP_PROP_OPENNI_",
|
|
"WINDOW_AUTOSIZE",
|
|
"CV_WND_PROP_",
|
|
"CV_WINDOW_",
|
|
"CV_EVENT_",
|
|
"CV_GUI_",
|
|
"CV_PUSH_BUTTON",
|
|
"CV_CHECKBOX",
|
|
"CV_RADIOBOX",
|
|
|
|
#attention! the following constants are added to this list using code automatic generation -- should be checked
|
|
"CV_CAP_ANY",
|
|
"CV_CAP_MIL",
|
|
"CV_CAP_VFW",
|
|
"CV_CAP_V4L",
|
|
"CV_CAP_V4L2",
|
|
"CV_CAP_FIREWARE",
|
|
"CV_CAP_FIREWIRE",
|
|
"CV_CAP_IEEE1394",
|
|
"CV_CAP_DC1394",
|
|
"CV_CAP_CMU1394",
|
|
"CV_CAP_STEREO",
|
|
"CV_CAP_TYZX",
|
|
"CV_TYZX_LEFT",
|
|
"CV_TYZX_RIGHT",
|
|
"CV_TYZX_COLOR",
|
|
"CV_TYZX_Z",
|
|
"CV_CAP_QT",
|
|
"CV_CAP_UNICAP",
|
|
"CV_CAP_DSHOW",
|
|
"CV_CAP_PVAPI",
|
|
"CV_CAP_PROP_DC1394_OFF",
|
|
"CV_CAP_PROP_DC1394_MODE_MANUAL",
|
|
"CV_CAP_PROP_DC1394_MODE_AUTO",
|
|
"CV_CAP_PROP_DC1394_MODE_ONE_PUSH_AUTO",
|
|
"CV_CAP_PROP_POS_MSEC",
|
|
"CV_CAP_PROP_POS_FRAMES",
|
|
"CV_CAP_PROP_POS_AVI_RATIO",
|
|
"CV_CAP_PROP_FPS",
|
|
"CV_CAP_PROP_FOURCC",
|
|
"CV_CAP_PROP_FRAME_COUNT",
|
|
"CV_CAP_PROP_FORMAT",
|
|
"CV_CAP_PROP_MODE",
|
|
"CV_CAP_PROP_BRIGHTNESS",
|
|
"CV_CAP_PROP_CONTRAST",
|
|
"CV_CAP_PROP_SATURATION",
|
|
"CV_CAP_PROP_HUE",
|
|
"CV_CAP_PROP_GAIN",
|
|
"CV_CAP_PROP_EXPOSURE",
|
|
"CV_CAP_PROP_CONVERT_RGB",
|
|
"CV_CAP_PROP_WHITE_BALANCE_BLUE_U",
|
|
"CV_CAP_PROP_RECTIFICATION",
|
|
"CV_CAP_PROP_MONOCROME",
|
|
"CV_CAP_PROP_SHARPNESS",
|
|
"CV_CAP_PROP_AUTO_EXPOSURE",
|
|
"CV_CAP_PROP_GAMMA",
|
|
"CV_CAP_PROP_TEMPERATURE",
|
|
"CV_CAP_PROP_TRIGGER",
|
|
"CV_CAP_PROP_TRIGGER_DELAY",
|
|
"CV_CAP_PROP_WHITE_BALANCE_RED_V",
|
|
"CV_CAP_PROP_MAX_DC1394",
|
|
"CV_CAP_GSTREAMER_QUEUE_LENGTH",
|
|
"CV_CAP_PROP_PVAPI_MULTICASTIP",
|
|
"CV_CAP_PROP_SUPPORTED_PREVIEW_SIZES_STRING",
|
|
"EVENT_.*",
|
|
"CV_L?(BGRA?|RGBA?|GRAY|XYZ|YCrCb|Luv|Lab|HLS|YUV|HSV)\d*2L?(BGRA?|RGBA?|GRAY|XYZ|YCrCb|Luv|Lab|HLS|YUV|HSV).*",
|
|
"CV_COLORCVT_MAX",
|
|
"CV_.*Bayer.*",
|
|
"CV_YUV420i2.+",
|
|
"CV_TM_.+",
|
|
"CV_FLOODFILL_.+",
|
|
"CV_ADAPTIVE_THRESH_.+",
|
|
)
|
|
|
|
const_private_list = (
|
|
"CV_MOP_.+",
|
|
"CV_INTER_.+",
|
|
"CV_THRESH_.+",
|
|
"CV_INPAINT_.+",
|
|
"CV_RETR_.+",
|
|
"CV_CHAIN_APPROX_.+",
|
|
)
|
|
|
|
# { Module : { public : [[name, val],...], private : [[]...] } }
|
|
missing_consts = \
|
|
{
|
|
'Core' :
|
|
{
|
|
'private' :
|
|
(
|
|
('CV_8U', 0 ), ('CV_8S', 1 ),
|
|
('CV_16U', 2 ), ('CV_16S', 3 ),
|
|
('CV_32S', 4 ),
|
|
('CV_32F', 5 ), ('CV_64F', 6 ),
|
|
('CV_USRTYPE1', 7 ),
|
|
) # private
|
|
}, # Core
|
|
|
|
"Imgproc":
|
|
{
|
|
'private' :
|
|
(
|
|
('IPL_BORDER_CONSTANT', 0 ),
|
|
('IPL_BORDER_REPLICATE', 1 ),
|
|
('IPL_BORDER_REFLECT', 2 ),
|
|
('IPL_BORDER_WRAP', 3 ),
|
|
('IPL_BORDER_REFLECT_101', 4 ),
|
|
('IPL_BORDER_TRANSPARENT', 5 ),
|
|
) # private
|
|
}, # Imgproc
|
|
|
|
"Calib3d":
|
|
{
|
|
'private' :
|
|
(
|
|
('CV_LMEDS', 4),
|
|
('CV_RANSAC', 8),
|
|
('CV_FM_LMEDS', 'CV_LMEDS'),
|
|
('CV_FM_RANSAC','CV_RANSAC'),
|
|
('CV_FM_7POINT', 1),
|
|
('CV_FM_8POINT', 2),
|
|
('CV_CALIB_USE_INTRINSIC_GUESS', 1),
|
|
('CV_CALIB_FIX_ASPECT_RATIO', 2),
|
|
('CV_CALIB_FIX_PRINCIPAL_POINT', 4),
|
|
('CV_CALIB_ZERO_TANGENT_DIST', 8),
|
|
('CV_CALIB_FIX_FOCAL_LENGTH', 16),
|
|
('CV_CALIB_FIX_K1', 32),
|
|
('CV_CALIB_FIX_K2', 64),
|
|
('CV_CALIB_FIX_K3', 128),
|
|
('CV_CALIB_FIX_K4', 2048),
|
|
('CV_CALIB_FIX_K5', 4096),
|
|
('CV_CALIB_FIX_K6', 8192),
|
|
('CV_CALIB_RATIONAL_MODEL', 16384),
|
|
('CV_CALIB_FIX_INTRINSIC', 256),
|
|
('CV_CALIB_SAME_FOCAL_LENGTH', 512),
|
|
('CV_CALIB_ZERO_DISPARITY', 1024),
|
|
) # public
|
|
}, # Calib3d
|
|
|
|
}
|
|
|
|
|
|
# c_type : { java/jni correspondence }
|
|
type_dict = {
|
|
# "simple" : { j_type : "?", jn_type : "?", jni_type : "?", suffix : "?" },
|
|
"" : { "j_type" : "", "jn_type" : "long", "jni_type" : "jlong" }, # c-tor ret_type
|
|
"void" : { "j_type" : "void", "jn_type" : "void", "jni_type" : "void" },
|
|
"env" : { "j_type" : "", "jn_type" : "", "jni_type" : "JNIEnv*"},
|
|
"cls" : { "j_type" : "", "jn_type" : "", "jni_type" : "jclass"},
|
|
"bool" : { "j_type" : "boolean", "jn_type" : "boolean", "jni_type" : "jboolean", "suffix" : "Z" },
|
|
"int" : { "j_type" : "int", "jn_type" : "int", "jni_type" : "jint", "suffix" : "I" },
|
|
"long" : { "j_type" : "int", "jn_type" : "int", "jni_type" : "jint", "suffix" : "I" },
|
|
"float" : { "j_type" : "float", "jn_type" : "float", "jni_type" : "jfloat", "suffix" : "F" },
|
|
"double" : { "j_type" : "double", "jn_type" : "double", "jni_type" : "jdouble", "suffix" : "D" },
|
|
"size_t" : { "j_type" : "long", "jn_type" : "long", "jni_type" : "jlong", "suffix" : "J" },
|
|
"__int64" : { "j_type" : "long", "jn_type" : "long", "jni_type" : "jlong", "suffix" : "J" },
|
|
"int64" : { "j_type" : "long", "jn_type" : "long", "jni_type" : "jlong", "suffix" : "J" },
|
|
"double[]": { "j_type" : "double[]", "jn_type" : "double[]", "jni_type" : "jdoubleArray", "suffix" : "_3D" },
|
|
|
|
# "complex" : { j_type : "?", jn_args : (("", ""),), jn_name : "", jni_var : "", jni_name : "", "suffix" : "?" },
|
|
|
|
"vector_Point": { "j_type" : "List<Point>", "jn_type" : "long", "jni_type" : "jlong", "jni_var" : "vector<Point> %(n)s", "suffix" : "J" },
|
|
"vector_Mat" : { "j_type" : "List<Mat>", "jn_type" : "long", "jni_type" : "jlong", "jni_var" : "vector<Mat> %(n)s", "suffix" : "J" },
|
|
"vector_KeyPoint" : { "j_type" : "List<KeyPoint>", "jn_type" : "long", "jni_type" : "jlong", "jni_var" : "vector<KeyPoint> %(n)s", "suffix" : "J" },
|
|
"vector_Rect" : { "j_type" : "List<Rect>", "jn_type" : "long", "jni_type" : "jlong", "jni_var" : "vector<Rect> %(n)s", "suffix" : "J" },
|
|
"vector_uchar" : { "j_type" : "List<Byte>", "jn_type" : "long", "jni_type" : "jlong", "jni_var" : "vector<uchar> %(n)s", "suffix" : "J" },
|
|
"vector_int" : { "j_type" : "List<Integer>", "jn_type" : "long", "jni_type" : "jlong", "jni_var" : "vector<int> %(n)s", "suffix" : "J" },
|
|
"vector_float" : { "j_type" : "List<Float>", "jn_type" : "long", "jni_type" : "jlong", "jni_var" : "vector<float> %(n)s", "suffix" : "J" },
|
|
"vector_double" : { "j_type" : "List<Double>", "jn_type" : "long", "jni_type" : "jlong", "jni_var" : "vector<double> %(n)s", "suffix" : "J" },
|
|
|
|
"Mat" : { "j_type" : "Mat", "jn_type" : "long", "jn_args" : (("__int64", ".nativeObj"),),
|
|
"jni_var" : "Mat& %(n)s = *((Mat*)%(n)s_nativeObj)",
|
|
"jni_type" : "jlong", #"jni_name" : "*%(n)s",
|
|
"suffix" : "J" },
|
|
"Point" : { "j_type" : "Point", "jn_args" : (("double", ".x"), ("double", ".y")),
|
|
"jni_var" : "Point %(n)s((int)%(n)s_x, (int)%(n)s_y)", "jni_type" : "jdoubleArray",
|
|
"suffix" : "DD"},
|
|
"Point2f" : { "j_type" : "Point", "jn_args" : (("double", ".x"), ("double", ".y")),
|
|
"jni_var" : "Point2f %(n)s((float)%(n)s_x, (float)%(n)s_y)", "jni_type" : "jdoubleArray",
|
|
"suffix" : "DD"},
|
|
"Point2d" : { "j_type" : "Point", "jn_args" : (("double", ".x"), ("double", ".y")),
|
|
"jni_var" : "Point2d %(n)s(%(n)s_x, %(n)s_y)", "jni_type" : "jdoubleArray",
|
|
"suffix" : "DD"},
|
|
"Point3i" : { "j_type" : "Point", "jn_args" : (("double", ".x"), ("double", ".y"), ("double", ".z")),
|
|
"jni_var" : "Point3i %(n)s((int)%(n)s_x, (int)%(n)s_y, (int)%(n)s_z)", "jni_type" : "jdoubleArray",
|
|
"suffix" : "DDD"},
|
|
"Point3f" : { "j_type" : "Point", "jn_args" : (("double", ".x"), ("double", ".y"), ("double", ".z")),
|
|
"jni_var" : "Point3f %(n)s((float)%(n)s_x, (float)%(n)s_y, (float)%(n)s_z)", "jni_type" : "jdoubleArray",
|
|
"suffix" : "DDD"},
|
|
"Point3d" : { "j_type" : "Point", "jn_args" : (("double", ".x"), ("double", ".y"), ("double", ".z")),
|
|
"jni_var" : "Point3d %(n)s(%(n)s_x, %(n)s_y, %(n)s_z)", "jni_type" : "jdoubleArray",
|
|
"suffix" : "DDD"},
|
|
"KeyPoint": { "j_type" : "KeyPoint", "jn_args" : (("float", ".x"), ("float", ".y"), ("float", ".size"),
|
|
("float", ".angle"), ("float", ".response"), ("int", ".octave"), ("int", ".class_id")),
|
|
"jni_var" : "KeyPoint %(n)s(%(n)s_x, %(n)s_y, %(n)s_size, %(n)s_angle, %(n)s_response, %(n)s_octave, %(n)s_class_id)",
|
|
"jni_type" : "jdoubleArray",
|
|
"suffix" : "FFFFFII"},
|
|
"Rect" : { "j_type" : "Rect", "jn_args" : (("int", ".x"), ("int", ".y"), ("int", ".width"), ("int", ".height")),
|
|
"jni_var" : "Rect %(n)s(%(n)s_x, %(n)s_y, %(n)s_width, %(n)s_height)", "jni_type" : "jdoubleArray",
|
|
"suffix" : "IIII"},
|
|
"Size" : { "j_type" : "Size", "jn_args" : (("double", ".width"), ("double", ".height")),
|
|
"jni_var" : "Size %(n)s((int)%(n)s_width, (int)%(n)s_height)", "jni_type" : "jdoubleArray",
|
|
"suffix" : "DD"},
|
|
"Size2f" : { "j_type" : "Size", "jn_args" : (("double", ".width"), ("double", ".height")),
|
|
"jni_var" : "Size2f %(n)s((float)%(n)s_width, (float)%(n)s_height)", "jni_type" : "jdoubleArray",
|
|
"suffix" : "DD"},
|
|
"RotatedRect": { "j_type" : "RotatedRect", "jn_args" : (("double", ".center.x"), ("double", ".center.y"), ("double", ".size.width"), ("double", ".size.height"), ("double", ".angle")),
|
|
"jni_var" : "RotatedRect %(n)s(cv::Point2f(%(n)s_center_x, %(n)s_center_y), cv::Size2f(%(n)s_size_width, %(n)s_size_height), %(n)s_angle)",
|
|
"jni_type" : "jdoubleArray", "suffix" : "DDDDD"},
|
|
"Scalar" : { "j_type" : "Scalar", "jn_args" : (("double", ".val[0]"), ("double", ".val[1]"), ("double", ".val[2]"), ("double", ".val[3]")),
|
|
"jni_var" : "Scalar %(n)s(%(n)s_val0, %(n)s_val1, %(n)s_val2, %(n)s_val3)", "jni_type" : "jdoubleArray",
|
|
"suffix" : "DDDD"},
|
|
"Range" : { "j_type" : "Range", "jn_args" : (("int", ".start"), ("int", ".end")),
|
|
"jni_var" : "Range %(n)s(%(n)s_start, %(n)s_end)", "jni_type" : "jdoubleArray",
|
|
"suffix" : "II"},
|
|
"CvSlice" : { "j_type" : "Range", "jn_args" : (("int", ".start"), ("int", ".end")),
|
|
"jni_var" : "Range %(n)s(%(n)s_start, %(n)s_end)", "jni_type" : "jdoubleArray",
|
|
"suffix" : "II"},
|
|
"string" : { "j_type" : "String", "jn_type" : "String",
|
|
"jni_type" : "jstring", "jni_name" : "n_%(n)s",
|
|
"jni_var" : 'const char* utf_%(n)s = env->GetStringUTFChars(%(n)s, 0); std::string n_%(n)s( utf_%(n)s ? utf_%(n)s : "" ); env->ReleaseStringUTFChars(%(n)s, utf_%(n)s)',
|
|
"suffix" : "Ljava_lang_String_2"},
|
|
"String" : { "j_type" : "String", "jn_type" : "String",
|
|
"jni_type" : "jstring", "jni_name" : "n_%(n)s",
|
|
"jni_var" : 'const char* utf_%(n)s = env->GetStringUTFChars(%(n)s, 0); String n_%(n)s( utf_%(n)s ? utf_%(n)s : "" ); env->ReleaseStringUTFChars(%(n)s, utf_%(n)s)',
|
|
"suffix" : "Ljava_lang_String_2"},
|
|
"c_string": { "j_type" : "String", "jn_type" : "String",
|
|
"jni_type" : "jstring", "jni_name" : "n_%(n)s.c_str()",
|
|
"jni_var" : 'const char* utf_%(n)s = env->GetStringUTFChars(%(n)s, 0); std::string n_%(n)s( utf_%(n)s ? utf_%(n)s : "" ); env->ReleaseStringUTFChars(%(n)s, utf_%(n)s)',
|
|
"suffix" : "Ljava_lang_String_2"},
|
|
"TermCriteria": { "j_type" : "TermCriteria", "jn_args" : (("int", ".type"), ("int", ".maxCount"), ("double", ".epsilon")),
|
|
"jni_var" : "TermCriteria %(n)s(%(n)s_type, %(n)s_maxCount, %(n)s_epsilon)",
|
|
"suffix" : "IID"},
|
|
"Vec3d" : { "j_type" : "double[]", "jn_args" : (("double", ".val[0]"), ("double", ".val[1]"), ("double", ".val[2]")),
|
|
"jn_type" : "double[]",
|
|
"jni_var" : "Vec3d %(n)s(%(n)s_val0, %(n)s_val1, %(n)s_val2)", "jni_type" : "jdoubleArray",
|
|
"suffix" : "DDD"},
|
|
|
|
}
|
|
|
|
# { class : { func : {j_code, jn_code, cpp_code} } }
|
|
ManualFuncs = {
|
|
'Core' :
|
|
{
|
|
'minMaxLoc' : {
|
|
'j_code' : """
|
|
// manual port
|
|
public static class MinMaxLocResult {
|
|
public double minVal;
|
|
public double maxVal;
|
|
public Point minLoc;
|
|
public Point maxLoc;
|
|
|
|
public MinMaxLocResult() {
|
|
minVal=0; maxVal=0;
|
|
minLoc=new Point();
|
|
maxLoc=new Point();
|
|
}
|
|
}
|
|
|
|
// C++: minMaxLoc(Mat src, double* minVal, double* maxVal=0, Point* minLoc=0, Point* maxLoc=0, InputArray mask=noArray())
|
|
|
|
//javadoc: minMaxLoc(src, mask)
|
|
public static MinMaxLocResult minMaxLoc(Mat src, Mat mask) {
|
|
MinMaxLocResult res = new MinMaxLocResult();
|
|
long maskNativeObj=0;
|
|
if (mask != null) {
|
|
maskNativeObj=mask.nativeObj;
|
|
}
|
|
double resarr[] = n_minMaxLocManual(src.nativeObj, maskNativeObj);
|
|
res.minVal=resarr[0];
|
|
res.maxVal=resarr[1];
|
|
res.minLoc.x=resarr[2];
|
|
res.minLoc.y=resarr[3];
|
|
res.maxLoc.x=resarr[4];
|
|
res.maxLoc.y=resarr[5];
|
|
return res;
|
|
}
|
|
|
|
//javadoc: minMaxLoc(src)
|
|
public static MinMaxLocResult minMaxLoc(Mat src) {
|
|
return minMaxLoc(src, null);
|
|
}
|
|
|
|
""",
|
|
'jn_code' :
|
|
""" private static native double[] n_minMaxLocManual(long src_nativeObj, long mask_nativeObj);\n""",
|
|
'cpp_code' :
|
|
"""
|
|
// C++: minMaxLoc(Mat src, double* minVal, double* maxVal=0, Point* minLoc=0, Point* maxLoc=0, InputArray mask=noArray())
|
|
|
|
JNIEXPORT jdoubleArray JNICALL Java_org_opencv_core_Core_n_1minMaxLocManual
|
|
(JNIEnv* env, jclass cls, jlong src_nativeObj, jlong mask_nativeObj)
|
|
{
|
|
try {
|
|
LOGD("Core::n_1minMaxLoc()");
|
|
jdoubleArray result;
|
|
result = env->NewDoubleArray(6);
|
|
if (result == NULL) {
|
|
return NULL; /* out of memory error thrown */
|
|
}
|
|
|
|
Mat& src = *((Mat*)src_nativeObj);
|
|
|
|
double minVal, maxVal;
|
|
Point minLoc, maxLoc;
|
|
if (mask_nativeObj != 0) {
|
|
Mat& mask = *((Mat*)mask_nativeObj);
|
|
minMaxLoc(src, &minVal, &maxVal, &minLoc, &maxLoc, mask);
|
|
} else {
|
|
minMaxLoc(src, &minVal, &maxVal, &minLoc, &maxLoc);
|
|
}
|
|
|
|
jdouble fill[6];
|
|
fill[0]=minVal;
|
|
fill[1]=maxVal;
|
|
fill[2]=minLoc.x;
|
|
fill[3]=minLoc.y;
|
|
fill[4]=maxLoc.x;
|
|
fill[5]=maxLoc.y;
|
|
|
|
env->SetDoubleArrayRegion(result, 0, 6, fill);
|
|
|
|
return result;
|
|
|
|
} catch(cv::Exception e) {
|
|
LOGD("Core::n_1minMaxLoc() catched cv::Exception: %s", e.what());
|
|
jclass je = env->FindClass("org/opencv/core/CvException");
|
|
if(!je) je = env->FindClass("java/lang/Exception");
|
|
env->ThrowNew(je, e.what());
|
|
return NULL;
|
|
} catch (...) {
|
|
LOGD("Core::n_1minMaxLoc() catched unknown exception (...)");
|
|
jclass je = env->FindClass("java/lang/Exception");
|
|
env->ThrowNew(je, "Unknown exception in JNI code {core::minMaxLoc()}");
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
""",
|
|
}, # minMaxLoc
|
|
|
|
'getTextSize' :
|
|
{
|
|
'j_code' :
|
|
"""
|
|
// C++: Size getTextSize(const string& text, int fontFace, double fontScale, int thickness, int* baseLine);
|
|
//javadoc:getTextSize(text, fontFace, fontScale, thickness, baseLine)
|
|
public static Size getTextSize(String text, int fontFace, double fontScale, int thickness, int[] baseLine) {
|
|
if(baseLine != null && baseLine.length != 1)
|
|
throw new java.lang.IllegalArgumentException("'baseLine' must be 'int[1]' or 'null'.");
|
|
Size retVal = new Size(n_getTextSize(text, fontFace, fontScale, thickness, baseLine));
|
|
return retVal;
|
|
}
|
|
""",
|
|
'jn_code' :
|
|
""" private static native double[] n_getTextSize(String text, int fontFace, double fontScale, int thickness, int[] baseLine);\n""",
|
|
'cpp_code' :
|
|
"""
|
|
// C++: Size getTextSize(const string& text, int fontFace, double fontScale, int thickness, int* baseLine);
|
|
|
|
JNIEXPORT jdoubleArray JNICALL Java_org_opencv_core_Core_n_1getTextSize
|
|
(JNIEnv* env, jclass cls, jstring text, jint fontFace, jdouble fontScale, jint thickness, jintArray baseLine)
|
|
{
|
|
try {
|
|
LOGD("Core::n_1getTextSize()");
|
|
jdoubleArray result;
|
|
result = env->NewDoubleArray(2);
|
|
if (result == NULL) {
|
|
return NULL; /* out of memory error thrown */
|
|
}
|
|
|
|
const char* utf_text = env->GetStringUTFChars(text, 0);
|
|
std::string n_text( utf_text ? utf_text : "" );
|
|
env->ReleaseStringUTFChars(text, utf_text);
|
|
|
|
int _baseLine;
|
|
int* pbaseLine = 0;
|
|
|
|
if (baseLine != NULL)
|
|
pbaseLine = &_baseLine;
|
|
|
|
cv::Size rsize = cv::getTextSize(n_text, (int)fontFace, (double)fontScale, (int)thickness, pbaseLine);
|
|
|
|
jdouble fill[2];
|
|
fill[0]=rsize.width;
|
|
fill[1]=rsize.height;
|
|
|
|
env->SetDoubleArrayRegion(result, 0, 2, fill);
|
|
|
|
if (baseLine != NULL)
|
|
env->SetIntArrayRegion(baseLine, 0, 1, pbaseLine);
|
|
|
|
return result;
|
|
|
|
} catch(cv::Exception e) {
|
|
LOGD("Core::n_1getTextSize() catched cv::Exception: %s", e.what());
|
|
jclass je = env->FindClass("org/opencv/core/CvException");
|
|
if(!je) je = env->FindClass("java/lang/Exception");
|
|
env->ThrowNew(je, e.what());
|
|
return NULL;
|
|
} catch (...) {
|
|
LOGD("Core::n_1getTextSize() catched unknown exception (...)");
|
|
jclass je = env->FindClass("java/lang/Exception");
|
|
env->ThrowNew(je, "Unknown exception in JNI code {core::getTextSize()}");
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
""",
|
|
}, # getTextSize
|
|
|
|
"checkHardwareSupport" : {'j_code' : '', 'jn_code' : '', 'cpp_code' : '' },
|
|
"setUseOptimized" : {'j_code' : '', 'jn_code' : '', 'cpp_code' : '' },
|
|
"useOptimized" : {'j_code' : '', 'jn_code' : '', 'cpp_code' : '' },
|
|
"vconcat" : {'j_code' : '', 'jn_code' : '', 'cpp_code' : '' },
|
|
|
|
}, # Core
|
|
|
|
'Highgui' :
|
|
{
|
|
"namedWindow" : {'j_code' : '', 'jn_code' : '', 'cpp_code' : '' },
|
|
"destroyWindow" : {'j_code' : '', 'jn_code' : '', 'cpp_code' : '' },
|
|
"destroyAllWindows" : {'j_code' : '', 'jn_code' : '', 'cpp_code' : '' },
|
|
"startWindowThread" : {'j_code' : '', 'jn_code' : '', 'cpp_code' : '' },
|
|
"setWindowProperty" : {'j_code' : '', 'jn_code' : '', 'cpp_code' : '' },
|
|
"getWindowProperty" : {'j_code' : '', 'jn_code' : '', 'cpp_code' : '' },
|
|
"getTrackbarPos" : {'j_code' : '', 'jn_code' : '', 'cpp_code' : '' },
|
|
"setTrackbarPos" : {'j_code' : '', 'jn_code' : '', 'cpp_code' : '' },
|
|
"imshow" : {'j_code' : '', 'jn_code' : '', 'cpp_code' : '' },
|
|
"waitKey" : {'j_code' : '', 'jn_code' : '', 'cpp_code' : '' },
|
|
}, # Highgui
|
|
}
|
|
|
|
# { class : { func : {arg_name : ctype} } }
|
|
func_arg_fix = {
|
|
'' : {
|
|
'randu' : { 'low' : 'Scalar', 'high' : 'Scalar', },
|
|
'randn' : { 'mean' : 'Scalar', 'stddev' : 'Scalar', },
|
|
'inRange' : { 'lowerb' : 'Scalar', 'upperb' : 'Scalar', },
|
|
}, # '', i.e. empty class
|
|
} # func_arg_fix
|
|
|
|
class ConstInfo(object):
|
|
def __init__(self, cname, name, val, addedManually=False):
|
|
self.cname = cname
|
|
self.name = re.sub(r"^Cv", "", name)
|
|
self.value = val
|
|
self.addedManually = addedManually
|
|
|
|
|
|
class ClassPropInfo(object):
|
|
def __init__(self, decl): # [f_ctype, f_name, '', '/RW']
|
|
self.ctype = decl[0]
|
|
self.name = decl[1]
|
|
self.rw = "/RW" in decl[3]
|
|
|
|
class ClassInfo(object):
|
|
def __init__(self, decl): # [ 'class/struct cname', [bases], [modlist] ]
|
|
name = decl[0]
|
|
name = name[name.find(" ")+1:].strip()
|
|
self.cname = self.name = self.jname = re.sub(r"^cv\.", "", name)
|
|
self.cname =self.cname.replace(".", "::")
|
|
self.methods = {}
|
|
self.consts = [] # using a list to save the occurence order
|
|
self.private_consts = []
|
|
self.imports = set()
|
|
self.props= []
|
|
for m in decl[2]:
|
|
if m.startswith("="):
|
|
self.jname = m[1:]
|
|
|
|
|
|
class ArgInfo(object):
|
|
def __init__(self, arg_tuple): # [ ctype, name, def val, [mod], argno ]
|
|
self.pointer = False
|
|
ctype = arg_tuple[0]
|
|
if ctype.endswith("*"):
|
|
ctype = ctype[:-1]
|
|
self.pointer = True
|
|
self.ctype = ctype
|
|
self.name = arg_tuple[1]
|
|
self.defval = arg_tuple[2]
|
|
self.out = ""
|
|
if "/O" in arg_tuple[3]:
|
|
self.out = "O"
|
|
if "/IO" in arg_tuple[3]:
|
|
self.out = "IO"
|
|
|
|
|
|
class FuncInfo(object):
|
|
def __init__(self, decl): # [ funcname, return_ctype, [modifiers], [args] ]
|
|
name = re.sub(r"^cv\.", "", decl[0])
|
|
self.cname = name.replace(".", "::")
|
|
classname = ""
|
|
dpos = name.rfind(".")
|
|
if dpos >= 0:
|
|
classname = name[:dpos]
|
|
name = name[dpos+1:]
|
|
self.classname = classname
|
|
self.jname = self.name = name
|
|
if "[" in name:
|
|
self.jname = "getelem"
|
|
for m in decl[2]:
|
|
if m.startswith("="):
|
|
self.jname = m[1:]
|
|
self.jn_name = "n_" + self.jname
|
|
self.jni_name= re.sub(r"_", "_1", self.jn_name)
|
|
self.static = ["","static"][ "/S" in decl[2] ]
|
|
self.ctype = decl[1] or ""
|
|
self.args = []
|
|
#self.jni_suffix = "__"
|
|
#if self.classname and self.ctype and not self.static: # non-static class methods except c-tors
|
|
# self.jni_suffix += "J" # artifical 'self'
|
|
arg_fix_map = func_arg_fix.get(classname, {}).get(self.jname, {})
|
|
for a in decl[3]:
|
|
arg = a[:]
|
|
arg[0] = arg_fix_map.get(arg[1], arg[0])
|
|
ai = ArgInfo(arg)
|
|
self.args.append(ai)
|
|
# self.jni_suffix += ctype2j.get(ai.ctype, ["","","",""])[3]
|
|
|
|
|
|
|
|
class FuncFamilyInfo(object):
|
|
def __init__(self, decl): # [ funcname, return_ctype, [modifiers], [args] ]
|
|
self.funcs = []
|
|
self.funcs.append( FuncInfo(decl) )
|
|
self.jname = self.funcs[0].jname
|
|
self.isconstructor = self.funcs[0].name == self.funcs[0].classname
|
|
|
|
|
|
|
|
def add_func(self, fi):
|
|
self.funcs.append( fi )
|
|
|
|
|
|
class JavaWrapperGenerator(object):
|
|
def __init__(self):
|
|
self.clear()
|
|
|
|
def clear(self):
|
|
self.classes = { "Mat" : ClassInfo([ 'class Mat', [], [] ]) }
|
|
self.module = ""
|
|
self.Module = ""
|
|
self.java_code= {} # { class : {j_code, jn_code} }
|
|
self.cpp_code = None
|
|
self.ported_func_list = []
|
|
self.skipped_func_list = []
|
|
|
|
def add_class_code_stream(self, class_name):
|
|
self.java_code[class_name] = { "j_code" : StringIO(), "jn_code" : StringIO(), }
|
|
if class_name != self.Module:
|
|
self.java_code[class_name]["j_code"].write("""
|
|
//
|
|
// This file is auto-generated. Please don't modify it!
|
|
//
|
|
package org.opencv.%(m)s;
|
|
|
|
$imports
|
|
|
|
// C++: class %(c)s
|
|
//javadoc: %(c)s
|
|
public class %(c)s {
|
|
|
|
protected final long nativeObj;
|
|
protected %(c)s(long addr) { nativeObj = addr; }
|
|
|
|
""" % { 'm' : self.module, 'c' : class_name } )
|
|
|
|
else: # class_name == self.Module
|
|
self.java_code[class_name]["j_code"].write("""
|
|
//
|
|
// This file is auto-generated. Please don't modify it!
|
|
//
|
|
package org.opencv.%(m)s;
|
|
|
|
$imports
|
|
|
|
public class %(c)s {
|
|
""" % { 'm' : self.module, 'c' : class_name } )
|
|
|
|
self.java_code[class_name]["jn_code"].write("""
|
|
//
|
|
// native stuff
|
|
//
|
|
static { System.loadLibrary("opencv_java"); }
|
|
""" )
|
|
|
|
|
|
|
|
def add_class(self, decl):
|
|
classinfo = ClassInfo(decl)
|
|
if classinfo.name in class_ignore_list:
|
|
return
|
|
if classinfo.name in self.classes:
|
|
print "Generator error: class %s (%s) is duplicated" % \
|
|
(classinfo.name, classinfo.cname)
|
|
sys.exit(-1)
|
|
self.classes[classinfo.name] = classinfo
|
|
if classinfo.name in type_dict:
|
|
print "Duplicated class: " + classinfo.name
|
|
sys.exit(-1)
|
|
type_dict[classinfo.name] = \
|
|
{ "j_type" : classinfo.name,
|
|
"jn_type" : "long", "jn_args" : (("__int64", ".nativeObj"),),
|
|
"jni_name" : "(*("+classinfo.name+"*)%(n)s_nativeObj)", "jni_type" : "jlong",
|
|
"suffix" : "J" }
|
|
|
|
# missing_consts { Module : { public : [[name, val],...], private : [[]...] } }
|
|
if classinfo.name in missing_consts:
|
|
if 'private' in missing_consts[classinfo.name]:
|
|
for (name, val) in missing_consts[classinfo.name]['private']:
|
|
classinfo.private_consts.append( ConstInfo(name, name, val, True) )
|
|
if 'public' in missing_consts[classinfo.name]:
|
|
for (name, val) in missing_consts[classinfo.name]['public']:
|
|
classinfo.consts.append( ConstInfo(name, name, val, True) )
|
|
|
|
# class props
|
|
for p in decl[3]:
|
|
if "vector" not in p[0]:
|
|
classinfo.props.append( ClassPropInfo(p) )
|
|
else:
|
|
print "Skipped property: [%s]" % classinfo.name, p
|
|
|
|
self.add_class_code_stream(classinfo.name)
|
|
|
|
|
|
def add_const(self, decl): # [ "const cname", val, [], [] ]
|
|
name = decl[0].replace("const ", "").strip()
|
|
name = re.sub(r"^cv\.", "", name)
|
|
cname = name.replace(".", "::")
|
|
for c in const_ignore_list:
|
|
if re.match(c, name):
|
|
return
|
|
# class member?
|
|
dpos = name.rfind(".")
|
|
if dpos >= 0:
|
|
classname = name[:dpos]
|
|
name = name[dpos+1:]
|
|
else:
|
|
classname = self.Module
|
|
if classname not in self.classes:
|
|
# this class isn't wrapped
|
|
# skipping this const
|
|
return
|
|
|
|
consts = self.classes[classname].consts
|
|
for c in const_private_list:
|
|
if re.match(c, name):
|
|
consts = self.classes[classname].private_consts
|
|
break
|
|
|
|
constinfo = ConstInfo(cname, name, decl[1])
|
|
# checking duplication
|
|
for list in self.classes[classname].consts, self.classes[classname].private_consts:
|
|
for c in list:
|
|
if c.name == constinfo.name:
|
|
if c.addedManually:
|
|
return
|
|
print "Generator error: constant %s (%s) is duplicated" \
|
|
% (constinfo.name, constinfo.cname)
|
|
sys.exit(-1)
|
|
|
|
consts.append(constinfo)
|
|
|
|
def add_func(self, decl):
|
|
ffi = FuncFamilyInfo(decl)
|
|
classname = ffi.funcs[0].classname or self.Module
|
|
if classname in class_ignore_list:
|
|
return
|
|
if classname in ManualFuncs and ffi.jname in ManualFuncs[classname]:
|
|
return
|
|
if classname not in self.classes:
|
|
print "Generator error: the class %s for method %s is missing" % \
|
|
(classname, ffi.jname)
|
|
sys.exit(-1)
|
|
func_map = self.classes[classname].methods
|
|
if ffi.jname in func_map:
|
|
func_map[ffi.jname].add_func(ffi.funcs[0])
|
|
else:
|
|
func_map[ffi.jname] = ffi
|
|
|
|
def save(self, path, buf):
|
|
f = open(path, "wt")
|
|
f.write(buf)
|
|
f.close()
|
|
|
|
def gen(self, srcfiles, module, output_path):
|
|
self.clear()
|
|
self.module = module
|
|
self.Module = module.capitalize()
|
|
parser = hdr_parser.CppHeaderParser()
|
|
|
|
self.add_class( ['class ' + self.Module, [], [], []] ) # [ 'class/struct cname', [bases], [modlist] [props] ]
|
|
|
|
# scan the headers and build more descriptive maps of classes, consts, functions
|
|
for hdr in srcfiles:
|
|
decls = parser.parse(hdr)
|
|
for decl in decls:
|
|
name = decl[0]
|
|
if name.startswith("struct") or name.startswith("class"):
|
|
self.add_class(decl)
|
|
elif name.startswith("const"):
|
|
self.add_const(decl)
|
|
else: # function
|
|
self.add_func(decl)
|
|
|
|
self.cpp_code = StringIO()
|
|
self.cpp_code.write("""
|
|
//
|
|
// This file is auto-generated, please don't edit!
|
|
//
|
|
|
|
#include <jni.h>
|
|
|
|
#include "converters.h"
|
|
|
|
#ifdef DEBUG
|
|
#include <android/log.h>
|
|
#define MODULE_LOG_TAG "OpenCV.%(m)s"
|
|
#define LOGD(...) ((void)__android_log_print(ANDROID_LOG_DEBUG, MODULE_LOG_TAG, __VA_ARGS__))
|
|
#else //DEBUG
|
|
#define LOGD(...)
|
|
#endif //DEBUG
|
|
|
|
#include "opencv2/%(m)s/%(m)s.hpp"
|
|
|
|
using namespace cv;
|
|
|
|
extern "C" {
|
|
|
|
""" % {'m' : module} )
|
|
|
|
# generate code for the classes
|
|
for name in self.classes.keys():
|
|
self.gen_class(name)
|
|
|
|
# saving code streams
|
|
for cls in self.java_code.keys():
|
|
imports = "\n".join([ "import %s;" % c for c in \
|
|
sorted(self.classes[cls].imports) if not c.startswith('org.opencv.'+self.module) ])
|
|
##imports = "import org.opencv.core.*;\nimport org.opencv.Converters;\n"
|
|
self.java_code[cls]["j_code"].write("\n\n%s\n}\n" % self.java_code[cls]["jn_code"].getvalue())
|
|
java_code = self.java_code[cls]["j_code"].getvalue()
|
|
java_code = Template(java_code).substitute(imports = imports)
|
|
self.save("%s/%s+%s.java" % (output_path, module, cls), java_code)
|
|
|
|
self.cpp_code.write( '\n} // extern "C"\n' )
|
|
self.save(output_path+"/"+module+".cpp", self.cpp_code.getvalue())
|
|
|
|
# report
|
|
report = StringIO()
|
|
report.write("PORTED FUNCs LIST (%i of %i):\n\n" % \
|
|
(len(self.ported_func_list), len(self.ported_func_list)+ len(self.skipped_func_list))
|
|
)
|
|
report.write("\n".join(self.ported_func_list))
|
|
report.write("\n\nSKIPPED FUNCs LIST (%i of %i):\n\n" % \
|
|
(len(self.skipped_func_list), len(self.ported_func_list)+ len(self.skipped_func_list))
|
|
)
|
|
report.write("".join(self.skipped_func_list))
|
|
self.save(output_path+"/"+module+".txt", report.getvalue())
|
|
|
|
print "Done %i of %i funcs." % (len(self.ported_func_list), len(self.ported_func_list)+ len(self.skipped_func_list))
|
|
|
|
|
|
|
|
def get_imports(self, scope_classname, ctype):
|
|
imports = self.classes[scope_classname or self.Module].imports
|
|
if ctype.startswith('vector'):
|
|
imports.add("java.util.List")
|
|
imports.add("org.opencv.core.Mat")
|
|
imports.add("org.opencv.Converters")
|
|
ctype = ctype.replace('vector_', '')
|
|
j_type = ''
|
|
if ctype in type_dict:
|
|
j_type = type_dict[ctype]['j_type']
|
|
if j_type in ( "CvType", "Mat", "Point", "Point3", "Range", "Rect", "RotatedRect", "Scalar", "Size", "TermCriteria" ):
|
|
imports.add("org.opencv.core." + j_type)
|
|
if j_type == 'String':
|
|
imports.add("java.lang.String")
|
|
|
|
|
|
|
|
def gen_func(self, fi, isoverload, prop_name=''):
|
|
j_code = self.java_code[fi.classname or self.Module]["j_code"]
|
|
jn_code = self.java_code[fi.classname or self.Module]["jn_code"]
|
|
cpp_code = self.cpp_code
|
|
|
|
# c_decl
|
|
# e.g: void add(Mat src1, Mat src2, Mat dst, Mat mask = Mat(), int dtype = -1)
|
|
if prop_name:
|
|
c_decl = "%s %s::%s" % (fi.ctype, fi.classname, prop_name)
|
|
else:
|
|
decl_args = []
|
|
for a in fi.args:
|
|
s = a.ctype
|
|
if a.pointer:
|
|
s += "*"
|
|
elif a.out:
|
|
s += "&"
|
|
s += " " + a.name
|
|
if a.defval:
|
|
s += " = "+a.defval
|
|
decl_args.append(s)
|
|
c_decl = "%s %s %s(%s)" % ( fi.static, fi.ctype, fi.cname, ", ".join(decl_args) )
|
|
|
|
# java comment
|
|
j_code.write( "\n //\n // C++: %s\n //\n\n" % c_decl )
|
|
# check if we 'know' all the types
|
|
if fi.ctype not in type_dict: # unsupported ret type
|
|
msg = "// Return type '%s' is not supported, skipping the function\n\n" % fi.ctype
|
|
self.skipped_func_list.append(c_decl + "\n" + msg)
|
|
j_code.write( " "*4 + msg )
|
|
print "SKIP:", c_decl, "\n\tdue to RET type", fi.ctype
|
|
return
|
|
for a in fi.args:
|
|
if a.ctype not in type_dict:
|
|
msg = "// Unknown type '%s' (%s), skipping the function\n\n" % (a.ctype, a.out or "I")
|
|
self.skipped_func_list.append(c_decl + "\n" + msg)
|
|
j_code.write( " "*4 + msg )
|
|
print "SKIP:", c_decl, "\n\tdue to ARG type", a.ctype, "/" + (a.out or "I")
|
|
return
|
|
|
|
self.ported_func_list.append(c_decl)
|
|
|
|
# jn & cpp comment
|
|
jn_code.write( "\n // C++: %s\n" % c_decl )
|
|
cpp_code.write( "\n//\n// %s\n//\n" % c_decl )
|
|
|
|
# java args
|
|
args = fi.args[:] # copy
|
|
if args and args[-1].defval:
|
|
isoverload = True
|
|
|
|
while True:
|
|
|
|
# java native method args
|
|
jn_args = []
|
|
# jni (cpp) function args
|
|
jni_args = [ArgInfo([ "env", "env", "", [], "" ]), ArgInfo([ "cls", "cls", "", [], "" ])]
|
|
suffix = "__"
|
|
j_prologue = []
|
|
j_epilogue = []
|
|
c_prologue = []
|
|
c_epilogue = []
|
|
if type_dict[fi.ctype]["jni_type"] == "jdoubleArray":
|
|
fields = type_dict[fi.ctype]["jn_args"]
|
|
c_epilogue.append( \
|
|
("jdoubleArray _da_retval_ = env->NewDoubleArray(%(cnt)i); " +
|
|
"jdouble _tmp_retval_[%(cnt)i] = {%(args)s}; " +
|
|
"env->SetDoubleArrayRegion(_da_retval_, 0, %(cnt)i, _tmp_retval_);") %
|
|
{ "cnt" : len(fields), "args" : ", ".join(["_retval_" + f[1] for f in fields]) } )
|
|
if fi.classname and fi.ctype and not fi.static: # non-static class method except c-tor
|
|
# adding 'self'
|
|
jn_args.append ( ArgInfo([ "__int64", "nativeObj", "", [], "" ]) )
|
|
jni_args.append( ArgInfo([ "__int64", "self", "", [], "" ]) )
|
|
suffix += "J"
|
|
self.get_imports(fi.classname, fi.ctype)
|
|
for a in args:
|
|
self.get_imports(fi.classname, a.ctype)
|
|
if a.pointer:
|
|
suffix += "_3D"
|
|
else:
|
|
suffix += type_dict[a.ctype].get("suffix") or ""
|
|
|
|
if "vector" in a.ctype: # pass as Mat
|
|
jn_args.append ( ArgInfo([ "__int64", "%s_mat.nativeObj" % a.name, "", [], "" ]) )
|
|
jni_args.append ( ArgInfo([ "__int64", "%s_mat_nativeObj" % a.name, "", [], "" ]) )
|
|
c_prologue.append( type_dict[a.ctype]["jni_var"] % {"n" : a.name} + ";" )
|
|
c_prologue.append( "Mat& %(n)s_mat = *((Mat*)%(n)s_mat_nativeObj)" % {"n" : a.name} + ";" )
|
|
if "I" in a.out or not a.out:
|
|
j_prologue.append( "Mat %(n)s_mat = Converters.%(t)s_to_Mat(%(n)s);" % {"n" : a.name, "t" : a.ctype} )
|
|
c_prologue.append( "Mat_to_%(t)s( %(n)s_mat, %(n)s );" % {"n" : a.name, "t" : a.ctype} )
|
|
else:
|
|
j_prologue.append( "Mat %s_mat = new Mat();" % a.name )
|
|
if "O" in a.out:
|
|
j_epilogue.append("Converters.Mat_to_%(t)s(%(n)s_mat, %(n)s);" % {"t" : a.ctype, "n" : a.name})
|
|
c_epilogue.append( "%(t)s_to_Mat( %(n)s, %(n)s_mat );" % {"n" : a.name, "t" : a.ctype} )
|
|
else:
|
|
fields = type_dict[a.ctype].get("jn_args", ((a.ctype, ""),))
|
|
if "I" in a.out or not a.out or a.ctype in self.classes: # input arg, pass by primitive fields
|
|
for f in fields:
|
|
jn_args.append ( ArgInfo([ f[0], a.name + f[1], "", [], "" ]) )
|
|
jni_args.append( ArgInfo([ f[0], a.name + f[1].replace(".","_").replace("[","").replace("]",""), "", [], "" ]) )
|
|
if a.out and a.ctype not in self.classes: # out arg, pass as double[]
|
|
jn_args.append ( ArgInfo([ "double[]", "%s_out" % a.name, "", [], "" ]) )
|
|
jni_args.append ( ArgInfo([ "double[]", "%s_out" % a.name, "", [], "" ]) )
|
|
j_prologue.append( "double[] %s_out = new double[%i];" % (a.name, len(fields)) )
|
|
set_vals = []
|
|
i = 0
|
|
for f in fields:
|
|
set_vals.append( "%(n)s%(f)s = %(t)s%(n)s_out[%(i)i]" %
|
|
{"n" : a.name, "t": ("("+type_dict[f[0]]["j_type"]+")", "")[f[0]=="double"], "f" : f[1], "i" : i}
|
|
)
|
|
i += 1
|
|
j_epilogue.append("; ".join(set_vals) + "; ")
|
|
c_epilogue.append( \
|
|
"jdouble tmp_%(n)s[%(cnt)i] = {%(args)s}; env->SetDoubleArrayRegion(%(n)s_out, 0, %(cnt)i, tmp_%(n)s);" %
|
|
{ "n" : a.name, "cnt" : len(fields), "args" : ", ".join([a.name + f[1] for f in fields]) } )
|
|
|
|
|
|
# java part:
|
|
# private java NATIVE method decl
|
|
# e.g.
|
|
# private static native void n_add(long src1, long src2, long dst, long mask, int dtype);
|
|
jn_code.write( Template(\
|
|
" private static native $jn_type $jn_name($jn_args);\n").substitute(\
|
|
jn_type = type_dict[fi.ctype].get("jn_type", "double[]"), \
|
|
jn_name = fi.jn_name, \
|
|
jn_args = ", ".join(["%s %s" % (type_dict[a.ctype]["jn_type"], a.name.replace(".","_").replace("[","").replace("]","")) for a in jn_args])
|
|
) );
|
|
|
|
# java part:
|
|
|
|
#java doc comment
|
|
f_name = fi.name
|
|
if fi.classname:
|
|
f_name = fi.classname + "::" + fi.name
|
|
java_doc = "//javadoc: " + f_name + "(%s)" % ", ".join([a.name for a in args])
|
|
j_code.write(" "*4 + java_doc + "\n")
|
|
|
|
# public java wrapper method impl (calling native one above)
|
|
# e.g.
|
|
# public static void add( Mat src1, Mat src2, Mat dst, Mat mask, int dtype )
|
|
# { n_add( src1.nativeObj, src2.nativeObj, dst.nativeObj, mask.nativeObj, dtype ); }
|
|
ret_val = type_dict[fi.ctype]["j_type"] + " retVal = "
|
|
tail = ""
|
|
ret = "return retVal;"
|
|
if fi.ctype == "void":
|
|
ret_val = ""
|
|
ret = "return;"
|
|
elif fi.ctype == "": # c-tor
|
|
ret_val = "nativeObj = "
|
|
ret = "return;"
|
|
elif fi.ctype in self.classes: # wrapped class
|
|
ret_val = type_dict[fi.ctype]["j_type"] + " retVal = new " + self.classes[fi.ctype].jname + "("
|
|
tail = ")"
|
|
elif "jn_type" not in type_dict[fi.ctype]:
|
|
ret_val = type_dict[fi.ctype]["j_type"] + " retVal = new " + type_dict[fi.ctype]["j_type"] + "("
|
|
tail = ")"
|
|
|
|
static = "static"
|
|
if fi.classname:
|
|
static = fi.static
|
|
|
|
j_code.write( Template(\
|
|
""" public $static $j_type $j_name($j_args)
|
|
{
|
|
$prologue
|
|
$ret_val$jn_name($jn_args_call)$tail;
|
|
$epilogue
|
|
$ret
|
|
}
|
|
|
|
"""
|
|
).substitute(\
|
|
ret = ret, \
|
|
ret_val = ret_val, \
|
|
tail = tail, \
|
|
prologue = " ".join(j_prologue), \
|
|
epilogue = " ".join(j_epilogue), \
|
|
static=static, \
|
|
j_type=type_dict[fi.ctype]["j_type"], \
|
|
j_name=fi.jname, \
|
|
j_args=", ".join(["%s %s" % (type_dict[a.ctype]["j_type"], a.name) for a in args]), \
|
|
jn_name=fi.jn_name, \
|
|
jn_args_call=", ".join( [a.name for a in jn_args] ),\
|
|
)
|
|
)
|
|
|
|
# cpp part:
|
|
# jni_func(..) { _retval_ = cv_func(..); return _retval_; }
|
|
ret = "return _retval_;"
|
|
default = "return 0;"
|
|
if fi.ctype == "void":
|
|
ret = "return;"
|
|
default = "return;"
|
|
elif not fi.ctype: # c-tor
|
|
ret = "return (jlong) _retval_;"
|
|
elif fi.ctype == "string":
|
|
ret = "return env->NewStringUTF(_retval_.c_str());"
|
|
default = 'return env->NewStringUTF("");'
|
|
elif fi.ctype in self.classes: # wrapped class:
|
|
ret = "return (jlong) new %s(_retval_);" % fi.ctype
|
|
elif type_dict[fi.ctype]["jni_type"] == "jdoubleArray":
|
|
ret = "return _da_retval_;"
|
|
|
|
# hack: replacing func call with property set/get
|
|
name = fi.name
|
|
if prop_name:
|
|
if args:
|
|
name = prop_name + " = "
|
|
else:
|
|
name = prop_name + ";//"
|
|
|
|
cvname = "cv::" + name
|
|
retval = fi.ctype + " _retval_ = "
|
|
if fi.ctype == "void":
|
|
retval = ""
|
|
if fi.classname:
|
|
if not fi.ctype: # c-tor
|
|
retval = fi.classname + "* _retval_ = "
|
|
cvname = "new " + fi.classname
|
|
elif fi.static:
|
|
cvname = "%s::%s" % (fi.classname, name)
|
|
else:
|
|
cvname = "me->" + name
|
|
c_prologue.append(\
|
|
"%(cls)s* me = (%(cls)s*) self; //TODO: check for NULL" \
|
|
% { "cls" : fi.classname} \
|
|
)
|
|
cvargs = []
|
|
for a in args:
|
|
if a.pointer:
|
|
jni_name = "&%(n)s"
|
|
else:
|
|
jni_name = "%(n)s"
|
|
cvargs.append( type_dict[a.ctype].get("jni_name", jni_name) % {"n" : a.name})
|
|
if "vector" not in a.ctype :
|
|
if ("I" in a.out or not a.out or a.ctype in self.classes) and "jni_var" in type_dict[a.ctype]: # complex type
|
|
c_prologue.append(type_dict[a.ctype]["jni_var"] % {"n" : a.name} + ";")
|
|
if a.out and "I" not in a.out and a.ctype not in self.classes:
|
|
c_prologue.append("%s %s;" % (a.ctype, a.name))
|
|
|
|
rtype = type_dict[fi.ctype].get("jni_type", "jdoubleArray")
|
|
cpp_code.write ( Template( \
|
|
"""
|
|
|
|
JNIEXPORT $rtype JNICALL Java_org_opencv_${module}_${clazz}_$fname
|
|
($args)
|
|
{
|
|
try {
|
|
LOGD("$module::$fname()");
|
|
$prologue
|
|
$retval$cvname( $cvargs );
|
|
$epilogue
|
|
$ret
|
|
} catch(cv::Exception e) {
|
|
LOGD("$module::$fname() catched cv::Exception: %s", e.what());
|
|
jclass je = env->FindClass("org/opencv/core/CvException");
|
|
if(!je) je = env->FindClass("java/lang/Exception");
|
|
env->ThrowNew(je, e.what());
|
|
$default
|
|
} catch (...) {
|
|
LOGD("$module::$fname() catched unknown exception (...)");
|
|
jclass je = env->FindClass("java/lang/Exception");
|
|
env->ThrowNew(je, "Unknown exception in JNI code {$module::$fname()}");
|
|
$default
|
|
}
|
|
}
|
|
|
|
|
|
""" ).substitute( \
|
|
rtype = rtype, \
|
|
module = self.module, \
|
|
clazz = fi.classname or self.Module, \
|
|
fname = fi.jni_name + ["",suffix][isoverload], \
|
|
args = ", ".join(["%s %s" % (type_dict[a.ctype].get("jni_type"), a.name) for a in jni_args]), \
|
|
prologue = "\n ".join(c_prologue), \
|
|
epilogue = " ".join(c_epilogue), \
|
|
ret = ret, \
|
|
cvname = cvname, \
|
|
cvargs = ", ".join([a for a in cvargs]), \
|
|
default = default, \
|
|
retval = retval, \
|
|
) )
|
|
|
|
# processing args with default values
|
|
if args and args[-1].defval:
|
|
a = args.pop()
|
|
else:
|
|
break
|
|
|
|
|
|
|
|
def gen_class(self, name):
|
|
# generate code for the class
|
|
if name == "Mat":
|
|
return
|
|
ci = self.classes[name]
|
|
# constants
|
|
if ci.private_consts:
|
|
self.java_code[name]['j_code'].write("""
|
|
private static final int
|
|
%s;\n\n""" % (",\n"+" "*12).join(["%s = %s" % (c.name, c.value) for c in ci.private_consts])
|
|
)
|
|
if ci.consts:
|
|
self.java_code[name]['j_code'].write("""
|
|
public static final int
|
|
%s;\n\n""" % (",\n"+" "*12).join(["%s = %s" % (c.name, c.value) for c in ci.consts])
|
|
)
|
|
# c-tors
|
|
fflist = ci.methods.items()
|
|
fflist.sort()
|
|
for n, ffi in fflist:
|
|
if ffi.isconstructor:
|
|
for fi in ffi.funcs:
|
|
self.gen_func(fi, len(ffi.funcs)>1)
|
|
# other methods
|
|
for n, ffi in fflist:
|
|
if not ffi.isconstructor:
|
|
for fi in ffi.funcs:
|
|
self.gen_func(fi, len(ffi.funcs)>1)
|
|
# props
|
|
for pi in ci.props:
|
|
# getter
|
|
getter_name = name + ".get_" + pi.name
|
|
#print getter_name
|
|
fi = FuncInfo( [getter_name, pi.ctype, [], []] ) # [ funcname, return_ctype, [modifiers], [args] ]
|
|
self.gen_func(fi, getter_name in ci.methods, pi.name)
|
|
if pi.rw:
|
|
#setter
|
|
setter_name = name + ".set_" + pi.name
|
|
#print setter_name
|
|
fi = FuncInfo( [ setter_name, "void", [], [ [pi.ctype, pi.name, "", [], ""] ] ] )
|
|
self.gen_func(fi, getter_name in ci.methods, pi.name)
|
|
|
|
# manual ports
|
|
if name in ManualFuncs:
|
|
for func in ManualFuncs[name].keys():
|
|
self.java_code[name]["j_code"].write ( ManualFuncs[name][func]["j_code"] )
|
|
self.java_code[name]["jn_code"].write( ManualFuncs[name][func]["jn_code"] )
|
|
self.cpp_code.write( ManualFuncs[name][func]["cpp_code"] )
|
|
|
|
if name != self.Module:
|
|
# finalize()
|
|
self.java_code[name]["j_code"].write(
|
|
"""
|
|
@Override
|
|
protected void finalize() throws Throwable {
|
|
n_delete(nativeObj);
|
|
super.finalize();
|
|
}
|
|
""" )
|
|
|
|
self.java_code[name]["jn_code"].write(
|
|
"""
|
|
// native support for java finalize()
|
|
private static native void n_delete(long nativeObj);
|
|
""" )
|
|
|
|
# native support for java finalize()
|
|
self.cpp_code.write( \
|
|
"""
|
|
//
|
|
// native support for java finalize()
|
|
// static void %(cls)s::n_delete( __int64 self )
|
|
//
|
|
|
|
JNIEXPORT void JNICALL Java_org_opencv_%(module)s_%(cls)s_n_1delete
|
|
(JNIEnv* env, jclass cls, jlong self)
|
|
{
|
|
delete (%(cls)s*) self;
|
|
}
|
|
|
|
""" % {"module" : module, "cls" : name}
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) < 4:
|
|
print "Usage:\n", \
|
|
os.path.basename(sys.argv[0]), \
|
|
"<full path to hdr_parser.py> <module name> <C++ header> [<C++ header>...]"
|
|
print "Current args are: ", ", ".join(["'"+a+"'" for a in sys.argv])
|
|
exit(0)
|
|
|
|
dstdir = "."
|
|
hdr_parser_path = os.path.abspath(sys.argv[1])
|
|
if hdr_parser_path.endswith(".py"):
|
|
hdr_parser_path = os.path.dirname(hdr_parser_path)
|
|
sys.path.append(hdr_parser_path)
|
|
import hdr_parser
|
|
module = sys.argv[2]
|
|
srcfiles = sys.argv[3:]
|
|
print "Generating module '" + module + "' from headers:\n\t" + "\n\t".join(srcfiles)
|
|
generator = JavaWrapperGenerator()
|
|
generator.gen(srcfiles, module, dstdir)
|
|
|