| [658] | 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 |  | 
|---|