1#ifndef DUNE_FEM_SCHEMES_GALERKIN_HH
2#define DUNE_FEM_SCHEMES_GALERKIN_HH
13#include <dune/common/hybridutilities.hh>
16#include <dune/grid/common/rangegenerators.hh>
18#include <dune/fem/function/localfunction/temporary.hh>
19#include <dune/fem/io/parameter/reader.hh>
20#include <dune/fem/operator/common/automaticdifferenceoperator.hh>
21#include <dune/fem/operator/common/differentiableoperator.hh>
22#include <dune/fem/operator/common/operator.hh>
23#include <dune/fem/operator/common/stencil.hh>
24#include <dune/fem/operator/common/temporarylocalmatrix.hh>
25#include <dune/fem/quadrature/cachingquadrature.hh>
26#include <dune/fem/quadrature/intersectionquadrature.hh>
27#include <dune/fem/common/bindguard.hh>
29#include <dune/fem/misc/threads/threaditerator.hh>
30#include <dune/fem/misc/threads/threadsafevalue.hh>
31#include <dune/fem/misc/hasboundaryintersection.hh>
32#include <dune/fem/misc/griddeclaration.hh>
34#include <dune/fem/operator/common/localmatrixcolumn.hh>
35#include <dune/fem/operator/common/localcontribution.hh>
36#include <dune/fem/operator/1order/localmassmatrix.hh>
37#include <dune/fem/schemes/integrands.hh>
38#include <dune/fem/schemes/dirichletwrapper.hh>
39#include <dune/fem/schemes/femscheme.hh>
41#include <dune/fem/space/common/capabilities.hh>
44#include <dune/fempy/quadrature/fempyquadratures.hh>
59 static int callOrder(
const F& f,
char)
62 std::cerr <<
"WARNING: no order method available on " <<
typeid(F).name() <<
", defaulting to 1!" << std::endl;
68 static auto callOrder(
const F& f,
int) ->
decltype( f.order() )
75 static int order (
const F& f ) {
return callOrder(f, 0); }
81 template <
class Space>
82 struct DefaultGalerkinOperatorQuadratureSelector
84 typedef typename Space :: GridPartType GridPartType;
85 typedef typename GridPartType :: GridType GridType;
88 struct AnotherQuadSelector
90 typedef CachingQuadrature< GridPartType, 0, Capabilities::DefaultQuadrature< Space > :: template DefaultQuadratureTraits > InteriorQuadratureType;
91 typedef CachingQuadrature< GridPartType, 1, Capabilities::DefaultQuadrature< Space > :: template DefaultQuadratureTraits > SurfaceQuadratureType;
95#if HAVE_DUNE_POLYGONGRID
97 struct AnotherQuadSelector<
Dune::PolygonGrid< ct > >
104 typedef typename AnotherQuadSelector< GridType >::InteriorQuadratureType InteriorQuadratureType;
105 typedef typename AnotherQuadSelector< GridType >::SurfaceQuadratureType SurfaceQuadratureType;
111 template<
class Integrands,
template <
class>
class QuadSelector = DefaultGalerkinOperatorQuadratureSelector >
112 struct LocalGalerkinOperator
114 typedef LocalGalerkinOperator<Integrands> ThisType;
115 typedef std::conditional_t< Fem::IntegrandsTraits< Integrands >::isFull, Integrands, FullIntegrands< Integrands > > IntegrandsType;
117 typedef typename IntegrandsType::GridPartType GridPartType;
119 typedef typename GridPartType::ctype ctype;
120 typedef typename GridPartType::template Codim< 0 >::EntityType EntityType;
123 template <
class Space>
124 using QuadratureSelector = QuadSelector< Space >;
127 template<
class... Args >
128 explicit LocalGalerkinOperator (
const GridPartType &gridPart, Args &&... args )
129 : gridPart_( gridPart ),
130 integrands_(
std::forward< Args >( args )... ),
131 defaultInteriorOrder_( [] (const int order) {
return 2 * order; } ),
132 defaultSurfaceOrder_ ( [] (
const int order) {
return 2 * order + 1; } ),
133 interiorQuadOrder_(0), surfaceQuadOrder_(0)
138 typedef typename IntegrandsType::DomainValueType DomainValueType;
139 typedef typename IntegrandsType::RangeValueType RangeValueType;
140 typedef std::make_index_sequence< std::tuple_size< DomainValueType >::value > DomainValueIndices;
141 typedef std::make_index_sequence< std::tuple_size< RangeValueType >::value > RangeValueIndices;
144 template< std::size_t... i >
145 static auto makeDomainValueVector ( std::size_t maxNumLocalDofs, std::index_sequence< i... > )
147 return std::make_tuple( std::vector< std::tuple_element_t< i, DomainValueType > >( maxNumLocalDofs )... );
150 static auto makeDomainValueVector ( std::size_t maxNumLocalDofs )
152 return makeDomainValueVector( maxNumLocalDofs, DomainValueIndices() );
155 template< std::size_t... i >
156 static auto makeRangeValueVector ( std::size_t maxNumLocalDofs, std::index_sequence< i... > )
158 return std::make_tuple( std::vector< std::tuple_element_t< i, RangeValueType > >( maxNumLocalDofs )... );
161 static auto makeRangeValueVector ( std::size_t maxNumLocalDofs )
163 return makeRangeValueVector( maxNumLocalDofs, RangeValueIndices() );
166 typedef decltype( makeDomainValueVector( 0u ) ) DomainValueVectorType;
167 typedef decltype( makeRangeValueVector( 0u ) ) RangeValueVectorType;
169 static void resizeDomainValueVector ( DomainValueVectorType& vec,
const std::size_t
size )
172 std::get< i >( vec ).resize(
size );
176 static void resizeRangeValueVector ( RangeValueVectorType& vec,
const std::size_t
size )
179 std::get< i >( vec ).resize(
size );
184 void prepare(
const std::size_t
size )
const
186 resizeDomainValueVector( phiIn_,
size );
187 resizeDomainValueVector( phiOut_,
size );
188 resizeDomainValueVector( basisValues_,
size );
189 resizeDomainValueVector( domainValues_,
size );
192 template<
class LocalFunction,
class Quadrature >
193 static void evaluateQuadrature (
const LocalFunction &u,
const Quadrature &quad, std::vector< typename LocalFunction::RangeType > &phi )
195 u.evaluateQuadrature( quad, phi );
198 template<
class LocalFunction,
class Quadrature>
199 static void evaluateQuadrature (
const LocalFunction &u,
const Quadrature &quad, std::vector< typename LocalFunction::JacobianRangeType > &phi )
201 u.jacobianQuadrature( quad, phi );
204 template<
class LocalFunction,
class Quadrature >
205 static void evaluateQuadrature (
const LocalFunction &u,
const Quadrature &quad, std::vector< typename LocalFunction::HessianRangeType > &phi )
207 u.hessianQuadrature( quad, phi );
211 template<
class LocalFunction,
class Po
int >
214 u.evaluate( x, phi );
217 template<
class LocalFunction,
class Po
int >
220 u.jacobian( x, phi );
223 template<
class LocalFunction,
class Po
int >
229 template<
class LocalFunction,
class Point,
class... T >
230 static void value (
const LocalFunction &u,
const Point &x, std::tuple< T... > &phi )
232 Hybrid::forEach( std::index_sequence_for< T... >(), [ &u, &x, &phi ] (
auto i ) { LocalGalerkinOperator::value( u, x, std::get< i >( phi ) ); } );
235 template<
class Basis,
class Po
int >
236 static void values (
const Basis &basis,
const Point &x, std::vector< typename Basis::RangeType > &phi )
238 basis.evaluateAll( x, phi );
241 template<
class Basis,
class Po
int >
242 static void values (
const Basis &basis,
const Point &x, std::vector< typename Basis::JacobianRangeType > &phi )
244 basis.jacobianAll( x, phi );
247 template<
class Basis,
class Po
int >
248 static void values (
const Basis &basis,
const Point &x, std::vector< typename Basis::HessianRangeType > &phi )
250 basis.hessianAll( x, phi );
253 template<
class Basis,
class Point,
class... T >
254 static void values (
const Basis &basis,
const Point &x, std::tuple< std::vector< T >... > &phi )
256 Hybrid::forEach( std::index_sequence_for< T... >(), [ &basis, &x, &phi ] (
auto i ) { LocalGalerkinOperator::values( basis, x, std::get< i >( phi ) ); } );
259 template<
class LocalFunction,
class Po
int >
260 static DomainValueType domainValue (
const LocalFunction &u,
const Point &x )
267 static DomainValueType domainValue (
const unsigned int qpIdx, DomainValueVectorType& vec)
270 Hybrid::forEach( DomainValueIndices(), [ &qpIdx, &vec, &phi ] (
auto i ) {
271 std::get< i > ( phi ) = std::get< i >( vec )[ qpIdx ];
276 template<
class LocalFunction,
class Quadrature >
277 static void domainValue (
const LocalFunction &u,
const Quadrature& quadrature, DomainValueVectorType &result )
279 Hybrid::forEach( DomainValueIndices(), [ &u, &quadrature, &result ] (
auto i ) {
280 auto& vec = std::get< i >( result );
281 vec.resize( quadrature.nop() );
282 ThisType::evaluateQuadrature( u, quadrature, vec );
286 template<
class Phi, std::size_t... i >
287 static auto value (
const Phi &phi, std::size_t col, std::index_sequence< i... > )
289 return std::make_tuple( std::get< i >( phi )[ col ]... );
292 template<
class... T >
293 static auto value (
const std::tuple< std::vector< T >... > &phi, std::size_t col )
295 return value( phi, col, std::index_sequence_for< T... >() );
298 static void assignRange( RangeValueVectorType& ranges,
const std::size_t idx,
const RangeValueType&
range )
301 std::get< i >( ranges )[ idx ] = std::get< i >(
range );
305 static void assignRange( RangeValueVectorType& ranges,
const std::size_t idx,
const RangeValueType&
range,
const W &weight )
308 std::get< i >( ranges )[ idx ] = std::get< i >(
range );
309 std::get< i >( ranges )[ idx ] *= weight;
313 static void assignDomain( DomainValueVectorType& domains,
const std::size_t idx,
const DomainValueType& domain )
315 Hybrid::forEach( DomainValueIndices(), [ &domains, &idx, &domain ] (
auto i ) {
316 std::get< i >( domains )[ idx ] = std::get< i >( domain );
320 template <
class W,
class Quadrature>
321 static void axpyQuadrature( W& w,
const Quadrature& quadrature, RangeValueVectorType& ranges )
323 Hybrid::forEach( RangeValueIndices(), [ &w, &quadrature, &ranges ] (
auto i ) {
324 w.axpyQuadrature( quadrature, std::get< i >( ranges ) );
331 template<
class U,
class W >
332 void addInteriorIntegral (
const U &u, W &w )
const
334 if( !integrands().init( u.entity() ) )
337 const auto& geometry = u.geometry();
339 typedef typename QuadratureSelector< typename W::DiscreteFunctionSpaceType > :: InteriorQuadratureType InteriorQuadratureType;
340 const InteriorQuadratureType quadrature( u.entity(), interiorQuadratureOrder(maxOrder(u, w)) );
343 DomainValueVectorType& domains = domainValues_;
344 domainValue( u, quadrature, domains );
346 auto& ranges = values_;
347 resizeRangeValueVector( ranges, quadrature.nop() );
350 for(
const auto qp : quadrature )
352 const ctype weight = qp.weight() * geometry.integrationElement( qp.position() );
353 assignRange( ranges, qp.index(), integrands().
interior( qp, domainValue( qp.index(), domains ) ), weight );
357 axpyQuadrature( w, quadrature, ranges );
358 integrands().unbind();
361 template<
class U,
class J >
362 void addLinearizedInteriorIntegral (
const U &u, J &j )
const
364 if( !integrands().init( u.entity() ) )
367 const auto &geometry = u.geometry();
368 const auto &domainBasis = j.domainBasisFunctionSet();
369 const auto &rangeBasis = j.rangeBasisFunctionSet();
371 typedef typename QuadratureSelector< typename J::RangeSpaceType > :: InteriorQuadratureType InteriorQuadratureType;
372 const InteriorQuadratureType quadrature( u.entity(), interiorQuadratureOrder( maxOrder( u, domainBasis, rangeBasis )) );
373 const size_t domainSize = domainBasis.size();
374 const size_t quadNop = quadrature.nop();
376 auto& basisValues = basisValues_;
377 resizeDomainValueVector( basisValues, domainSize );
380 auto& rangeValues = rangeValues_;
381 DomainValueVectorType& domains = domainValues_;
382 domainValue( u, quadrature, domains );
384 rangeValues.resize( domainSize );
385 for( std::size_t col = 0; col < domainSize; ++col )
387 resizeRangeValueVector( rangeValues[ col ], quadNop );
391 for(
const auto qp : quadrature )
393 values( domainBasis, qp, basisValues );
394 const auto weight = qp.weight() * geometry.integrationElement( qp.position() );
395 auto integrand = integrands().linearizedInterior( qp, domainValue( qp.index(), domains ) );
396 for( std::size_t col = 0; col < domainSize; ++col )
398 assignRange( rangeValues[ col ], qp.index(), integrand( value( basisValues, col ) ), weight );
403 for( std::size_t col = 0; col < domainSize; ++col )
405 LocalMatrixColumn< J > jCol( j, col );
406 axpyQuadrature( jCol, quadrature, rangeValues[ col ] );
408 integrands().unbind();
413 template<
class Intersection,
class U,
class W >
414 void addBoundaryIntegral (
const Intersection &intersection,
const U &u, W &w )
const
416 if( !integrands().init( intersection ) )
419 const auto geometry = intersection.geometry();
420 typedef typename QuadratureSelector< typename W::DiscreteFunctionSpaceType > :: SurfaceQuadratureType SurfaceQuadratureType;
421 const SurfaceQuadratureType quadrature( gridPart(), intersection, surfaceQuadratureOrder(maxOrder( u, w )), SurfaceQuadratureType::INSIDE );
422 for(
const auto qp : quadrature )
424 const ctype weight = qp.weight() * geometry.integrationElement( qp.localPosition() );
426 RangeValueType integrand = integrands().boundary( qp, domainValue( u, qp ) );
428 Hybrid::forEach( RangeValueIndices(), [ &qp, &w, &integrand, weight ] (
auto i ) {
429 std::get< i >( integrand ) *= weight;
430 w.axpy( qp, std::get< i >( integrand ) );
433 integrands().unbind();
436 template<
class Intersection,
class U,
class J >
437 void addLinearizedBoundaryIntegral (
const Intersection &intersection,
const U &u, J &j )
const
439 if( !integrands().init( intersection ) )
442 DomainValueVectorType &phi = phiIn_;
444 const auto geometry = intersection.geometry();
445 const auto &domainBasis = j.domainBasisFunctionSet();
446 const auto &rangeBasis = j.rangeBasisFunctionSet();
448 typedef typename QuadratureSelector< typename J::RangeSpaceType > :: SurfaceQuadratureType SurfaceQuadratureType;
449 const SurfaceQuadratureType quadrature( gridPart(), intersection, surfaceQuadratureOrder(maxOrder(u, domainBasis, rangeBasis )), SurfaceQuadratureType::INSIDE );
450 for(
const auto qp : quadrature )
452 const ctype weight = qp.weight() * geometry.integrationElement( qp.localPosition() );
454 values( domainBasis, qp, phi );
455 auto integrand = integrands().linearizedBoundary( qp, domainValue( u, qp ) );
457 for( std::size_t col = 0, cols = domainBasis.size(); col < cols; ++col )
459 LocalMatrixColumn< J > jCol( j, col );
460 RangeValueType intPhi = integrand( value( phi, col ) );
462 Hybrid::forEach( RangeValueIndices(), [ &qp, &jCol, &intPhi, weight ] (
auto i ) {
463 std::get< i >( intPhi ) *= weight;
464 jCol.axpy( qp, std::get< i >( intPhi ) );
468 integrands().unbind();
474 template<
bool conforming,
class Intersection,
class U,
class W >
475 void addSkeletonIntegral (
const Intersection &intersection,
const U &uIn,
const U &uOut, W &wIn )
const
477 const auto geometry = intersection.geometry();
479 typedef typename QuadratureSelector< typename W::DiscreteFunctionSpaceType > :: SurfaceQuadratureType SurfaceQuadratureType;
480 typedef IntersectionQuadrature< SurfaceQuadratureType, conforming > IntersectionQuadratureType;
481 const IntersectionQuadratureType quadrature( gridPart(), intersection, surfaceQuadratureOrder(maxOrder( uIn, uOut, wIn)),
false );
482 for( std::size_t qp = 0, nop = quadrature.nop(); qp != nop; ++qp )
484 const ctype weight = quadrature.weight( qp ) * geometry.integrationElement( quadrature.localPoint( qp ) );
486 const auto qpIn = quadrature.inside()[ qp ];
487 const auto qpOut = quadrature.outside()[ qp ];
488 std::pair< RangeValueType, RangeValueType > integrand = integrands().skeleton( qpIn, domainValue( uIn, qpIn ), qpOut, domainValue( uOut, qpOut ) );
490 Hybrid::forEach( RangeValueIndices(), [ &qpIn, &wIn, &integrand, weight ] (
auto i ) {
491 std::get< i >( integrand.first ) *= weight;
492 wIn.axpy( qpIn, std::get< i >( integrand.first ) );
497 template<
bool conforming,
class Intersection,
class U,
class W >
498 void addSkeletonIntegral (
const Intersection &intersection,
const U &uIn,
const U &uOut, W &wIn, W &wOut )
const
500 const auto geometry = intersection.geometry();
501 typedef typename QuadratureSelector< typename W::DiscreteFunctionSpaceType > :: SurfaceQuadratureType SurfaceQuadratureType;
502 typedef IntersectionQuadrature< SurfaceQuadratureType, conforming > IntersectionQuadratureType;
503 const IntersectionQuadratureType quadrature( gridPart(), intersection, surfaceQuadratureOrder(maxOrder( uIn, uOut, wIn, wOut)),
false );
504 for( std::size_t qp = 0, nop = quadrature.nop(); qp != nop; ++qp )
506 const ctype weight = quadrature.weight( qp ) * geometry.integrationElement( quadrature.localPoint( qp ) );
508 const auto qpIn = quadrature.inside()[ qp ];
509 const auto qpOut = quadrature.outside()[ qp ];
510 std::pair< RangeValueType, RangeValueType > integrand = integrands().skeleton( qpIn, domainValue( uIn, qpIn ), qpOut, domainValue( uOut, qpOut ) );
512 Hybrid::forEach( RangeValueIndices(), [ &qpIn, &wIn, &qpOut, &wOut, &integrand, weight ] (
auto i ) {
513 std::get< i >( integrand.first ) *= weight;
514 wIn.axpy( qpIn, std::get< i >( integrand.first ) );
516 std::get< i >( integrand.second ) *= weight;
517 wOut.axpy( qpOut, std::get< i >( integrand.second ) );
522 template<
bool conforming,
class Intersection,
class U,
class J >
523 void addLinearizedSkeletonIntegral (
const Intersection &intersection,
524 const U &uIn,
const U &uOut, J &jInIn, J &jOutIn )
const
526 DomainValueVectorType &phiIn = phiIn_;
527 DomainValueVectorType &phiOut = phiOut_;
529 const auto &domainBasisIn = jInIn.domainBasisFunctionSet();
530 const auto &domainBasisOut = jOutIn.domainBasisFunctionSet();
532 const auto &rangeBasisIn = jInIn.rangeBasisFunctionSet();
534 const int order = std::max( maxOrder(uIn, uOut), maxOrder( domainBasisIn, domainBasisOut, rangeBasisIn ));
536 const auto geometry = intersection.geometry();
537 typedef typename QuadratureSelector< typename J::RangeSpaceType > :: SurfaceQuadratureType SurfaceQuadratureType;
538 typedef IntersectionQuadrature< SurfaceQuadratureType, conforming > IntersectionQuadratureType;
539 const IntersectionQuadratureType quadrature( gridPart(), intersection, surfaceQuadratureOrder(order),
false );
540 for( std::size_t qp = 0, nop = quadrature.nop(); qp != nop; ++qp )
542 const ctype weight = quadrature.weight( qp ) * geometry.integrationElement( quadrature.localPoint( qp ) );
544 const auto qpIn = quadrature.inside()[ qp ];
545 const auto qpOut = quadrature.outside()[ qp ];
547 values( domainBasisIn, qpIn, phiIn );
548 values( domainBasisOut, qpOut, phiOut );
550 auto integrand = integrands().linearizedSkeleton( qpIn, domainValue( uIn, qpIn ), qpOut, domainValue( uOut, qpOut ) );
551 for( std::size_t col = 0, cols = domainBasisIn.size(); col < cols; ++col )
553 LocalMatrixColumn< J > jInInCol( jInIn, col );
554 std::pair< RangeValueType, RangeValueType > intPhi = integrand.first( value( phiIn, col ) );
556 Hybrid::forEach( RangeValueIndices(), [ &qpIn, &jInInCol, &intPhi, weight ] (
auto i ) {
557 std::get< i >( intPhi.first ) *= weight;
558 jInInCol.axpy( qpIn, std::get< i >( intPhi.first ) );
561 for( std::size_t col = 0, cols = domainBasisOut.size(); col < cols; ++col )
563 LocalMatrixColumn< J > jOutInCol( jOutIn, col );
564 std::pair< RangeValueType, RangeValueType > intPhi = integrand.second( value( phiOut, col ) );
566 Hybrid::forEach( RangeValueIndices(), [ &qpIn, &jOutInCol, &intPhi, weight ] (
auto i ) {
567 std::get< i >( intPhi.first ) *= weight;
568 jOutInCol.axpy( qpIn, std::get< i >( intPhi.first ) );
574 template<
bool conforming,
class Intersection,
class U,
class J >
575 void addLinearizedSkeletonIntegral (
const Intersection &intersection,
const U &uIn,
const U &uOut,
576 J &jInIn, J &jOutIn, J &jInOut, J &jOutOut )
const
578 DomainValueVectorType &phiIn = phiIn_;
579 DomainValueVectorType &phiOut = phiOut_;
581 const auto &domainBasisIn = jInIn.domainBasisFunctionSet();
582 const auto &domainBasisOut = jOutIn.domainBasisFunctionSet();
584 const auto &rangeBasisIn = jInIn.rangeBasisFunctionSet();
585 const auto &rangeBasisOut = jInOut.rangeBasisFunctionSet();
587 const int order = std::max( maxOrder(uIn, uOut), maxOrder( domainBasisIn, domainBasisOut, rangeBasisIn, rangeBasisOut ));
589 const auto geometry = intersection.geometry();
590 typedef typename QuadratureSelector< typename J::RangeSpaceType > :: SurfaceQuadratureType SurfaceQuadratureType;
591 typedef IntersectionQuadrature< SurfaceQuadratureType, conforming > IntersectionQuadratureType;
592 const IntersectionQuadratureType quadrature( gridPart(), intersection, surfaceQuadratureOrder(order),
false );
593 for( std::size_t qp = 0, nop = quadrature.nop(); qp != nop; ++qp )
595 const ctype weight = quadrature.weight( qp ) * geometry.integrationElement( quadrature.localPoint( qp ) );
597 const auto qpIn = quadrature.inside()[ qp ];
598 const auto qpOut = quadrature.outside()[ qp ];
600 values( domainBasisIn, qpIn, phiIn );
601 values( domainBasisOut, qpOut, phiOut );
603 auto integrand = integrands().linearizedSkeleton( qpIn, domainValue( uIn, qpIn ), qpOut, domainValue( uOut, qpOut ) );
604 for( std::size_t col = 0, cols = domainBasisIn.size(); col < cols; ++col )
606 LocalMatrixColumn< J > jInInCol( jInIn, col );
607 LocalMatrixColumn< J > jInOutCol( jInOut, col );
608 std::pair< RangeValueType, RangeValueType > intPhi = integrand.first( value( phiIn, col ) );
610 Hybrid::forEach( RangeValueIndices(), [ &qpIn, &jInInCol, &qpOut, &jInOutCol, &intPhi, weight ] (
auto i ) {
611 std::get< i >( intPhi.first ) *= weight;
612 jInInCol.axpy( qpIn, std::get< i >( intPhi.first ) );
614 std::get< i >( intPhi.second ) *= weight;
615 jInOutCol.axpy( qpOut, std::get< i >( intPhi.second ) );
618 for( std::size_t col = 0, cols = domainBasisOut.size(); col < cols; ++col )
620 LocalMatrixColumn< J > jOutInCol( jOutIn, col );
621 LocalMatrixColumn< J > jOutOutCol( jOutOut, col );
622 std::pair< RangeValueType, RangeValueType > intPhi = integrand.second( value( phiOut, col ) );
624 Hybrid::forEach( RangeValueIndices(), [ &qpIn, &jOutInCol, &qpOut, &jOutOutCol, &intPhi, weight ] (
auto i ) {
625 std::get< i >( intPhi.first ) *= weight;
626 jOutInCol.axpy( qpIn, std::get< i >( intPhi.first ) );
628 std::get< i >( intPhi.second ) *= weight;
629 jOutOutCol.axpy( qpOut, std::get< i >( intPhi.second ) );
636 template<
class Intersection,
class U,
class... W >
637 void addSkeletonIntegral (
const Intersection &intersection,
const U &uIn,
const U &uOut, W &... w )
const
639 if( !integrands().init( intersection ) )
642 if( intersection.conforming() )
643 addSkeletonIntegral< true >( intersection, uIn, uOut, w... );
645 addSkeletonIntegral< false >( intersection, uIn, uOut, w... );
646 integrands().unbind();
649 template<
class Intersection,
class U,
class... J >
650 void addLinearizedSkeletonIntegral (
const Intersection &intersection,
const U &uIn,
const U &uOut, J &... j )
const
652 if( !integrands().init( intersection ) )
655 if( intersection.conforming() )
656 addLinearizedSkeletonIntegral< true >( intersection, uIn, uOut, j... );
658 addLinearizedSkeletonIntegral< false >( intersection, uIn, uOut, j... );
659 integrands().unbind();
662 void setQuadratureOrders(
unsigned int interior,
unsigned int surface)
665 surfaceQuadOrder_ = surface;
668 void setQuadratureOrderFunctions( std::function<
int(
const int)> interiorOrder,
669 std::function<
int(
const int)> surfaceOrder )
671 defaultInteriorOrder_ = interiorOrder;
672 defaultSurfaceOrder_ = surfaceOrder;
675 IntegrandsType& model()
const
679 bool nonlinear()
const {
return model().nonlinear(); }
680 bool hasInterior()
const {
return model().hasInterior(); }
681 bool hasSkeleton()
const {
return model().hasSkeleton(); }
682 bool hasBoundary()
const {
return model().hasBoundary(); }
685 IntegrandsType& integrands()
const
692 const GridPartType &gridPart ()
const {
return gridPart_; }
694 unsigned int interiorQuadratureOrder(
unsigned int order)
const {
return interiorQuadOrder_ == 0 ? defaultInteriorOrder_(order) : interiorQuadOrder_; }
695 unsigned int surfaceQuadratureOrder(
unsigned int order)
const {
return surfaceQuadOrder_ == 0 ? defaultSurfaceOrder_ (order) : surfaceQuadOrder_; }
699 int maxOrder(
const U& u )
const
701 return CallOrder< U > :: order( u );
704 template<
class U,
class W >
705 int maxOrder(
const U& u,
const W& w )
const
707 return std::max( maxOrder( u ), maxOrder( w ) );
710 template<
class U,
class V,
class W >
711 int maxOrder(
const U& u,
const V& v,
const W& w )
const
713 return std::max( maxOrder( u, v ), maxOrder( w ) );
716 template<
class U,
class V,
class W,
class X >
717 int maxOrder(
const U& u,
const V& v,
const W& w,
const X& x )
const
719 return std::max( maxOrder( u, v ), maxOrder( w, x) );
723 const GridPartType &gridPart_;
725 mutable IntegrandsType integrands_;
727 mutable std::function<int(
const int)> defaultInteriorOrder_;
728 mutable std::function<int(
const int)> defaultSurfaceOrder_;
730 unsigned int interiorQuadOrder_;
731 unsigned int surfaceQuadOrder_;
733 mutable std::vector< RangeValueVectorType > rangeValues_;
734 mutable RangeValueVectorType values_;
735 mutable DomainValueVectorType phiIn_;
736 mutable DomainValueVectorType phiOut_;
737 mutable DomainValueVectorType basisValues_;
738 mutable DomainValueVectorType domainValues_;
746 template<
class Gr
idPart >
748 struct GalerkinOperator
750 typedef GridPart GridPartType;
751 typedef GalerkinOperator< GridPartType > ThisType;
753 typedef typename GridPartType::ctype ctype;
754 typedef typename GridPartType::template Codim< 0 >::EntityType EntityType;
757 explicit GalerkinOperator (
const GridPartType &gridPart )
758 : gridPart_( gridPart ),
759 gridSizeInterior_( 0 )
764 template <
class IntegrandsTuple>
765 bool hasBoundary(
const IntegrandsTuple& integrandsTuple )
const
767 typedef std::make_index_sequence< std::tuple_size< IntegrandsTuple >::value > Indices;
768 bool hasBoundary = false ;
769 Hybrid::forEach( Indices(), [&integrandsTuple, &hasBoundary](
auto i ) {
770 if( std::get< i > (integrandsTuple).hasBoundary() )
779 template<
class Gr
idFunction,
class DiscreteFunction,
class Iterators,
class IntegrandsTuple,
class Functor,
bool hasSkeleton >
780 void evaluateImpl (
const GridFunction &u, DiscreteFunction &w,
const Iterators& iterators,
781 const IntegrandsTuple& integrandsTuple, Functor& addLocalDofs, std::integral_constant<bool, hasSkeleton> )
const
783 Dune::Fem::ConstLocalFunction< GridFunction > uInside( u );
784 Dune::Fem::ConstLocalFunction< GridFunction > uOutside( u );
786 typedef typename DiscreteFunction::DiscreteFunctionSpaceType DiscreteFunctionSpaceType;
790 gridSizeInterior_ = 0;
792 typedef std::make_index_sequence< std::tuple_size< IntegrandsTuple >::value > Indices;
795 const bool hasBnd = hasBoundary( integrandsTuple );
797 const auto &indexSet = gridPart().indexSet();
798 const auto end = iterators.end();
799 for(
auto it = iterators.begin(); it != end; ++it )
802 const EntityType inside = *it ;
807 auto uGuard = bindGuard( uInside, inside );
808 auto wGuard = bindGuard( wInside, inside );
811 auto addInteriorIntegral = [&integrandsTuple, &uInside, &wInside](
auto i )
813 const auto& integrands = std::get< i >( integrandsTuple );
814 if( integrands.hasInterior() )
815 integrands.addInteriorIntegral( uInside, wInside );
820 if( hasSkeleton || (hasBnd && HasBoundaryIntersection<GridPartType>::apply(inside) ) )
822 for(
const auto &intersection : intersections( gridPart(), inside ) )
824 bool neighbor =
false;
825 if constexpr ( hasSkeleton )
829 if( intersection.neighbor() )
832 const EntityType outside = intersection.outside();
836 auto uOutGuard = bindGuard( uOutside, outside );
838 auto addSkeletonIntegral = [&integrandsTuple, &intersection, &uInside, &uOutside, &wInside] (
auto i )
840 const auto& integrands = std::get< i >( integrandsTuple );
841 if( integrands.hasSkeleton() )
842 integrands.addSkeletonIntegral( intersection, uInside, uOutside, wInside );
847 else if( indexSet.index( inside ) < indexSet.index( outside ) )
849 auto uOutGuard = bindGuard( uOutside, outside );
850 auto wOutGuard = bindGuard( wOutside, outside );
853 auto addSkeletonIntegral = [&integrandsTuple, &intersection, &uInside, &uOutside, &wInside, &wOutside] (
auto i )
855 const auto& integrands = std::get< i >( integrandsTuple );
856 if( integrands.hasSkeleton() )
857 integrands.addSkeletonIntegral( intersection, uInside, uOutside, wInside, wOutside );
864 addLocalDofs( outside, wOutside );
869 if( ! neighbor && intersection.boundary() )
871 auto addBoundaryIntegral = [&integrandsTuple, &intersection, &uInside, &wInside](
auto i )
873 const auto& integrands = std::get< i >( integrandsTuple );
874 if( integrands.hasBoundary() )
875 integrands.addBoundaryIntegral( intersection, uInside, wInside );
883 addLocalDofs( inside, wInside );
887 template <
class Space>
890 typedef typename Space::EntityType EntityType;
891 template <
class Iterators>
892 InsideEntity(
const Space &space,
const Iterators& iterators)
893 : space_(space), dofThread_(space.
size(),-1)
894 , thread_( iterators.thread() )
896 const auto& mapper = space_.blockMapper();
897 for (
const auto &entity : space_)
899 int t=iterators.threadParallel(entity);
900 mapper.mapEach(entity, [
this, t ] (
int local,
auto global )
901 { dofThread_[global] = (dofThread_[global]==t || dofThread_[global]==-1)?
905 bool operator()(
const EntityType &entity)
const
907 bool needsLocking =
false;
908 space_.blockMapper().mapEach(entity,
909 [
this, &needsLocking ] (
int local,
auto global )
910 { needsLocking = (needsLocking || dofThread_[global]!=thread_); });
911 return !needsLocking;
914 std::vector<int> dofThread_;
918 template <
class DiscreteFunction>
919 struct AddLocalEvaluate
921 AddLocalEvaluate(DiscreteFunction &w)
923 template <
class LocalDofs>
924 void operator () (
const EntityType& entity,
const LocalDofs& wLocal )
const
926 w_.addLocalDofs( entity, wLocal.localDofVector() );
928 DiscreteFunction &w_;
931 template <
class DiscreteFunction>
932 struct AddLocalEvaluateLocked :
public AddLocalEvaluate<DiscreteFunction>
934 typedef AddLocalEvaluate<DiscreteFunction> BaseType;
936 std::shared_mutex& mutex_;
937 InsideEntity<typename DiscreteFunction::DiscreteFunctionSpaceType> inside_;
939 template <
class Iterators>
940 AddLocalEvaluateLocked(DiscreteFunction &w, std::shared_mutex& mtx,
const Iterators &iterators)
941 : BaseType(w), mutex_(mtx), inside_(w.space(),iterators) {}
943 template <
class LocalDofs>
944 void operator () (
const EntityType& entity,
const LocalDofs& wLocal )
const
949 std::shared_lock<std::shared_mutex> guard ( mutex_ );
950 BaseType::operator()( entity, wLocal );
955 std::lock_guard<std::shared_mutex> guard ( mutex_ );
956 BaseType::operator()( entity, wLocal );
961 template<
class Gr
idFunction,
class DiscreteFunction,
class Iterators,
class IntegrandsTuple,
class Functor >
962 void evaluate (
const GridFunction &u, DiscreteFunction &w,
const Iterators& iterators,
963 const IntegrandsTuple& integrandsTuple, Functor& addLocalDofs )
const
965 static_assert( std::is_same< typename GridFunction::GridPartType, GridPartType >::value,
"Argument 'u' and Integrands must be defined on the same grid part." );
966 static_assert( std::is_same< typename DiscreteFunction::GridPartType, GridPartType >::value,
"Argument 'w' and Integrands must be defined on the same grid part." );
968 if( hasSkeleton( integrandsTuple ) )
969 evaluateImpl( u, w, iterators, integrandsTuple, addLocalDofs, std::true_type() );
971 evaluateImpl( u, w, iterators, integrandsTuple, addLocalDofs, std::false_type() );
975 template <
class IntegrandsTuple>
976 bool hasSkeleton(
const IntegrandsTuple& integrandsTuple )
const
978 typedef std::make_index_sequence< std::tuple_size< IntegrandsTuple >::value > Indices;
979 bool hasSkeleton = false ;
980 Hybrid::forEach( Indices(), [&integrandsTuple, &hasSkeleton] (
auto i ) {
981 if( std::get< i >( integrandsTuple ).hasSkeleton() )
990 template<
class Gr
idFunction,
class DiscreteFunction,
class Iterators,
class IntegrandsTuple >
991 void evaluate (
const GridFunction &u, DiscreteFunction &w,
const Iterators& iterators,
992 const IntegrandsTuple& integrandsTuple, std::shared_mutex& mtx )
const
994 AddLocalEvaluateLocked<DiscreteFunction> addLocalEvaluate(w,mtx,iterators);
995 evaluate( u, w, iterators, integrandsTuple, addLocalEvaluate );
998 template<
class Gr
idFunction,
class DiscreteFunction,
class Iterators,
class IntegrandsTuple >
999 void evaluate (
const GridFunction &u, DiscreteFunction &w,
const Iterators& iterators,
const IntegrandsTuple& integrandsTuple )
const
1001 AddLocalEvaluate<DiscreteFunction> addLocalEvaluate(w);
1002 evaluate( u, w, iterators, integrandsTuple, addLocalEvaluate );
1006 template<
class T,
int length>
1011 FiniteStack () : _f(0) {}
1014 bool empty ()
const {
return _f <= 0; }
1017 bool full ()
const {
return (_f >= length); }
1020 void clear() { _f = 0; }
1023 void push (
const T& t)
1025 assert ( _f < length );
1042 int size ()
const {
return _f; }
1050 template <
class JacobianOperator>
1051 struct AddLocalAssemble
1053 typedef typename JacobianOperator::DomainSpaceType DomainSpaceType;
1054 typedef typename JacobianOperator::RangeSpaceType RangeSpaceType;
1055 typedef TemporaryLocalMatrix< DomainSpaceType, RangeSpaceType > TemporaryLocalMatrixType;
1056 JacobianOperator &jOp_;
1057 std::vector< TemporaryLocalMatrixType > jOpLocal_;
1059 FiniteStack< TemporaryLocalMatrixType*, 12 > jOpLocalFinalized_;
1060 FiniteStack< TemporaryLocalMatrixType*, 12 > jOpLocalFree_;
1062 std::size_t locked, notLocked, timesLocked;
1063 AddLocalAssemble(JacobianOperator& jOp)
1065 , jOpLocal_(12, TemporaryLocalMatrixType(jOp_.domainSpace(), jOp_.rangeSpace()))
1066 , jOpLocalFinalized_()
1068 , locked(0), notLocked(0), timesLocked(0)
1070 for(
auto& jOpLocal : jOpLocal_ )
1071 jOpLocalFree_.push( &jOpLocal );
1074 TemporaryLocalMatrixType& bind(
const EntityType& dE,
const EntityType& rE)
1076 assert( ! jOpLocalFree_.empty() );
1077 TemporaryLocalMatrixType& lop = *(jOpLocalFree_.pop());
1083 void unbind(TemporaryLocalMatrixType &lop)
1086 jOp_.addLocalMatrix( lop.domainEntity(), lop.rangeEntity(), lop );
1088 jOpLocalFree_.push( &lop );
1093 locked += jOpLocalFinalized_.size();
1094 while ( ! jOpLocalFinalized_.empty() )
1096 TemporaryLocalMatrixType &lop = *(jOpLocalFinalized_.pop());
1097 jOp_.addLocalMatrix( lop.domainEntity(), lop.rangeEntity(), lop );
1099 jOpLocalFree_.push( &lop );
1104 template <
class JacobianOperator>
1105 struct AddLocalAssembleLocked :
public AddLocalAssemble<JacobianOperator>
1107 typedef AddLocalAssemble<JacobianOperator> BaseType;
1108 typedef typename BaseType::TemporaryLocalMatrixType TemporaryLocalMatrixType;
1109 using BaseType::jOpLocalFinalized_;
1110 using BaseType::jOpLocalFree_;
1112 std::shared_mutex& mutex_;
1113 InsideEntity<typename JacobianOperator::DomainSpaceType> insideDomain_;
1114 InsideEntity<typename JacobianOperator::RangeSpaceType> insideRange_;
1116 template <
class Iterators>
1117 AddLocalAssembleLocked(JacobianOperator &jOp, std::shared_mutex &mtx,
const Iterators &iterators)
1120 , insideDomain_(jOp.domainSpace(),iterators)
1121 , insideRange_(jOp.rangeSpace(),iterators)
1127 ++BaseType::timesLocked;
1128 std::lock_guard<std::shared_mutex> guard ( mutex_ );
1129 BaseType::finalize();
1132 TemporaryLocalMatrixType& bind(
const EntityType& dE,
const EntityType& rE)
1134 if ( jOpLocalFree_.empty() )
1138 return BaseType::bind(dE,rE);
1141 void unbind(TemporaryLocalMatrixType &lop)
1150 if ( insideDomain_(lop.domainEntity()) &&
1151 insideRange_(lop.rangeEntity()) )
1153 std::shared_lock<std::shared_mutex> guard ( mutex_ );
1154 BaseType::unbind(lop);
1158 jOpLocalFinalized_.push( &lop );
1163 template<
class Gr
idFunction,
class JacobianOperator,
class Iterators,
class IntegrandsTuple,
class Functor,
bool hasSkeleton >
1164 void assembleImpl (
const GridFunction &u, JacobianOperator &jOp,
const Iterators& iterators,
const IntegrandsTuple& integrandsTuple,
1165 Functor& addLocalMatrix, std::integral_constant<bool, hasSkeleton> )
const
1167 typedef typename JacobianOperator::DomainSpaceType DomainSpaceType;
1168 typedef typename JacobianOperator::RangeSpaceType RangeSpaceType;
1170 typedef TemporaryLocalMatrix< DomainSpaceType, RangeSpaceType > TemporaryLocalMatrixType;
1172 Dune::Fem::ConstLocalFunction< GridFunction > uIn( u );
1173 Dune::Fem::ConstLocalFunction< GridFunction > uOut( u );
1175 typedef std::make_index_sequence< std::tuple_size< IntegrandsTuple >::value > Indices;
1176 const std::size_t maxNumLocalDofs = jOp.domainSpace().blockMapper().maxNumDofs() * jOp.domainSpace().localBlockSize;
1179 Hybrid::forEach( Indices(), [&integrandsTuple, &maxNumLocalDofs] (
auto i ) {
1180 const auto& integrands = std::get< i >( integrandsTuple );
1181 integrands.prepare( maxNumLocalDofs );
1185 gridSizeInterior_ = 0;
1188 const bool hasBnd = hasBoundary( integrandsTuple );
1190 const auto &indexSet = gridPart().indexSet();
1192 const auto end = iterators.end();
1193 for(
auto it = iterators.begin(); it != end; ++it )
1196 ++gridSizeInterior_;
1198 const EntityType inside = *it;
1200 auto uiGuard = bindGuard( uIn, inside );
1202 TemporaryLocalMatrixType& jOpInIn = addLocalMatrix.bind( inside, inside );
1203 auto addLinearizedInteriorIntegral = [&integrandsTuple, &uIn, &jOpInIn](
auto i )
1205 const auto& integrands = std::get< i >( integrandsTuple );
1206 if( integrands.hasInterior() )
1207 integrands.addLinearizedInteriorIntegral( uIn, jOpInIn );
1212 if( hasSkeleton || (hasBnd && HasBoundaryIntersection<GridPartType>::apply(inside) ) )
1214 for(
const auto &intersection : intersections( gridPart(), inside ) )
1216 bool neighbor = false ;
1219 if constexpr ( hasSkeleton )
1221 if( intersection.neighbor() )
1224 const EntityType &outside = intersection.outside();
1226 TemporaryLocalMatrixType &jOpOutIn = addLocalMatrix.bind( outside, inside );
1228 auto uoGuard = bindGuard( uOut, outside );
1232 auto addLinearizedSkeletonIntegral = [&integrandsTuple, &intersection, &uIn, &uOut, &jOpInIn, &jOpOutIn](
auto i )
1234 const auto& integrands = std::get< i >( integrandsTuple );
1235 if( integrands.hasSkeleton() )
1236 integrands.addLinearizedSkeletonIntegral( intersection, uIn, uOut, jOpInIn, jOpOutIn );
1241 else if( indexSet.index( inside ) < indexSet.index( outside ) )
1243 TemporaryLocalMatrixType &jOpInOut = addLocalMatrix.bind( inside, outside );
1244 TemporaryLocalMatrixType &jOpOutOut = addLocalMatrix.bind( outside, outside );
1246 auto addLinearizedSkeletonIntegral = [&integrandsTuple, &intersection, &uIn, &uOut, &jOpInIn, &jOpOutIn, &jOpInOut, &jOpOutOut](
auto i )
1248 const auto& integrands = std::get< i >( integrandsTuple );
1249 if( integrands.hasSkeleton() )
1250 integrands.addLinearizedSkeletonIntegral( intersection, uIn, uOut, jOpInIn, jOpOutIn, jOpInOut, jOpOutOut );
1255 addLocalMatrix.unbind(jOpInOut);
1256 addLocalMatrix.unbind(jOpOutOut);
1259 addLocalMatrix.unbind(jOpOutIn);
1263 if( !neighbor && intersection.boundary() )
1265 auto addLinearizedBoundaryIntegral = [&integrandsTuple, &intersection, &uIn, &jOpInIn](
auto i )
1267 const auto& integrands = std::get< i >( integrandsTuple );
1268 if( integrands.hasBoundary() )
1269 integrands.addLinearizedBoundaryIntegral( intersection, uIn, jOpInIn );
1277 addLocalMatrix.unbind(jOpInIn);
1281 addLocalMatrix.finalize();
1285 template<
class Gr
idFunction,
class JacobianOperator,
class Iterators,
class IntegrandsTuple,
class Functor >
1286 void assemble (
const GridFunction &u, JacobianOperator &jOp,
const Iterators& iterators,
1287 const IntegrandsTuple& integrandsTuple, Functor& addLocalMatrix,
int )
const
1289 static_assert( std::is_same< typename GridFunction::GridPartType, GridPartType >::value,
"Argument 'u' and Integrands must be defined on the same grid part." );
1290 static_assert( std::is_same< typename JacobianOperator::DomainSpaceType::GridPartType, GridPartType >::value,
"Argument 'jOp' and Integrands must be defined on the same grid part." );
1291 static_assert( std::is_same< typename JacobianOperator::RangeSpaceType::GridPartType, GridPartType >::value,
"Argument 'jOp' and Integrands must be defined on the same grid part." );
1293 if( hasSkeleton( integrandsTuple ) )
1294 assembleImpl( u, jOp, iterators, integrandsTuple ,addLocalMatrix, std::true_type() );
1296 assembleImpl( u, jOp, iterators, integrandsTuple, addLocalMatrix, std::false_type() );
1300 template<
class Gr
idFunction,
class JacobianOperator,
class Iterators,
class IntegrandsTuple>
1301 void assemble (
const GridFunction &u, JacobianOperator &jOp,
const Iterators& iterators,
1302 const IntegrandsTuple& integrandsTuple, std::shared_mutex& mtx)
const
1304 AddLocalAssembleLocked<JacobianOperator> addLocalAssemble( jOp, mtx, iterators);
1305 assemble( u, jOp, iterators, integrandsTuple, addLocalAssemble, 10 );
1307 std::lock_guard guard ( mtx );
1308 std::cout << MPIManager::thread() <<
" : "
1309 << addLocalAssemble.locked <<
" " << addLocalAssemble.notLocked <<
" "
1310 << addLocalAssemble.timesLocked << std::endl;
1314 template<
class Gr
idFunction,
class JacobianOperator,
class Iterators,
class IntegrandsTuple>
1315 void assemble (
const GridFunction &u, JacobianOperator &jOp,
const Iterators& iterators,
const IntegrandsTuple& integrandsTuple )
const
1317 AddLocalAssemble<JacobianOperator> addLocalAssemble(jOp);
1318 assemble( u, jOp, iterators, integrandsTuple, addLocalAssemble, 10 );
1322 const GridPartType &gridPart ()
const {
return gridPart_; }
1324 std::size_t gridSizeInterior ()
const {
return gridSizeInterior_; }
1327 const GridPartType &gridPart_;
1328 mutable std::size_t gridSizeInterior_;
1332 template <
class GalerkinOperator >
1333 static std::size_t accumulateGridSize(
const ThreadSafeValue< GalerkinOperator >& ops )
1335 std::size_t s = ops.size();
1336 std::size_t sum = 0;
1337 for( std::size_t i=0; i<s; ++i )
1338 sum += ops[ i ].gridSizeInterior();
1351 template<
class Integrands,
class DomainFunction,
class RangeFunction = DomainFunction >
1352 struct GalerkinOperator
1353 :
public virtual Operator< DomainFunction, RangeFunction >
1355 typedef DomainFunction DomainFunctionType;
1356 typedef RangeFunction RangeFunctionType;
1358 typedef typename RangeFunctionType::GridPartType GridPartType;
1360 typedef Impl::LocalGalerkinOperator< Integrands > LocalGalerkinOperatorImplType;
1361 typedef Impl::GalerkinOperator< GridPartType > GalerkinOperatorImplType;
1363 static_assert( std::is_same< typename DomainFunctionType::GridPartType, typename RangeFunctionType::GridPartType >::value,
"DomainFunction and RangeFunction must be defined on the same grid part." );
1367 template<
class... Args >
1368 explicit GalerkinOperator (
const GridPartType &gridPart, Args &&... args )
1369 : iterators_( gridPart ),
1370 opImpl_( gridPart ),
1371 localOp_( gridPart,
std::forward< Args >( args )... ),
1372 gridSizeInterior_( 0 ),
1373 communicate_( true )
1377 void setCommunicate(
const bool communicate )
1379 communicate_ = communicate;
1382 std::cout <<
"GalerkinOperator::setCommunicate: communicate was disabled!" << std::endl;
1386 void setQuadratureOrders(
unsigned int interior,
unsigned int surface)
1388 size_t size = localOp_.size();
1389 for(
size_t i=0; i<
size; ++i )
1390 localOp_[ i ].setQuadratureOrders(interior,surface);
1393 virtual bool nonlinear() const final
override
1395 return localOperator().nonlinear();
1398 virtual void operator() (
const DomainFunctionType &u, RangeFunctionType &w )
const final override
1403 template<
class Gr
idFunction >
1404 void operator() (
const GridFunction &u, RangeFunctionType &w )
const
1409 const GridPartType &gridPart ()
const {
return op().gridPart(); }
1411 typedef Integrands ModelType;
1412 typedef Integrands DirichletModelType;
1413 ModelType &model()
const {
return localOperator().model(); }
1415 [[deprecated(
"Use localOperator instead!")]]
1416 const LocalGalerkinOperatorImplType& impl()
const {
return localOperator(); }
1419 const LocalGalerkinOperatorImplType& localOperator()
const {
return *localOp_; }
1421 std::size_t gridSizeInterior ()
const {
return gridSizeInterior_; }
1425 const GalerkinOperatorImplType& op()
const {
return *opImpl_; }
1427 template <
class Gr
idFunction >
1428 void evaluate(
const GridFunction &u, RangeFunctionType &w )
const
1430 iterators_.update();
1433 std::shared_mutex mutex;
1435 auto doEval = [
this, &u, &w, &mutex] ()
1438 std::tuple< const LocalGalerkinOperatorImplType& > integrands( localOperator() );
1439 this->op().evaluate( u, w, this->iterators_, integrands, mutex );
1444 MPIManager :: run ( doEval );
1447 gridSizeInterior_ = Impl::accumulateGridSize( opImpl_ );
1449 catch (
const SingleThreadModeError& e )
1454 std::tuple< const LocalGalerkinOperatorImplType& > integrands( localOperator() );
1455 op().evaluate( u, w, iterators_, integrands );
1458 gridSizeInterior_ = op().gridSizeInterior();
1466 mutable ThreadIteratorType iterators_;
1470 mutable std::size_t gridSizeInterior_;
1479 template<
class Integrands,
class JacobianOperator >
1480 class DifferentiableGalerkinOperator
1481 :
public GalerkinOperator< Integrands, typename JacobianOperator::DomainFunctionType, typename JacobianOperator::RangeFunctionType >,
1482 public DifferentiableOperator< JacobianOperator >
1484 typedef GalerkinOperator< Integrands, typename JacobianOperator::DomainFunctionType, typename JacobianOperator::RangeFunctionType > BaseType;
1486 typedef typename BaseType :: LocalGalerkinOperatorImplType LocalGalerkinOperatorImplType;
1488 typedef JacobianOperator JacobianOperatorType;
1490 typedef typename BaseType::DomainFunctionType DomainFunctionType;
1491 typedef typename BaseType::RangeFunctionType RangeFunctionType;
1492 typedef typename DomainFunctionType::DiscreteFunctionSpaceType DomainDiscreteFunctionSpaceType;
1493 typedef typename RangeFunctionType::DiscreteFunctionSpaceType RangeDiscreteFunctionSpaceType;
1495 typedef DiagonalAndNeighborStencil< DomainDiscreteFunctionSpaceType, RangeDiscreteFunctionSpaceType > DiagonalAndNeighborStencilType;
1496 typedef DiagonalStencil< DomainDiscreteFunctionSpaceType, RangeDiscreteFunctionSpaceType > DiagonalStencilType;
1498 typedef typename BaseType::GridPartType GridPartType;
1500 template<
class... Args >
1501 explicit DifferentiableGalerkinOperator (
const DomainDiscreteFunctionSpaceType &dSpace,
1502 const RangeDiscreteFunctionSpaceType &rSpace,
1504 : BaseType( rSpace.gridPart(),
std::forward< Args >( args )... ),
1505 dSpace_(dSpace), rSpace_(rSpace),
1506 domainSpaceSequence_(dSpace.sequence()),
1507 rangeSpaceSequence_(rSpace.sequence()),
1508 stencilDAN_(), stencilD_()
1511 stencilDAN_.reset(
new DiagonalAndNeighborStencilType( dSpace_, rSpace_ ) );
1513 stencilD_.reset(
new DiagonalStencilType( dSpace_, rSpace_ ) );
1516 virtual void jacobian (
const DomainFunctionType &u, JacobianOperatorType &jOp )
const final override
1521 template<
class Gr
idFunction >
1522 void jacobian (
const GridFunction &u, JacobianOperatorType &jOp )
const
1527 const DomainDiscreteFunctionSpaceType& domainSpace()
const
1531 const RangeDiscreteFunctionSpaceType& rangeSpace()
const
1536 using BaseType::localOperator;
1537 using BaseType::nonlinear;
1542 bool hasSkeleton()
const
1544 std::tuple< const LocalGalerkinOperatorImplType& > integrands( localOperator() );
1545 return op().hasSkeleton( integrands );
1548 void prepare( JacobianOperatorType& jOp )
const
1550 if ( domainSpaceSequence_ != domainSpace().sequence()
1551 || rangeSpaceSequence_ != rangeSpace().sequence() )
1553 domainSpaceSequence_ = domainSpace().sequence();
1554 rangeSpaceSequence_ = rangeSpace().sequence();
1557 assert( stencilDAN_ );
1558 stencilDAN_->update();
1562 assert( stencilD_ );
1563 stencilD_->update();
1567 jOp.reserve( *stencilDAN_ );
1569 jOp.reserve( *stencilD_ );
1574 template <
class Gr
idFunction >
1575 void assemble(
const GridFunction &u, JacobianOperatorType &jOp )
const
1580 iterators_.update();
1583 std::shared_mutex mutex;
1585 auto doAssemble = [
this, &u, &jOp, &mutex] ()
1587 std::tuple< const LocalGalerkinOperatorImplType& > integrands( localOperator() );
1588 this->op().assemble( u, jOp, this->iterators_, integrands, mutex );
1593 MPIManager :: run ( doAssemble );
1596 gridSizeInterior_ = Impl::accumulateGridSize( this->opImpl_ );
1598 catch (
const SingleThreadModeError& e )
1602 std::tuple< const LocalGalerkinOperatorImplType& > integrands( localOperator() );
1603 op().assemble( u, jOp, iterators_, integrands );
1605 gridSizeInterior_ = op().gridSizeInterior();
1610 jOp.flushAssembly();
1613 using BaseType::iterators_;
1614 using BaseType::gridSizeInterior_;
1616 const DomainDiscreteFunctionSpaceType &dSpace_;
1617 const RangeDiscreteFunctionSpaceType &rSpace_;
1619 mutable int domainSpaceSequence_, rangeSpaceSequence_;
1621 mutable std::unique_ptr< DiagonalAndNeighborStencilType > stencilDAN_;
1622 mutable std::unique_ptr< DiagonalStencilType > stencilD_;
1630 template<
class Integrands,
class DomainFunction,
class RangeFunction >
1631 class AutomaticDifferenceGalerkinOperator
1632 :
public GalerkinOperator< Integrands, DomainFunction, RangeFunction >,
1633 public AutomaticDifferenceOperator< DomainFunction, RangeFunction >
1635 typedef GalerkinOperator< Integrands, DomainFunction, RangeFunction > BaseType;
1639 typedef typename BaseType::GridPartType GridPartType;
1641 template<
class... Args >
1642 explicit AutomaticDifferenceGalerkinOperator (
const GridPartType &gridPart, Args &&... args )
1643 : BaseType( gridPart,
std::forward< Args >( args )... ), AutomaticDifferenceOperatorType()
1652 template <
class LinearOperator,
class ModelIntegrands >
1653 struct ModelDifferentiableGalerkinOperator
1654 :
public DifferentiableGalerkinOperator< ModelIntegrands, LinearOperator >
1656 typedef DifferentiableGalerkinOperator< ModelIntegrands, LinearOperator > BaseType;
1658 typedef typename ModelIntegrands::ModelType ModelType;
1661 typedef typename LinearOperator::RangeSpaceType DiscreteFunctionSpaceType;
1663 ModelDifferentiableGalerkinOperator ( ModelType &model,
const DiscreteFunctionSpaceType &dfSpace )
1664 : BaseType( dfSpace.gridPart(), model )
1667 template<
class Gr
idFunction >
1668 void apply (
const GridFunction &u, RangeFunctionType &w )
const
1673 template<
class Gr
idFunction >
1674 void apply (
const GridFunction &u, LinearOperator &jOp )
const
1676 (*this).jacobian( u, jOp );
1685 template<
class Integrands,
class LinearOperator,
bool addDirichletBC,
1686 template <
class,
class>
class DifferentiableGalerkinOperatorImpl >
1687 struct GalerkinSchemeTraits
1689 template <
class O,
bool addDBC>
1690 struct DirichletBlockSelector {
using type = void; };
1692 struct DirichletBlockSelector<O,true> {
using type =
typename O::DirichletBlockVector; };
1694 using DifferentiableOperatorType = std::conditional_t< addDirichletBC,
1695 DirichletWrapperOperator< DifferentiableGalerkinOperatorImpl< Integrands, LinearOperator >>,
1696 DifferentiableGalerkinOperatorImpl< Integrands, LinearOperator > >;
1697 using DirichletBlockVector =
typename DirichletBlockSelector<
1698 DirichletWrapperOperator<
1699 DifferentiableGalerkinOperatorImpl< Integrands, LinearOperator >>,
1700 addDirichletBC>::type;
1702 typedef DifferentiableOperatorType type;
1705 template<
class Integrands,
class LinearOperator,
class LinearInverseOperator,
bool addDirichletBC,
1706 template <
class,
class>
class DifferentiableGalerkinOperatorImpl = DifferentiableGalerkinOperator >
1707 struct GalerkinSchemeImpl :
public FemScheme< typename
1708 GalerkinSchemeTraits< Integrands, LinearOperator,
1709 addDirichletBC, DifferentiableGalerkinOperatorImpl>::type,
1710 LinearInverseOperator >
1712 typedef FemScheme<
typename GalerkinSchemeTraits< Integrands, LinearOperator,
1713 addDirichletBC, DifferentiableGalerkinOperatorImpl>::type,
1714 LinearInverseOperator >
1717 typedef typename BaseType :: DiscreteFunctionSpaceType DiscreteFunctionSpaceType;
1719 GalerkinSchemeImpl (
const DiscreteFunctionSpaceType &dfSpace,
1720 const Integrands &integrands,
1721 const ParameterReader& parameter = Parameter::container() )
1724 std::move(integrands))
1733 template<
class Integrands,
class LinearOperator,
class InverseOperator,
bool addDirichletBC >
1734 using GalerkinScheme = Impl::GalerkinSchemeImpl< Integrands, LinearOperator, InverseOperator, addDirichletBC,
1735 DifferentiableGalerkinOperator >;
FunctionSpaceType::RangeType RangeType
type of range vectors, i.e., type of function values
Definition: localfunction.hh:110
FunctionSpaceType::HessianRangeType HessianRangeType
type of the Hessian
Definition: localfunction.hh:114
FunctionSpaceType::JacobianRangeType JacobianRangeType
type of the Jacobian, i.e., type of evaluated Jacobian matrix
Definition: localfunction.hh:112
static bool verbose()
obtain the cached value for fem.verbose with default verbosity level 2
Definition: parameter.hh:466
quadrature on the codim-0 reference element
actual interface class for quadratures
constexpr void forEach(Range &&range, F &&f)
Range based for loop.
Definition: hybridutilities.hh:261
static constexpr IntegralRange< std::decay_t< T > > range(T &&from, U &&to) noexcept
free standing function for setting up a range based for loop over an integer range for (auto i: range...
Definition: rangeutilities.hh:288
constexpr Interior interior
PartitionSet for the interior partition.
Definition: partitionset.hh:271
Dune namespace
Definition: alignedallocator.hh:13
constexpr std::integral_constant< std::size_t, sizeof...(II)> size(std::integer_sequence< T, II... >)
Return the size of the sequence.
Definition: integersequence.hh:75
constexpr std::bool_constant<(sizeof...(II)==0)> empty(std::integer_sequence< T, II... >)
Checks whether the sequence is empty.
Definition: integersequence.hh:80
DomainFunction DomainFunctionType
type of discrete function in the operator's domain
Definition: operator.hh:36