patternengine.hh

#ifndef DUNE_PDELAB_GRIDOPERATOR_FASTDG_PATTERNENGINE_HH
#define DUNE_PDELAB_GRIDOPERATOR_FASTDG_PATTERNENGINE_HH
 
#include <dune/pdelab/gridfunctionspace/localvector.hh>
#include <dune/pdelab/gridoperator/common/localmatrix.hh>
#include <dune/pdelab/gridoperator/common/assemblerutilities.hh>
#include <dune/pdelab/gridoperator/common/localassemblerenginebase.hh>
#include <dune/pdelab/constraints/common/constraints.hh>
#include <dune/pdelab/localoperator/callswitch.hh>
 
namespace Dune{
  namespace PDELab{
 
    template<typename LA>
    class FastDGLocalPatternAssemblerEngine
      : public LocalAssemblerEngineBase
    {
    public:
 
      template<typename TrialConstraintsContainer, typename TestConstraintsContainer>
      bool needsConstraintsCaching(const TrialConstraintsContainer& cu, const TestConstraintsContainer& cv) const
      {
        return cu.containsNonDirichletConstraints() || cv.containsNonDirichletConstraints();
      }
 
      typedef LA LocalAssembler;
 
      typedef typename LA::LocalOperator LOP;
 
      typedef typename LA::LFSU LFSU;
      typedef typename LA::LFSUCache LFSUCache;
      typedef typename LFSU::Traits::GridFunctionSpace GFSU;
      typedef typename LA::LFSV LFSV;
      typedef typename LA::LFSVCache LFSVCache;
      typedef typename LFSV::Traits::GridFunctionSpace GFSV;
 
      typedef typename LA::Traits::BorderDOFExchanger BorderDOFExchanger;
      typedef typename BorderDOFExchanger::BorderPattern BorderPattern;
 
      typedef typename LA::Traits::MatrixPattern Pattern;
 
      typedef Dune::PDELab::LocalSparsityPattern LocalPattern;
 
      typedef std::size_t size_type;
 
      FastDGLocalPatternAssemblerEngine(const LocalAssembler & local_assembler_,
                                        std::shared_ptr<typename LA::Traits::BorderDOFExchanger> border_dof_exchanger)
        : local_assembler(local_assembler_)
        , lop(local_assembler.localOperator())
        , pattern(nullptr)
        , _border_dof_exchanger(border_dof_exchanger)
      {}
 
      const LocalAssembler & localAssembler() const
      {
        return local_assembler;
      }
 
      const typename LocalAssembler::Traits::TrialGridFunctionSpaceConstraints& trialConstraints() const
      {
        return localAssembler().trialConstraints();
      }
 
      const typename LocalAssembler::Traits::TestGridFunctionSpaceConstraints& testConstraints() const
      {
        return localAssembler().testConstraints();
      }
 
      void setPattern(Pattern & pattern_)
      {
        pattern = &pattern_;
      }
 
 
      bool requireSkeleton() const
      {
        return local_assembler.doPatternSkeleton();
      }
 
      bool requireUVVolume() const
      {
        return local_assembler.doPatternVolume();
      }
 
      bool requireUVSkeleton() const
      {
        return local_assembler.doPatternSkeleton();
      }
 
      bool requireUVBoundary() const
      {
        return local_assembler.doPatternBoundary();
      }
 
      bool requireUVVolumePostSkeleton() const
      {
        return local_assembler.doPatternVolumePostSkeleton();
      }
 
      auto partition() const
      {
        return Partitions::all;
      }
 
 
 
      template<typename LFSVC, typename LFSUC>
      void add_border_pattern(std::true_type, const LFSVC& lfsv_cache, const LFSUC& lfsu_cache,
                              const LocalPattern& p)
      {
        if (local_assembler.reconstructBorderEntries() &&
            !communicationCache().initialized())
          {
            communicationCache().addEntries(lfsv_cache,lfsu_cache,p);
          }
      }
 
      template<typename LFSVC, typename LFSUC>
      void add_border_pattern(std::false_type, const LFSVC& lfsv_cache, const LFSUC& lfsu_cache,
                              const LocalPattern& p)
      {}
 
      template<typename LFSVC, typename LFSUC>
      void add_pattern(const LFSVC& lfsv_cache, const LFSUC& lfsu_cache, const LocalPattern& p)
      {
        if (p.size() > 0)
        {
          local_assembler.add_entry(*pattern,
                                    lfsv_cache,p[0].i(),
                                    lfsu_cache,p[0].j()
                                    );
 
        }
 
        add_border_pattern(std::integral_constant<bool,LocalAssembler::isNonOverlapping>(),
                           lfsv_cache,
                           lfsu_cache,
                           p);
      }
 
 
      template<typename EG, typename LFSUC, typename LFSVC>
      void onUnbindLFSUV(const EG & eg, const LFSUC & lfsu_cache, const LFSVC & lfsv_cache)
      {
        add_pattern(lfsv_cache,lfsu_cache,localpattern);
        localpattern.clear();
      }
 
      template<typename IG, typename LFSUC, typename LFSVC>
      void onUnbindLFSUVOutside(const IG& ig,
                                const LFSUC& lfsu_s_cache, const LFSVC& lfsv_s_cache,
                                const LFSUC& lfsu_n_cache, const LFSVC& lfsv_n_cache)
      {
        add_pattern(lfsv_s_cache,lfsu_n_cache,localpattern_sn);
        localpattern_sn.clear();
        add_pattern(lfsv_n_cache,lfsu_s_cache,localpattern_ns);
        localpattern_ns.clear();
      }
 
 
 
      template<typename EG, typename LFSUC, typename LFSVC>
      void assembleUVVolume(const EG & eg, const LFSUC & lfsu_cache, const LFSVC & lfsv_cache)
      {
        Dune::PDELab::LocalAssemblerCallSwitch<LOP,LOP::doPatternVolume>::
          pattern_volume(lop,lfsu_cache.localFunctionSpace(),lfsv_cache.localFunctionSpace(),localpattern);
      }
 
      template<typename IG, typename LFSUC, typename LFSVC>
      void assembleUVSkeleton(const IG & ig, const LFSUC & lfsu_s_cache, const LFSVC & lfsv_s_cache,
                              const LFSUC & lfsu_n_cache, const LFSVC & lfsv_n_cache)
      {
        Dune::PDELab::LocalAssemblerCallSwitch<LOP,LOP::doPatternSkeleton>::
          pattern_skeleton(lop,
                           lfsu_s_cache.localFunctionSpace(),lfsv_s_cache.localFunctionSpace(),
                           lfsu_n_cache.localFunctionSpace(),lfsv_n_cache.localFunctionSpace(),
                           localpattern_sn, localpattern_ns);
      }
 
      template<typename IG, typename LFSUC, typename LFSVC>
      void assembleUVBoundary(const IG & ig, const LFSUC & lfsu_s_cache, const LFSVC & lfsv_s_cache)
      {
        Dune::PDELab::LocalAssemblerCallSwitch<LOP,LOP::doPatternBoundary>::
          pattern_boundary(lop,lfsu_s_cache.localFunctionSpace(),lfsv_s_cache.localFunctionSpace(),localpattern);
      }
 
      template<typename IG, typename LFSUC, typename LFSVC>
      static void assembleUVEnrichedCoupling(const IG & ig,
                                             const LFSUC & lfsu_s_cache, const LFSVC & lfsv_s_cache,
                                             const LFSUC & lfsu_n_cache, const LFSVC & lfsv_n_cache,
                                             const LFSUC & lfsu_coupling_cache, const LFSVC & lfsv_coupling_cache)
      {
        DUNE_THROW(Dune::NotImplemented,"Assembling of coupling spaces is not implemented for ");
      }
 
      template<typename IG, typename LFSVC>
      static void assembleVEnrichedCoupling(const IG & ig,
                                            const LFSVC & lfsv_s_cache,
                                            const LFSVC & lfsv_n_cache,
                                            const LFSVC & lfsv_coupling_cache)
      {
        DUNE_THROW(Dune::NotImplemented,"Assembling of coupling spaces is not implemented for ");
      }
 
      template<typename EG, typename LFSUC, typename LFSVC>
      void assembleUVVolumePostSkeleton(const EG & eg, const LFSUC & lfsu_cache, const LFSVC & lfsv_cache)
      {
        Dune::PDELab::LocalAssemblerCallSwitch<LOP,LOP::doPatternVolumePostSkeleton>::
          pattern_volume_post_skeleton(lop,lfsu_cache.localFunctionSpace(),lfsv_cache.localFunctionSpace(),localpattern);
      }
 
 
      void postAssembly(const GFSU& gfsu, const GFSV& gfsv){
        post_border_pattern_assembly(std::integral_constant<bool,LocalAssembler::isNonOverlapping>(),
                                     gfsu,
                                     gfsv);
      }
 
      void post_border_pattern_assembly(std::true_type, const GFSU& gfsu, const GFSV& gfsv)
      {
        if(local_assembler.doPostProcessing() &&
           local_assembler.reconstructBorderEntries())
          {
            communicationCache().finishInitialization();
 
            typename LA::Traits::BorderDOFExchanger::template PatternExtender<Pattern>
              data_handle(*_border_dof_exchanger,gfsu,gfsv,*pattern);
            gfsv.gridView().communicate(data_handle,
                                        InteriorBorder_InteriorBorder_Interface,
                                        ForwardCommunication);
          }
      }
 
      void post_border_pattern_assembly(std::false_type, const GFSU& gfsu, const GFSV& gfsv)
      {}
 
 
 
    private:
 
      typename LA::Traits::BorderDOFExchanger::CommunicationCache&
      communicationCache()
      {
        return _border_dof_exchanger->communicationCache();
      }
 
      const typename LA::Traits::BorderDOFExchanger::CommunicationCache&
      communicationCache() const
      {
        return _border_dof_exchanger->communicationCache();
      }
 
      const LocalAssembler & local_assembler;
 
      const LOP & lop;
 
      Pattern * pattern;
 
      LocalPattern localpattern;
      LocalPattern localpattern_sn, localpattern_ns;
 
      BorderPattern _border_pattern;
 
      std::shared_ptr<BorderDOFExchanger> _border_dof_exchanger;
 
    }; // End of class FastDGLocalPatternAssemblerEngine
 
  }
}
#endif