scalarfe.hh

Go to the documentation of this file.

// SPDX-FileCopyrightText: 2021-2024 The Ikarus Developers mueller@ibb.uni-stuttgart.de
// SPDX-License-Identifier: LGPL-3.0-or-later
 
#pragma once
 
#include <ikarus/finiteelements/fetraits.hh>
 
namespace Ikarus {
 
  template <typename Basis_>
  class ScalarFieldFE {
  public:
    using Basis                   = Basis_;                          
    using LocalView               = typename Basis::LocalView;       
    using GlobalIndex             = typename LocalView::MultiIndex;  
    using GridElement             = typename LocalView::Element;     
    using Traits                  = FETraits<GridElement>;           
    static constexpr int worlddim = Traits::worlddim;                
    explicit ScalarFieldFE(const Basis& basis, const typename LocalView::Element& element)
        : localView_{basis.localView()} {
      static_assert(Basis::PreBasis::Node::CHILDREN == 0, "This is no scalar basis!");
      localView_.bind(element);
    }
 
    [[nodiscard]] int size() const { return localView_.size(); }
 
    void globalFlatIndices(std::vector<GlobalIndex>& globalIndices) const {
      const auto& fe = localView_.tree().finiteElement();
      for (size_t i = 0; i < fe.size(); ++i)
        globalIndices.push_back(localView_.index(localView_.tree().localIndex(i)));
    }
 
    const GridElement& gridElement() const { return localView_.element(); }
 
    const LocalView& localView() const { return localView_; }
 
    LocalView& localView() { return localView_; }
 
  private:
    LocalView localView_; 
  };
}  // namespace Ikarus