subset_user_data.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: Julian Hilbert
 * 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 SUBSET_USER_DATA_H
#define SUBSET_USER_DATA_H
#include "lib_disc/spatial_disc/user_data/linker/linker_traits.h"
#include "lib_disc/spatial_disc/user_data/linker/linker.h"
 
namespace ug{
 
 
template <typename TData, typename TDomain, typename TRet = TData>
class SubsetUserData: public StdDataLinker<SubsetUserData<TData, TDomain, TRet>, TRet, TDomain::dim> {
  public:
  static const int dim = TDomain::dim;
  typedef CplUserData<TData, dim> TUserData;
  typedef StdDataLinker<SubsetUserData<TData, TDomain, TRet>, TRet, dim> base_type;
  SubsetUserData(){}
 
 
  void set_domain(SmartPtr<TDomain> domain) {
    m_sp_domain = domain;
  }
  void insert(SmartPtr<TUserData> data, size_t index ){
 
 
 
      m_sp_data.push_back(data);
      m_index_map[index] = base_type::num_input();
      base_type::set_num_input(base_type::num_input() + 1);
 
 
      base_type::set_input(m_index_map[index], data, data);
 
  }
 
 
 
  inline void evaluate (TData& value,
                          const MathVector<dim>& globIP,
                          number time, int si) const
  {
 
   (*m_sp_data[m_index_map.at(si)])(value, globIP, time, si);
 
 
  }
 
  template <int refDim>
    inline void evaluate(TData vValue[],
               const MathVector<dim> vGlobIP[],
               number time, int si,
               GridObject* elem,
               const MathVector<dim> vCornerCoords[],
               const MathVector<refDim> vLocIP[],
               const size_t nip,
               LocalVector* u,
               const MathMatrix<refDim, dim>* vJT = NULL) const {
 
    int elem_si = m_sp_domain->subset_handler()->get_subset_index(elem);
 
 
    (*m_sp_data[m_index_map.at(elem_si)])(vValue, vGlobIP, time, si,
                                   elem, vCornerCoords,
                                   vLocIP, nip, u,vJT);
 
    }
  template <int refDim>
    void eval_and_deriv(TData vValue[],
                        const MathVector<dim> vGlobIP[],
                        number time, int si,
                        GridObject* elem,
                        const MathVector<dim> vCornerCoords[],
                        const MathVector<refDim> vLocIP[],
                        const size_t nip,
                        LocalVector* u,
                        bool bDeriv,
                        int s,
                        std::vector<std::vector<TData> > vvvDeriv[],
                        const MathMatrix<refDim, dim>* vJT = NULL) const
  {
    int elem_si = m_sp_domain->subset_handler()->get_subset_index(elem);
    (*m_sp_data[m_index_map.at(elem_si)])(vValue, vGlobIP, time, si,
                                   elem, vCornerCoords,
                                   vLocIP, nip, u,vJT);
 
    this->set_zero(vvvDeriv, nip);
 
    if(!m_sp_data[m_index_map.at(elem_si)]->zero_derivative())
    {
      for(size_t ip = 0; ip < nip; ++ip)
      {
        //  loop functions
        for(size_t fct = 0; fct < m_sp_data[m_index_map.at(elem_si)]->num_fct(); ++fct)
        {
          //  get common fct id for this function
          const size_t commonFct = this->input_common_fct(elem_si, fct);
 
          //  loop dofs
          for(size_t sh = 0; sh < this->num_sh(fct); ++sh)
          {
            linker_traits<TData, number,TRet>::
            mult_add(vvvDeriv[ip][commonFct][sh],(m_sp_data[m_index_map.at(elem_si)].template cast_dynamic<DependentUserData<TData, dim>>())->deriv(this->series_id(si, s), ip, fct, sh), 1.0);
 
          }
        }
      }
    }
  }
  private:
  std::map<int, int> m_index_map;
  std::vector<SmartPtr<TUserData > > m_sp_data;
  SmartPtr<TDomain> m_sp_domain;
};
 
 
}
#endif //SUBSET_USER_DATA_H