temporary.hh

#ifndef DUNE_FEM_FUNCTION_LOCALFUNCTION_TEMPORARY_HH
#define DUNE_FEM_FUNCTION_LOCALFUNCTION_TEMPORARY_HH
 
#include <dune/common/ftraits.hh>
#include <dune/common/dynvector.hh>
 
#include <dune/fem/function/localfunction/localfunction.hh>
#include <dune/fem/common/intersectionside.hh>
 
 
namespace Dune
{
 
  namespace Fem
  {
 
    // internal forward declarations
    // -----------------------------
 
    template< class DiscreteFunctionSpace, class DoFVector >
    class BasicTemporaryLocalFunction;
 
    template< class DiscreteFunctionSpace, class Dof >
    class TemporaryLocalFunction;
 
 
    template< class DiscreteFunctionSpace, class DofVector >
    class BasicTemporaryLocalFunction
    : public LocalFunction < typename DiscreteFunctionSpace :: BasisFunctionSetType, DofVector >
    {
      typedef BasicTemporaryLocalFunction< DiscreteFunctionSpace, DofVector > ThisType;
      typedef LocalFunction < typename DiscreteFunctionSpace :: BasisFunctionSetType, DofVector > BaseType;
 
    public:
      typedef DiscreteFunctionSpace  DiscreteFunctionSpaceType;
 
      typedef typename BaseType :: EntityType EntityType;
 
      typedef typename BaseType :: BasisFunctionSetType BasisFunctionSetType;
 
      typedef typename BaseType :: LocalDofVectorType LocalDofVectorType;
 
      /* \copydoc Dune::Fem::LocalFunction :: localDofVector */
      using BaseType::localDofVector;
 
      explicit BasicTemporaryLocalFunction ( const DiscreteFunctionSpaceType &dfSpace,
                                             const LocalDofVectorType &dofVector = LocalDofVectorType() )
      : BaseType( dofVector ),
        dfSpace_( dfSpace )
      {
        localDofVector().reserve( DiscreteFunctionSpaceType::localBlockSize * dfSpace_.blockMapper().maxNumDofs() );
      }
 
      BasicTemporaryLocalFunction ( const DiscreteFunctionSpaceType &dfSpace, const EntityType &entity,
                                    const LocalDofVectorType &dofVector = LocalDofVectorType() )
      : BaseType( dofVector ),
        dfSpace_( dfSpace )
      {
        localDofVector().reserve( DiscreteFunctionSpaceType::localBlockSize * dfSpace_.blockMapper().maxNumDofs() );
        init( entity );
      }
 
      void init ( const EntityType &entity )
      {
        BaseType::init( dfSpace_.basisFunctionSet( entity ) );
      }
 
      void bind ( const EntityType &entity ) { init( entity ); }
 
      void unbind ()
      {
        BaseType::unbind();
      }
 
      template <class IntersectionType>
      void bind(const IntersectionType &intersection, IntersectionSide side)
      {
        // store local copy to avoid problems with casting to temporary types
        const EntityType entity = side==IntersectionSide::in? intersection.inside(): intersection.outside();
        bind( entity );
      }
 
      const DiscreteFunctionSpaceType &space() const
      {
        return dfSpace_;
      }
 
    protected:
      const DiscreteFunctionSpaceType &dfSpace_;
    };
 
 
    template< class DiscreteFunctionSpace, class Dof >
    class TemporaryLocalFunction;
  }
 
  template< class DiscreteFunctionSpace, class Dof  >
  struct FieldTraits< Fem::TemporaryLocalFunction<DiscreteFunctionSpace,Dof> >
  : public FieldTraits< Dof >
  {};
 
  namespace Fem
  {
    template< class DiscreteFunctionSpace, class Dof = typename DiscreteFunctionSpace::RangeFieldType >
    class TemporaryLocalFunction
    : public BasicTemporaryLocalFunction< DiscreteFunctionSpace, Dune::DynamicVector< Dof > >
    {
      typedef TemporaryLocalFunction< DiscreteFunctionSpace, Dof > ThisType;
      typedef BasicTemporaryLocalFunction< DiscreteFunctionSpace, Dune::DynamicVector< Dof > > BaseType;
 
    public:
      typedef typename BaseType :: EntityType EntityType;
 
      typedef typename BaseType :: DiscreteFunctionSpaceType DiscreteFunctionSpaceType;
 
      explicit TemporaryLocalFunction ( const DiscreteFunctionSpaceType &dfSpace )
      : BaseType( dfSpace ) {}
 
      TemporaryLocalFunction ( const DiscreteFunctionSpaceType &dfSpace, const EntityType &entity )
      : BaseType( dfSpace, entity ){}
    };
 
  } // namespace Fem
 
} // namespace Dune
 
#endif // #ifndef DUNE_FEM_FUNCTION_LOCALFUNCTION_TEMPORARY_HH