Shuto Naruse / Mbed 2 deprecated Eurobot2012_Primary

Eurobot2012_Primary

Dependencies:   mbed Eurobot_2012_Primary

tvmet/xpr/MMProduct.h@25:143b19c1fb05, 2012-10-17 (annotated)

Committer:
narshu
Date:
Wed Oct 17 22:22:28 2012 +0000
Revision:
25:143b19c1fb05
Commit before publishing;

Who changed what in which revision?

UserRevisionLine numberNew contents of line
narshu 25:143b19c1fb05 1 /*
narshu 25:143b19c1fb05 2 * Tiny Vector Matrix Library
narshu 25:143b19c1fb05 3 * Dense Vector Matrix Libary of Tiny size using Expression Templates
narshu 25:143b19c1fb05 4 *
narshu 25:143b19c1fb05 5 * Copyright (C) 2001 - 2007 Olaf Petzold <opetzold@users.sourceforge.net>
narshu 25:143b19c1fb05 6 *
narshu 25:143b19c1fb05 7 * This library is free software; you can redistribute it and/or
narshu 25:143b19c1fb05 8 * modify it under the terms of the GNU Lesser General Public
narshu 25:143b19c1fb05 9 * License as published by the Free Software Foundation; either
narshu 25:143b19c1fb05 10 * version 2.1 of the License, or (at your option) any later version.
narshu 25:143b19c1fb05 11 *
narshu 25:143b19c1fb05 12 * This library is distributed in the hope that it will be useful,
narshu 25:143b19c1fb05 13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
narshu 25:143b19c1fb05 14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
narshu 25:143b19c1fb05 15 * Lesser General Public License for more details.
narshu 25:143b19c1fb05 16 *
narshu 25:143b19c1fb05 17 * You should have received a copy of the GNU Lesser General Public
narshu 25:143b19c1fb05 18 * License along with this library; if not, write to the Free Software
narshu 25:143b19c1fb05 19 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
narshu 25:143b19c1fb05 20 *
narshu 25:143b19c1fb05 21 * $Id: MMProduct.h,v 1.24 2007-06-23 15:58:59 opetzold Exp $
narshu 25:143b19c1fb05 22 */
narshu 25:143b19c1fb05 23
narshu 25:143b19c1fb05 24 #ifndef TVMET_XPR_MMPRODUCT_H
narshu 25:143b19c1fb05 25 #define TVMET_XPR_MMPRODUCT_H
narshu 25:143b19c1fb05 26
narshu 25:143b19c1fb05 27 #include <tvmet/meta/Gemm.h>
narshu 25:143b19c1fb05 28 #include <tvmet/loop/Gemm.h>
narshu 25:143b19c1fb05 29
narshu 25:143b19c1fb05 30 namespace tvmet {
narshu 25:143b19c1fb05 31
narshu 25:143b19c1fb05 32
narshu 25:143b19c1fb05 33 /**
narshu 25:143b19c1fb05 34 * \class XprMMProduct MMProduct.h "tvmet/xpr/MMProduct.h"
narshu 25:143b19c1fb05 35 * \brief Expression for matrix-matrix product.
narshu 25:143b19c1fb05 36 * Using formula:
narshu 25:143b19c1fb05 37 * \f[
narshu 25:143b19c1fb05 38 * M_1\,M_2
narshu 25:143b19c1fb05 39 * \f]
narshu 25:143b19c1fb05 40 * \note The Rows2 has to be equal to Cols1.
narshu 25:143b19c1fb05 41 */
narshu 25:143b19c1fb05 42 template<class E1, std::size_t Rows1, std::size_t Cols1,
narshu 25:143b19c1fb05 43 class E2, std::size_t Cols2>
narshu 25:143b19c1fb05 44 class XprMMProduct
narshu 25:143b19c1fb05 45 : public TvmetBase< XprMMProduct<E1, Rows1, Cols1, E2, Cols2> >
narshu 25:143b19c1fb05 46 {
narshu 25:143b19c1fb05 47 private:
narshu 25:143b19c1fb05 48 XprMMProduct();
narshu 25:143b19c1fb05 49 XprMMProduct& operator=(const XprMMProduct&);
narshu 25:143b19c1fb05 50
narshu 25:143b19c1fb05 51 public:
narshu 25:143b19c1fb05 52 typedef typename PromoteTraits<
narshu 25:143b19c1fb05 53 typename E1::value_type,
narshu 25:143b19c1fb05 54 typename E2::value_type
narshu 25:143b19c1fb05 55 >::value_type value_type;
narshu 25:143b19c1fb05 56
narshu 25:143b19c1fb05 57 public:
narshu 25:143b19c1fb05 58 /** Complexity counter. */
narshu 25:143b19c1fb05 59 enum {
narshu 25:143b19c1fb05 60 ops_lhs = E1::ops,
narshu 25:143b19c1fb05 61 ops_rhs = E2::ops,
narshu 25:143b19c1fb05 62 M = Rows1 * Cols1 * Cols2,
narshu 25:143b19c1fb05 63 N = Rows1 * (Cols1 - 1) * Cols2,
narshu 25:143b19c1fb05 64 ops_plus = M * NumericTraits<value_type>::ops_plus,
narshu 25:143b19c1fb05 65 ops_muls = N * NumericTraits<value_type>::ops_muls,
narshu 25:143b19c1fb05 66 ops = ops_plus + ops_muls,
narshu 25:143b19c1fb05 67 use_meta = Rows1*Cols2 < TVMET_COMPLEXITY_MM_TRIGGER ? true : false
narshu 25:143b19c1fb05 68 };
narshu 25:143b19c1fb05 69
narshu 25:143b19c1fb05 70 public:
narshu 25:143b19c1fb05 71 /** Constructor. */
narshu 25:143b19c1fb05 72 explicit XprMMProduct(const E1& lhs, const E2& rhs)
narshu 25:143b19c1fb05 73 : m_lhs(lhs), m_rhs(rhs)
narshu 25:143b19c1fb05 74 { }
narshu 25:143b19c1fb05 75
narshu 25:143b19c1fb05 76 /** Copy Constructor. Not explicit! */
narshu 25:143b19c1fb05 77 #if defined(TVMET_OPTIMIZE_XPR_MANUAL_CCTOR)
narshu 25:143b19c1fb05 78 XprMMProduct(const XprMMProduct& e)
narshu 25:143b19c1fb05 79 : m_lhs(e.m_lhs), m_rhs(e.m_rhs)
narshu 25:143b19c1fb05 80 { }
narshu 25:143b19c1fb05 81 #endif
narshu 25:143b19c1fb05 82
narshu 25:143b19c1fb05 83 private:
narshu 25:143b19c1fb05 84 /** Wrapper for meta gemm. */
narshu 25:143b19c1fb05 85 static inline
narshu 25:143b19c1fb05 86 value_type do_gemm(dispatch<true>, const E1& lhs, const E2& rhs, std::size_t i, std::size_t j) {
narshu 25:143b19c1fb05 87 return meta::gemm<Rows1, Cols1,
narshu 25:143b19c1fb05 88 Cols2,
narshu 25:143b19c1fb05 89 0>::prod(lhs, rhs, i, j);
narshu 25:143b19c1fb05 90 }
narshu 25:143b19c1fb05 91
narshu 25:143b19c1fb05 92 /** Wrapper for loop gemm. */
narshu 25:143b19c1fb05 93 static inline
narshu 25:143b19c1fb05 94 value_type do_gemm(dispatch<false>, const E1& lhs, const E2& rhs, std::size_t i, std::size_t j) {
narshu 25:143b19c1fb05 95 return loop::gemm<Rows1, Cols1, Cols2>::prod(lhs, rhs, i, j);
narshu 25:143b19c1fb05 96 }
narshu 25:143b19c1fb05 97
narshu 25:143b19c1fb05 98 public:
narshu 25:143b19c1fb05 99 /** index operator for arrays/matrices */
narshu 25:143b19c1fb05 100 value_type operator()(std::size_t i, std::size_t j) const {
narshu 25:143b19c1fb05 101 TVMET_RT_CONDITION((i < Rows1) && (j < Cols2), "XprMMProduct Bounce Violation")
narshu 25:143b19c1fb05 102 return do_gemm(dispatch<use_meta>(), m_lhs, m_rhs, i, j);
narshu 25:143b19c1fb05 103 }
narshu 25:143b19c1fb05 104
narshu 25:143b19c1fb05 105 public: // debugging Xpr parse tree
narshu 25:143b19c1fb05 106 void print_xpr(std::ostream& os, std::size_t l=0) const {
narshu 25:143b19c1fb05 107 os << IndentLevel(l++)
narshu 25:143b19c1fb05 108 << "XprMMProduct["
narshu 25:143b19c1fb05 109 << (use_meta ? "M" : "L") << ", O=" << ops
narshu 25:143b19c1fb05 110 << ", (O1=" << ops_lhs << ", O2=" << ops_rhs << ")]<"
narshu 25:143b19c1fb05 111 << std::endl;
narshu 25:143b19c1fb05 112 m_lhs.print_xpr(os, l);
narshu 25:143b19c1fb05 113 os << IndentLevel(l)
narshu 25:143b19c1fb05 114 << "R1=" << Rows1 << ", C1=" << Cols1 << ",\n";
narshu 25:143b19c1fb05 115 m_rhs.print_xpr(os, l);
narshu 25:143b19c1fb05 116 os << IndentLevel(l)
narshu 25:143b19c1fb05 117 << "C2=" << Cols2 << ",\n";
narshu 25:143b19c1fb05 118 os << IndentLevel(--l)
narshu 25:143b19c1fb05 119 << ">," << std::endl;
narshu 25:143b19c1fb05 120 }
narshu 25:143b19c1fb05 121
narshu 25:143b19c1fb05 122 private:
narshu 25:143b19c1fb05 123 const E1 m_lhs;
narshu 25:143b19c1fb05 124 const E2 m_rhs;
narshu 25:143b19c1fb05 125 };
narshu 25:143b19c1fb05 126
narshu 25:143b19c1fb05 127
narshu 25:143b19c1fb05 128 } // namespace tvmet
narshu 25:143b19c1fb05 129
narshu 25:143b19c1fb05 130 #endif // TVMET_XPR_MMPRODUCT_H
narshu 25:143b19c1fb05 131
narshu 25:143b19c1fb05 132 // Local Variables:
narshu 25:143b19c1fb05 133 // mode:C++
narshu 25:143b19c1fb05 134 // tab-width:8
narshu 25:143b19c1fb05 135 // End: