dune-localfunctions  2.2.1
meta/power/basis.hh
Go to the documentation of this file.
1 // -*- tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*-
2 // vi: set et ts=8 sw=2 sts=2:
3 
4 #ifndef DUNE_LOCALFUNCTIONS_META_POWER_BASIS_HH
5 #define DUNE_LOCALFUNCTIONS_META_POWER_BASIS_HH
6 
7 #include <cstddef>
8 #include <vector>
9 
10 #include <dune/common/fmatrix.hh>
11 #include <dune/common/fvector.hh>
12 #include <dune/common/static_assert.hh>
13 
14 namespace Dune {
15 
17 
23  template<class Backend, std::size_t dimR>
24  class PowerBasis {
25  dune_static_assert(Backend::Traits::dimRange == 1, "PowerBasis works only "
26  "with scalar backends");
27 
28  // don't use a reference here so this class stays copyable
29  const Backend *backend;
30 
31  public:
33  struct Traits : public Backend::Traits
34  {
36  static const std::size_t dimRange = dimR;
38  typedef FieldVector<typename Traits::RangeField, dimR> Range;
39 
41 
45  typedef FieldMatrix<typename Traits::RangeField, dimR,
46  Traits::dimDomainGlobal> Jacobian;
47 
49  static const std::size_t diffOrder =
50  Backend::Traits::diffOrder > 0 ? 1 : 0;
51  };
52 
54 
60  PowerBasis(const Backend &backend_) : backend(&backend_) { }
61 
63  std::size_t size () const { return backend->size()*dimR; }
65  std::size_t order () const{ return backend->order(); }
66 
68  void evaluateFunction(const typename Traits::DomainLocal& in,
69  std::vector<typename Traits::Range>& out) const
70  {
71  std::vector<typename Backend::Traits::Range> backendValues;
72  backend->evaluateFunction(in, backendValues);
73  out.assign(size(), typename Traits::Range(0));
74  for(std::size_t d = 0; d < dimR; ++d)
75  for(std::size_t i = 0; i < backend->size(); ++i)
76  out[d*backend->size()+i][d] = backendValues[i][0];
77  }
78 
80  void evaluateJacobian(const typename Traits::DomainLocal& in,
81  std::vector<typename Traits::Jacobian>& out) const
82  {
83  std::vector<typename Backend::Traits::Jacobian> backendValues;
84  backend->evaluateJacobian(in, backendValues);
85  out.assign(size(), typename Traits::Jacobian(0));
86  for(std::size_t d = 0; d < dimR; ++d)
87  for(std::size_t i = 0; i < backend->size(); ++i)
88  out[d*backend->size()+i][d] = backendValues[i][0];
89  }
90  };
91 
92 } // namespace Dune
93 
94 #endif // DUNE_LOCALFUNCTIONS_META_POWER_BASIS_HH