| 1 | /* | 
|---|
| 2 | * $Id: vecdot.cc,v 1.1.1.1 1999-11-26 16:37:05 ansari Exp $ | 
|---|
| 3 | * | 
|---|
| 4 | * Copyright (C) 1997 Todd Veldhuizen <tveldhui@seurat.uwaterloo.ca> | 
|---|
| 5 | * All rights reserved.  Please see <blitz/blitz.h> for terms and | 
|---|
| 6 | * conditions of use. | 
|---|
| 7 | * | 
|---|
| 8 | * $Log: not supported by cvs2svn $ | 
|---|
| 9 | // Revision 1.1.1.1  1999/04/09  17:58:58  ansari | 
|---|
| 10 | // Creation module DPC/Blitz (blitz 0.4) Reza 09/04/99 | 
|---|
| 11 | // | 
|---|
| 12 | * Revision 1.4  1998/03/14 00:04:47  tveldhui | 
|---|
| 13 | * 0.2-alpha-05 | 
|---|
| 14 | * | 
|---|
| 15 | * Revision 1.3  1997/07/16 14:51:20  tveldhui | 
|---|
| 16 | * Update: Alpha release 0.2 (Arrays) | 
|---|
| 17 | * | 
|---|
| 18 | * Revision 1.2  1997/01/24 14:42:00  tveldhui | 
|---|
| 19 | * Periodic RCS update | 
|---|
| 20 | * | 
|---|
| 21 | */ | 
|---|
| 22 |  | 
|---|
| 23 | #ifndef BZ_VECDOT_CC | 
|---|
| 24 | #define BZ_VECDOT_CC | 
|---|
| 25 |  | 
|---|
| 26 | #ifndef BZ_VECGLOBS_H | 
|---|
| 27 | #error <blitz/vecdot.cc> must be included via <blitz/vecglobs.h> | 
|---|
| 28 | #endif | 
|---|
| 29 |  | 
|---|
| 30 | BZ_NAMESPACE(blitz) | 
|---|
| 31 |  | 
|---|
| 32 | template<class P1, class P2> | 
|---|
| 33 | inline | 
|---|
| 34 | BZ_SUMTYPE(BZ_PROMOTE(_bz_typename P1::T_numtype, _bz_typename P2::T_numtype)) | 
|---|
| 35 | _bz_dot(P1 vector1, P2 vector2) | 
|---|
| 36 | { | 
|---|
| 37 | BZPRECONDITION(vector1._bz_suggestLength() == vector2._bz_suggestLength()); | 
|---|
| 38 |  | 
|---|
| 39 | typedef BZ_SUMTYPE(BZ_PROMOTE(_bz_typename P1::T_numtype, | 
|---|
| 40 | _bz_typename P2::T_numtype))  T_sumtype; | 
|---|
| 41 |  | 
|---|
| 42 | T_sumtype sum = 0; | 
|---|
| 43 | int length = vector1._bz_suggestLength(); | 
|---|
| 44 |  | 
|---|
| 45 | if (vector1._bz_hasFastAccess() && vector2._bz_hasFastAccess()) | 
|---|
| 46 | { | 
|---|
| 47 | for (int i=0; i < length; ++i) | 
|---|
| 48 | sum += vector1._bz_fastAccess(i) | 
|---|
| 49 | * vector2._bz_fastAccess(i); | 
|---|
| 50 | } | 
|---|
| 51 | else { | 
|---|
| 52 | for (int i=0; i < length; ++i) | 
|---|
| 53 | sum += vector1[i] * vector2[i]; | 
|---|
| 54 | } | 
|---|
| 55 |  | 
|---|
| 56 | return sum; | 
|---|
| 57 | } | 
|---|
| 58 |  | 
|---|
| 59 |  | 
|---|
| 60 | // dot() | 
|---|
| 61 | template<class P_numtype1, class P_numtype2> | 
|---|
| 62 | inline | 
|---|
| 63 | BZ_SUMTYPE(BZ_PROMOTE(P_numtype1,P_numtype2)) | 
|---|
| 64 | dot(const Vector<P_numtype1>& a, const Vector<P_numtype2>& b) | 
|---|
| 65 | { | 
|---|
| 66 | return _bz_dot(a, b); | 
|---|
| 67 | } | 
|---|
| 68 |  | 
|---|
| 69 | // dot(expr,expr) | 
|---|
| 70 | template<class P_expr1, class P_expr2> | 
|---|
| 71 | inline | 
|---|
| 72 | BZ_SUMTYPE(BZ_PROMOTE(_bz_typename P_expr1::T_numtype, | 
|---|
| 73 | _bz_typename P_expr2::T_numtype)) | 
|---|
| 74 | dot(_bz_VecExpr<P_expr1> expr1, _bz_VecExpr<P_expr2> expr2) | 
|---|
| 75 | { | 
|---|
| 76 | return _bz_dot(expr1, expr2); | 
|---|
| 77 | } | 
|---|
| 78 |  | 
|---|
| 79 | // dot(expr,vec) | 
|---|
| 80 | template<class P_expr1, class P_numtype2> | 
|---|
| 81 | inline | 
|---|
| 82 | BZ_SUMTYPE(BZ_PROMOTE(_bz_typename P_expr1::T_numtype, P_numtype2)) | 
|---|
| 83 | dot(_bz_VecExpr<P_expr1> expr1, const Vector<P_numtype2>& vector2) | 
|---|
| 84 | { | 
|---|
| 85 | return _bz_dot(vector2, expr1); | 
|---|
| 86 | } | 
|---|
| 87 |  | 
|---|
| 88 | // dot(vec,expr) | 
|---|
| 89 | template<class P_numtype1, class P_expr2> | 
|---|
| 90 | inline | 
|---|
| 91 | BZ_SUMTYPE(BZ_PROMOTE(P_numtype1, _bz_typename P_expr2::T_numtype)) | 
|---|
| 92 | dot(const Vector<P_numtype1>& vector1, _bz_VecExpr<P_expr2> expr2) | 
|---|
| 93 | { | 
|---|
| 94 | return _bz_dot(vector1, expr2); | 
|---|
| 95 | } | 
|---|
| 96 |  | 
|---|
| 97 | // dot(vec,vecpick) | 
|---|
| 98 | template<class P_numtype1, class P_numtype2> | 
|---|
| 99 | inline | 
|---|
| 100 | BZ_SUMTYPE(BZ_PROMOTE(P_numtype1, P_numtype2)) | 
|---|
| 101 | dot(const Vector<P_numtype1>& vector1, const VectorPick<P_numtype2>& vector2) | 
|---|
| 102 | { | 
|---|
| 103 | return _bz_dot(vector1, vector2); | 
|---|
| 104 | } | 
|---|
| 105 |  | 
|---|
| 106 | // dot(vecpick,vec) | 
|---|
| 107 | template<class P_numtype1, class P_numtype2> | 
|---|
| 108 | inline | 
|---|
| 109 | BZ_SUMTYPE(BZ_PROMOTE(P_numtype1, P_numtype2)) | 
|---|
| 110 | dot(const VectorPick<P_numtype1>& vector1, const Vector<P_numtype2>& vector2) | 
|---|
| 111 | { | 
|---|
| 112 | return _bz_dot(vector1, vector2); | 
|---|
| 113 | } | 
|---|
| 114 |  | 
|---|
| 115 | // dot(vecpick,vecpick) | 
|---|
| 116 | template<class P_numtype1, class P_numtype2> | 
|---|
| 117 | inline | 
|---|
| 118 | BZ_SUMTYPE(BZ_PROMOTE(P_numtype1, P_numtype2)) | 
|---|
| 119 | dot(const VectorPick<P_numtype1>& vector1, const VectorPick<P_numtype2>& vector2) | 
|---|
| 120 | { | 
|---|
| 121 | return _bz_dot(vector1, vector2); | 
|---|
| 122 | } | 
|---|
| 123 |  | 
|---|
| 124 | // dot(expr, vecpick) | 
|---|
| 125 | template<class P_expr1, class P_numtype2> | 
|---|
| 126 | inline | 
|---|
| 127 | BZ_SUMTYPE(BZ_PROMOTE(_bz_typename P_expr1::T_numtype, P_numtype2)) | 
|---|
| 128 | dot(_bz_VecExpr<P_expr1> expr1, const VectorPick<P_numtype2>& vector2) | 
|---|
| 129 | { | 
|---|
| 130 | return _bz_dot(expr1, vector2); | 
|---|
| 131 | } | 
|---|
| 132 |  | 
|---|
| 133 | // dot(vecpick, expr) | 
|---|
| 134 | template<class P_numtype1, class P_expr2> | 
|---|
| 135 | inline | 
|---|
| 136 | BZ_SUMTYPE(BZ_PROMOTE(P_numtype1, _bz_typename P_expr2::T_numtype)) | 
|---|
| 137 | dot(const VectorPick<P_numtype1>& vector1, _bz_VecExpr<P_expr2> expr2) | 
|---|
| 138 | { | 
|---|
| 139 | return _bz_dot(vector1, expr2); | 
|---|
| 140 | } | 
|---|
| 141 |  | 
|---|
| 142 | BZ_NAMESPACE_END | 
|---|
| 143 |  | 
|---|
| 144 | #endif // BZ_VECDOT_CC | 
|---|
| 145 |  | 
|---|