fe_collection.h

// ---------------------------------------------------------------------
//
// Copyright (C) 2003 - 2016 by the deal.II authors
//
// This file is part of the deal.II library.
//
// The deal.II library is free software; you can use it, redistribute
// it, and/or modify it under the terms of the GNU Lesser General
// Public License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
// The full text of the license can be found in the file LICENSE at
// the top level of the deal.II distribution.
//
// ---------------------------------------------------------------------

#ifndef dealii__fe_collection_h
#define dealii__fe_collection_h

#include <deal.II/base/config.h>
#include <deal.II/base/std_cxx11/shared_ptr.h>
#include <deal.II/fe/fe.h>
#include <deal.II/fe/fe_values_extractors.h>
#include <deal.II/fe/component_mask.h>

DEAL_II_NAMESPACE_OPEN

namespace hp
{

  template <int dim, int spacedim=dim>
  class FECollection : public Subscriptor
  {
  public:
    FECollection ();

    explicit FECollection (const FiniteElement<dim,spacedim> &fe);

    FECollection (const FiniteElement<dim,spacedim> &fe1,
                  const FiniteElement<dim,spacedim> &fe2);

    FECollection (const FiniteElement<dim,spacedim> &fe1,
                  const FiniteElement<dim,spacedim> &fe2,
                  const FiniteElement<dim,spacedim> &fe3);

    FECollection (const FiniteElement<dim,spacedim> &fe1,
                  const FiniteElement<dim,spacedim> &fe2,
                  const FiniteElement<dim,spacedim> &fe3,
                  const FiniteElement<dim,spacedim> &fe4);

    FECollection (const FiniteElement<dim,spacedim> &fe1,
                  const FiniteElement<dim,spacedim> &fe2,
                  const FiniteElement<dim,spacedim> &fe3,
                  const FiniteElement<dim,spacedim> &fe4,
                  const FiniteElement<dim,spacedim> &fe5);

    FECollection (const std::vector<const FiniteElement<dim,spacedim>*> &fes);

    FECollection (const FECollection<dim,spacedim> &fe_collection);

    void push_back (const FiniteElement<dim,spacedim> &new_fe);

    const FiniteElement<dim,spacedim> &
    operator[] (const unsigned int index) const;

    unsigned int size () const;

    unsigned int n_components () const;

    unsigned int n_blocks () const;

    unsigned int max_dofs_per_vertex () const;

    unsigned int max_dofs_per_line () const;

    unsigned int max_dofs_per_quad () const;

    unsigned int max_dofs_per_hex () const;

    unsigned int max_dofs_per_face () const;

    unsigned int max_dofs_per_cell () const;

    std::size_t memory_consumption () const;


    bool hp_constraints_are_implemented () const;

    unsigned int
    find_least_face_dominating_fe (const std::set<unsigned int> &fes) const;

    ComponentMask
    component_mask (const FEValuesExtractors::Scalar &scalar) const;

    ComponentMask
    component_mask (const FEValuesExtractors::Vector &vector) const;

    ComponentMask
    component_mask (const FEValuesExtractors::SymmetricTensor<2> &sym_tensor) const;

    ComponentMask
    component_mask (const BlockMask &block_mask) const;

    BlockMask
    block_mask (const FEValuesExtractors::Scalar &scalar) const;

    BlockMask
    block_mask (const FEValuesExtractors::Vector &vector) const;

    BlockMask
    block_mask (const FEValuesExtractors::SymmetricTensor<2> &sym_tensor) const;

    BlockMask
    block_mask (const ComponentMask &component_mask) const;


    DeclException0 (ExcNoFiniteElements);

  private:
    std::vector<std_cxx11::shared_ptr<const FiniteElement<dim,spacedim> > > finite_elements;
  };



  /* --------------- inline functions ------------------- */

  template <int dim, int spacedim>
  inline
  unsigned int
  FECollection<dim,spacedim>::size () const
  {
    return finite_elements.size();
  }


