powerlocalrestrictprolong.hh

#ifndef DUNE_FEM_SPACE_COMBINEDSPACE_POWERLOCALRESTRICPROLONG_HH
#define DUNE_FEM_SPACE_COMBINEDSPACE_POWERLOCALRESTRICPROLONG_HH
 
#include <algorithm>
 
#include <dune/common/exceptions.hh>
 
#include <dune/fem/function/localfunction/const.hh>
#include <dune/fem/function/localfunction/localfunction.hh>
#include <dune/fem/space/common/localrestrictprolong.hh>
#include <dune/fem/storage/subvector.hh>
 
namespace Dune
{
 
  namespace Fem
  {
 
    // PowerLocalRestricProlong
    // ------------------------
 
    template< class DiscreteFunctionSpace, int N >
    class PowerLocalRestrictProlong
    {
      typedef PowerLocalRestrictProlong< DiscreteFunctionSpace, N > ThisType;
 
      // type of contained DefaultLocalRestrictProlong
      typedef DefaultLocalRestrictProlong< DiscreteFunctionSpace > LocalRestrictProlongType;
 
    public:
      // type of DomainField
      typedef typename LocalRestrictProlongType::DomainFieldType DomainFieldType;
 
      PowerLocalRestrictProlong ( const DiscreteFunctionSpace &space )
        : localRestrictProlong_( space )
      {}
 
      void setFatherChildWeight ( const DomainFieldType &weight )
      {
        localRestrictProlong_.setFatherChildWeight( weight );
      }
 
      template< class LFFather, class LFSon, class LocalGeometry >
      void restrictLocal ( LFFather &lfFather, const LFSon &lfSon,
                           const LocalGeometry &geometryInFather, bool initialize ) const
      {
        typedef SubVector< const typename LFSon::LocalDofVectorType, OffsetSubMapper > SubDofVectorTypeSon;
        typedef SubVector< typename LFFather::LocalDofVectorType, OffsetSubMapper > SubDofVectorTypeFather;
 
        typedef typename LFSon::BasisFunctionSetType::ScalarBasisFunctionSetType SubSonBasisFunctionSetType;
        typedef typename LFFather::BasisFunctionSetType::ScalarBasisFunctionSetType SubFatherBasisFunctionSetType;
 
        SubFatherBasisFunctionSetType subFatherBasisFunctionSet = lfFather.basisFunctionSet().scalarBasisFunctionSet();
        SubSonBasisFunctionSetType subSonBasisFunctionSet = lfSon.basisFunctionSet().scalarBasisFunctionSet();
 
        for( std::size_t i = 0; i < N; ++i )
        {
          std::size_t sonBasisSetSize = subSonBasisFunctionSet.size();
          std::size_t fatherBasisSetsize = subFatherBasisFunctionSet.size();
 
          SubDofVectorTypeSon sonSubDofVector( lfSon.localDofVector(), OffsetSubMapper( sonBasisSetSize, sonBasisSetSize * i ) );
          SubDofVectorTypeFather fatherSubDofVector( lfFather.localDofVector(),
                                                     OffsetSubMapper( fatherBasisSetsize, fatherBasisSetsize * i ) );
 
          BasicConstLocalFunction< SubSonBasisFunctionSetType, SubDofVectorTypeSon > subLFSon( subSonBasisFunctionSet, sonSubDofVector );
          LocalFunction< SubFatherBasisFunctionSetType, SubDofVectorTypeFather > subLFFather( subFatherBasisFunctionSet,
                                                                                              fatherSubDofVector );
 
          localRestrictProlong_.restrictLocal( subLFFather, subLFSon, geometryInFather, initialize );
        }
      }
 
 
      template< class LFFather, class LFSon, class LocalGeometry >
      void prolongLocal ( const LFFather &lfFather, LFSon &lfSon,
                          const LocalGeometry &geometryInFather, bool initialize ) const
      {
        typedef SubVector< typename LFSon::LocalDofVectorType, OffsetSubMapper > SubDofVectorTypeSon;
        typedef SubVector< const typename LFFather::LocalDofVectorType, OffsetSubMapper > SubDofVectorTypeFather;
 
        typedef typename LFSon::BasisFunctionSetType::ScalarBasisFunctionSetType SubSonBasisFunctionSetType;
        typedef typename LFFather::BasisFunctionSetType::ScalarBasisFunctionSetType SubFatherBasisFunctionSetType;
 
        SubSonBasisFunctionSetType subSonBasisFunctionSet = lfSon.basisFunctionSet().scalarBasisFunctionSet();
        SubFatherBasisFunctionSetType subFatherBasisFunctionSet = lfFather.basisFunctionSet().scalarBasisFunctionSet();
 
        for( std::size_t i = 0; i < N; ++i )
        {
          std::size_t sonBasisSetSize = subSonBasisFunctionSet.size();
          std::size_t fatherBasisSetsize = subFatherBasisFunctionSet.size();
 
          SubDofVectorTypeSon sonSubDofVector( lfSon.localDofVector(), OffsetSubMapper( sonBasisSetSize, sonBasisSetSize * i ) );
          SubDofVectorTypeFather fatherSubDofVector( lfFather.localDofVector(),
                                                     OffsetSubMapper( fatherBasisSetsize, fatherBasisSetsize * i ) );
 
          LocalFunction< SubSonBasisFunctionSetType, SubDofVectorTypeSon > subLFSon( subSonBasisFunctionSet, sonSubDofVector );
          BasicConstLocalFunction< SubFatherBasisFunctionSetType, SubDofVectorTypeFather > subLFFather( subFatherBasisFunctionSet,
                                                                                                        fatherSubDofVector );
 
          localRestrictProlong_.prolongLocal( subLFFather, subLFSon, geometryInFather, initialize );
        }
      }
 
      template <class LFFather>
      void restrictFinalize ( LFFather &lfFather ) const
      {
        DUNE_THROW(Dune::NotImplemented,"PowerLocalRestrictProlong::restrictFinalize is missing!");
      }
 
      bool needCommunication () const
      {
        return localRestrictProlong_.needCommunication();
      }
 
    private:
      LocalRestrictProlongType localRestrictProlong_;
    };
 
  } // namespace Fem
 
} // namespace Dune
 
#endif // #ifndef DUNE_FEM_SPACE_COMBINEDSPACE_POWERLOCALRESTRICPROLONG_HH