matrix.h 3.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121
  1. /***********************************************************************
  2. * Software License Agreement (BSD License)
  3. *
  4. * Copyright 2008-2009 Marius Muja (mariusm@cs.ubc.ca). All rights reserved.
  5. * Copyright 2008-2009 David G. Lowe (lowe@cs.ubc.ca). All rights reserved.
  6. *
  7. * THE BSD LICENSE
  8. *
  9. * Redistribution and use in source and binary forms, with or without
  10. * modification, are permitted provided that the following conditions
  11. * are met:
  12. *
  13. * 1. Redistributions of source code must retain the above copyright
  14. * notice, this list of conditions and the following disclaimer.
  15. * 2. Redistributions in binary form must reproduce the above copyright
  16. * notice, this list of conditions and the following disclaimer in the
  17. * documentation and/or other materials provided with the distribution.
  18. *
  19. * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  20. * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  21. * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  22. * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  23. * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  24. * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  25. * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  26. * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  27. * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  28. * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  29. *************************************************************************/
  30. #ifndef OPENCV_FLANN_DATASET_H_
  31. #define OPENCV_FLANN_DATASET_H_
  32. //! @cond IGNORED
  33. #include <stdio.h>
  34. #include "opencv2/core/cvdef.h"
  35. #include "opencv2/flann/defines.h"
  36. namespace cvflann
  37. {
  38. /**
  39. * Class that implements a simple rectangular matrix stored in a memory buffer and
  40. * provides convenient matrix-like access using the [] operators.
  41. */
  42. template <typename T>
  43. class Matrix
  44. {
  45. public:
  46. typedef T type;
  47. size_t rows;
  48. size_t cols;
  49. size_t stride;
  50. T* data;
  51. Matrix() : rows(0), cols(0), stride(0), data(NULL)
  52. {
  53. }
  54. Matrix(T* data_, size_t rows_, size_t cols_, size_t stride_ = 0) :
  55. rows(rows_), cols(cols_), stride(stride_), data(data_)
  56. {
  57. if (stride==0) stride = cols;
  58. }
  59. /**
  60. * Convenience function for deallocating the storage data.
  61. */
  62. CV_DEPRECATED void free()
  63. {
  64. fprintf(stderr, "The cvflann::Matrix<T>::free() method is deprecated "
  65. "and it does not do any memory deallocation any more. You are"
  66. "responsible for deallocating the matrix memory (by doing"
  67. "'delete[] matrix.data' for example)");
  68. }
  69. /**
  70. * Operator that return a (pointer to a) row of the data.
  71. */
  72. T* operator[](size_t index) const
  73. {
  74. return data+index*stride;
  75. }
  76. };
  77. class UntypedMatrix
  78. {
  79. public:
  80. size_t rows;
  81. size_t cols;
  82. void* data;
  83. flann_datatype_t type;
  84. UntypedMatrix(void* data_, long rows_, long cols_) :
  85. rows(rows_), cols(cols_), data(data_)
  86. {
  87. }
  88. ~UntypedMatrix()
  89. {
  90. }
  91. template<typename T>
  92. Matrix<T> as()
  93. {
  94. return Matrix<T>((T*)data, rows, cols);
  95. }
  96. };
  97. }
  98. //! @endcond
  99. #endif //OPENCV_FLANN_DATASET_H_