00001
00002
00003
00004
00005 #ifndef PLAYA_VECTORDECL_HPP
00006 #define PLAYA_VECTORDECL_HPP
00007
00008 #include "PlayaDefs.hpp"
00009 #include "PlayaHandle.hpp"
00010 #include "PlayaVectorBaseDecl.hpp"
00011 #include "PlayaVectorSpaceDecl.hpp"
00012 #include "PlayaVectorFunctorsDecl.hpp"
00013 #include "Teuchos_TimeMonitor.hpp"
00014
00015 namespace Playa
00016 {
00017
00018 template <class Scalar, int> class LCN;
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035
00036
00037
00038
00039
00040
00041
00042
00043
00044
00045
00046
00047
00048
00049
00050
00051
00052
00053
00054
00055
00056
00057
00058
00059
00060
00061
00062 template <class Scalar>
00063 class Vector : public Playa::Handle<VectorBase<Scalar> >
00064 {
00065 public:
00066
00067
00068 HANDLE_CTORS(Vector<Scalar>, VectorBase<Scalar>);
00069
00070
00071 template <int N>
00072 Vector<Scalar>(const LCN<Scalar, N>& x);
00073
00074
00075
00076 Vector<Scalar>& operator=(const LCN<Scalar, 1>& x);
00077
00078
00079 Vector<Scalar>& operator=(const LCN<Scalar, 2>& x);
00080
00081
00082 Vector<Scalar>& operator=(const LCN<Scalar, 3>& x);
00083
00084
00085 template <int N>
00086 Vector<Scalar>& operator=(const LCN<Scalar, N>& x);
00087
00088
00089
00090
00091
00092
00093 Vector<Scalar>& operator+=(const Vector<Scalar>& other);
00094
00095
00096 Vector<Scalar>& operator+=(const LCN<Scalar, 1>& x);
00097
00098
00099 Vector<Scalar>& operator+=(const LCN<Scalar, 2>& x);
00100
00101
00102 template <int N>
00103 Vector<Scalar>& operator+=(const LCN<Scalar, N>& x);
00104
00105
00106 Vector<Scalar>& operator-=(const Vector<Scalar>& other);
00107
00108
00109 Vector<Scalar>& operator-=(const LCN<Scalar, 1>& x);
00110
00111
00112 Vector<Scalar>& operator-=(const LCN<Scalar, 2>& x);
00113
00114
00115 template <int N>
00116 Vector<Scalar>& operator-=(const LCN<Scalar, N>& x);
00117
00118
00119 Vector<Scalar>& operator*=(const Scalar& a);
00120
00121
00122 Vector<Scalar>& operator/=(const Scalar& a);
00123
00124
00125
00126
00127
00128 VectorSpace<Scalar> space() const
00129 {return this->ptr()->space();}
00130
00131
00132 const MPIComm& comm() const
00133 {return this->space().comm();}
00134
00135
00136 int dim() const
00137 {
00138 return this->ptr()->space()->dim();
00139 }
00140
00141
00142
00143
00144
00145
00146 int numBlocks() const ;
00147
00148
00149 void setBlock(int i, const Vector<Scalar>& v);
00150
00151
00152 const Vector<Scalar>& getBlock(int i) const;
00153
00154
00155 Vector<Scalar> getNonConstBlock(int i) ;
00156
00157
00158 const Vector<Scalar>& getBlock(const BlockIterator<Scalar>& b) const;
00159
00160
00161 Vector<Scalar> getNonConstBlock(const BlockIterator<Scalar>& b);
00162
00163
00164
00165
00166
00167
00168
00169 ConstDataChunk<Scalar> nextConstChunk() const ;
00170
00171
00172 NonConstDataChunk<Scalar> nextChunk() ;
00173
00174
00175 bool hasMoreChunks() const ;
00176
00177
00178
00179 void rewind() const ;
00180
00181
00182
00183
00184
00185
00186
00187
00188
00189
00190
00191 const Scalar& operator[](int localIndex) const ;
00192
00193
00194
00195
00196
00197
00198
00199
00200 Scalar& operator[](int localIndex);
00201
00202
00203
00204
00205
00206
00207 const Scalar& operator()(const BlockIterator<Scalar>& b,
00208 int localIndexWithinBlock) const ;
00209
00210
00211
00212
00213
00214
00215 Scalar& operator()(const BlockIterator<Scalar>& b,
00216 int localIndexWithinBlock) ;
00217
00218
00219
00220
00221
00222 std::string description() const ;
00223
00224
00225 void print(std::ostream& os) const ;
00226
00227
00228
00229
00230
00231
00232
00233
00234
00235
00236 Vector<Scalar>& scale(const Scalar& alpha);
00237
00238
00239
00240
00241
00242 Vector<Scalar>& update(const Scalar& alpha, const Vector<Scalar>& x,
00243 const Scalar& gamma=1.0);
00244
00245
00246
00247
00248 Vector<Scalar>& update(const Scalar& alpha, const Vector<Scalar>& x,
00249 const Scalar& beta, const Vector<Scalar>& y,
00250 const Scalar& gamma);
00251
00252
00253
00254
00255 Vector<Scalar>& update(const Scalar& alpha, const Vector<Scalar>& x,
00256 const Scalar& beta, const Vector<Scalar>& y,
00257 const Scalar& gamma, const Vector<Scalar>& z,
00258 const Scalar& delta);
00259
00260
00261
00262
00263
00264
00265
00266 Vector<Scalar>& acceptCopyOf(const Vector<Scalar>& x);
00267
00268
00269
00270
00271 Vector<Scalar> copy() const ;
00272
00273
00274
00275
00276 Vector<Scalar>& selfDotStar(const Vector<Scalar>& other) ;
00277
00278
00279
00280
00281 Vector<Scalar>& selfDotSlash(const Vector<Scalar>& other) ;
00282
00283
00284
00285
00286 Vector<Scalar> dotStar(const Vector<Scalar>& other) const ;
00287
00288
00289
00290
00291 Vector<Scalar> dotSlash(const Vector<Scalar>& other) const ;
00292
00293
00294
00295
00296 Vector<Scalar> reciprocal() const ;
00297
00298
00299
00300
00301
00302 Vector<Scalar> abs() const ;
00303
00304
00305
00306
00307 Vector<Scalar>& selfReciprocal() ;
00308
00309
00310
00311
00312 Vector<Scalar>& selfAbs() ;
00313
00314
00315
00316
00317 Vector<Scalar>& randomize() ;
00318
00319
00320
00321
00322 void setToConstant(const Scalar& alpha) ;
00323
00324
00325
00326
00327
00328 Scalar dot(const Vector<Scalar>& other) const ;
00329
00330
00331
00332
00333 Scalar operator*(const Vector<Scalar>& other) const ;
00334
00335
00336
00337
00338 Scalar norm1() const ;
00339
00340
00341
00342
00343 Scalar norm2() const ;
00344
00345
00346
00347
00348 Scalar norm2(const Vector<Scalar>& weights) const ;
00349
00350
00351
00352
00353
00354 Scalar normInf() const ;
00355
00356
00357
00358
00359 void zero();
00360
00361
00362
00363 Scalar max() const;
00364
00365
00366 Scalar min()const;
00367
00368
00369
00370
00371
00372
00373
00374
00375 static RCP<Time>& opTimer()
00376 {
00377 static RCP<Time> rtn
00378 = TimeMonitor::getNewTimer("Low-level vector operations");
00379 return rtn;
00380 }
00381
00382
00383
00384
00385
00386 template <class UF>
00387 Vector<Scalar>& applyUnaryFunctor(const UF& functor);
00388
00389
00390
00391 template <class UF>
00392 Vector<Scalar>& acceptUnaryFunctor(const UF& functor,
00393 const Vector<Scalar>& other);
00394
00395
00396
00397
00398 template <class VF>
00399 Vector<Scalar>& applyBinaryFunctor(const VF& functor,
00400 const Vector<Scalar>& other);
00401
00402
00403
00404
00405 template <class VF>
00406 Vector<Scalar>& applyTernaryFunctor(const VF& functor,
00407 const Vector<Scalar>& x, const Vector<Scalar>& y);
00408
00409
00410 template <class RF>
00411 typename PlayaFunctors::VectorFunctorTraits<Scalar, RF>::ReturnType
00412 applyUnaryReductionFunctor(
00413 const RF& functor)
00414 const ;
00415
00416
00417 template <class RF>
00418 typename PlayaFunctors::VectorFunctorTraits<Scalar, RF>::ReturnType
00419 applyBinaryReductionFunctor(const RF& functor, const Vector<Scalar>& other)
00420 const ;
00421
00422
00423
00424
00425
00426 protected:
00427
00428
00429 const Vector<Scalar>& getBlock(const std::deque<int>& iter) const ;
00430
00431
00432 Vector<Scalar> getNonConstBlock(const std::deque<int>& iter) ;
00433
00434
00435 private:
00436
00437 };
00438
00439
00440 template <class Scalar> class LoadableVector;
00441
00442
00443 template <class Scalar>
00444 LoadableVector<Scalar>* loadable(Vector<Scalar> vec) ;
00445
00446
00447
00448
00449
00450 template <class Scalar>
00451 Scalar* dataPtr(Vector<Scalar> vec) ;
00452
00453
00454
00455
00456 template <class Scalar>
00457 const Scalar* dataPtr(const Vector<Scalar>& vec) ;
00458
00459
00460 }
00461
00462 template <class Scalar> inline
00463 std::ostream& operator<<(std::ostream& os, const Playa::Vector<Scalar>& x)
00464 {
00465 x.print(os);
00466 return os;
00467 }
00468
00469
00470
00471
00472 #endif