fn_conv.hpp 2.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  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 fn_conv
  16. //! @{
  17. //! Convolution, which is also equivalent to polynomial multiplication and FIR digital filtering.
  18. template<typename T1, typename T2>
  19. arma_warn_unused
  20. inline
  21. typename
  22. enable_if2
  23. <
  24. (is_arma_type<T1>::value && is_arma_type<T2>::value && is_same_type<typename T1::elem_type, typename T2::elem_type>::value),
  25. const Glue<T1, T2, glue_conv>
  26. >::result
  27. conv(const T1& A, const T2& B, const char* shape = "full")
  28. {
  29. arma_extra_debug_sigprint();
  30. const char sig = (shape != NULL) ? shape[0] : char(0);
  31. arma_debug_check( ((sig != 'f') && (sig != 's')), "conv(): unsupported value of 'shape' parameter" );
  32. const uword mode = (sig == 's') ? uword(1) : uword(0);
  33. return Glue<T1, T2, glue_conv>(A, B, mode);
  34. }
  35. template<typename T1, typename T2>
  36. arma_warn_unused
  37. inline
  38. typename
  39. enable_if2
  40. <
  41. (is_arma_type<T1>::value && is_arma_type<T2>::value && is_same_type<typename T1::elem_type, typename T2::elem_type>::value),
  42. const Glue<T1, T2, glue_conv2>
  43. >::result
  44. conv2(const T1& A, const T2& B, const char* shape = "full")
  45. {
  46. arma_extra_debug_sigprint();
  47. const char sig = (shape != NULL) ? shape[0] : char(0);
  48. arma_debug_check( ((sig != 'f') && (sig != 's')), "conv2(): unsupported value of 'shape' parameter" );
  49. const uword mode = (sig == 's') ? uword(1) : uword(0);
  50. return Glue<T1, T2, glue_conv2>(A, B, mode);
  51. }
  52. //! @}