field_profiles.hpp

Go to the documentation of this file.

/*
 * This file is part of BubbleProfiler.
 *
 * BubbleProfiler is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * BubbleProfiler is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with BubbleProfiler.  If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef BUBBLEPROFILER_FIELD_PROFILES_HPP_INCLUDED
#define BUBBLEPROFILER_FIELD_PROFILES_HPP_INCLUDED

#include "gsl_interpolator.hpp"

#include <Eigen/Core>

#include <vector>

namespace BubbleProfiler {

class Field_profiles {
public:
   Field_profiles() = default;

   Field_profiles(const Eigen::VectorXd& gridpoints_,
                  const Eigen::MatrixXd& profiles_,
                  double interpolation_points_fraction_ = 1.0);

   Field_profiles(int num_fields_, double domain_start_,
                  double domain_end_, int num_steps_,
                  double interpolation_points_fraction_ = 1.0);

   Field_profiles(const Eigen::MatrixXd& profiles_,
                  double domain_start_, double domain_end_,
                  double interpolation_points_fraction_ = 1.0);

   ~Field_profiles() = default;
   Field_profiles(const Field_profiles&) = default;
   Field_profiles& operator=(const Field_profiles&) = default;
   Field_profiles(Field_profiles&&) = default;
   Field_profiles& operator=(Field_profiles&&) = default;

   void set_number_of_dimensions(int d) { n_spatial_dims = d; }
   void set_interpolation_points_fraction(double f);
   void set_field_profile(int i, const Eigen::VectorXd& p);

   int get_number_of_fields() const { return profiles.cols(); }
   int get_number_of_dimensions() const { return n_spatial_dims; }

   int get_number_of_grid_points() const { return gridpoints.size(); }
   double get_domain_start() const { return gridpoints(0); }
   double get_domain_end() const { return gridpoints(gridpoints.size() - 1); }

   const Eigen::VectorXd& get_spatial_grid() const { return gridpoints; }

   const Eigen::MatrixXd& get_field_profiles() const { return profiles; }

   Eigen::VectorXd evaluate_at(double rho) const;

   double evaluate_at(int field, double rho) const;

   double derivative_at(int field, int order, double rho) const;

private:
   Eigen::VectorXd gridpoints{};
   Eigen::MatrixXd profiles{};
   double n_spatial_dims{3};
   bool must_update_coords{true};
   double interpolation_points_fraction{1.0};
   Eigen::VectorXd interpolation_grid_values{};
   Eigen::MatrixXd interpolation_field_values{};
   std::vector<GSL_interpolator> splines{};

   void initialize_interpolation_grid_values();
   void initialize_interpolation_field_values(int);
   void initialize_interpolation_field_values();
   void build_spline_for_field(int);
   void build_splines();
};

} // namespace BubbleProfiler

#endif