Skip to content
Snippets Groups Projects
Commit e27c2b45 authored by Dr. Felix Tobias Schindler's avatar Dr. Felix Tobias Schindler
Browse files

[P|functions] add gradient

parent 7858e20b
No related branches found
No related tags found
2 merge requests!41Fix compilation with icc, several other changes,!39Make functions copyable
...@@ -22,6 +22,7 @@ dune_pybindxi_add_module(_functions_function_as_grid_function EXCLUDE_FROM_ALL f ...@@ -22,6 +22,7 @@ dune_pybindxi_add_module(_functions_function_as_grid_function EXCLUDE_FROM_ALL f
dune_pybindxi_add_module(_functions_function_interface_1d EXCLUDE_FROM_ALL function-interface-1d.cc) dune_pybindxi_add_module(_functions_function_interface_1d EXCLUDE_FROM_ALL function-interface-1d.cc)
dune_pybindxi_add_module(_functions_function_interface_2d EXCLUDE_FROM_ALL function-interface-2d.cc) dune_pybindxi_add_module(_functions_function_interface_2d EXCLUDE_FROM_ALL function-interface-2d.cc)
dune_pybindxi_add_module(_functions_function_interface_3d EXCLUDE_FROM_ALL function-interface-3d.cc) dune_pybindxi_add_module(_functions_function_interface_3d EXCLUDE_FROM_ALL function-interface-3d.cc)
dune_pybindxi_add_module(_functions_gradient EXCLUDE_FROM_ALL gradient.cc)
dune_pybindxi_add_module(_functions_gridfunction EXCLUDE_FROM_ALL gridfunction.cc) dune_pybindxi_add_module(_functions_gridfunction EXCLUDE_FROM_ALL gridfunction.cc)
dune_pybindxi_add_module(_functions_indicator EXCLUDE_FROM_ALL indicator.cc) dune_pybindxi_add_module(_functions_indicator EXCLUDE_FROM_ALL indicator.cc)
dune_pybindxi_add_module(_functions_interfaces_grid_function_1d EXCLUDE_FROM_ALL interfaces/grid-function_1d.cc) dune_pybindxi_add_module(_functions_interfaces_grid_function_1d EXCLUDE_FROM_ALL interfaces/grid-function_1d.cc)
......
...@@ -28,6 +28,7 @@ for mod_name in ( ...@@ -28,6 +28,7 @@ for mod_name in (
'_functions_divergence', '_functions_divergence',
'_functions_function_as_grid_function', '_functions_function_as_grid_function',
'_functions_function_interface_3d', '_functions_function_interface_3d',
'_functions_gradient',
'_functions_gridfunction', '_functions_gridfunction',
'_functions_indicator', '_functions_indicator',
'_functions_interfaces_grid_function_1d', '_functions_interfaces_grid_function_1d',
......
// This file is part of the dune-xt project:
// https://github.com/dune-community/dune-xt
// Copyright 2009-2018 dune-xt developers and contributors. All rights reserved.
// License: Dual licensed as BSD 2-Clause License (http://opensource.org/licenses/BSD-2-Clause)
// or GPL-2.0+ (http://opensource.org/licenses/gpl-license)
// with "runtime exception" (http://www.dune-project.org/license.html)
// Authors:
// Felix Schindler (2019)
#include "config.h"
#include <string>
#include <vector>
#include <dune/common/parallel/mpihelper.hh>
#include <dune/pybindxi/pybind11.h>
#include <dune/pybindxi/functional.h>
#include <dune/pybindxi/stl.h>
#include <dune/xt/common/string.hh>
#include <dune/xt/grid/gridprovider/provider.hh>
#include <dune/xt/grid/type_traits.hh>
#include <dune/xt/functions/base/derivatives-of-grid-functions.hh>
#include <python/dune/xt/common/parameter.hh>
#include <python/dune/xt/common/fvector.hh>
#include <python/dune/xt/common/fmatrix.hh>
#include <python/dune/xt/common/bindings.hh>
#include <python/dune/xt/grid/traits.hh>
#include <python/dune/xt/common/exceptions.bindings.hh>
namespace Dune {
namespace XT {
namespace Functions {
namespace bindings {
template <class G, class E>
class GradientGridFunction
{
using GP = XT::Grid::GridProvider<G>;
static const size_t d = G::dimension;
public:
using type = Functions::GradientGridFunction<GridFunctionInterface<E>>;
using base_type = Functions::GridFunctionInterface<E, d>;
using bound_type = pybind11::class_<type, base_type>;
public:
static bound_type bind(pybind11::module& m,
const std::string& grid_id = Grid::bindings::grid_name<G>::value(),
const std::string& layer_id = "",
const std::string& class_id = "gradient_grid_function")
{
namespace py = pybind11;
using namespace pybind11::literals;
std::string class_name = class_id;
class_name += "_" + grid_id;
if (!layer_id.empty())
class_name += "_" + layer_id;
class_name += "_to_" + Common::to_string(size_t(d)) + "d";
const auto ClassName = Common::to_camel_case(class_name);
bound_type c(m, ClassName.c_str(), Common::to_camel_case(class_id).c_str());
c.def(py::init([](const Functions::GridFunctionInterface<E>& grid_function, const std::string& logging_prefix) {
return new type(grid_function, logging_prefix);
}),
"grid_function"_a,
"name"_a = "GradientGridFunction");
m.def(
"gradient",
[](const GridFunctionInterface<E>& grid_function, const std::string& name) {
return new type(grid_function, name);
},
"grid_function"_a,
"name"_a = "GradientGridFunction");
return c;
}
}; // class GridFunction
} // namespace bindings
} // namespace Functions
} // namespace XT
} // namespace Dune
template <class GridTypes = Dune::XT::Grid::AvailableGridTypes>
struct GradientGridFunction_for_all_grids
{
using G = typename GridTypes::head_type;
using GV = typename G::LeafGridView;
using E = Dune::XT::Grid::extract_entity_t<GV>;
static void bind(pybind11::module& m)
{
using Dune::XT::Functions::bindings::GradientGridFunction;
using Dune::XT::Grid::bindings::grid_name;
GradientGridFunction<G, E>::bind(m, grid_name<G>::value());
GradientGridFunction_for_all_grids<typename GridTypes::tail_type>::bind(m);
}
};
template <>
struct GradientGridFunction_for_all_grids<boost::tuples::null_type>
{
static void bind(pybind11::module& /*m*/) {}
};
PYBIND11_MODULE(_functions_gradient, m)
{
namespace py = pybind11;
py::module::import("dune.xt.common");
py::module::import("dune.xt.grid");
py::module::import("dune.xt.la");
py::module::import("dune.xt.functions._functions_interfaces_grid_function_1d");
py::module::import("dune.xt.functions._functions_interfaces_grid_function_2d");
py::module::import("dune.xt.functions._functions_interfaces_grid_function_3d");
py::module::import("dune.xt.functions._functions_gridfunction");
GradientGridFunction_for_all_grids<>::bind(m);
} // PYBIND11_MODULE(...)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment