extravol.h

Go to the documentation of this file.

/* 
 * SPDX-FileCopyrightText: 2025 Gesellschaft fuer Anlagen- und Reaktorsicherheit gGmbH
 * SPDX-License-Identifier: EUPL-1.2
 * SPDX-FileContributor: Dmitry Logashenko
 * SPDX-FileContributor: Goethe Universität Frankfurt
 * SPDX-FileType: SOURCE
 * 
 * This file is part of d3f++.
 * d3f++ is an extension for UG4. Licensing information and citation requirements of UG4 are provided in LICENSES/UG4-LGPL_2.1
 */
 
 
/*
 * extravol.h - data structures for the volume stabilization of flow equations.
 */
#ifndef __H__UG__PLUGINS__D3F__EXTRAVOL__
#define __H__UG__PLUGINS__D3F__EXTRAVOL__
 
namespace ug {
namespace d3f {
 
template <typename TDomain>
class FlowVolStabData
{
public:
 
  typedef TDomain domain_type;
 
  typedef typename TDomain::grid_type grid_type;
 
  typedef AVector2 t_aVector2;
 
  typedef typename Grid::VertexAttachmentAccessor<t_aVector2> t_aaVector2;
  
  typedef ABool t_aFlag;
  
  typedef typename Grid::VertexAttachmentAccessor<t_aFlag> t_aaFlag;
 
public:
 
  FlowVolStabData
  (
    SmartPtr<domain_type> dom
  )
  : m_spDomain (dom), m_bFluxActive (false), m_bFlagsActive (false)
  {}
  
  void reset_flux ()
  {
    if (! m_bFluxActive) return;
    grid_type & grid = * m_spDomain->grid ();
    SetAttachmentValues (m_aExtraVol, grid.vertices_begin (), grid.vertices_end (), 0);
  }
  
  void activate_flux ()
  {
    if (m_bFluxActive) return;
    grid_type & grid = * m_spDomain->grid ();
    grid.template attach_to_dv<Vertex> (m_extraVol, 0);
    m_aExtraVol.access (grid, m_extraVol);
    m_bFluxActive = true;
  }
  
  bool flux_is_active () {return m_bFluxActive;}
  
  void reset_flags ()
  {
    if (! m_bFlagsActive) return;
    grid_type & grid = * m_spDomain->grid ();
    SetAttachmentValues (m_aFlags, grid.vertices_begin (), grid.vertices_end (), false);
  }
  
  void activate_flags ()
  {
    if (m_bFlagsActive) return;
    grid_type & grid = * m_spDomain->grid ();
    grid.template attach_to_dv<Vertex> (m_flags, 0);
    m_aFlags.access (grid, m_flags);
    m_bFlagsActive = true;
  }
  
  bool flags_are_active () {return m_bFlagsActive;}
  
  number & stiff
  (
    const Vertex * vrt
  )
  {
    return m_aExtraVol[vrt][0];
  }
  
  number & mass
  (
    const Vertex * vrt
  )
  {
    return m_aExtraVol[vrt][1];
  }
  
  MathVector<2> & value
  (
    const Vertex * vrt
  )
  {
    return m_aExtraVol[vrt];
  }
  
  bool flag
  (
    const Vertex * vrt
  )
  {
    return m_aFlags[vrt];
  }
  
  void set_flag
  (
    const Vertex * vrt,
    bool v
  )
  {
    m_aFlags[vrt] = v;
  }
  
private:
 
  SmartPtr<domain_type> m_spDomain;
 
  bool m_bFluxActive;
  t_aVector2 m_extraVol;
  t_aaVector2 m_aExtraVol;
  
  bool m_bFlagsActive;
  t_aFlag m_flags;
  t_aaFlag m_aFlags;
};
 
} // namespace d3f
} // end namespace ug
 
#endif // __H__UG__PLUGINS__D3F__EXTRAVOL__
 
/* End of File */