finite_element_dual.hpp

Go to the documentation of this file.

// Copyright (c) Lawrence Livermore National Security, LLC and
// other Smith Project Developers. See the top-level LICENSE file for
// details.
//
// SPDX-License-Identifier: (BSD-3-Clause)
 
#pragma once
 
#include <memory>
 
#include "mfem.hpp"
 
#include "smith/physics/state/finite_element_vector.hpp"
 
namespace smith {
 
class FiniteElementDual : public FiniteElementVector {
 public:
  using FiniteElementVector::FiniteElementVector;
  using FiniteElementVector::operator=;
  using mfem::Vector::Print;
 
  FiniteElementDual(const FiniteElementDual& rhs) : FiniteElementVector(rhs)
  {
    if (rhs.linear_form_) {
      linearForm();
      *linear_form_ = *rhs.linear_form_;
    }
  }
 
  FiniteElementDual(FiniteElementDual&& rhs) : FiniteElementVector(rhs)
  {
    this->linear_form_ = std::move(rhs.linear_form_);
  }
 
  FiniteElementDual& operator=(FiniteElementDual&& rhs)
  {
    FiniteElementVector::operator=(rhs);
    this->linear_form_ = std::move(rhs.linear_form_);
    return *this;
  }
 
  FiniteElementDual& operator=(const FiniteElementDual& rhs)
  {
    FiniteElementVector::operator=(rhs);
    if (rhs.linear_form_) {
      linearForm();
      *linear_form_ = *rhs.linear_form_;
    }
    return *this;
  }
 
  void fillLinearForm(mfem::ParLinearForm& linear_form) const
  {
    space_->GetRestrictionMatrix()->MultTranspose(*this, linear_form);
  }
 
  void setFromLinearForm(const mfem::ParLinearForm& linear_form)
  {
    const_cast<mfem::ParLinearForm&>(linear_form).ParallelAssemble(*this);
  }
 
  mfem::ParLinearForm& linearForm() const
  {
    if (!linear_form_) {
      linear_form_ = std::make_unique<mfem::ParLinearForm>(space_.get());
    }
 
    fillLinearForm(*linear_form_);
    return *linear_form_;
  }
 
 protected:
  mutable std::unique_ptr<mfem::ParLinearForm> linear_form_;
};
 
}  // namespace smith