fe_variable_collection.h

Go to the documentation of this file.

/*
  Copyright (C) 2016 - 2024 by the authors of the ASPECT code.

  This file is part of ASPECT.

  ASPECT is free software; you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation; either version 2, or (at your option)
  any later version.

  ASPECT 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 General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with ASPECT; see the file LICENSE.  If not see
  <http://www.gnu.org/licenses/>.
*/

#ifndef _aspect_fe_variable_system_h
#define _aspect_fe_variable_system_h

#include <aspect/global.h>
#include <deal.II/fe/component_mask.h>
#include <deal.II/fe/fe_values_extractors.h>
#include <deal.II/fe/fe.h>

namespace aspect
{
  template <int dim>
  struct VariableDeclaration
  {
    VariableDeclaration(const std::string &name,
                        const std::shared_ptr<FiniteElement<dim>> &fe,
                        const unsigned int multiplicity,
                        const unsigned int n_blocks);

    VariableDeclaration();

    unsigned int n_components() const;

    std::string name;

    std::shared_ptr<FiniteElement<dim>> fe;

    unsigned int multiplicity;

    unsigned int n_blocks;
  };

  template <int dim>
  struct FEVariable: public VariableDeclaration<dim>
  {
      FEVariable(const VariableDeclaration<dim> &fe_variable,
                 const unsigned int component_index,
                 const unsigned int block_index,
                 const unsigned int base_index);

      unsigned int first_component_index;

      unsigned int block_index;
      unsigned int base_index;

      ComponentMask component_mask;

      const FEValuesExtractors::Scalar &extractor_scalar() const;

      const FEValuesExtractors::Vector &extractor_vector() const;

    private:
      FEValuesExtractors::Scalar scalar_extractor;

      FEValuesExtractors::Vector vector_extractor;
  };


  template <int dim>
  class FEVariableCollection
  {
    public:
      FEVariableCollection();

      FEVariableCollection(const std::vector<VariableDeclaration<dim>> &variable_definitions);

      void initialize(const std::vector<VariableDeclaration<dim>> &variable_definitions);

      const FEVariable<dim> &variable(const std::string &name) const;

      std::vector<const FEVariable<dim>*> variables_with_name(const std::string &name) const;

      bool variable_exists(const std::string &name) const;

      const std::vector<FEVariable<dim>> &get_variables() const;

      unsigned int n_components() const;

      unsigned int n_blocks() const;

      const std::vector<const FiniteElement<dim> *> &get_fes() const;

      const std::vector<unsigned int> &get_multiplicities() const;

      const std::vector<unsigned int> &get_components_to_blocks() const;

    protected:
      std::vector<FEVariable<dim>> variables;

      unsigned int n_components_;

      unsigned int n_blocks_;

      std::vector<const FiniteElement<dim> *> fes;

      std::vector<unsigned int> multiplicities;

      std::vector<unsigned int> components_to_blocks;
  };

} // namespace aspect

#endif