| [658] | 1 | /*************************************************************************** | 
|---|
|  | 2 | * blitz/meta/dot.h      Tiny vector dot product metaprogram | 
|---|
|  | 3 | * | 
|---|
|  | 4 | * $Id: dot.h,v 1.1.1.1 1999-11-26 16:37:07 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.1.1.1  1999/04/09  17:59:03  ansari | 
|---|
|  | 27 | * Creation module DPC/Blitz (blitz 0.4) Reza 09/04/99 | 
|---|
|  | 28 | * | 
|---|
|  | 29 | * Revision 1.2  1998/03/14 00:08:44  tveldhui | 
|---|
|  | 30 | * 0.2-alpha-05 | 
|---|
|  | 31 | * | 
|---|
|  | 32 | * Revision 1.1  1997/07/16 14:51:20  tveldhui | 
|---|
|  | 33 | * Update: Alpha release 0.2 (Arrays) | 
|---|
|  | 34 | * | 
|---|
|  | 35 | */ | 
|---|
|  | 36 |  | 
|---|
|  | 37 | #ifndef BZ_META_DOT_H | 
|---|
|  | 38 | #define BZ_META_DOT_H | 
|---|
|  | 39 |  | 
|---|
|  | 40 | #ifndef BZ_PROMOTE_H | 
|---|
|  | 41 | #include <blitz/promote.h> | 
|---|
|  | 42 | #endif | 
|---|
|  | 43 |  | 
|---|
|  | 44 | #ifndef BZ_METAPROG_H | 
|---|
|  | 45 | #include <blitz/meta/metaprog.h> | 
|---|
|  | 46 | #endif | 
|---|
|  | 47 |  | 
|---|
|  | 48 | BZ_NAMESPACE(blitz) | 
|---|
|  | 49 |  | 
|---|
|  | 50 | template<int N, int I> | 
|---|
|  | 51 | class _bz_meta_vectorDot { | 
|---|
|  | 52 | public: | 
|---|
|  | 53 | enum { loopFlag = (I < N-1) ? 1 : 0 }; | 
|---|
|  | 54 |  | 
|---|
|  | 55 | template<class T_expr1, class T_expr2> | 
|---|
|  | 56 | static inline BZ_PROMOTE(_bz_typename T_expr1::T_numtype, | 
|---|
|  | 57 | _bz_typename T_expr2::T_numtype) | 
|---|
|  | 58 | f(const T_expr1& a, const T_expr2& b) | 
|---|
|  | 59 | { | 
|---|
|  | 60 | return a[I] * b[I] | 
|---|
|  | 61 | + _bz_meta_vectorDot<loopFlag * N, loopFlag * (I+1)>::f(a,b); | 
|---|
|  | 62 | } | 
|---|
|  | 63 |  | 
|---|
|  | 64 | template<class T_expr1, class T_expr2> | 
|---|
|  | 65 | static inline BZ_PROMOTE(_bz_typename T_expr1::T_numtype, | 
|---|
|  | 66 | _bz_typename T_expr2::T_numtype) | 
|---|
|  | 67 | f_value_ref(T_expr1 a, const T_expr2& b) | 
|---|
|  | 68 | { | 
|---|
|  | 69 | return a[I] * b[I] | 
|---|
|  | 70 | + _bz_meta_vectorDot<loopFlag * N, loopFlag * (I+1)>::f(a,b); | 
|---|
|  | 71 | } | 
|---|
|  | 72 |  | 
|---|
|  | 73 | template<class T_expr1, class T_expr2> | 
|---|
|  | 74 | static inline BZ_PROMOTE(_bz_typename T_expr1::T_numtype, | 
|---|
|  | 75 | _bz_typename T_expr2::T_numtype) | 
|---|
|  | 76 | f_ref_value(const T_expr1& a, T_expr2 b) | 
|---|
|  | 77 | { | 
|---|
|  | 78 | return a[I] * b[I] | 
|---|
|  | 79 | + _bz_meta_vectorDot<loopFlag * N, loopFlag * (I+1)>::f(a,b); | 
|---|
|  | 80 | } | 
|---|
|  | 81 |  | 
|---|
|  | 82 | template<class T_expr1, class P_numtype2> | 
|---|
|  | 83 | static inline BZ_PROMOTE(_bz_typename T_expr1::T_numtype, | 
|---|
|  | 84 | P_numtype2) | 
|---|
|  | 85 | dotWithArgs(const T_expr1& a, P_numtype2 i1, P_numtype2 i2=0, | 
|---|
|  | 86 | P_numtype2 i3=0, P_numtype2 i4=0, P_numtype2 i5=0, P_numtype2 i6=0, | 
|---|
|  | 87 | P_numtype2 i7=0, P_numtype2 i8=0, P_numtype2 i9=0, P_numtype2 i10=0) | 
|---|
|  | 88 | { | 
|---|
|  | 89 | return a[I] * i1 | 
|---|
|  | 90 | + _bz_meta_vectorDot<loopFlag * N, loopFlag * (I+1)>::dotWithArgs | 
|---|
|  | 91 | (a, i2, i3, i4, i5, i6, i7, i8, i9); | 
|---|
|  | 92 | } | 
|---|
|  | 93 | }; | 
|---|
|  | 94 |  | 
|---|
|  | 95 | template<> | 
|---|
|  | 96 | class _bz_meta_vectorDot<0,0> { | 
|---|
|  | 97 | public: | 
|---|
|  | 98 | template<class T_expr1, class T_expr2> | 
|---|
|  | 99 | static inline _bz_meta_nullOperand f(const T_expr1&, const T_expr2&) | 
|---|
|  | 100 | { return _bz_meta_nullOperand(); } | 
|---|
|  | 101 |  | 
|---|
|  | 102 | template<class T_expr1, class P_numtype2> | 
|---|
|  | 103 | static inline _bz_meta_nullOperand | 
|---|
|  | 104 | dotWithArgs(const T_expr1& a, P_numtype2 i1, P_numtype2 i2=0, | 
|---|
|  | 105 | P_numtype2 i3=0, P_numtype2 i4=0, P_numtype2 i5=0, P_numtype2 i6=0, | 
|---|
|  | 106 | P_numtype2 i7=0, P_numtype2 i8=0, P_numtype2 i9=0, P_numtype2 i10=0) | 
|---|
|  | 107 | { | 
|---|
|  | 108 | return _bz_meta_nullOperand(); | 
|---|
|  | 109 | } | 
|---|
|  | 110 |  | 
|---|
|  | 111 | }; | 
|---|
|  | 112 |  | 
|---|
|  | 113 | BZ_NAMESPACE_END | 
|---|
|  | 114 |  | 
|---|
|  | 115 | #endif // BZ_META_DOT_H | 
|---|