Operations.cpp 10 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340
  1. #include "Operations.h"
  2. using namespace OBJREC;
  3. using namespace std;
  4. using namespace NICE;
  5. Operation::Operation()
  6. {
  7. values = NULL;
  8. maxtypes = 1000;
  9. }
  10. void Operation::set ( int _x1, int _y1, int _z1, int _x2, int _y2, int _z2, int _channel1, int _channel2, ValueAccess *_values )
  11. {
  12. x1 = _x1;
  13. y1 = _y1;
  14. z1 = _z1;
  15. x2 = _x2;
  16. y2 = _y2;
  17. z2 = _z2;
  18. channel1 = _channel1;
  19. channel2 = _channel2;
  20. values = _values;
  21. }
  22. void Operation::setContext ( bool _context )
  23. {
  24. context = _context;
  25. }
  26. bool Operation::getContext()
  27. {
  28. return context;
  29. }
  30. void Operation::setFeatType ( int _featType )
  31. {
  32. featType = _featType;
  33. }
  34. int Operation::getFeatType()
  35. {
  36. return featType;
  37. }
  38. void Operation::getXYZ ( const Features &feats, int &xsize, int &ysize, int &zsize )
  39. {
  40. xsize = feats.feats->width();
  41. ysize = feats.feats->height();
  42. zsize = feats.feats->depth();
  43. }
  44. void Operation::store ( std::ostream & os )
  45. {
  46. os << x1 << " " << x2 << " " << y1 << " " << y2 << " " << z1 << " " << z2 << " " << channel1 << " " << channel2 << " " << featType << std::endl;
  47. if ( values == NULL )
  48. os << -1 << std::endl;
  49. else
  50. os << values->getType() << std::endl;
  51. }
  52. void Operation::restore ( std::istream &is )
  53. {
  54. is >> x1;
  55. is >> x2;
  56. is >> y1;
  57. is >> y2;
  58. is >> z1;
  59. is >> z2;
  60. is >> channel1;
  61. is >> channel2;
  62. is >> featType;
  63. int tmp;
  64. is >> tmp;
  65. if ( tmp >= 0 )
  66. {
  67. if ( tmp == RAWFEAT )
  68. {
  69. values = new MCImageAccess();
  70. }
  71. else if ( tmp == CONTEXT )
  72. {
  73. values = new ClassificationResultAccess();
  74. }
  75. else
  76. {
  77. throw ( "no valid ValueAccess" );
  78. }
  79. }
  80. else
  81. {
  82. values = NULL;
  83. }
  84. }
  85. std::string Operation::writeInfos()
  86. {
  87. std::stringstream ss;
  88. ss << " x1: " << x1 << " y1: " << y1 << " z1: " << z1 << " x2: " << x2 << " y2: " << y2 << " z2: " << z2 << " c1: " << channel1 << " c2: " << channel2;
  89. return ss.str();
  90. }
  91. double RegionFeat::getVal ( const Features &feats, const int &x, const int &y, const int &z )
  92. {
  93. return (*feats.rProbs)[(*feats.feats)(x,y,z,channel1)][channel2];
  94. }
  95. double Minus::getVal ( const Features &feats, const int &x, const int &y, const int &z )
  96. {
  97. int xsize, ysize, zsize;
  98. getXYZ ( feats, xsize, ysize, zsize );
  99. double v1 = values->getVal ( feats, BOUND ( x + x1, 0, xsize - 1 ), BOUND ( y + y1, 0, ysize - 1 ), BOUND ( z + z1, 0, zsize - 1 ), channel1 );
  100. double v2 = values->getVal ( feats, BOUND ( x + x2, 0, xsize - 1 ), BOUND ( y + y2, 0, ysize - 1 ), BOUND ( z + z2, 0, zsize - 1 ), channel2 );
  101. return v1 -v2;
  102. }
  103. double MinusAbs::getVal ( const Features &feats, const int &x, const int &y, const int &z )
  104. {
  105. int xsize, ysize, zsize;
  106. getXYZ ( feats, xsize, ysize, zsize );
  107. double v1 = values->getVal ( feats, BOUND ( x + x1, 0, xsize - 1 ), BOUND ( y + y1, 0, ysize - 1 ), BOUND ( z + z1, 0, zsize - 1 ), channel1 );
  108. double v2 = values->getVal ( feats, BOUND ( x + x2, 0, xsize - 1 ), BOUND ( y + y2, 0, ysize - 1 ), BOUND ( z + z2, 0, zsize - 1 ), channel2 );
  109. return abs ( v1 -v2 );
  110. }
  111. double Addition::getVal ( const Features &feats, const int &x, const int &y, const int &z )
  112. {
  113. int xsize, ysize, zsize;
  114. getXYZ ( feats, xsize, ysize, zsize );
  115. double v1 = values->getVal ( feats, BOUND ( x + x1, 0, xsize - 1 ), BOUND ( y + y1, 0, ysize - 1 ), BOUND ( z + z1, 0, zsize - 1 ), channel1 );
  116. double v2 = values->getVal ( feats, BOUND ( x + x2, 0, xsize - 1 ), BOUND ( y + y2, 0, ysize - 1 ), BOUND ( z + z2, 0, zsize - 1 ), channel2 );
  117. return v1 + v2;
  118. }
  119. double Only1::getVal ( const Features &feats, const int &x, const int &y, const int &z )
  120. {
  121. int xsize, ysize, zsize;
  122. getXYZ ( feats, xsize, ysize, zsize );
  123. double v1 = values->getVal ( feats, BOUND ( x + x1, 0, xsize - 1 ), BOUND ( y + y1, 0, ysize - 1 ), BOUND ( z + z1, 0, zsize - 1 ), channel1 );
  124. return v1;
  125. }
  126. double RelativeXPosition::getVal ( const Features &feats, const int &x, const int &y, const int &z )
  127. {
  128. int xsize, ysize, zsize;
  129. getXYZ ( feats, xsize, ysize, zsize );
  130. return ( double ) x / ( double ) xsize;
  131. }
  132. double RelativeYPosition::getVal ( const Features &feats, const int &x, const int &y, const int &z )
  133. {
  134. int xsize, ysize, zsize;
  135. getXYZ ( feats, xsize, ysize, zsize );
  136. return ( double ) y / ( double ) ysize;
  137. }
  138. double RelativeZPosition::getVal ( const Features &feats, const int &x, const int &y, const int &z )
  139. {
  140. int xsize, ysize, zsize;
  141. getXYZ ( feats, xsize, ysize, zsize );
  142. return ( double ) z / ( double ) zsize;
  143. }
  144. double IntegralOps::getVal ( const Features &feats, const int &x, const int &y, const int &z )
  145. {
  146. return feats.feats->getIntegralValue(x + x1, y + y1, z + z1, x + x2, y + y2, z + z2, channel1);
  147. }
  148. double GlobalFeats::getVal ( const Features &feats, const int &x, const int &y, const int &z )
  149. {
  150. int xsize, ysize, zsize;
  151. getXYZ ( feats, xsize, ysize, zsize );
  152. return feats.feats->getIntegralValue( 0, 0, 0, xsize - 1, ysize - 1, zsize - 1, channel1 );
  153. }
  154. double IntegralCenteredOps::getVal ( const Features &feats, const int &x, const int &y, const int &z )
  155. {
  156. return feats.feats->getIntegralValue(x - x1, y - y1, z - z1, x + x1, y + y1, z + z1, channel1);
  157. }
  158. double BiIntegralCenteredOps::getVal ( const Features &feats, const int &x, const int &y, const int &z )
  159. {
  160. return feats.feats->getIntegralValue(x - x1, y - y1, z - z1, x + x1, y + y1, z + z1, channel1 ) - feats.feats->getIntegralValue(x - x2, y - y2, z - z2, x + x2, y + y2, z + z2, channel1);
  161. }
  162. double HaarHorizontal::getVal ( const Features &feats, const int &x, const int &y, const int &z )
  163. {
  164. int tlx = x - x1;
  165. int tly = y - y1;
  166. int tlz = z - z1;
  167. int lrx = x + x1;
  168. int lry = y + y1;
  169. int lrz = z + z1;
  170. return feats.feats->getIntegralValue(tlx, tly, tlz, lrx, y, lrz, channel1 ) - feats.feats->getIntegralValue(tlx, y, tlz, lrx, lry, lrz, channel1);
  171. }
  172. double HaarVertical::getVal ( const Features &feats, const int &x, const int &y, const int &z )
  173. {
  174. int tlx = x - x1;
  175. int tly = y - y1;
  176. int tlz = z - z1;
  177. int lrx = x + x1;
  178. int lry = y + y1;
  179. int lrz = z + z1;
  180. return feats.feats->getIntegralValue(tlx, tly, tlz, x, lry, lrz, channel1) - feats.feats->getIntegralValue(x, tly, tlz, lrx, lry, lrz, channel1);
  181. }
  182. double HaarStacked::getVal ( const Features &feats, const int &x, const int &y, const int &z )
  183. {
  184. int tlx = x - x1;
  185. int tly = y - y1;
  186. int tlz = z - z1;
  187. int lrx = x + x1;
  188. int lry = y + y1;
  189. int lrz = z + z1;
  190. return feats.feats->getIntegralValue(tlx, tly, tlz, lrx, lry, z, channel1) - feats.feats->getIntegralValue(tlx, tly, z, lrx, lry, lrz, channel1);
  191. }
  192. double HaarDiagXY::getVal ( const Features &feats, const int &x, const int &y, const int &z )
  193. {
  194. int tlx = x - x1;
  195. int tly = y - y1;
  196. int tlz = z - z1;
  197. int lrx = x + x1;
  198. int lry = y + y1;
  199. int lrz = z + z1;
  200. return feats.feats->getIntegralValue(tlx, tly, tlz, x, y, lrz, channel1) + feats.feats->getIntegralValue(x, y, tlz, lrx, lry, lrz, channel1) - feats.feats->getIntegralValue(tlx, y, tlz, x, lry, lrz, channel1) - feats.feats->getIntegralValue(x, tly, tlz, lrx, y, lrz, channel1);
  201. }
  202. double HaarDiagXZ::getVal ( const Features &feats, const int &x, const int &y, const int &z )
  203. {
  204. int tlx = x - x1;
  205. int tly = y - y1;
  206. int tlz = z - z1;
  207. int lrx = x + x1;
  208. int lry = y + y1;
  209. int lrz = z + z1;
  210. return feats.feats->getIntegralValue(tlx, tly, tlz, x, lry, z, channel1) + feats.feats->getIntegralValue(x, tly, z, lrx, lry, lrz, channel1) - feats.feats->getIntegralValue(tlx, tly, z, x, lry, lrz, channel1) - feats.feats->getIntegralValue(x, tly, tlz, lrx, lry, z, channel1);
  211. }
  212. double HaarDiagYZ::getVal ( const Features &feats, const int &x, const int &y, const int &z )
  213. {
  214. int tlx = x - x1;
  215. int tly = y - y1;
  216. int tlz = z - z1;
  217. int lrx = x + x1;
  218. int lry = y + y1;
  219. int lrz = z + z1;
  220. return feats.feats->getIntegralValue(tlx, tly, tlz, lrx, y, z, channel1) + feats.feats->getIntegralValue(tlx, y, z, lrx, lry, lrz, channel1) - feats.feats->getIntegralValue(tlx, tly, z, lrx, y, lrz, channel1) - feats.feats->getIntegralValue(tlx, y, tlz, lrx, lry, z, channel1);
  221. }
  222. double Haar3Horiz::getVal ( const Features &feats, const int &x, const int &y, const int &z )
  223. {
  224. int tlx = x - x2;
  225. int tly = y - y2;
  226. int tlz = z - z2;
  227. int mtly = y - y1;
  228. int mlry = y + y1;
  229. int lrx = x + x2;
  230. int lry = y + y2;
  231. int lrz = z + z2;
  232. return feats.feats->getIntegralValue(tlx, tly, tlz, lrx, mtly, lrz, channel1) - feats.feats->getIntegralValue(tlx, mtly, tlz, lrx, mlry, lrz, channel1) + feats.feats->getIntegralValue(tlx, mlry, tlz, lrx, lry, lrz, channel1);
  233. }
  234. double Haar3Vert::getVal ( const Features &feats, const int &x, const int &y, const int &z )
  235. {
  236. int tlx = x - x2;
  237. int tly = y - y2;
  238. int tlz = z - z2;
  239. int mtlx = x - x1;
  240. int mlrx = x + x1;
  241. int lrx = x + x2;
  242. int lry = y + y2;
  243. int lrz = z + z2;
  244. return feats.feats->getIntegralValue(tlx, tly, tlz, mtlx, lry, lrz, channel1) - feats.feats->getIntegralValue(mtlx, tly, tlz, mlrx, lry, lrz, channel1) + feats.feats->getIntegralValue(mlrx, tly, tlz, lrx, lry, lrz, channel1);
  245. }
  246. double Haar3Stack::getVal ( const Features &feats, const int &x, const int &y, const int &z )
  247. {
  248. int tlx = x - x2;
  249. int tly = y - y2;
  250. int tlz = z - z2;
  251. int mtlz = z - z1;
  252. int mlrz = z + z1;
  253. int lrx = x + x2;
  254. int lry = y + y2;
  255. int lrz = z + z2;
  256. return feats.feats->getIntegralValue(tlx, tly, tlz, lrx, lry, mtlz, channel1) - feats.feats->getIntegralValue(tlx, tly, mtlz, lrx, lry, mlrz, channel1) + feats.feats->getIntegralValue(tlx, tly, mlrz, lrx, lry, lrz, channel1);
  257. }
  258. void IntegralOps::set ( int _x1, int _y1, int _z1, int _x2, int _y2, int _z2, int _channel1, int _channel2, ValueAccess *_values )
  259. {
  260. x1 = std::min ( _x1, _x2 );
  261. y1 = std::min ( _y1, _y2 );
  262. z1 = std::min ( _z1, _z2 );
  263. x2 = std::max ( _x1, _x2 );
  264. y2 = std::max ( _y1, _y2 );
  265. z2 = std::max ( _z1, _z2 );
  266. channel1 = _channel1;
  267. channel2 = _channel2;
  268. values = _values;
  269. }
  270. void IntegralCenteredOps::set ( int _x1, int _y1, int _z1, int _x2, int _y2, int _z2, int _channel1, int _channel2, ValueAccess *_values )
  271. {
  272. x1 = abs ( _x1 );
  273. y1 = abs ( _y1 );
  274. z1 = abs ( _z1 );
  275. x2 = abs ( _x2 );
  276. y2 = abs ( _y2 );
  277. z2 = abs ( _z2 );
  278. channel1 = _channel1;
  279. channel2 = _channel2;
  280. values = _values;
  281. }
  282. void BiIntegralCenteredOps::set ( int _x1, int _y1, int _z1, int _x2, int _y2, int _z2, int _channel1, int _channel2, ValueAccess *_values )
  283. {
  284. x1 = std::min ( abs ( _x1 ), abs ( _x2 ) );
  285. y1 = std::min ( abs ( _y1 ), abs ( _y2 ) );
  286. z1 = std::min ( abs ( _z1 ), abs ( _z2 ) );
  287. x2 = std::max ( abs ( _x1 ), abs ( _x2 ) );
  288. y2 = std::max ( abs ( _y1 ), abs ( _y2 ) );
  289. z2 = std::max ( abs ( _z1 ), abs ( _z2 ) );
  290. channel1 = _channel1;
  291. channel2 = _channel2;
  292. values = _values;
  293. }