segment_network.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: Sebastian Reiter
 * 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
 */
 
 
#ifndef __H__UG_segment_network
#define __H__UG_segment_network
 
#include "lib_grid/grid/grid.h"
#include "lib_grid/common_attachments.h"
#include "lib_grid/tools/subset_handler_grid.h"
#include "lib_disc/spatial_disc/user_data/user_data.h"
 
namespace ug{
namespace d3f{
 
 
class SegmentNetwork
{
  public:
    static const int dim = 3;
    
    typedef vector2         vector_t;
    typedef Attachment<vector_t>  position_attachment_t;
    typedef Grid::VertexAttachmentAccessor<position_attachment_t> position_accessor_t;
 
    typedef number          gauge_t;
    typedef Attachment<gauge_t>   gauge_attachment_t;
    typedef Grid::VertexAttachmentAccessor<gauge_attachment_t>    gauge_accessor_t;
    
  public:
    SegmentNetwork ()
    {
      m_grid.attach_to_vertices(m_aPos);
      m_aaPos.access(m_grid, m_aPos);
      m_grid.attach_to_vertices(m_aGauge);
      m_aaGauge.access(m_grid, m_aGauge);
      m_SH.assign_grid(&m_grid);
    }
  
    ~SegmentNetwork () {}
 
    void load_from_file
    (
      const char* name, 
      number maxEdgeLen 
    );
    
    void set_gauge_values
    (
      SmartPtr<UserData<number, dim> > h 
    );
 
    void save_to_file_with_
    (
      gauge_accessor_t& z_acc, 
      const char* name 
    );
 
    void save_to_file
    (
      const char* name 
    )
    {
      save_to_file_with_ (m_aaGauge, name);
    }
 
 
    size_t find_vertices
    (
      std::vector<Vertex*>& vrtsOut,
      const vector_t& center,
      number radius
    ) const;
 
    inline Grid& grid()                 {return m_grid;}
    
    inline position_attachment_t position_attachment () {return m_aPos;}
    inline position_accessor_t position_accessor ()   {return m_aaPos;}
 
    inline gauge_attachment_t gauge_attachment()    {return m_aGauge;}
    inline gauge_accessor_t gauge_accessor()      {return m_aaGauge;}
    
    inline GridSubsetHandler& subset_handler()      {return m_SH;}
 
  private:
    Grid          m_grid;
    position_attachment_t m_aPos;
    position_accessor_t   m_aaPos;
    gauge_attachment_t    m_aGauge;
    gauge_accessor_t    m_aaGauge;
    GridSubsetHandler   m_SH;
};
 
}// end of namespace
}// end of namespace
 
#endif  //__H__UG_segment_network