operator_minus.hpp 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568
  1. // Copyright 2008-2016 Conrad Sanderson (http://conradsanderson.id.au)
  2. // Copyright 2008-2016 National ICT Australia (NICTA)
  3. //
  4. // Licensed under the Apache License, Version 2.0 (the "License");
  5. // you may not use this file except in compliance with the License.
  6. // You may obtain a copy of the License at
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. // ------------------------------------------------------------------------
  15. //! \addtogroup operator_minus
  16. //! @{
  17. //! unary -
  18. template<typename T1>
  19. arma_inline
  20. typename
  21. enable_if2< is_arma_type<T1>::value, const eOp<T1, eop_neg> >::result
  22. operator-
  23. (const T1& X)
  24. {
  25. arma_extra_debug_sigprint();
  26. return eOp<T1,eop_neg>(X);
  27. }
  28. //! Base - scalar
  29. template<typename T1>
  30. arma_inline
  31. typename
  32. enable_if2< is_arma_type<T1>::value, const eOp<T1, eop_scalar_minus_post> >::result
  33. operator-
  34. (
  35. const T1& X,
  36. const typename T1::elem_type k
  37. )
  38. {
  39. arma_extra_debug_sigprint();
  40. return eOp<T1, eop_scalar_minus_post>(X, k);
  41. }
  42. //! scalar - Base
  43. template<typename T1>
  44. arma_inline
  45. typename
  46. enable_if2< is_arma_type<T1>::value, const eOp<T1, eop_scalar_minus_pre> >::result
  47. operator-
  48. (
  49. const typename T1::elem_type k,
  50. const T1& X
  51. )
  52. {
  53. arma_extra_debug_sigprint();
  54. return eOp<T1, eop_scalar_minus_pre>(X, k);
  55. }
  56. //! complex scalar - non-complex Base
  57. template<typename T1>
  58. arma_inline
  59. typename
  60. enable_if2
  61. <
  62. (is_arma_type<T1>::value && is_cx<typename T1::elem_type>::no),
  63. const mtOp<typename std::complex<typename T1::pod_type>, T1, op_cx_scalar_minus_pre>
  64. >::result
  65. operator-
  66. (
  67. const std::complex<typename T1::pod_type>& k,
  68. const T1& X
  69. )
  70. {
  71. arma_extra_debug_sigprint();
  72. return mtOp<typename std::complex<typename T1::pod_type>, T1, op_cx_scalar_minus_pre>('j', X, k);
  73. }
  74. //! non-complex Base - complex scalar
  75. template<typename T1>
  76. arma_inline
  77. typename
  78. enable_if2
  79. <
  80. (is_arma_type<T1>::value && is_cx<typename T1::elem_type>::no),
  81. const mtOp<typename std::complex<typename T1::pod_type>, T1, op_cx_scalar_minus_post>
  82. >::result
  83. operator-
  84. (
  85. const T1& X,
  86. const std::complex<typename T1::pod_type>& k
  87. )
  88. {
  89. arma_extra_debug_sigprint();
  90. return mtOp<typename std::complex<typename T1::pod_type>, T1, op_cx_scalar_minus_post>('j', X, k);
  91. }
  92. //! subtraction of Base objects with same element type
  93. template<typename T1, typename T2>
  94. arma_inline
  95. typename
  96. enable_if2
  97. <
  98. is_arma_type<T1>::value && is_arma_type<T2>::value && is_same_type<typename T1::elem_type, typename T2::elem_type>::value,
  99. const eGlue<T1, T2, eglue_minus>
  100. >::result
  101. operator-
  102. (
  103. const T1& X,
  104. const T2& Y
  105. )
  106. {
  107. arma_extra_debug_sigprint();
  108. return eGlue<T1, T2, eglue_minus>(X, Y);
  109. }
  110. //! subtraction of Base objects with different element types
  111. template<typename T1, typename T2>
  112. inline
  113. typename
  114. enable_if2
  115. <
  116. (is_arma_type<T1>::value && is_arma_type<T2>::value && (is_same_type<typename T1::elem_type, typename T2::elem_type>::no)),
  117. const mtGlue<typename promote_type<typename T1::elem_type, typename T2::elem_type>::result, T1, T2, glue_mixed_minus>
  118. >::result
  119. operator-
  120. (
  121. const T1& X,
  122. const T2& Y
  123. )
  124. {
  125. arma_extra_debug_sigprint();
  126. typedef typename T1::elem_type eT1;
  127. typedef typename T2::elem_type eT2;
  128. typedef typename promote_type<eT1,eT2>::result out_eT;
  129. promote_type<eT1,eT2>::check();
  130. return mtGlue<out_eT, T1, T2, glue_mixed_minus>( X, Y );
  131. }
  132. //! unary "-" for sparse objects
  133. template<typename T1>
  134. inline
  135. typename
  136. enable_if2
  137. <
  138. is_arma_sparse_type<T1>::value && is_signed<typename T1::elem_type>::value,
  139. SpOp<T1,spop_scalar_times>
  140. >::result
  141. operator-
  142. (const T1& X)
  143. {
  144. arma_extra_debug_sigprint();
  145. typedef typename T1::elem_type eT;
  146. return SpOp<T1,spop_scalar_times>(X, eT(-1));
  147. }
  148. //! subtraction of two sparse objects
  149. template<typename T1, typename T2>
  150. inline
  151. typename
  152. enable_if2
  153. <
  154. (is_arma_sparse_type<T1>::value && is_arma_sparse_type<T2>::value && is_same_type<typename T1::elem_type, typename T2::elem_type>::value),
  155. const SpGlue<T1,T2,spglue_minus>
  156. >::result
  157. operator-
  158. (
  159. const T1& X,
  160. const T2& Y
  161. )
  162. {
  163. arma_extra_debug_sigprint();
  164. return SpGlue<T1,T2,spglue_minus>(X,Y);
  165. }
  166. //! subtraction of one sparse and one dense object
  167. template<typename T1, typename T2>
  168. inline
  169. typename
  170. enable_if2
  171. <
  172. (is_arma_sparse_type<T1>::value && is_arma_type<T2>::value && is_same_type<typename T1::elem_type, typename T2::elem_type>::value),
  173. Mat<typename T1::elem_type>
  174. >::result
  175. operator-
  176. (
  177. const T1& x,
  178. const T2& y
  179. )
  180. {
  181. arma_extra_debug_sigprint();
  182. const SpProxy<T1> pa(x);
  183. Mat<typename T1::elem_type> result(-y);
  184. arma_debug_assert_same_size( pa.get_n_rows(), pa.get_n_cols(), result.n_rows, result.n_cols, "subtraction" );
  185. typename SpProxy<T1>::const_iterator_type it = pa.begin();
  186. typename SpProxy<T1>::const_iterator_type it_end = pa.end();
  187. while(it != it_end)
  188. {
  189. result.at(it.row(), it.col()) += (*it);
  190. ++it;
  191. }
  192. return result;
  193. }
  194. //! subtraction of one dense and one sparse object
  195. template<typename T1, typename T2>
  196. inline
  197. typename
  198. enable_if2
  199. <
  200. (is_arma_type<T1>::value && is_arma_sparse_type<T2>::value && is_same_type<typename T1::elem_type, typename T2::elem_type>::value),
  201. Mat<typename T1::elem_type>
  202. >::result
  203. operator-
  204. (
  205. const T1& x,
  206. const T2& y
  207. )
  208. {
  209. arma_extra_debug_sigprint();
  210. Mat<typename T1::elem_type> result(x);
  211. const SpProxy<T2> pb(y);
  212. arma_debug_assert_same_size( result.n_rows, result.n_cols, pb.get_n_rows(), pb.get_n_cols(), "subtraction" );
  213. typename SpProxy<T2>::const_iterator_type it = pb.begin();
  214. typename SpProxy<T2>::const_iterator_type it_end = pb.end();
  215. while(it != it_end)
  216. {
  217. result.at(it.row(), it.col()) -= (*it);
  218. ++it;
  219. }
  220. return result;
  221. }
  222. //! subtraction of two sparse objects with different element types
  223. template<typename T1, typename T2>
  224. inline
  225. typename
  226. enable_if2
  227. <
  228. (is_arma_sparse_type<T1>::value && is_arma_sparse_type<T2>::value && is_same_type<typename T1::elem_type, typename T2::elem_type>::no),
  229. const mtSpGlue< typename promote_type<typename T1::elem_type, typename T2::elem_type>::result, T1, T2, spglue_minus_mixed >
  230. >::result
  231. operator-
  232. (
  233. const T1& X,
  234. const T2& Y
  235. )
  236. {
  237. arma_extra_debug_sigprint();
  238. typedef typename T1::elem_type eT1;
  239. typedef typename T2::elem_type eT2;
  240. typedef typename promote_type<eT1,eT2>::result out_eT;
  241. promote_type<eT1,eT2>::check();
  242. return mtSpGlue<out_eT, T1, T2, spglue_minus_mixed>( X, Y );
  243. }
  244. //! subtraction of sparse and non-sparse objects with different element types
  245. template<typename T1, typename T2>
  246. inline
  247. typename
  248. enable_if2
  249. <
  250. (is_arma_sparse_type<T1>::value && is_arma_type<T2>::value && is_same_type<typename T1::elem_type, typename T2::elem_type>::no),
  251. Mat< typename promote_type<typename T1::elem_type, typename T2::elem_type>::result >
  252. >::result
  253. operator-
  254. (
  255. const T1& x,
  256. const T2& y
  257. )
  258. {
  259. arma_extra_debug_sigprint();
  260. Mat< typename promote_type<typename T1::elem_type, typename T2::elem_type>::result > out;
  261. spglue_minus_mixed::sparse_minus_dense(out, x, y);
  262. return out;
  263. }
  264. //! subtraction of sparse and non-sparse objects with different element types
  265. template<typename T1, typename T2>
  266. inline
  267. typename
  268. enable_if2
  269. <
  270. (is_arma_type<T1>::value && is_arma_sparse_type<T2>::value && is_same_type<typename T1::elem_type, typename T2::elem_type>::no),
  271. Mat< typename promote_type<typename T1::elem_type, typename T2::elem_type>::result >
  272. >::result
  273. operator-
  274. (
  275. const T1& x,
  276. const T2& y
  277. )
  278. {
  279. arma_extra_debug_sigprint();
  280. Mat< typename promote_type<typename T1::elem_type, typename T2::elem_type>::result > out;
  281. spglue_minus_mixed::dense_minus_sparse(out, x, y);
  282. return out;
  283. }
  284. //! sparse - scalar
  285. template<typename T1>
  286. arma_inline
  287. typename
  288. enable_if2< is_arma_sparse_type<T1>::value, const SpToDOp<T1, op_sp_minus_post> >::result
  289. operator-
  290. (
  291. const T1& X,
  292. const typename T1::elem_type k
  293. )
  294. {
  295. arma_extra_debug_sigprint();
  296. return SpToDOp<T1, op_sp_minus_post>(X, k);
  297. }
  298. //! scalar - sparse
  299. template<typename T1>
  300. arma_inline
  301. typename
  302. enable_if2< is_arma_sparse_type<T1>::value, const SpToDOp<T1, op_sp_minus_pre> >::result
  303. operator-
  304. (
  305. const typename T1::elem_type k,
  306. const T1& X
  307. )
  308. {
  309. arma_extra_debug_sigprint();
  310. return SpToDOp<T1, op_sp_minus_pre>(X, k);
  311. }
  312. // TODO: this is an uncommon use case; remove?
  313. //! multiple applications of add/subtract scalars can be condensed
  314. template<typename T1, typename op_type>
  315. inline
  316. typename
  317. enable_if2
  318. <
  319. (is_arma_sparse_type<T1>::value &&
  320. (is_same_type<op_type, op_sp_plus>::value ||
  321. is_same_type<op_type, op_sp_minus_post>::value)),
  322. const SpToDOp<T1, op_sp_minus_post>
  323. >::result
  324. operator-
  325. (
  326. const SpToDOp<T1, op_type>& x,
  327. const typename T1::elem_type k
  328. )
  329. {
  330. arma_extra_debug_sigprint();
  331. const typename T1::elem_type aux = (is_same_type<op_type, op_sp_plus>::value) ? -x.aux : x.aux;
  332. return SpToDOp<T1, op_sp_minus_post>(x.m, aux + k);
  333. }
  334. // TODO: this is an uncommon use case; remove?
  335. //! multiple applications of add/subtract scalars can be condensed
  336. template<typename T1, typename op_type>
  337. inline
  338. typename
  339. enable_if2
  340. <
  341. (is_arma_sparse_type<T1>::value &&
  342. (is_same_type<op_type, op_sp_plus>::value ||
  343. is_same_type<op_type, op_sp_minus_post>::value)),
  344. const SpToDOp<T1, op_sp_minus_pre>
  345. >::result
  346. operator-
  347. (
  348. const typename T1::elem_type k,
  349. const SpToDOp<T1, op_type>& x
  350. )
  351. {
  352. arma_extra_debug_sigprint();
  353. const typename T1::elem_type aux = (is_same_type<op_type, op_sp_plus>::value) ? -x.aux : x.aux;
  354. return SpToDOp<T1, op_sp_minus_pre>(x.m, k + aux);
  355. }
  356. // TODO: this is an uncommon use case; remove?
  357. //! multiple applications of add/subtract scalars can be condensed
  358. template<typename T1, typename op_type>
  359. inline
  360. typename
  361. enable_if2
  362. <
  363. (is_arma_sparse_type<T1>::value &&
  364. is_same_type<op_type, op_sp_minus_pre>::value),
  365. const SpToDOp<T1, op_sp_minus_pre>
  366. >::result
  367. operator-
  368. (
  369. const SpToDOp<T1, op_type>& x,
  370. const typename T1::elem_type k
  371. )
  372. {
  373. arma_extra_debug_sigprint();
  374. return SpToDOp<T1, op_sp_minus_pre>(x.m, x.aux - k);
  375. }
  376. // TODO: this is an uncommon use case; remove?
  377. //! multiple applications of add/subtract scalars can be condensed
  378. template<typename T1, typename op_type>
  379. inline
  380. typename
  381. enable_if2
  382. <
  383. (is_arma_sparse_type<T1>::value &&
  384. is_same_type<op_type, op_sp_minus_pre>::value),
  385. const SpToDOp<T1, op_sp_plus>
  386. >::result
  387. operator-
  388. (
  389. const typename T1::elem_type k,
  390. const SpToDOp<T1, op_type>& x
  391. )
  392. {
  393. arma_extra_debug_sigprint();
  394. return SpToDOp<T1, op_sp_plus>(x.m, k - x.aux);
  395. }
  396. template<typename parent, unsigned int mode, typename T2>
  397. arma_inline
  398. Mat<typename parent::elem_type>
  399. operator-
  400. (
  401. const subview_each1<parent,mode>& X,
  402. const Base<typename parent::elem_type,T2>& Y
  403. )
  404. {
  405. arma_extra_debug_sigprint();
  406. return subview_each1_aux::operator_minus(X, Y.get_ref());
  407. }
  408. template<typename T1, typename parent, unsigned int mode>
  409. arma_inline
  410. Mat<typename parent::elem_type>
  411. operator-
  412. (
  413. const Base<typename parent::elem_type,T1>& X,
  414. const subview_each1<parent,mode>& Y
  415. )
  416. {
  417. arma_extra_debug_sigprint();
  418. return subview_each1_aux::operator_minus(X.get_ref(), Y);
  419. }
  420. template<typename parent, unsigned int mode, typename TB, typename T2>
  421. arma_inline
  422. Mat<typename parent::elem_type>
  423. operator-
  424. (
  425. const subview_each2<parent,mode,TB>& X,
  426. const Base<typename parent::elem_type,T2>& Y
  427. )
  428. {
  429. arma_extra_debug_sigprint();
  430. return subview_each2_aux::operator_minus(X, Y.get_ref());
  431. }
  432. template<typename T1, typename parent, unsigned int mode, typename TB>
  433. arma_inline
  434. Mat<typename parent::elem_type>
  435. operator-
  436. (
  437. const Base<typename parent::elem_type,T1>& X,
  438. const subview_each2<parent,mode,TB>& Y
  439. )
  440. {
  441. arma_extra_debug_sigprint();
  442. return subview_each2_aux::operator_minus(X.get_ref(), Y);
  443. }
  444. //! @}