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(LINEAR) {}; 00043 00044 flann_algorithm_t getIndexType() const { return algorithm; } 00045 00046 void print() const 00047 { 00048 logger().info("Index type: %d\n",(int)algorithm); 00049 } 00050 }; 00051 00052 00053 template <typename ELEM_TYPE, typename DIST_TYPE = typename DistType<ELEM_TYPE>::type > 00054 class LinearIndex : public NNIndex<ELEM_TYPE> 00055 { 00056 const Matrix<ELEM_TYPE> dataset; 00057 const LinearIndexParams& index_params; 00058 00059 public: 00060 00061 LinearIndex(const Matrix<ELEM_TYPE>& inputData, const LinearIndexParams& params = LinearIndexParams() ) : 00062 dataset(inputData), index_params(params) 00063 { 00064 } 00065 00066 flann_algorithm_t getType() const 00067 { 00068 return LINEAR; 00069 } 00070 00071 00072 size_t size() const 00073 { 00074 return dataset.rows; 00075 } 00076 00077 size_t veclen() const 00078 { 00079 return dataset.cols; 00080 } 00081 00082 00083 int usedMemory() const 00084 { 00085 return 0; 00086 } 00087 00088 void buildIndex() 00089 { 00090 /* nothing to do here for linear search */ 00091 } 00092 00093 void saveIndex(FILE*) 00094 { 00095 /* nothing to do here for linear search */ 00096 } 00097 00098 00099 void loadIndex(FILE*) 00100 { 00101 /* nothing to do here for linear search */ 00102 } 00103 00104 void findNeighbors(ResultSet<ELEM_TYPE>& resultSet, const ELEM_TYPE*, const SearchParams&) 00105 { 00106 for (size_t i=0;i<dataset.rows;++i) { 00107 resultSet.addPoint(dataset[i],(int)i); 00108 } 00109 } 00110 00111 const IndexParams* getParameters() const 00112 { 00113 return &index_params; 00114 } 00115 00116 }; 00117 00118 } // namespace cvflann 00119 00120 #endif // _OPENCV_LINEARSEARCH_H_