Home | History | Annotate | Download | only in cpp 
      1 #define LOG_TAG "org.opencv.utils.Converters"
      2 #include "common.h"
      3 #include "features2d_converters.hpp"
      4 
      5 using namespace cv;
      6 
      7 #define CHECK_MAT(cond) if(!(cond)){ LOGD("FAILED: " #cond); return; }
      8 
      9 
     10 //vector_KeyPoint
     11 void Mat_to_vector_KeyPoint(Mat& mat, std::vector<KeyPoint>& v_kp)
     12 {
     13     v_kp.clear();
     14     CHECK_MAT(mat.type()==CV_32FC(7) && mat.cols==1);
     15     for(int i=0; i<mat.rows; i++)
     16     {
     17         Vec<float, 7> v = mat.at< Vec<float, 7> >(i, 0);
     18         KeyPoint kp(v[0], v[1], v[2], v[3], v[4], (int)v[5], (int)v[6]);
     19         v_kp.push_back(kp);
     20     }
     21     return;
     22 }
     23 
     24 
     25 void vector_KeyPoint_to_Mat(std::vector<KeyPoint>& v_kp, Mat& mat)
     26 {
     27     int count = (int)v_kp.size();
     28     mat.create(count, 1, CV_32FC(7));
     29     for(int i=0; i<count; i++)
     30     {
     31         KeyPoint kp = v_kp[i];
     32         mat.at< Vec<float, 7> >(i, 0) = Vec<float, 7>(kp.pt.x, kp.pt.y, kp.size, kp.angle, kp.response, (float)kp.octave, (float)kp.class_id);
     33     }
     34 }
     35 
     36 //vector_DMatch
     37 void Mat_to_vector_DMatch(Mat& mat, std::vector<DMatch>& v_dm)
     38 {
     39     v_dm.clear();
     40     CHECK_MAT(mat.type()==CV_32FC4 && mat.cols==1);
     41     for(int i=0; i<mat.rows; i++)
     42     {
     43         Vec<float, 4> v = mat.at< Vec<float, 4> >(i, 0);
     44         DMatch dm((int)v[0], (int)v[1], (int)v[2], v[3]);
     45         v_dm.push_back(dm);
     46     }
     47     return;
     48 }
     49 
     50 
     51 void vector_DMatch_to_Mat(std::vector<DMatch>& v_dm, Mat& mat)
     52 {
     53     int count = (int)v_dm.size();
     54     mat.create(count, 1, CV_32FC4);
     55     for(int i=0; i<count; i++)
     56     {
     57         DMatch dm = v_dm[i];
     58         mat.at< Vec<float, 4> >(i, 0) = Vec<float, 4>((float)dm.queryIdx, (float)dm.trainIdx, (float)dm.imgIdx, dm.distance);
     59     }
     60 }
     61 
     62 void Mat_to_vector_vector_KeyPoint(Mat& mat, std::vector< std::vector< KeyPoint > >& vv_kp)
     63 {
     64     std::vector<Mat> vm;
     65     vm.reserve( mat.rows );
     66     Mat_to_vector_Mat(mat, vm);
     67     for(size_t i=0; i<vm.size(); i++)
     68     {
     69         std::vector<KeyPoint> vkp;
     70         Mat_to_vector_KeyPoint(vm[i], vkp);
     71         vv_kp.push_back(vkp);
     72     }
     73 }
     74 
     75 void vector_vector_KeyPoint_to_Mat(std::vector< std::vector< KeyPoint > >& vv_kp, Mat& mat)
     76 {
     77     std::vector<Mat> vm;
     78     vm.reserve( vv_kp.size() );
     79     for(size_t i=0; i<vv_kp.size(); i++)
     80     {
     81         Mat m;
     82         vector_KeyPoint_to_Mat(vv_kp[i], m);
     83         vm.push_back(m);
     84     }
     85     vector_Mat_to_Mat(vm, mat);
     86 }
     87 
     88 void Mat_to_vector_vector_DMatch(Mat& mat, std::vector< std::vector< DMatch > >& vv_dm)
     89 {
     90     std::vector<Mat> vm;
     91     vm.reserve( mat.rows );
     92     Mat_to_vector_Mat(mat, vm);
     93     for(size_t i=0; i<vm.size(); i++)
     94     {
     95         std::vector<DMatch> vdm;
     96         Mat_to_vector_DMatch(vm[i], vdm);
     97         vv_dm.push_back(vdm);
     98     }
     99 }
    100 
    101 void vector_vector_DMatch_to_Mat(std::vector< std::vector< DMatch > >& vv_dm, Mat& mat)
    102 {
    103     std::vector<Mat> vm;
    104     vm.reserve( vv_dm.size() );
    105     for(size_t i=0; i<vv_dm.size(); i++)
    106     {
    107         Mat m;
    108         vector_DMatch_to_Mat(vv_dm[i], m);
    109         vm.push_back(m);
    110     }
    111     vector_Mat_to_Mat(vm, mat);
    112 }
    113