assembler.hh

#ifndef DUNE_PDELAB_GRIDOPERATOR_COMMON_ASSEMBLER_HH
#define DUNE_PDELAB_GRIDOPERATOR_COMMON_ASSEMBLER_HH
#include <gridoperatorutilities.hh>
#include <assemblerutilties.hh>
 
namespace Dune {
  namespace PDELab {
 
 
    class AssemblerInterface{
    public:
      template<class LocalAssemblerEngine>
      void assemble(LocalAssemblerEngine & local_assembler_engine);
    };
 
 
    class LocalAssemblerEngine{
    public:
      typedef LocalAssemblerInterface LocalAssembler;
 
      const LocalAssembler & localAssembler();
 
      bool requireSkeleton() const;
      bool requireSkeletonTwoSided() const;
      bool requireUVVolume() const;
      bool requireVVolume() const;
      bool requireUVSkeleton() const;
      bool requireVSkeleton() const;
      bool requireUVBoundary() const;
      bool requireVBoundary() const;
      bool requireUVProcessor() const;
      bool requireVProcessor() const;
      bool requireUVEnrichedCoupling() const;
      bool requireVEnrichedCoupling() const;
      bool requireUVVolumePostSkeleton() const;
      bool requireVVolumePostSkeleton() const;
 
      template<typename EG>
      bool assembleCell(const EG & eg);
 
      template<typename EG, typename LFSU, typename LFSV>
      void assembleUVVolume(const EG & eg, const LFSU & lfsu, const LFSV & lfsv);
 
      template<typename EG, typename LFSV>
      void assembleVVolume(const EG & eg, const LFSV & lfsv);
 
      template<typename IG, typename LFSU_S, typename LFSV_S, typename LFSU_N, typename LFSV_N>
      void assembleUVSkeleton(const IG & ig, const LFSU_S & lfsu_s, const LFSV_S & lfsv_s,
                              const LFSU_N & lfsu_n, const LFSV_N & lfsv_n);
 
      template<typename IG, typename LFSV_S, typename LFSV_N>
      void assembleVSkeleton(const IG & ig, const LFSV_S & lfsv_s, const LFSV_N & lfsv_n);
 
      template<typename IG, typename LFSU_S, typename LFSV_S>
      void assembleUVBoundary(const IG & ig, const LFSU_S & lfsu_s, const LFSV_S & lfsv_s);
 
      template<typename IG, typename LFSV_S>
      void assembleVBoundary(const IG & ig, const LFSV_S & lfsv_s);
 
      template<typename IG, typename LFSU_S, typename LFSV_S>
      void assembleUVProcessor(const IG & ig, const LFSU_S & lfsu_s, const LFSV_S & lfsv_s);
 
      template<typename IG, typename LFSV_S>
      void assembleVProcessor(const IG & ig, const LFSV_S & lfsv_s);
 
      template<typename IG, typename LFSU_S, typename LFSV_S, typename LFSU_N, typename LFSV_N,
               typename LFSU_C, typename LFSV_C>
      void assembleUVEnrichedCoupling(const IG & ig,
                                      const LFSU_S & lfsu_s, const LFSV_S & lfsv_s,
                                      const LFSU_N & lfsu_n, const LFSV_N & lfsv_n,
                                      const LFSU_C & lfsu_c, const LFSV_C & lfsv_c);
 
      template<typename IG, typename LFSV_S, typename LFSV_N, typename LFSV_C>
      void assembleVEnrichedCoupling(const IG & ig,
                                     const LFSV_S & lfsv_s,
                                     const LFSV_N & lfsv_n,
                                     const LFSV_C & lfsv_c);
 
      template<typename EG, typename LFSU, typename LFSV>
      void assembleUVVolumePostSkeleton(const EG & eg, const LFSU & lfsu, const LFSV & lfsv);
 
      template<typename EG, typename LFSV>
      void assembleVVolumePostSkeleton(const EG & eg, const LFSV & lfsv);
 
      void preAssembly();
 
      void postAssembly();
 
      void onBindLFSUV(const EG & eg,
                       const LFSU_S & lfsu_s, const LFSV_S & lfsv_s);
      void onBindLFSV(const EG & eg,
                      const LFSV_S & lfsv_s);
      void onBindLFSUVInside(const IG & ig,
                             const LFSU_S & lfsu_s, const LFSV_S & lfsv_s);
      void onBindLFSVInside(const IG & ig,
                            const LFSV_S & lfsv_s);
      void onBindLFSUVOutside(const IG & ig,
                              const LFSU_S & lfsu_s, const LFSV_S & lfsv_s,
                              const LFSU_N & lfsu_n, const LFSV_N & lfsv_n);
      void onBindLFSVOutside(const IG & ig,
                             const LFSV_S & lfsv_s,
                             const LFSV_N & lfsv_n);
      void onBindLFSUVCoupling(const IG & ig,
                               const LFSU_S & lfsu_s, const LFSV_S & lfsv_s,
                               const LFSU_N & lfsu_n, const LFSV_N & lfsv_n
                               const LFSU_Coupling & lfsu_coupling, const LFSV_Coupling & lfsv_coupling);
      void onBindLFSVCoupling(const IG & ig,
                              const LFSV_S & lfsv_s,
                              const LFSV_N & lfsv_n,
                              const LFSV_Coupling & lfsv_coupling);
 