  template <int dim, int spacedim>
  inline
  unsigned int
  FECollection<dim,spacedim>::n_components () const
  {
    Assert (finite_elements.size () > 0, ExcNoFiniteElements());

    // note that there is no need
    // here to enforce that indeed
    // all elements have the same
    // number of components since we
    // have already done this when
    // adding a new element to the
    // collection.

    return finite_elements[0]->n_components ();
  }


  template <int dim, int spacedim>
  inline
  const FiniteElement<dim,spacedim> &
  FECollection<dim,spacedim>::operator[] (const unsigned int index) const
  {
    Assert (index < finite_elements.size(),
            ExcIndexRange (index, 0, finite_elements.size()));
    return *finite_elements[index];
  }



  template <int dim, int spacedim>
  unsigned int
  FECollection<dim,spacedim>::max_dofs_per_vertex () const
  {
    Assert (finite_elements.size() > 0, ExcNoFiniteElements());

    unsigned int max = 0;
    for (unsigned int i=0; i<finite_elements.size(); ++i)
      if (finite_elements[i]->dofs_per_vertex > max)
        max = finite_elements[i]->dofs_per_vertex;

    return max;
  }



  template <int dim, int spacedim>
  unsigned int
  FECollection<dim,spacedim>::max_dofs_per_line () const
  {
    Assert (finite_elements.size() > 0, ExcNoFiniteElements());

    unsigned int max = 0;
    for (unsigned int i=0; i<finite_elements.size(); ++i)
      if (finite_elements[i]->dofs_per_line > max)
        max = finite_elements[i]->dofs_per_line;

    return max;
  }



  template <int dim, int spacedim>
  unsigned int
  FECollection<dim,spacedim>::max_dofs_per_quad () const
  {
    Assert (finite_elements.size() > 0, ExcNoFiniteElements());

    unsigned int max = 0;
    for (unsigned int i=0; i<finite_elements.size(); ++i)
      if (finite_elements[i]->dofs_per_quad > max)
        max = finite_elements[i]->dofs_per_quad;

    return max;
  }



  template <int dim, int spacedim>
  unsigned int
  FECollection<dim,spacedim>::max_dofs_per_hex () const
  {
    Assert (finite_elements.size() > 0, ExcNoFiniteElements());

    unsigned int max = 0;
    for (unsigned int i=0; i<finite_elements.size(); ++i)
      if (finite_elements[i]->dofs_per_hex > max)
        max = finite_elements[i]->dofs_per_hex;

    return max;
  }



  template <int dim, int spacedim>
  unsigned int
  FECollection<dim,spacedim>::max_dofs_per_face () const
  {
    Assert (finite_elements.size() > 0, ExcNoFiniteElements());

    unsigned int max = 0;
    for (unsigned int i=0; i<finite_elements.size(); ++i)
      if (finite_elements[i]->dofs_per_face > max)
        max = finite_elements[i]->dofs_per_face;

    return max;
  }



  template <int dim, int spacedim>
  unsigned int
  FECollection<dim,spacedim>::max_dofs_per_cell () const
  {
    Assert (finite_elements.size() > 0, ExcNoFiniteElements());

    unsigned int max = 0;
    for (unsigned int i=0; i<finite_elements.size(); ++i)
      if (finite_elements[i]->dofs_per_cell > max)
        max = finite_elements[i]->dofs_per_cell;

    return max;
  }


  template <int dim, int spacedim>
  bool
  FECollection<dim,spacedim>::hp_constraints_are_implemented () const
  {
    Assert (finite_elements.size() > 0, ExcNoFiniteElements());

    bool hp_constraints = true;
    for (unsigned int i=0; i<finite_elements.size(); ++i)
      hp_constraints = hp_constraints &&
                       finite_elements[i]->hp_constraints_are_implemented();

    return hp_constraints;
  }


} // namespace hp

DEAL_II_NAMESPACE_CLOSE

#endif