package org.opencv.utils; import java.util.ArrayList; import java.util.List; import org.opencv.core.CvType; import org.opencv.core.Mat; import org.opencv.core.MatOfByte; import org.opencv.core.MatOfDMatch; import org.opencv.core.MatOfKeyPoint; import org.opencv.core.MatOfPoint; import org.opencv.core.MatOfPoint2f; import org.opencv.core.MatOfPoint3f; import org.opencv.core.Point; import org.opencv.core.Point3; import org.opencv.core.Rect; import org.opencv.features2d.DMatch; import org.opencv.features2d.KeyPoint; public class Converters { public static Mat vector_Point_to_Mat(List<Point> pts) { return vector_Point_to_Mat(pts, CvType.CV_32S); } public static Mat vector_Point2f_to_Mat(List<Point> pts) { return vector_Point_to_Mat(pts, CvType.CV_32F); } public static Mat vector_Point2d_to_Mat(List<Point> pts) { return vector_Point_to_Mat(pts, CvType.CV_64F); } public static Mat vector_Point_to_Mat(List<Point> pts, int typeDepth) { Mat res; int count = (pts != null) ? pts.size() : 0; if (count > 0) { switch (typeDepth) { case CvType.CV_32S: { res = new Mat(count, 1, CvType.CV_32SC2); int[] buff = new int[count * 2]; for (int i = 0; i < count; i++) { Point p = pts.get(i); buff[i * 2] = (int) p.x; buff[i * 2 + 1] = (int) p.y; } res.put(0, 0, buff); } break; case CvType.CV_32F: { res = new Mat(count, 1, CvType.CV_32FC2); float[] buff = new float[count * 2]; for (int i = 0; i < count; i++) { Point p = pts.get(i); buff[i * 2] = (float) p.x; buff[i * 2 + 1] = (float) p.y; } res.put(0, 0, buff); } break; case CvType.CV_64F: { res = new Mat(count, 1, CvType.CV_64FC2); double[] buff = new double[count * 2]; for (int i = 0; i < count; i++) { Point p = pts.get(i); buff[i * 2] = p.x; buff[i * 2 + 1] = p.y; } res.put(0, 0, buff); } break; default: throw new IllegalArgumentException("'typeDepth' can be CV_32S, CV_32F or CV_64F"); } } else { res = new Mat(); } return res; } public static Mat vector_Point3i_to_Mat(List<Point3> pts) { return vector_Point3_to_Mat(pts, CvType.CV_32S); } public static Mat vector_Point3f_to_Mat(List<Point3> pts) { return vector_Point3_to_Mat(pts, CvType.CV_32F); } public static Mat vector_Point3d_to_Mat(List<Point3> pts) { return vector_Point3_to_Mat(pts, CvType.CV_64F); } public static Mat vector_Point3_to_Mat(List<Point3> pts, int typeDepth) { Mat res; int count = (pts != null) ? pts.size() : 0; if (count > 0) { switch (typeDepth) { case CvType.CV_32S: { res = new Mat(count, 1, CvType.CV_32SC3); int[] buff = new int[count * 3]; for (int i = 0; i < count; i++) { Point3 p = pts.get(i); buff[i * 3] = (int) p.x; buff[i * 3 + 1] = (int) p.y; buff[i * 3 + 2] = (int) p.z; } res.put(0, 0, buff); } break; case CvType.CV_32F: { res = new Mat(count, 1, CvType.CV_32FC3); float[] buff = new float[count * 3]; for (int i = 0; i < count; i++) { Point3 p = pts.get(i); buff[i * 3] = (float) p.x; buff[i * 3 + 1] = (float) p.y; buff[i * 3 + 2] = (float) p.z; } res.put(0, 0, buff); } break; case CvType.CV_64F: { res = new Mat(count, 1, CvType.CV_64FC3); double[] buff = new double[count * 3]; for (int i = 0; i < count; i++) { Point3 p = pts.get(i); buff[i * 3] = p.x; buff[i * 3 + 1] = p.y; buff[i * 3 + 2] = p.z; } res.put(0, 0, buff); } break; default: throw new IllegalArgumentException("'typeDepth' can be CV_32S, CV_32F or CV_64F"); } } else { res = new Mat(); } return res; } public static void Mat_to_vector_Point2f(Mat m, List<Point> pts) { Mat_to_vector_Point(m, pts); } public static void Mat_to_vector_Point2d(Mat m, List<Point> pts) { Mat_to_vector_Point(m, pts); } public static void Mat_to_vector_Point(Mat m, List<Point> pts) { if (pts == null) throw new java.lang.IllegalArgumentException("Output List can't be null"); int count = m.rows(); int type = m.type(); if (m.cols() != 1) throw new java.lang.IllegalArgumentException("Input Mat should have one column\n" + m); pts.clear(); if (type == CvType.CV_32SC2) { int[] buff = new int[2 * count]; m.get(0, 0, buff); for (int i = 0; i < count; i++) { pts.add(new Point(buff[i * 2], buff[i * 2 + 1])); } } else if (type == CvType.CV_32FC2) { float[] buff = new float[2 * count]; m.get(0, 0, buff); for (int i = 0; i < count; i++) { pts.add(new Point(buff[i * 2], buff[i * 2 + 1])); } } else if (type == CvType.CV_64FC2) { double[] buff = new double[2 * count]; m.get(0, 0, buff); for (int i = 0; i < count; i++) { pts.add(new Point(buff[i * 2], buff[i * 2 + 1])); } } else { throw new java.lang.IllegalArgumentException( "Input Mat should be of CV_32SC2, CV_32FC2 or CV_64FC2 type\n" + m); } } public static void Mat_to_vector_Point3i(Mat m, List<Point3> pts) { Mat_to_vector_Point3(m, pts); } public static void Mat_to_vector_Point3f(Mat m, List<Point3> pts) { Mat_to_vector_Point3(m, pts); } public static void Mat_to_vector_Point3d(Mat m, List<Point3> pts) { Mat_to_vector_Point3(m, pts); } public static void Mat_to_vector_Point3(Mat m, List<Point3> pts) { if (pts == null) throw new java.lang.IllegalArgumentException("Output List can't be null"); int count = m.rows(); int type = m.type(); if (m.cols() != 1) throw new java.lang.IllegalArgumentException("Input Mat should have one column\n" + m); pts.clear(); if (type == CvType.CV_32SC3) { int[] buff = new int[3 * count]; m.get(0, 0, buff); for (int i = 0; i < count; i++) { pts.add(new Point3(buff[i * 3], buff[i * 3 + 1], buff[i * 3 + 2])); } } else if (type == CvType.CV_32FC3) { float[] buff = new float[3 * count]; m.get(0, 0, buff); for (int i = 0; i < count; i++) { pts.add(new Point3(buff[i * 3], buff[i * 3 + 1], buff[i * 3 + 2])); } } else if (type == CvType.CV_64FC3) { double[] buff = new double[3 * count]; m.get(0, 0, buff); for (int i = 0; i < count; i++) { pts.add(new Point3(buff[i * 3], buff[i * 3 + 1], buff[i * 3 + 2])); } } else { throw new java.lang.IllegalArgumentException( "Input Mat should be of CV_32SC3, CV_32FC3 or CV_64FC3 type\n" + m); } } public static Mat vector_Mat_to_Mat(List<Mat> mats) { Mat res; int count = (mats != null) ? mats.size() : 0; if (count > 0) { res = new Mat(count, 1, CvType.CV_32SC2); int[] buff = new int[count * 2]; for (int i = 0; i < count; i++) { long addr = mats.get(i).nativeObj; buff[i * 2] = (int) (addr >> 32); buff[i * 2 + 1] = (int) (addr & 0xffffffff); } res.put(0, 0, buff); } else { res = new Mat(); } return res; } public static void Mat_to_vector_Mat(Mat m, List<Mat> mats) { if (mats == null) throw new java.lang.IllegalArgumentException("mats == null"); int count = m.rows(); if (CvType.CV_32SC2 != m.type() || m.cols() != 1) throw new java.lang.IllegalArgumentException( "CvType.CV_32SC2 != m.type() || m.cols()!=1\n" + m); mats.clear(); int[] buff = new int[count * 2]; m.get(0, 0, buff); for (int i = 0; i < count; i++) { long addr = (((long) buff[i * 2]) << 32) | (((long) buff[i * 2 + 1]) & 0xffffffffL); mats.add(new Mat(addr)); } } public static Mat vector_float_to_Mat(List<Float> fs) { Mat res; int count = (fs != null) ? fs.size() : 0; if (count > 0) { res = new Mat(count, 1, CvType.CV_32FC1); float[] buff = new float[count]; for (int i = 0; i < count; i++) { float f = fs.get(i); buff[i] = f; } res.put(0, 0, buff); } else { res = new Mat(); } return res; } public static void Mat_to_vector_float(Mat m, List<Float> fs) { if (fs == null) throw new java.lang.IllegalArgumentException("fs == null"); int count = m.rows(); if (CvType.CV_32FC1 != m.type() || m.cols() != 1) throw new java.lang.IllegalArgumentException( "CvType.CV_32FC1 != m.type() || m.cols()!=1\n" + m); fs.clear(); float[] buff = new float[count]; m.get(0, 0, buff); for (int i = 0; i < count; i++) { fs.add(buff[i]); } } public static Mat vector_uchar_to_Mat(List<Byte> bs) { Mat res; int count = (bs != null) ? bs.size() : 0; if (count > 0) { res = new Mat(count, 1, CvType.CV_8UC1); byte[] buff = new byte[count]; for (int i = 0; i < count; i++) { byte b = bs.get(i); buff[i] = b; } res.put(0, 0, buff); } else { res = new Mat(); } return res; } public static void Mat_to_vector_uchar(Mat m, List<Byte> us) { if (us == null) throw new java.lang.IllegalArgumentException("Output List can't be null"); int count = m.rows(); if (CvType.CV_8UC1 != m.type() || m.cols() != 1) throw new java.lang.IllegalArgumentException( "CvType.CV_8UC1 != m.type() || m.cols()!=1\n" + m); us.clear(); byte[] buff = new byte[count]; m.get(0, 0, buff); for (int i = 0; i < count; i++) { us.add(buff[i]); } } public static Mat vector_char_to_Mat(List<Byte> bs) { Mat res; int count = (bs != null) ? bs.size() : 0; if (count > 0) { res = new Mat(count, 1, CvType.CV_8SC1); byte[] buff = new byte[count]; for (int i = 0; i < count; i++) { byte b = bs.get(i); buff[i] = b; } res.put(0, 0, buff); } else { res = new Mat(); } return res; } public static Mat vector_int_to_Mat(List<Integer> is) { Mat res; int count = (is != null) ? is.size() : 0; if (count > 0) { res = new Mat(count, 1, CvType.CV_32SC1); int[] buff = new int[count]; for (int i = 0; i < count; i++) { int v = is.get(i); buff[i] = v; } res.put(0, 0, buff); } else { res = new Mat(); } return res; } public static void Mat_to_vector_int(Mat m, List<Integer> is) { if (is == null) throw new java.lang.IllegalArgumentException("is == null"); int count = m.rows(); if (CvType.CV_32SC1 != m.type() || m.cols() != 1) throw new java.lang.IllegalArgumentException( "CvType.CV_32SC1 != m.type() || m.cols()!=1\n" + m); is.clear(); int[] buff = new int[count]; m.get(0, 0, buff); for (int i = 0; i < count; i++) { is.add(buff[i]); } } public static void Mat_to_vector_char(Mat m, List<Byte> bs) { if (bs == null) throw new java.lang.IllegalArgumentException("Output List can't be null"); int count = m.rows(); if (CvType.CV_8SC1 != m.type() || m.cols() != 1) throw new java.lang.IllegalArgumentException( "CvType.CV_8SC1 != m.type() || m.cols()!=1\n" + m); bs.clear(); byte[] buff = new byte[count]; m.get(0, 0, buff); for (int i = 0; i < count; i++) { bs.add(buff[i]); } } public static Mat vector_Rect_to_Mat(List<Rect> rs) { Mat res; int count = (rs != null) ? rs.size() : 0; if (count > 0) { res = new Mat(count, 1, CvType.CV_32SC4); int[] buff = new int[4 * count]; for (int i = 0; i < count; i++) { Rect r = rs.get(i); buff[4 * i] = r.x; buff[4 * i + 1] = r.y; buff[4 * i + 2] = r.width; buff[4 * i + 3] = r.height; } res.put(0, 0, buff); } else { res = new Mat(); } return res; } public static void Mat_to_vector_Rect(Mat m, List<Rect> rs) { if (rs == null) throw new java.lang.IllegalArgumentException("rs == null"); int count = m.rows(); if (CvType.CV_32SC4 != m.type() || m.cols() != 1) throw new java.lang.IllegalArgumentException( "CvType.CV_32SC4 != m.type() || m.rows()!=1\n" + m); rs.clear(); int[] buff = new int[4 * count]; m.get(0, 0, buff); for (int i = 0; i < count; i++) { rs.add(new Rect(buff[4 * i], buff[4 * i + 1], buff[4 * i + 2], buff[4 * i + 3])); } } public static Mat vector_KeyPoint_to_Mat(List<KeyPoint> kps) { Mat res; int count = (kps != null) ? kps.size() : 0; if (count > 0) { res = new Mat(count, 1, CvType.CV_64FC(7)); double[] buff = new double[count * 7]; for (int i = 0; i < count; i++) { KeyPoint kp = kps.get(i); buff[7 * i] = kp.pt.x; buff[7 * i + 1] = kp.pt.y; buff[7 * i + 2] = kp.size; buff[7 * i + 3] = kp.angle; buff[7 * i + 4] = kp.response; buff[7 * i + 5] = kp.octave; buff[7 * i + 6] = kp.class_id; } res.put(0, 0, buff); } else { res = new Mat(); } return res; } public static void Mat_to_vector_KeyPoint(Mat m, List<KeyPoint> kps) { if (kps == null) throw new java.lang.IllegalArgumentException("Output List can't be null"); int count = m.rows(); if (CvType.CV_64FC(7) != m.type() || m.cols() != 1) throw new java.lang.IllegalArgumentException( "CvType.CV_64FC(7) != m.type() || m.cols()!=1\n" + m); kps.clear(); double[] buff = new double[7 * count]; m.get(0, 0, buff); for (int i = 0; i < count; i++) { kps.add(new KeyPoint((float) buff[7 * i], (float) buff[7 * i + 1], (float) buff[7 * i + 2], (float) buff[7 * i + 3], (float) buff[7 * i + 4], (int) buff[7 * i + 5], (int) buff[7 * i + 6])); } } // vector_vector_Point public static Mat vector_vector_Point_to_Mat(List<MatOfPoint> pts, List<Mat> mats) { Mat res; int lCount = (pts != null) ? pts.size() : 0; if (lCount > 0) { for (MatOfPoint vpt : pts) mats.add(vpt); res = vector_Mat_to_Mat(mats); } else { res = new Mat(); } return res; } public static void Mat_to_vector_vector_Point(Mat m, List<MatOfPoint> pts) { if (pts == null) throw new java.lang.IllegalArgumentException("Output List can't be null"); if (m == null) throw new java.lang.IllegalArgumentException("Input Mat can't be null"); List<Mat> mats = new ArrayList<Mat>(m.rows()); Mat_to_vector_Mat(m, mats); for (Mat mi : mats) { MatOfPoint pt = new MatOfPoint(mi); pts.add(pt); } } // vector_vector_Point2f public static void Mat_to_vector_vector_Point2f(Mat m, List<MatOfPoint2f> pts) { if (pts == null) throw new java.lang.IllegalArgumentException("Output List can't be null"); if (m == null) throw new java.lang.IllegalArgumentException("Input Mat can't be null"); List<Mat> mats = new ArrayList<Mat>(m.rows()); Mat_to_vector_Mat(m, mats); for (Mat mi : mats) { MatOfPoint2f pt = new MatOfPoint2f(mi); pts.add(pt); } } // vector_vector_Point2f public static Mat vector_vector_Point2f_to_Mat(List<MatOfPoint2f> pts, List<Mat> mats) { Mat res; int lCount = (pts != null) ? pts.size() : 0; if (lCount > 0) { for (MatOfPoint2f vpt : pts) mats.add(vpt); res = vector_Mat_to_Mat(mats); } else { res = new Mat(); } return res; } // vector_vector_Point3f public static void Mat_to_vector_vector_Point3f(Mat m, List<MatOfPoint3f> pts) { if (pts == null) throw new java.lang.IllegalArgumentException("Output List can't be null"); if (m == null) throw new java.lang.IllegalArgumentException("Input Mat can't be null"); List<Mat> mats = new ArrayList<Mat>(m.rows()); Mat_to_vector_Mat(m, mats); for (Mat mi : mats) { MatOfPoint3f pt = new MatOfPoint3f(mi); pts.add(pt); } } // vector_vector_Point3f public static Mat vector_vector_Point3f_to_Mat(List<MatOfPoint3f> pts, List<Mat> mats) { Mat res; int lCount = (pts != null) ? pts.size() : 0; if (lCount > 0) { for (MatOfPoint3f vpt : pts) mats.add(vpt); res = vector_Mat_to_Mat(mats); } else { res = new Mat(); } return res; } // vector_vector_KeyPoint public static Mat vector_vector_KeyPoint_to_Mat(List<MatOfKeyPoint> kps, List<Mat> mats) { Mat res; int lCount = (kps != null) ? kps.size() : 0; if (lCount > 0) { for (MatOfKeyPoint vkp : kps) mats.add(vkp); res = vector_Mat_to_Mat(mats); } else { res = new Mat(); } return res; } public static void Mat_to_vector_vector_KeyPoint(Mat m, List<MatOfKeyPoint> kps) { if (kps == null) throw new java.lang.IllegalArgumentException("Output List can't be null"); if (m == null) throw new java.lang.IllegalArgumentException("Input Mat can't be null"); List<Mat> mats = new ArrayList<Mat>(m.rows()); Mat_to_vector_Mat(m, mats); for (Mat mi : mats) { MatOfKeyPoint vkp = new MatOfKeyPoint(mi); kps.add(vkp); } } public static Mat vector_double_to_Mat(List<Double> ds) { Mat res; int count = (ds != null) ? ds.size() : 0; if (count > 0) { res = new Mat(count, 1, CvType.CV_64FC1); double[] buff = new double[count]; for (int i = 0; i < count; i++) { double v = ds.get(i); buff[i] = v; } res.put(0, 0, buff); } else { res = new Mat(); } return res; } public static void Mat_to_vector_double(Mat m, List<Double> ds) { if (ds == null) throw new java.lang.IllegalArgumentException("ds == null"); int count = m.rows(); if (CvType.CV_64FC1 != m.type() || m.cols() != 1) throw new java.lang.IllegalArgumentException( "CvType.CV_64FC1 != m.type() || m.cols()!=1\n" + m); ds.clear(); double[] buff = new double[count]; m.get(0, 0, buff); for (int i = 0; i < count; i++) { ds.add(buff[i]); } } public static Mat vector_DMatch_to_Mat(List<DMatch> matches) { Mat res; int count = (matches != null) ? matches.size() : 0; if (count > 0) { res = new Mat(count, 1, CvType.CV_64FC4); double[] buff = new double[count * 4]; for (int i = 0; i < count; i++) { DMatch m = matches.get(i); buff[4 * i] = m.queryIdx; buff[4 * i + 1] = m.trainIdx; buff[4 * i + 2] = m.imgIdx; buff[4 * i + 3] = m.distance; } res.put(0, 0, buff); } else { res = new Mat(); } return res; } public static void Mat_to_vector_DMatch(Mat m, List<DMatch> matches) { if (matches == null) throw new java.lang.IllegalArgumentException("Output List can't be null"); int count = m.rows(); if (CvType.CV_64FC4 != m.type() || m.cols() != 1) throw new java.lang.IllegalArgumentException( "CvType.CV_64FC4 != m.type() || m.cols()!=1\n" + m); matches.clear(); double[] buff = new double[4 * count]; m.get(0, 0, buff); for (int i = 0; i < count; i++) { matches.add(new DMatch((int) buff[4 * i], (int) buff[4 * i + 1], (int) buff[4 * i + 2], (float) buff[4 * i + 3])); } } // vector_vector_DMatch public static Mat vector_vector_DMatch_to_Mat(List<MatOfDMatch> lvdm, List<Mat> mats) { Mat res; int lCount = (lvdm != null) ? lvdm.size() : 0; if (lCount > 0) { for (MatOfDMatch vdm : lvdm) mats.add(vdm); res = vector_Mat_to_Mat(mats); } else { res = new Mat(); } return res; } public static void Mat_to_vector_vector_DMatch(Mat m, List<MatOfDMatch> lvdm) { if (lvdm == null) throw new java.lang.IllegalArgumentException("Output List can't be null"); if (m == null) throw new java.lang.IllegalArgumentException("Input Mat can't be null"); List<Mat> mats = new ArrayList<Mat>(m.rows()); Mat_to_vector_Mat(m, mats); lvdm.clear(); for (Mat mi : mats) { MatOfDMatch vdm = new MatOfDMatch(mi); lvdm.add(vdm); } } // vector_vector_char public static Mat vector_vector_char_to_Mat(List<MatOfByte> lvb, List<Mat> mats) { Mat res; int lCount = (lvb != null) ? lvb.size() : 0; if (lCount > 0) { for (MatOfByte vb : lvb) mats.add(vb); res = vector_Mat_to_Mat(mats); } else { res = new Mat(); } return res; } public static void Mat_to_vector_vector_char(Mat m, List<List<Byte>> llb) { if (llb == null) throw new java.lang.IllegalArgumentException("Output List can't be null"); if (m == null) throw new java.lang.IllegalArgumentException("Input Mat can't be null"); List<Mat> mats = new ArrayList<Mat>(m.rows()); Mat_to_vector_Mat(m, mats); for (Mat mi : mats) { List<Byte> lb = new ArrayList<Byte>(); Mat_to_vector_char(mi, lb); llb.add(lb); } } }