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