nedelec_transfer.h

Go to the documentation of this file.

/*
 * Copyright (c) 2013-2014:  G-CSC, Goethe University Frankfurt
 * Author: Dmitry Logashenko
 * 
 * This file is part of UG4.
 * 
 * UG4 is free software: you can redistribute it and/or modify it under the
 * terms of the GNU Lesser General Public License version 3 (as published by the
 * Free Software Foundation) with the following additional attribution
 * requirements (according to LGPL/GPL v3 §7):
 * 
 * (1) The following notice must be displayed in the Appropriate Legal Notices
 * of covered and combined works: "Based on UG4 (www.ug4.org/license)".
 * 
 * (2) The following notice must be displayed at a prominent place in the
 * terminal output of covered works: "Based on UG4 (www.ug4.org/license)".
 * 
 * (3) The following bibliography is recommended for citation and must be
 * preserved in all covered files:
 * "Reiter, S., Vogel, A., Heppner, I., Rupp, M., and Wittum, G. A massively
 *   parallel geometric multigrid solver on hierarchically distributed grids.
 *   Computing and visualization in science 16, 4 (2013), 151-164"
 * "Vogel, A., Reiter, S., Rupp, M., Nägel, A., and Wittum, G. UG4 -- a novel
 *   flexible software system for simulating pde based models on high performance
 *   computers. Computing and visualization in science 16, 4 (2013), 165-179"
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Lesser General Public License for more details.
 */
 
#ifndef __H__UG__PLUGINS__ELECTROMAGNETISM__NEDELEC_TRANSFER__
#define __H__UG__PLUGINS__ELECTROMAGNETISM__NEDELEC_TRANSFER__
 
#include "common/common.h"
#include "lib_disc/operator/linear_operator/transfer_interface.h"
 
#ifdef UG_PARALLEL
#include "lib_disc/parallelization/parallelization_util.h"
#endif
 
namespace ug{
namespace Electromagnetism{
 
template <typename TDomain, typename TAlgebra, typename TElem>
class NedelecProlongationMatrixHelper
{
  typedef TDomain domain_type;
  
  typedef TAlgebra algebra_type;
 
  typedef typename TAlgebra::matrix_type matrix_type;
 
  static const int WDim = TDomain::dim;
 
  typedef typename reference_element_traits<TElem>::reference_element_type ref_elem_type;
  
  static const int dim = ref_elem_type::dim;
 
  typedef typename TDomain::position_type position_type;
 
  static void GetRegularLocalCoordinate
  (
    const MultiGrid * mg, 
    Vertex * vrt, 
    TElem * base, 
    MathVector<TElem::dim> & local 
  );
  
public:
 
  static void assemble_prolongation_matrix
  (
    const domain_type & domain, 
    const DoFDistribution & coarseDD, 
    const DoFDistribution & fineDD, 
    matrix_type & mat, 
    std::vector<bool> & vIsRestricted 
  );
};
 
template <typename TDomain, typename TAlgebra>
class NedelecProlongationMatrixHelper<TDomain, TAlgebra, RegularEdge>
{
  typedef TDomain domain_type;
  
  typedef TAlgebra algebra_type;
 
  typedef typename TAlgebra::matrix_type matrix_type;
 
  static const int WDim = TDomain::dim;
 
public:
 
  static void assemble_prolongation_matrix
  (
    const domain_type & domain, 
    const DoFDistribution & coarseDD, 
    const DoFDistribution & fineDD, 
    matrix_type & mat, 
    std::vector<bool> & vIsRestricted 
  );
};
 
template <typename TDomain, typename TAlgebra>
class NedelecTransfer: public ITransferOperator<TDomain, TAlgebra>
{
public:
  typedef NedelecTransfer<TDomain, TAlgebra> this_type;
 
  typedef ITransferOperator<TDomain, TAlgebra> base_type;
 
  typedef TDomain domain_type;
  
  typedef TAlgebra algebra_type;
 
  typedef typename TAlgebra::vector_type vector_type;
 
  typedef typename TAlgebra::matrix_type matrix_type;
 
  static const int WDim = TDomain::dim;
  
public:
 
  NedelecTransfer (SmartPtr<ApproximationSpace<TDomain> > approxSpace)
  : ITransferOperator<TDomain, TAlgebra>(), m_bInit (false), m_spApproxSpace (approxSpace)
    {};
  
  virtual void set_levels (GridLevel coarseLevel, GridLevel fineLevel);
  
  void init ();
  
  void prolongate (vector_type & uFineOut, const vector_type & uCoarse);
  
  void do_restrict (vector_type & uCoarse, const vector_type & uFine);
  
  SmartPtr<ITransferOperator<TDomain, TAlgebra> > clone ();
  
private:
  
  void check_approximation_space ();
  
  template <typename TElem>
  void assemble_prolongation_matrix
  (
    matrix_type & mat, 
    std::vector<bool> & vIsRestricted, 
    const DoFDistribution & coarseDD, 
    const DoFDistribution & fineDD 
  );
  
  struct AssembleProlongationMatrix
  {
    this_type * m_pThis;
    const domain_type & m_domain;
    const DoFDistribution & m_coarseDD;
    const DoFDistribution & m_fineDD;
    
    AssembleProlongationMatrix
    (
      this_type* pThis,
      const DoFDistribution & coarseDD,
      const DoFDistribution & fineDD
    )
    : m_pThis (pThis), m_domain (* m_pThis->m_spApproxSpace->domain().get ()),
      m_coarseDD (coarseDD), m_fineDD (fineDD) {}
    
    template <typename TElem> void operator() (TElem &)
    {
      NedelecProlongationMatrixHelper<TDomain, TAlgebra, TElem>::assemble_prolongation_matrix
        (m_domain, m_coarseDD, m_fineDD,
          m_pThis->m_prolongation_matrix, m_pThis->m_vIsRestricted);
    }
  };
  
private:
 
  bool m_bInit;
  
  matrix_type m_prolongation_matrix;
 
  std::vector<bool> m_vIsRestricted;
 
  using base_type::m_vConstraint;
  
  SmartPtr<ApproximationSpace<TDomain> > m_spApproxSpace;
  
  GridLevel m_fineLevel;
  
  GridLevel m_coarseLevel;
};
 
} // end namespace Electromagnetism
} // end namespace ug
 
#include "nedelec_transfer_impl.h"
 
#endif // __H__UG__PLUGINS__ELECTROMAGNETISM__NEDELEC_TRANSFER__
 
/* End of File */