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 #include "NOX_Epetra_Vector.H"
00043 #include "NOX_Epetra_MultiVector.H"
00044 #include "Epetra_Vector.h"
00045 #include "NOX_Epetra_VectorSpace_L2.H"
00046
00047 NOX::Epetra::Vector::
00048 Vector(const Teuchos::RCP<Epetra_Vector>& source,
00049 NOX::Epetra::Vector::MemoryType memoryType,
00050 NOX::CopyType type,
00051 Teuchos::RCP<NOX::Epetra::VectorSpace> vs)
00052 {
00053 if (Teuchos::is_null(vs))
00054 vectorSpace = Teuchos::rcp(new NOX::Epetra::VectorSpaceL2);
00055 else
00056 vectorSpace = vs;
00057
00058 if (memoryType == NOX::Epetra::Vector::CreateView)
00059 epetraVec = source;
00060 else {
00061
00062 switch (type) {
00063
00064 case DeepCopy:
00065
00066 epetraVec = Teuchos::rcp(new Epetra_Vector(*source));
00067 break;
00068
00069 case ShapeCopy:
00070
00071 epetraVec = Teuchos::rcp(new Epetra_Vector(source->Map()));
00072 break;
00073 }
00074
00075 }
00076 }
00077
00078 NOX::Epetra::Vector::Vector(const Epetra_Vector& source, NOX::CopyType type,
00079 Teuchos::RCP<NOX::Epetra::VectorSpace> vs)
00080 {
00081 if (Teuchos::is_null(vs))
00082 vectorSpace = Teuchos::rcp(new NOX::Epetra::VectorSpaceL2);
00083 else
00084 vectorSpace = vs;
00085
00086 switch (type) {
00087
00088 case DeepCopy:
00089
00090 epetraVec = Teuchos::rcp(new Epetra_Vector(source));
00091 break;
00092
00093 case ShapeCopy:
00094
00095 epetraVec = Teuchos::rcp(new Epetra_Vector(source.Map()));
00096 break;
00097
00098 }
00099 }
00100
00101 NOX::Epetra::Vector::Vector(const NOX::Epetra::Vector& source,
00102 NOX::CopyType type)
00103 {
00104 vectorSpace = source.vectorSpace;
00105
00106 switch (type) {
00107
00108 case DeepCopy:
00109
00110 epetraVec = Teuchos::rcp(new Epetra_Vector(source.getEpetraVector()));
00111 break;
00112
00113 case ShapeCopy:
00114
00115 epetraVec =
00116 Teuchos::rcp(new Epetra_Vector(source.getEpetraVector().Map()));
00117 break;
00118
00119 }
00120 }
00121
00122 NOX::Epetra::Vector::~Vector()
00123 {
00124
00125 }
00126
00127 NOX::Abstract::Vector& NOX::Epetra::Vector::operator=(const Epetra_Vector& source)
00128 {
00129 epetraVec->Scale(1.0, source);
00130 return *this;
00131 }
00132
00133 NOX::Abstract::Vector& NOX::Epetra::Vector::operator=(const NOX::Abstract::Vector& source)
00134 {
00135 return operator=(dynamic_cast<const NOX::Epetra::Vector&>(source));
00136 }
00137
00138 NOX::Abstract::Vector& NOX::Epetra::Vector::operator=(const NOX::Epetra::Vector& source)
00139 {
00140 epetraVec->Scale(1.0, source.getEpetraVector());
00141 return *this;
00142 }
00143
00144 Epetra_Vector& NOX::Epetra::Vector::getEpetraVector()
00145 {
00146 return *epetraVec;
00147 }
00148
00149 const Epetra_Vector& NOX::Epetra::Vector::getEpetraVector() const
00150 {
00151 return *epetraVec;
00152 }
00153
00154 NOX::Abstract::Vector& NOX::Epetra::Vector::init(double value)
00155 {
00156 epetraVec->PutScalar(value);
00157 return *this;
00158 }
00159
00160 NOX::Abstract::Vector& NOX::Epetra::Vector::random(bool useSeed, int seed)
00161 {
00162 if (useSeed)
00163 epetraVec->SetSeed(seed);
00164 epetraVec->Random();
00165 return *this;
00166 }
00167
00168 NOX::Abstract::Vector& NOX::Epetra::Vector::abs(const NOX::Abstract::Vector& base)
00169 {
00170 return abs(dynamic_cast<const NOX::Epetra::Vector&>(base));
00171 }
00172
00173 NOX::Abstract::Vector& NOX::Epetra::Vector::abs(const NOX::Epetra::Vector& base)
00174 {
00175 epetraVec->Abs(base.getEpetraVector());
00176 return *this;
00177 }
00178
00179 NOX::Abstract::Vector& NOX::Epetra::Vector::reciprocal(const NOX::Abstract::Vector& base)
00180 {
00181 return reciprocal(dynamic_cast<const NOX::Epetra::Vector&>(base));
00182 }
00183
00184 NOX::Abstract::Vector& NOX::Epetra::Vector::reciprocal(const NOX::Epetra::Vector& base)
00185 {
00186 epetraVec->Reciprocal(base.getEpetraVector());
00187 return *this;
00188 }
00189
00190 NOX::Abstract::Vector& NOX::Epetra::Vector::scale(double alpha)
00191 {
00192 epetraVec->Scale(alpha);
00193 return *this;
00194 }
00195
00196 NOX::Abstract::Vector& NOX::Epetra::Vector::
00197 update(double alpha, const NOX::Abstract::Vector& a, double gamma)
00198 {
00199 return update(alpha, dynamic_cast<const NOX::Epetra::Vector&>(a), gamma);
00200 }
00201
00202 NOX::Abstract::Vector& NOX::Epetra::Vector::
00203 update(double alpha, const NOX::Epetra::Vector& a, double gamma)
00204 {
00205 epetraVec->Update(alpha, a.getEpetraVector(), gamma);
00206 return *this;
00207 }
00208
00209 NOX::Abstract::Vector& NOX::Epetra::Vector::
00210 update(double alpha, const NOX::Abstract::Vector& a,
00211 double beta, const NOX::Abstract::Vector& b,
00212 double gamma)
00213 {
00214 return update(alpha, dynamic_cast<const NOX::Epetra::Vector&>(a),
00215 beta, dynamic_cast<const NOX::Epetra::Vector&>(b), gamma);
00216 }
00217
00218 NOX::Abstract::Vector& NOX::Epetra::Vector::
00219 update(double alpha, const NOX::Epetra::Vector& a,
00220 double beta, const NOX::Epetra::Vector& b,
00221 double gamma)
00222 {
00223 epetraVec->Update(alpha, a.getEpetraVector(), beta, b.getEpetraVector(), gamma);
00224 return *this;
00225 }
00226
00227 NOX::Abstract::Vector& NOX::Epetra::Vector::scale(const NOX::Abstract::Vector& a)
00228 {
00229 return scale(dynamic_cast<const Epetra::Vector&>(a));
00230 }
00231
00232 NOX::Abstract::Vector& NOX::Epetra::Vector::scale(const NOX::Epetra::Vector& a)
00233 {
00234 epetraVec->Multiply(1.0, *epetraVec, a.getEpetraVector(), 0.0);
00235 return *this;
00236 }
00237
00238 Teuchos::RCP<NOX::Abstract::Vector> NOX::Epetra::Vector::
00239 clone(CopyType type) const
00240 {
00241 Teuchos::RCP<NOX::Abstract::Vector> newVec =
00242 Teuchos::rcp(new NOX::Epetra::Vector(*epetraVec, type, vectorSpace));
00243 return newVec;
00244 }
00245
00246 Teuchos::RCP<NOX::Abstract::MultiVector>
00247 NOX::Epetra::Vector::createMultiVector(
00248 const NOX::Abstract::Vector* const* vecs,
00249 int numVecs, NOX::CopyType type) const
00250 {
00251 if (numVecs < 0) {
00252 cerr << "NOX::Epetra::Vector::createMultiVector: Error! Multivector"
00253 << " must have postive number of columns!" << endl;
00254 throw "NOX Error";
00255 }
00256
00257 double** v = new double*[numVecs+1];
00258 const Epetra_BlockMap& map = epetraVec->Map();
00259 const NOX::Epetra::Vector* noxEpetraVecPtr;
00260 const Epetra_Vector* vecPtr;
00261
00262 epetraVec->ExtractView(&(v[0]));
00263 for (int i=0; i<numVecs; i++) {
00264 noxEpetraVecPtr = dynamic_cast<const NOX::Epetra::Vector*>(vecs[i]);
00265 vecPtr = &(noxEpetraVecPtr->getEpetraVector());
00266 vecPtr->ExtractView(&(v[i+1]));
00267 }
00268
00269 Epetra_MultiVector epetra_mv(View, map, v, numVecs+1);
00270
00271 Teuchos::RCP<NOX::Epetra::MultiVector> mv =
00272 Teuchos::rcp(new NOX::Epetra::MultiVector(epetra_mv, type));
00273
00274 delete [] v;
00275
00276 return mv;
00277 }
00278
00279 Teuchos::RCP<NOX::Abstract::MultiVector>
00280 NOX::Epetra::Vector::createMultiVector(int numVecs, NOX::CopyType type) const
00281 {
00282 if (numVecs <= 0) {
00283 cerr << "NOX::Epetra::Vector::createMultiVector: Error! Multivector"
00284 << " must have postive number of columns!" << endl;
00285 throw "NOX Error";
00286 }
00287
00288 const Epetra_BlockMap& map = epetraVec->Map();
00289 Epetra_MultiVector *epetra_mv;
00290
00291 if (type == NOX::ShapeCopy)
00292 epetra_mv = new Epetra_MultiVector(map, numVecs, true);
00293 else {
00294 epetra_mv = new Epetra_MultiVector(map, numVecs, false);
00295 Epetra_Vector* v;
00296 for (int i=0; i<numVecs; i++) {
00297 v = (*epetra_mv)(i);
00298 *v = *epetraVec;
00299 }
00300 }
00301
00302 Teuchos::RCP<NOX::Epetra::MultiVector> mv =
00303 Teuchos::rcp(new NOX::Epetra::MultiVector(*epetra_mv, type));
00304
00305 delete epetra_mv;
00306
00307 return mv;
00308 }
00309
00310 double NOX::Epetra::Vector::norm(NOX::Abstract::Vector::NormType type) const
00311 {
00312 return vectorSpace->norm(*epetraVec, type);
00313 }
00314
00315 double NOX::Epetra::Vector::norm(const NOX::Abstract::Vector& weights) const
00316 {
00317 return norm(dynamic_cast<const NOX::Epetra::Vector&>(weights));
00318 }
00319
00320 double NOX::Epetra::Vector::norm(const NOX::Epetra::Vector& weights) const
00321 {
00322 cerr << "NOX::Epetra::Vector - Weighted norm not supported" << endl;
00323 throw "NOX-Epetra Error";
00324 }
00325
00326 double NOX::Epetra::Vector::innerProduct(const NOX::Abstract::Vector& y) const
00327 {
00328 return innerProduct(dynamic_cast<const NOX::Epetra::Vector&>(y));
00329 }
00330
00331 double NOX::Epetra::Vector::innerProduct(const NOX::Epetra::Vector& y) const
00332 {
00333 return vectorSpace->innerProduct(*epetraVec, y.getEpetraVector());
00334 }
00335
00336 int NOX::Epetra::Vector::length() const
00337 {
00338 return epetraVec->GlobalLength();
00339 }
00340
00341 void NOX::Epetra::Vector::print(std::ostream& stream) const
00342 {
00343 epetraVec->Print(stream);
00344 return;
00345 }
00346
00347 Teuchos::RCP<NOX::Epetra::VectorSpace>
00348 NOX::Epetra::Vector::getVectorSpace() const
00349 {
00350 return vectorSpace;
00351 }