estimate.h

Go to the documentation of this file.

/*
This file is part of the Ristra portage project.
Please see the license file at the root of this repository, or at:
    https://github.com/laristra/portage/blob/master/LICENSE
*/
#ifndef ESTIMATE_H_INC_
#define ESTIMATE_H_INC_

#include <vector>
#include <string>
#include <cassert>

#include "portage/swarm/swarm_state.h"
#include "portage/support/portage.h"

namespace Portage {
namespace Meshfree {

using std::string;

template<int dim, class TargetSwarmState>
class Estimate {
 public:
  Estimate(SwarmState<dim> const& source_state):
    source_state_(source_state)
  {}

  void set_interpolation_variable(std::string const & interp_var_name, size_t derivative=0) {
    var_name_ = interp_var_name;
    derivative_ = derivative;
  }

  double operator()(int const target_index,
                    std::vector<Weights_t> const &sources_and_mults) const
  {    
    int nsrc = sources_and_mults.size();
    if (nsrc > 0) assert(derivative_ < sources_and_mults[0].weights.size());

    double result = 0.;
    for (int i = 0; i < nsrc; i++) {
      Weights_t const& wt = sources_and_mults[i];
      int p = wt.entityID;
      std::vector<double> const& shape_vec = wt.weights;
      result += source_vals_[p] * shape_vec[derivative_];
    }
    return result;
  }

  void set_variable(std::string const & var_name, size_t derivin=0) {
    var_name_ = var_name;
    derivative_ = derivin;
    source_vals_ = source_state_.get_field(var_name_);
  }

 private:
  SwarmState<dim> const& source_state_;
  std::string var_name_;
  size_t derivative_;
  Portage::vector<double> source_vals_;
};

}
}

#endif // ESTIMATE_H_INC