| 1 | /*************************************************************************** | 
|---|
| 2 | * blitz/meta/dot.h      Tiny vector dot product metaprogram | 
|---|
| 3 | * | 
|---|
| 4 | * $Id: dot.h,v 1.1.1.1 1999-04-09 17:59:03 ansari Exp $ | 
|---|
| 5 | * | 
|---|
| 6 | * Copyright (C) 1997,1998 Todd Veldhuizen <tveldhui@seurat.uwaterloo.ca> | 
|---|
| 7 | * | 
|---|
| 8 | * This program is free software; you can redistribute it and/or | 
|---|
| 9 | * modify it under the terms of the GNU General Public License | 
|---|
| 10 | * as published by the Free Software Foundation; either version 2 | 
|---|
| 11 | * of the License, or (at your option) any later version. | 
|---|
| 12 | * | 
|---|
| 13 | * This program is distributed in the hope that it will be useful, | 
|---|
| 14 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | 
|---|
| 15 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | 
|---|
| 16 | * GNU General Public License for more details. | 
|---|
| 17 | * | 
|---|
| 18 | * Suggestions:          blitz-suggest@cybervision.com | 
|---|
| 19 | * Bugs:                 blitz-bugs@cybervision.com | 
|---|
| 20 | * | 
|---|
| 21 | * For more information, please see the Blitz++ Home Page: | 
|---|
| 22 | *    http://seurat.uwaterloo.ca/blitz/ | 
|---|
| 23 | * | 
|---|
| 24 | *************************************************************************** | 
|---|
| 25 | * $Log: not supported by cvs2svn $ | 
|---|
| 26 | * Revision 1.2  1998/03/14 00:08:44  tveldhui | 
|---|
| 27 | * 0.2-alpha-05 | 
|---|
| 28 | * | 
|---|
| 29 | * Revision 1.1  1997/07/16 14:51:20  tveldhui | 
|---|
| 30 | * Update: Alpha release 0.2 (Arrays) | 
|---|
| 31 | * | 
|---|
| 32 | */ | 
|---|
| 33 |  | 
|---|
| 34 | #ifndef BZ_META_DOT_H | 
|---|
| 35 | #define BZ_META_DOT_H | 
|---|
| 36 |  | 
|---|
| 37 | #ifndef BZ_PROMOTE_H | 
|---|
| 38 | #include <blitz/promote.h> | 
|---|
| 39 | #endif | 
|---|
| 40 |  | 
|---|
| 41 | #ifndef BZ_METAPROG_H | 
|---|
| 42 | #include <blitz/meta/metaprog.h> | 
|---|
| 43 | #endif | 
|---|
| 44 |  | 
|---|
| 45 | BZ_NAMESPACE(blitz) | 
|---|
| 46 |  | 
|---|
| 47 | template<int N, int I> | 
|---|
| 48 | class _bz_meta_vectorDot { | 
|---|
| 49 | public: | 
|---|
| 50 | enum { loopFlag = (I < N-1) ? 1 : 0 }; | 
|---|
| 51 |  | 
|---|
| 52 | template<class T_expr1, class T_expr2> | 
|---|
| 53 | static inline BZ_PROMOTE(_bz_typename T_expr1::T_numtype, | 
|---|
| 54 | _bz_typename T_expr2::T_numtype) | 
|---|
| 55 | f(const T_expr1& a, const T_expr2& b) | 
|---|
| 56 | { | 
|---|
| 57 | return a[I] * b[I] | 
|---|
| 58 | + _bz_meta_vectorDot<loopFlag * N, loopFlag * (I+1)>::f(a,b); | 
|---|
| 59 | } | 
|---|
| 60 |  | 
|---|
| 61 | template<class T_expr1, class T_expr2> | 
|---|
| 62 | static inline BZ_PROMOTE(_bz_typename T_expr1::T_numtype, | 
|---|
| 63 | _bz_typename T_expr2::T_numtype) | 
|---|
| 64 | f_value_ref(T_expr1 a, const T_expr2& b) | 
|---|
| 65 | { | 
|---|
| 66 | return a[I] * b[I] | 
|---|
| 67 | + _bz_meta_vectorDot<loopFlag * N, loopFlag * (I+1)>::f(a,b); | 
|---|
| 68 | } | 
|---|
| 69 |  | 
|---|
| 70 | template<class T_expr1, class T_expr2> | 
|---|
| 71 | static inline BZ_PROMOTE(_bz_typename T_expr1::T_numtype, | 
|---|
| 72 | _bz_typename T_expr2::T_numtype) | 
|---|
| 73 | f_ref_value(const T_expr1& a, T_expr2 b) | 
|---|
| 74 | { | 
|---|
| 75 | return a[I] * b[I] | 
|---|
| 76 | + _bz_meta_vectorDot<loopFlag * N, loopFlag * (I+1)>::f(a,b); | 
|---|
| 77 | } | 
|---|
| 78 |  | 
|---|
| 79 | template<class T_expr1, class P_numtype2> | 
|---|
| 80 | static inline BZ_PROMOTE(_bz_typename T_expr1::T_numtype, | 
|---|
| 81 | P_numtype2) | 
|---|
| 82 | dotWithArgs(const T_expr1& a, P_numtype2 i1, P_numtype2 i2=0, | 
|---|
| 83 | P_numtype2 i3=0, P_numtype2 i4=0, P_numtype2 i5=0, P_numtype2 i6=0, | 
|---|
| 84 | P_numtype2 i7=0, P_numtype2 i8=0, P_numtype2 i9=0, P_numtype2 i10=0) | 
|---|
| 85 | { | 
|---|
| 86 | return a[I] * i1 | 
|---|
| 87 | + _bz_meta_vectorDot<loopFlag * N, loopFlag * (I+1)>::dotWithArgs | 
|---|
| 88 | (a, i2, i3, i4, i5, i6, i7, i8, i9); | 
|---|
| 89 | } | 
|---|
| 90 | }; | 
|---|
| 91 |  | 
|---|
| 92 | template<> | 
|---|
| 93 | class _bz_meta_vectorDot<0,0> { | 
|---|
| 94 | public: | 
|---|
| 95 | template<class T_expr1, class T_expr2> | 
|---|
| 96 | static inline _bz_meta_nullOperand f(const T_expr1&, const T_expr2&) | 
|---|
| 97 | { return _bz_meta_nullOperand(); } | 
|---|
| 98 |  | 
|---|
| 99 | template<class T_expr1, class P_numtype2> | 
|---|
| 100 | static inline _bz_meta_nullOperand | 
|---|
| 101 | dotWithArgs(const T_expr1& a, P_numtype2 i1, P_numtype2 i2=0, | 
|---|
| 102 | P_numtype2 i3=0, P_numtype2 i4=0, P_numtype2 i5=0, P_numtype2 i6=0, | 
|---|
| 103 | P_numtype2 i7=0, P_numtype2 i8=0, P_numtype2 i9=0, P_numtype2 i10=0) | 
|---|
| 104 | { | 
|---|
| 105 | return _bz_meta_nullOperand(); | 
|---|
| 106 | } | 
|---|
| 107 |  | 
|---|
| 108 | }; | 
|---|
| 109 |  | 
|---|
| 110 | BZ_NAMESPACE_END | 
|---|
| 111 |  | 
|---|
| 112 | #endif // BZ_META_DOT_H | 
|---|