escript  Revision_
speckley/src/WaveAssembler3D.h
Go to the documentation of this file.
1 
2 /*****************************************************************************
3 *
4 * Copyright (c) 2003-2020 by The University of Queensland
5 * http://www.uq.edu.au
6 *
7 * Primary Business: Queensland, Australia
8 * Licensed under the Apache License, version 2.0
9 * http://www.apache.org/licenses/LICENSE-2.0
10 *
11 * Development until 2012 by Earth Systems Science Computational Center (ESSCC)
12 * Development 2012-2013 by School of Earth Sciences
13 * Development from 2014-2017 by Centre for Geoscience Computing (GeoComp)
14 * Development from 2019 by School of Earth and Environmental Sciences
15 **
16 *****************************************************************************/
17 #ifndef __SPECKLEY_WAVE_ASSEMBLER_3D_H__
18 #define __SPECKLEY_WAVE_ASSEMBLER_3D_H__
19 
20 #include <speckley/Speckley.h>
21 #include <speckley/AbstractAssembler.h>
22 #include <speckley/Brick.h>
23 #include <speckley/SpeckleyException.h>
24 
25 #include <escript/Data.h>
26 
27 #include <map>
28 
29 namespace speckley {
30 
31 
32 class WaveAssembler3D : public AbstractAssembler
33 {
34 public:
35  WaveAssembler3D(escript::const_Domain_ptr dom, const double *dx,
36  const dim_t *NE, const dim_t *NN, const DataMap& c)
37  : AbstractAssembler(),
38  m_dx(dx),
39  m_NE(NE),
40  m_NN(NN)
41  {
42  domain = REFCOUNTNS::static_pointer_cast<const Brick>(dom);
43  isHTI = isVTI = false;
44  DataMap::const_iterator a = c.find("c12"), b = c.find("c23");
45  if (c.find("c11") == c.end()
46  || c.find("c13") == c.end() || c.find("c33") == c.end()
47  || c.find("c44") == c.end() || c.find("c66") == c.end()
48  || (a == c.end() && b == c.end()))
49  throw SpeckleyException("required constants missing for WaveAssembler");
50 
51  if (a != c.end() && b != c.end()) {
52  throw SpeckleyException("WaveAssembler3D() doesn't support general form waves");
53  } else if (a == c.end()) {
54  c23 = b->second;
55  isHTI = true;
57  throw SpeckleyException("C tensor elements must be reduced");
58  }
59  if (c23.isEmpty()) {
60  throw SpeckleyException("C tensor elements must not be empty");
61  }
62  } else if (b == c.end()) {
63  c12 = a->second;
64  isVTI = true;
66  throw SpeckleyException("C tensor elements must be reduced");
67  }
68  if (c12.isEmpty()) {
69  throw SpeckleyException("C tensor elements must not be empty");
70  }
71  } // final else case taken care of with the missing constants above
72  c11 = c.find("c11")->second;
73  c13 = c.find("c13")->second;
74  c33 = c.find("c33")->second;
75  c44 = c.find("c44")->second;
76  c66 = c.find("c66")->second;
82  throw SpeckleyException("C tensor elements must be reduced");
83  }
84  if (c11.isEmpty()
85  || c13.isEmpty()
86  || c33.isEmpty()
87  || c44.isEmpty()
88  || c66.isEmpty()) {
89  throw SpeckleyException("C tensor elements must not be empty");
90  }
91  }
92 
93  ~WaveAssembler3D() {}
94 
95  /* The default SpeckleyDomain assemblers, with original signatures */
96 
100  escript::Data& rhs, const escript::Data& A,
101  const escript::Data& B, const escript::Data& C,
102  const escript::Data& D, const escript::Data& X,
103  const escript::Data& Y) const;
104 
108  escript::Data& rhs, const escript::Data& d,
109  const escript::Data& y) const;
110 
114  escript::Data& rhs, const escript::Data& A, const escript::Data& B,
115  const escript::Data& C, const escript::Data& D,
116  const escript::Data& X, const escript::Data& Y) const;
117 
122  escript::Data& rhs, const escript::Data& d,
123  const escript::Data& y) const;
124 
128  escript::Data& rhs, const escript::Data& A,
129  const escript::Data& B, const escript::Data& C,
130  const escript::Data& D, const escript::Data& X,
131  const escript::Data& Y) const;
132 
136  escript::Data& rhs, const escript::Data& d,
137  const escript::Data& y) const;
138 
142  escript::Data& rhs, const escript::Data& A, const escript::Data& B,
143  const escript::Data& C, const escript::Data& D,
144  const escript::Data& X, const escript::Data& Y) const;
145 
150  escript::Data& rhs, const escript::Data& d,
151  const escript::Data& y) const;
152 
153  /* The new interface for assemblers */
154 
156  escript::Data& rhs, const DataMap& coefs) const;
158  escript::Data& rhs, const DataMap& coefs) const;
160  escript::Data& rhs, const DataMap& coefs) const;
163  escript::Data& rhs, const DataMap& coefs) const;
165  escript::Data& rhs, const DataMap& coefs) const;
167  escript::Data& rhs, const DataMap& coefs) const;
169  escript::Data& rhs, const DataMap& coefs) const;
172  escript::Data& rhs, const DataMap& coefs) const;
173 
174  void collateFunctionSpaceTypes(std::vector<int>& fsTypes,
175  const DataMap& coefs) const;
176 
177 protected:
178  POINTER_WRAPPER_CLASS(const Brick) domain;
179  const double *m_dx;
180  const dim_t *m_NE;
181  const dim_t *m_NN;
182  bool isHTI, isVTI;
184 };
185 
186 } // namespace speckley
187 
188 #endif // __SPECKLEY_WAVE_ASSEMBLER_3D_H__
189 
speckley::WaveAssembler3D::c11
escript::Data c11
Definition: speckley/src/WaveAssembler3D.h:211
speckley::WaveAssembler3D::c33
escript::Data c33
Definition: speckley/src/WaveAssembler3D.h:211
speckley::WaveAssembler3D::c44
escript::Data c44
Definition: speckley/src/WaveAssembler3D.h:211
speckley::WaveAssembler3D::assemblePDEBoundarySingle
virtual void assemblePDEBoundarySingle(escript::AbstractSystemMatrix *mat, escript::Data &rhs, const escript::Data &d, const escript::Data &y) const
Definition: speckley/src/WaveAssembler3D.cpp:516
speckley::WaveAssembler3D::WaveAssembler3D
WaveAssembler3D(escript::const_Domain_ptr dom, const double *dx, const dim_t *NE, const dim_t *NN, const DataMap &c)
Definition: speckley/src/WaveAssembler3D.h:63
speckley::WaveAssembler3D::assemblePDEBoundarySystemReduced
virtual void assemblePDEBoundarySystemReduced(escript::AbstractSystemMatrix *mat, escript::Data &rhs, const escript::Data &d, const escript::Data &y) const
Definition: speckley/src/WaveAssembler3D.cpp:537
escript::Data::getFunctionSpace
const FunctionSpace & getFunctionSpace() const
Return the function space.
Definition: Data.h:463
speckley::WaveAssembler3D::isHTI
bool isHTI
Definition: speckley/src/WaveAssembler3D.h:210
speckley
Definition: AbstractAssembler.cpp:19
speckley::WaveAssembler3D::assemblePDEBoundarySingleReduced
virtual void assemblePDEBoundarySingleReduced(escript::AbstractSystemMatrix *mat, escript::Data &rhs, const escript::Data &d, const escript::Data &y) const
Definition: speckley/src/WaveAssembler3D.cpp:523
escript::Data
Data represents a collection of datapoints.
Definition: Data.h:64
speckley::WaveAssembler3D::collateFunctionSpaceTypes
void collateFunctionSpaceTypes(std::vector< int > &fsTypes, const DataMap &coefs) const
Definition: speckley/src/WaveAssembler3D.cpp:116
escript::DataTypes::dim_t
index_t dim_t
Definition: DataTypes.h:93
speckley::WaveAssembler3D::c23
escript::Data c23
Definition: speckley/src/WaveAssembler3D.h:211
escript::AbstractSystemMatrix
Base class for escript system matrices.
Definition: AbstractSystemMatrix.h:57
speckley::Brick
Brick is the 3-dimensional implementation of a SpeckleyDomain.
Definition: speckley/src/Brick.h:47
speckley::WaveAssembler3D::c66
escript::Data c66
Definition: speckley/src/WaveAssembler3D.h:211
speckley::ReducedElements
@ ReducedElements
Definition: Speckley.h:81
escript::Data::isEmpty
bool isEmpty() const
Definition: Data.cpp:1132
speckley::WaveAssembler3D::assemblePDEBoundarySystem
virtual void assemblePDEBoundarySystem(escript::AbstractSystemMatrix *mat, escript::Data &rhs, const escript::Data &d, const escript::Data &y) const
Definition: speckley/src/WaveAssembler3D.cpp:530
speckley::WaveAssembler3D::POINTER_WRAPPER_CLASS
POINTER_WRAPPER_CLASS(const Brick) domain
speckley::WaveAssembler3D::assemblePDESingle
virtual void assemblePDESingle(escript::AbstractSystemMatrix *mat, escript::Data &rhs, const escript::Data &A, const escript::Data &B, const escript::Data &C, const escript::Data &D, const escript::Data &X, const escript::Data &Y) const
Definition: speckley/src/WaveAssembler3D.cpp:545
speckley::WaveAssembler3D::m_NE
const dim_t * m_NE
Definition: speckley/src/WaveAssembler3D.h:208
speckley::WaveAssembler3D::c13
escript::Data c13
Definition: speckley/src/WaveAssembler3D.h:211
speckley::WaveAssembler3D::assemblePDESystem
virtual void assemblePDESystem(escript::AbstractSystemMatrix *mat, escript::Data &rhs, const escript::Data &A, const escript::Data &B, const escript::Data &C, const escript::Data &D, const escript::Data &X, const escript::Data &Y) const
Definition: speckley/src/WaveAssembler3D.cpp:211
speckley::WaveAssembler3D::m_NN
const dim_t * m_NN
Definition: speckley/src/WaveAssembler3D.h:209
speckley::WaveAssembler3D::isVTI
bool isVTI
Definition: speckley/src/WaveAssembler3D.h:210
escript::FunctionSpace::getTypeCode
int getTypeCode() const
Returns the function space type code.
Definition: FunctionSpace.cpp:95
speckley::WaveAssembler3D::~WaveAssembler3D
~WaveAssembler3D()
Definition: speckley/src/WaveAssembler3D.h:121
speckley::DataMap
std::map< std::string, escript::Data > DataMap
Definition: speckley/src/domainhelpers.h:39
escript::const_Domain_ptr
boost::shared_ptr< const AbstractDomain > const_Domain_ptr
Definition: AbstractDomain.h:58
speckley::WaveAssembler3D::m_dx
const double * m_dx
Definition: speckley/src/WaveAssembler3D.h:207
speckley::WaveAssembler3D::c12
escript::Data c12
Definition: speckley/src/WaveAssembler3D.h:211
speckley::SpeckleyException
SpeckleyException exception class.
Definition: SpeckleyException.h:44
speckley::WaveAssembler3D::assemblePDESystemReduced
virtual void assemblePDESystemReduced(escript::AbstractSystemMatrix *mat, escript::Data &rhs, const escript::Data &A, const escript::Data &B, const escript::Data &C, const escript::Data &D, const escript::Data &X, const escript::Data &Y) const
Definition: speckley/src/WaveAssembler3D.cpp:506
speckley::WaveAssembler3D::assemblePDESingleReduced
virtual void assemblePDESingleReduced(escript::AbstractSystemMatrix *mat, escript::Data &rhs, const escript::Data &A, const escript::Data &B, const escript::Data &C, const escript::Data &D, const escript::Data &X, const escript::Data &Y) const
Definition: speckley/src/WaveAssembler3D.cpp:554