00001 /*********************************************************************** 00002 * Software License Agreement (BSD License) 00003 * 00004 * Copyright 2008-2009 Marius Muja (mariusm@cs.ubc.ca). All rights reserved. 00005 * Copyright 2008-2009 David G. Lowe (lowe@cs.ubc.ca). All rights reserved. 00006 * 00007 * THE BSD LICENSE 00008 * 00009 * Redistribution and use in source and binary forms, with or without 00010 * modification, are permitted provided that the following conditions 00011 * are met: 00012 * 00013 * 1. Redistributions of source code must retain the above copyright 00014 * notice, this list of conditions and the following disclaimer. 00015 * 2. Redistributions in binary form must reproduce the above copyright 00016 * notice, this list of conditions and the following disclaimer in the 00017 * documentation and/or other materials provided with the distribution. 00018 * 00019 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 00020 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 00021 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 00022 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 00023 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 00024 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 00025 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 00026 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 00027 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 00028 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 00029 *************************************************************************/ 00030 00031 #ifndef _OPENCV_LINEARSEARCH_H_ 00032 #define _OPENCV_LINEARSEARCH_H_ 00033 00034 #include "opencv2/flann/general.h" 00035 #include "opencv2/flann/nn_index.h" 00036 00037 00038 namespace cvflann 00039 { 00040 00041 struct CV_EXPORTS LinearIndexParams : public IndexParams { 00042 LinearIndexParams() : IndexParams(FLANN_INDEX_LINEAR) {}; 00043 00044 void print() const 00045 { 00046 logger().info("Index type: %d\n",(int)algorithm); 00047 } 00048 }; 00049 00050 00051 template <typename ELEM_TYPE, typename DIST_TYPE = typename DistType<ELEM_TYPE>::type > 00052 class LinearIndex : public NNIndex<ELEM_TYPE> 00053 { 00054 const Matrix<ELEM_TYPE> dataset; 00055 const LinearIndexParams& index_params; 00056 00057 LinearIndex(const LinearIndex&); 00058 LinearIndex& operator=(const LinearIndex&); 00059 00060 public: 00061 00062 LinearIndex(const Matrix<ELEM_TYPE>& inputData, const LinearIndexParams& params = LinearIndexParams() ) : 00063 dataset(inputData), index_params(params) 00064 { 00065 } 00066 00067 flann_algorithm_t getType() const 00068 { 00069 return FLANN_INDEX_LINEAR; 00070 } 00071 00072 00073 size_t size() const 00074 { 00075 return dataset.rows; 00076 } 00077 00078 size_t veclen() const 00079 { 00080 return dataset.cols; 00081 } 00082 00083 00084 int usedMemory() const 00085 { 00086 return 0; 00087 } 00088 00089 void buildIndex() 00090 { 00091 /* nothing to do here for linear search */ 00092 } 00093 00094 void saveIndex(FILE*) 00095 { 00096 /* nothing to do here for linear search */ 00097 } 00098 00099 00100 void loadIndex(FILE*) 00101 { 00102 /* nothing to do here for linear search */ 00103 } 00104 00105 void findNeighbors(ResultSet<ELEM_TYPE>& resultSet, const ELEM_TYPE*, const SearchParams&) 00106 { 00107 for (size_t i=0;i<dataset.rows;++i) { 00108 resultSet.addPoint(dataset[i],(int)i); 00109 } 00110 } 00111 00112 const IndexParams* getParameters() const 00113 { 00114 return &index_params; 00115 } 00116 00117 }; 00118 00119 } // namespace cvflann 00120 00121 #endif // _OPENCV_LINEARSEARCH_H_