00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035
00036
00037
00038
00039
00040
00041
00042 #ifndef LOCA_EXTENDED_MULTIVECTOR_H
00043 #define LOCA_EXTENDED_MULTIVECTOR_H
00044
00045 #include "NOX_Common.H"
00046 #include "NOX_Abstract_MultiVector.H"
00047
00048
00049 namespace LOCA {
00050 class GlobalData;
00051 namespace Extended {
00052 class Vector;
00053 }
00054 }
00055
00056 namespace LOCA {
00057
00058 namespace Extended {
00059
00065
00066
00067
00068
00069
00070
00071 class MultiVector : public NOX::Abstract::MultiVector {
00072
00074 friend class Vector;
00075
00076 public:
00077
00079 MultiVector(const MultiVector& source,
00080 NOX::CopyType type = NOX::DeepCopy);
00081
00086 MultiVector(const MultiVector& source, int nColumns);
00087
00092 MultiVector(const MultiVector& source, const vector<int>& index,
00093 bool view);
00094
00096 virtual ~MultiVector();
00097
00099
00101 virtual NOX::Abstract::MultiVector&
00102 init(double gamma);
00103
00105 virtual NOX::Abstract::MultiVector&
00106 random(bool useSeed = false, int seed = 1);
00107
00109 virtual NOX::Abstract::MultiVector&
00110 operator=(const NOX::Abstract::MultiVector& source);
00111
00113 virtual MultiVector& operator=(const MultiVector& y);
00114
00120 virtual NOX::Abstract::MultiVector&
00121 setBlock(const NOX::Abstract::MultiVector& source,
00122 const vector<int>& index);
00123 virtual NOX::Abstract::MultiVector&
00124 setBlock(const MultiVector& source, const vector<int>& index);
00125
00127 virtual NOX::Abstract::MultiVector&
00128 augment(const NOX::Abstract::MultiVector& source);
00129 virtual NOX::Abstract::MultiVector&
00130 augment(const MultiVector& source);
00131
00136 virtual NOX::Abstract::Vector& operator [] (int i);
00137
00142 virtual const NOX::Abstract::Vector& operator [] (int i) const;
00143
00145 virtual NOX::Abstract::MultiVector&
00146 scale(double gamma);
00147
00152 virtual NOX::Abstract::MultiVector&
00153 update(double alpha, const NOX::Abstract::MultiVector& a,
00154 double gamma = 0.0);
00155 virtual NOX::Abstract::MultiVector&
00156 update(double alpha, const MultiVector& a,
00157 double gamma = 0.0);
00158
00163 virtual NOX::Abstract::MultiVector&
00164 update(double alpha, const NOX::Abstract::MultiVector& a,
00165 double beta, const NOX::Abstract::MultiVector& b,
00166 double gamma = 0.0);
00167 virtual NOX::Abstract::MultiVector&
00168 update(double alpha, const MultiVector& a,
00169 double beta, const MultiVector& b,
00170 double gamma = 0.0);
00171
00178 virtual NOX::Abstract::MultiVector&
00179 update(Teuchos::ETransp transb, double alpha,
00180 const NOX::Abstract::MultiVector& a,
00181 const NOX::Abstract::MultiVector::DenseMatrix& b,
00182 double gamma = 0.0);
00183 virtual NOX::Abstract::MultiVector&
00184 update(Teuchos::ETransp transb, double alpha, const MultiVector& a,
00185 const NOX::Abstract::MultiVector::DenseMatrix& b,
00186 double gamma = 0.0);
00187
00201 virtual Teuchos::RCP<NOX::Abstract::MultiVector>
00202 clone(NOX::CopyType type = NOX::DeepCopy) const;
00203
00205 virtual Teuchos::RCP<NOX::Abstract::MultiVector>
00206 clone(int numvecs) const;
00207
00212 virtual Teuchos::RCP<NOX::Abstract::MultiVector>
00213 subCopy(const vector<int>& index) const;
00214
00219 virtual Teuchos::RCP<NOX::Abstract::MultiVector>
00220 subView(const vector<int>& index) const;
00221
00223 virtual void
00224 norm(vector<double>& result,
00225 NOX::Abstract::Vector::NormType type =
00226 NOX::Abstract::Vector::TwoNorm) const;
00227
00231 virtual void multiply(double alpha, const NOX::Abstract::MultiVector& y,
00232 NOX::Abstract::MultiVector::DenseMatrix& b) const;
00233 virtual void multiply(double alpha, const MultiVector& y,
00234 NOX::Abstract::MultiVector::DenseMatrix& b) const;
00235
00237 virtual int length() const;
00238
00240 virtual int numVectors() const;
00241
00243 virtual void print(std::ostream& stream) const;
00244
00246
00248
00250 virtual Teuchos::RCP<const NOX::Abstract::MultiVector>
00251 getMultiVector(int i) const;
00252
00254 virtual Teuchos::RCP<NOX::Abstract::MultiVector>
00255 getMultiVector(int i);
00256
00258 virtual
00259 Teuchos::RCP<const NOX::Abstract::MultiVector::DenseMatrix>
00260 getScalars() const;
00261
00263 virtual Teuchos::RCP<NOX::Abstract::MultiVector::DenseMatrix>
00264 getScalars();
00265
00270 virtual
00271 Teuchos::RCP<const NOX::Abstract::MultiVector::DenseMatrix>
00272 getScalarRows(int num_rows, int row) const;
00273
00278 virtual Teuchos::RCP<NOX::Abstract::MultiVector::DenseMatrix>
00279 getScalarRows(int num_rows, int row);
00280
00282 virtual const double& getScalar(int i, int j) const;
00283
00285 virtual double& getScalar(int i, int j);
00286
00291 virtual Teuchos::RCP<LOCA::Extended::Vector> getVector(int i);
00292
00297 virtual Teuchos::RCP<const LOCA::Extended::Vector>
00298 getVector(int i) const;
00299
00301 virtual int getNumScalarRows() const;
00302
00304 virtual int getNumMultiVectors() const;
00305
00307
00308 protected:
00309
00311 MultiVector(const Teuchos::RCP<LOCA::GlobalData>& global_data,
00312 int nColumns, int nVectorRows, int nScalarRows);
00313
00315
00317
00323 virtual Teuchos::RCP<LOCA::Extended::Vector>
00324 generateVector(int nVecs, int nScalarRows) const;
00325
00327
00329
00331 void setMultiVectorPtr(
00332 int i,
00333 Teuchos::RCP<NOX::Abstract::MultiVector> v);
00334
00336 void checkDimensions(const string& callingFunction,
00337 const LOCA::Extended::MultiVector& a) const;
00338
00340 void checkIndex(const string& callingFunction, int i) const;
00341
00343 void checkVectorRowIndex(const string& callingFunction, int i) const;
00344
00346 void checkIndex(const string& callingFunction, int i, int j) const;
00347
00349 bool isContiguous(const vector<int>& index) const;
00350
00352
00353 protected:
00354
00356 Teuchos::RCP<LOCA::GlobalData> globalData;
00357
00362 int numColumns;
00363
00365 int numMultiVecRows;
00366
00368 int numScalarRows;
00369
00371 vector< Teuchos::RCP<NOX::Abstract::MultiVector> > multiVectorPtrs;
00372
00374 Teuchos::RCP<NOX::Abstract::MultiVector::DenseMatrix> scalarsPtr;
00375
00377 mutable vector< Teuchos::RCP<LOCA::Extended::Vector> > extendedVectorPtrs;
00378
00380 bool isView;
00381
00382 };
00383 }
00384 }
00385
00386 #endif