      void onUnbindLFSUV(const EG & eg,
                         const LFSU_S & lfsu_s, const LFSV_S & lfsv_s);
      void onUnbindLFSV(const EG & eg,
                        const LFSV_S & lfsv_s);
      void onUnbindLFSUVInside(const IG & ig,
                               const LFSU_S & lfsu_s, const LFSV_S & lfsv_s);
      void onUnbindLFSVInside(const IG & ig,
                              const LFSV_S & lfsv_s);
      void onUnbindLFSUVOutside(const IG & ig,
                                const LFSU_S & lfsu_s, const LFSV_S & lfsv_s,
                                const LFSU_N & lfsu_n, const LFSV_N & lfsv_n);
      void onUnbindLFSVOutside(const IG & ig,
                               const LFSV_S & lfsv_s,
                               const LFSV_N & lfsv_n);
      void onUnbindLFSUVCoupling(const IG & ig,
                                 const LFSU_S & lfsu_s, const LFSV_S & lfsv_s,
                                 const LFSU_N & lfsu_n, const LFSV_N & lfsv_n,
                                 const LFSU_Coupling & lfsu_coupling, const LFSV_Coupling & lfsv_coupling);
      void onUnbindLFSVCoupling(const IG & ig,
                                const LFSV_S & lfsv_s,
                                const LFSV_N & lfsv_n,
                                const LFSV_Coupling & lfsv_coupling);
 
      void loadCoefficientsLFSUInside(const LFSU_S & lfsu_s);
      void loadCoefficientsLFSUOutside(const LFSU_N & lfsu_n);
      void loadCoefficientsLFSUCoupling(const LFSU_Coupling & lfsu_coupling);
      void setSolution(const X& x);
      void setPattern(const P& p);
      void setJacobian(const J & j);
      void setResidual(const R& r);
    };
 
    template<typename B, typename CU, typename CV>
    class LocalAssemblerInterface : public LocalAssemblerBase<B,CU,CV>
    {
    public:
 
      template<class TT>
      void setTime(TT time);
 
      template<typename TT>
      void preStep (TT time, TT dt, std::size_t stages);
 
      void postStep ();
 
      template<typename TT>
      void preStage (TT time, std::size_t stage);
 
      void postStage ();
 
      template<typename TT>
      TT suggestTimestep (TT dt) const;
 
      template<class RF>
      void setWeight(RF weight);
 
      LocalPatternAssemblerEngine & localPatternAssemblerEngine(P & p);
      LocalResidualAssemblerEngine & localResidualAssemblerEngine(R & r, const X & x);
      LocalJacobianAssemblerEngine & localJacobianAssemblerEngine(A & a, const X & x);
      LocalResidualJacobianAssemblerEngine & localResidualJacobianAssemblerEngine(R & r, A & a, const X & x);
      class LocalPatternAssemblerEngine : public LocalAssemblerEngine {};
      class LocalResidualAssemblerEngine : public LocalAssemblerEngine {};
      class LocalJacobianAssemblerEngine : public LocalAssemblerEngine {};
      class LocalResidualJacobianAssemblerEngine : public LocalAssemblerEngine {};
    };
 
    template<typename GFSU, typename GFSV,
             typename MB, typename DF, typename RF, typename JF>
    class GridOperatorInterface{
    public:
 
      typedef GridOperatorTraits
      <GFSU,GFSV,MB,DF,RF,JF,CU,CV,AssemblerInterface,LocalAssemblerInterface> Traits;
 
      template<typename P>
      void fill_pattern (P& globalpattern) const;
 
      template<typename X, typename R>
      void residual (const X& x, R& r) const;
 
      template<typename X, typename A>
      void jacobian (const X& x, A& a) const;
 
      Assembler & assembler();
      LocalAssemblerInterface & localAssembler();
 
      const GFSU& trialGridFunctionSpace() const;
      const GFSV& testGridFunctionSpace() const;
      typename GFSU::Traits::SizeType globalSizeU () const;
      typename GFSV::Traits::SizeType globalSizeV () const;
 
 
      template<typename F>
      void interpolate(const typename Traits::Domain& xold,
                       const F& f,
                       typename Traits::Domain& xnew);
 
 
      template<typename GridOperatorTuple>
      static void setupGridOperators(GridOperatorTuple& tuple);
 
    };
  };
};
 
#endif // DUNE_PDELAB_GRIDOPERATOR_COMMON_ASSEMBLER_HH