Stokhos Package Browser (Single Doxygen Collection)  Version of the Day
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
Stokhos_EpetraMultiVectorOrthogPoly.cpp
Go to the documentation of this file.
1 // @HEADER
2 // *****************************************************************************
3 // Stokhos Package
4 //
5 // Copyright 2009 NTESS and the Stokhos contributors.
6 // SPDX-License-Identifier: BSD-3-Clause
7 // *****************************************************************************
8 // @HEADER
9 
10 #include "Stokhos_EpetraMultiVectorOrthogPoly.hpp"
11 
12 Stokhos::EpetraMultiVectorOrthogPoly::
13 EpetraMultiVectorOrthogPoly() :
14  ProductContainer<Epetra_MultiVector>(),
15  VectorOrthogPoly<Epetra_MultiVector>(),
16  ProductEpetraMultiVector()
17 {
18 }
19 
20 Stokhos::EpetraMultiVectorOrthogPoly::
21 EpetraMultiVectorOrthogPoly(
22  const Teuchos::RCP<const Stokhos::OrthogPolyBasis<int, double> >& basis,
23  const Teuchos::RCP<const Epetra_BlockMap>& block_map) :
24  ProductContainer<Epetra_MultiVector>(block_map),
25  VectorOrthogPoly<Epetra_MultiVector>(basis, block_map),
26  ProductEpetraMultiVector(block_map)
27 {
28 }
29 
30 Stokhos::EpetraMultiVectorOrthogPoly::
31 EpetraMultiVectorOrthogPoly(
32  const Teuchos::RCP<const Stokhos::OrthogPolyBasis<int, double> >& basis,
33  const Teuchos::RCP<const Epetra_BlockMap>& block_map,
34  const Teuchos::RCP<const Epetra_BlockMap>& _coeff_map,
35  const Teuchos::RCP<const EpetraExt::MultiComm>& _product_comm,
36  int num_vectors) :
37  ProductContainer<Epetra_MultiVector>(block_map),
38  VectorOrthogPoly<Epetra_MultiVector>(basis, block_map),
39  ProductEpetraMultiVector(block_map, _coeff_map, _product_comm, num_vectors)
40 {
41 }
42 
43 Stokhos::EpetraMultiVectorOrthogPoly::
44 EpetraMultiVectorOrthogPoly(
45  const Teuchos::RCP<const Stokhos::OrthogPolyBasis<int, double> >& basis,
46  const Teuchos::RCP<const Epetra_BlockMap>& block_map,
47  const Teuchos::RCP<const Epetra_BlockMap>& _coeff_map,
48  const Teuchos::RCP<const Epetra_BlockMap>& _product_map,
49  const Teuchos::RCP<const EpetraExt::MultiComm>& _product_comm,
50  int num_vectors) :
51  ProductContainer<Epetra_MultiVector>(block_map),
52  VectorOrthogPoly<Epetra_MultiVector>(basis, block_map),
53  ProductEpetraMultiVector(block_map, _coeff_map, _product_map, _product_comm,
54  num_vectors)
55 {
56 }
57 
58 Stokhos::EpetraMultiVectorOrthogPoly::
59 EpetraMultiVectorOrthogPoly(
60  const Teuchos::RCP<const Stokhos::OrthogPolyBasis<int, double> >& basis,
61  const Teuchos::RCP<const Epetra_BlockMap>& block_map,
62  const Teuchos::RCP<const Epetra_BlockMap>& _coeff_map,
63  const Teuchos::RCP<const Epetra_BlockMap>& _product_map,
64  const Teuchos::RCP<const EpetraExt::MultiComm>& _product_comm,
65  Epetra_DataAccess CV,
66  const Epetra_MultiVector& block_vector) :
67  ProductContainer<Epetra_MultiVector>(block_map),
68  VectorOrthogPoly<Epetra_MultiVector>(basis, block_map),
69  ProductEpetraMultiVector(block_map, _coeff_map, _product_map, _product_comm, CV,
70  block_vector)
71 {
72 }
73 
74 Stokhos::EpetraMultiVectorOrthogPoly::
75 EpetraMultiVectorOrthogPoly(const Stokhos::EpetraMultiVectorOrthogPoly& v) :
76  ProductContainer<Epetra_MultiVector>(v),
77  VectorOrthogPoly<Epetra_MultiVector>(v),
78  ProductEpetraMultiVector(v)
79 {
80 }
81 
82 Stokhos::EpetraMultiVectorOrthogPoly::
83 ~EpetraMultiVectorOrthogPoly() {}
84 
85 Stokhos::EpetraMultiVectorOrthogPoly&
86 Stokhos::EpetraMultiVectorOrthogPoly::
87 operator=(const Stokhos::EpetraMultiVectorOrthogPoly& v) {
88  ProductEpetraMultiVector::operator=(v);
89  this->basis_ = v.basis_;
90  return *this;
91 }
92 
93 void
94 Stokhos::EpetraMultiVectorOrthogPoly::
95 reset(
96  const Teuchos::RCP<const Stokhos::OrthogPolyBasis<int, double> >& new_basis,
97  const Teuchos::RCP<const Epetra_BlockMap>& block_map,
98  const Teuchos::RCP<const Epetra_BlockMap>& _coeff_map,
99  const Teuchos::RCP<const EpetraExt::MultiComm>& _product_comm,
100  int num_vectors)
101 {
102  ProductEpetraMultiVector::reset(block_map, _coeff_map, _product_comm,
103  num_vectors);
104  this->basis_ = new_basis;
105 }
106 
107 void
108 Stokhos::EpetraMultiVectorOrthogPoly::
109 reset(
110  const Teuchos::RCP<const Stokhos::OrthogPolyBasis<int, double> >& new_basis,
111  const Teuchos::RCP<const Epetra_BlockMap>& block_map,
112  const Teuchos::RCP<const Epetra_BlockMap>& _coeff_map,
113  const Teuchos::RCP<const Epetra_BlockMap>& _product_map,
114  const Teuchos::RCP<const EpetraExt::MultiComm>& _product_comm,
115  int num_vectors)
116 {
117  ProductEpetraMultiVector::reset(block_map, _coeff_map, _product_map,
118  _product_comm, num_vectors);
119  this->basis_ = new_basis;
120 }
121 
122 void
123 Stokhos::EpetraMultiVectorOrthogPoly::
124 computeMean(Epetra_MultiVector& v) const
125 {
126  int lid = this->map_->LID(0);
127  v.Scale(1.0, *(this->coeff_[lid]));
128 }
129 
130 void
131 Stokhos::EpetraMultiVectorOrthogPoly::
132 computeVariance(Epetra_MultiVector& v) const
133 {
134  const Teuchos::Array<double>& nrm2 = this->basis_->norm_squared();
135  v.PutScalar(0.0);
136  int i_gid;
137  for (int i=1; i<this->size(); i++) {
138  i_gid = this->map_->GID(i);
139  v.Multiply(nrm2[i_gid], *(this->coeff_[i]), *(this->coeff_[i]), 1.0);
140  }
141 }
142 
143 void
144 Stokhos::EpetraMultiVectorOrthogPoly::
145 computeStandardDeviation(Epetra_MultiVector& v) const
146 {
147  computeVariance(v);
148 
149  for (int j=0; j<v.NumVectors(); j++)
150  for (int i=0; i<v.MyLength(); i++)
151  v[j][i] = std::sqrt(v[j][i]);
152 }
Sacado::UQ::sqrt
KOKKOS_INLINE_FUNCTION PCE< Storage > sqrt(const PCE< Storage > &a)
Definition: Sacado_UQ_PCE_Imp.hpp:847
Epetra_MultiVector
Stokhos::EpetraMultiVectorOrthogPoly::computeMean
void computeMean(Epetra_MultiVector &v) const
Compute mean.
Definition: Stokhos_EpetraMultiVectorOrthogPoly.cpp:124
Stokhos::EpetraMultiVectorOrthogPoly::operator=
EpetraMultiVectorOrthogPoly & operator=(const EpetraMultiVectorOrthogPoly &v)
Assignment.
Definition: Stokhos_EpetraMultiVectorOrthogPoly.cpp:87
Stokhos::EpetraMultiVectorOrthogPoly::computeStandardDeviation
void computeStandardDeviation(Epetra_MultiVector &v) const
Compute standard deviation.
Definition: Stokhos_EpetraMultiVectorOrthogPoly.cpp:145
Stokhos::EpetraMultiVectorOrthogPoly::reset
void reset(const Teuchos::RCP< const Stokhos::OrthogPolyBasis< int, double > > &basis, const Teuchos::RCP< const Epetra_BlockMap > &block_map, const Teuchos::RCP< const Epetra_BlockMap > &coeff_map, const Teuchos::RCP< const EpetraExt::MultiComm > &product_comm, int num_vectors)
Reset to a new basis.
Definition: Stokhos_EpetraMultiVectorOrthogPoly.cpp:95
Stokhos::ProductEpetraMultiVector::operator=
ProductEpetraMultiVector & operator=(const ProductEpetraMultiVector &v)
Assignment.
Definition: Stokhos_ProductEpetraMultiVector.cpp:95
j
j
Definition: Sacado_Fad_Exp_MP_Vector.hpp:495
Stokhos::EpetraMultiVectorOrthogPoly::computeVariance
void computeVariance(Epetra_MultiVector &v) const
Compute variance.
Definition: Stokhos_EpetraMultiVectorOrthogPoly.cpp:132
Stokhos::ProductEpetraMultiVector::reset
void reset(const Teuchos::RCP< const Epetra_BlockMap > &block_map, const Teuchos::RCP< const Epetra_BlockMap > &coeff_map, const Teuchos::RCP< const EpetraExt::MultiComm > &product_comm, int num_vectors)
Reset to a new size.
Definition: Stokhos_ProductEpetraMultiVector.cpp:179
Stokhos::ProductEpetraMultiVector
A container class storing products of Epetra_MultiVector&#39;s.
Definition: Stokhos_ProductEpetraMultiVector.hpp:24
Stokhos::VectorOrthogPoly::basis_
Teuchos::RCP< const Stokhos::OrthogPolyBasis< ordinal_type, value_type > > basis_
Basis.
Definition: Stokhos_VectorOrthogPoly.hpp:114
Stokhos::ProductContainer
A product (in the mathematical sense) container class whose coefficients are vectors, operators, or in general any type that would have an expensive copy constructor.
Definition: Stokhos_ProductContainer.hpp:29
Stokhos::VectorOrthogPoly
A container class storing an orthogonal polynomial whose coefficients are vectors, operators, or in general any type that would have an expensive copy constructor.
Definition: Stokhos_VectorOrthogPoly.hpp:24
Stokhos_EpetraMultiVectorOrthogPoly.hpp
Epetra_DataAccess
Epetra_DataAccess
Stokhos::EpetraMultiVectorOrthogPoly
A container class storing an orthogonal polynomial whose coefficients are vectors, operators, or in general any type that would have an expensive copy constructor.
Definition: Stokhos_EpetraMultiVectorOrthogPoly.hpp:23
Generated by   doxygen 1.8.5