smile_rescale.hpp

Go to the documentation of this file.

 
#ifndef _SMILE_RESCALE_HPP_
#define _SMILE_RESCALE_HPP_
 
#include <vector>
 
#include "common/common.h"
#include "lib_disc/function_spaces/grid_function.h"
#include "lib_disc/spatial_disc/user_data/user_data.h"
 
#include "smile_cloud.hpp"
 
namespace ug {
namespace smile {
 
template <typename TDomain, typename TAlgebra>
class SmartRescaler
{
  typedef SmartRescaler<TDomain, TAlgebra> this_type;
  
public:
  typedef TDomain domain_type;
  
  typedef typename TDomain::grid_type grid_type;
  
  typedef typename domain_type::subset_handler_type subset_handler_type;
  
  typedef typename domain_type::position_accessor_type position_accessor_type;
  
  typedef typename domain_type::position_type position_type;
  
  static const int WDim = TDomain::dim;
  
  typedef TAlgebra algebra_type;
  
  typedef UserData<number, WDim> factor_type;
  
  typedef ANumber a_factor_type;
  
  typedef Grid::VertexAttachmentAccessor<a_factor_type> aa_factor_type;
  
  typedef GridFunction<domain_type, algebra_type> gf_type;
 
public:
 
  SmartRescaler () {};
  
  void set_factor (SmartPtr<factor_type> factor)
  {
    m_spFactor = factor;
  }
 
  void multiply
  (
    SmartPtr<gf_type> gf, 
    const char * cmps 
  );
  
  void divide
  (
    SmartPtr<gf_type> gf, 
    const char * cmps 
  );
  
private:
 
  template <typename TElem>
  void get_scaling_factors_for_
  (
    int si, 
    gf_type & gf, 
    aa_factor_type & factor 
  );
  struct GetScalingFactors
  {
    this_type * m_pThis;
    int m_si;
    gf_type & m_gf;
    aa_factor_type & m_aa_factor;
    
    GetScalingFactors
    (
      this_type * pThis, 
      int si, 
      gf_type & gf, 
      aa_factor_type & aa_factor 
    )
    : m_pThis (pThis), m_si (si), m_gf (gf), m_aa_factor (aa_factor) {}
    
    template <typename TElem> void operator() (TElem &)
    {
      m_pThis->template get_scaling_factors_for_<TElem> (m_si, m_gf, m_aa_factor);
    }
  };
  void get_scaling_factors
  (
    gf_type & gf, 
    a_factor_type & factor 
  );
  
private:
 
  SmartPtr<factor_type> m_spFactor;
};
 
} // end namespace smile
} // end namespace ug
 
#include "smile_rescale_impl.hpp"
 
#endif // _SMILE_RESCALE_HPP_
 
/* End of File */