diff --git a/python/dune/xt/functions/CMakeLists.txt b/python/dune/xt/functions/CMakeLists.txt index b307518a7b66e50fde2233b9fe39825e1f235a1b..3212c5ac29bc06393c19c3c67cece795df13d8ef 100644 --- a/python/dune/xt/functions/CMakeLists.txt +++ b/python/dune/xt/functions/CMakeLists.txt @@ -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_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_gradient EXCLUDE_FROM_ALL gradient.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_interfaces_grid_function_1d EXCLUDE_FROM_ALL interfaces/grid-function_1d.cc) diff --git a/python/dune/xt/functions/__init__.py b/python/dune/xt/functions/__init__.py index 0b8e02c2dc7f5dda097b9e9e1e55831551437071..2a9d186cd883395dd02b7c96f05b04abc02e5d27 100644 --- a/python/dune/xt/functions/__init__.py +++ b/python/dune/xt/functions/__init__.py @@ -28,6 +28,7 @@ for mod_name in ( '_functions_divergence', '_functions_function_as_grid_function', '_functions_function_interface_3d', + '_functions_gradient', '_functions_gridfunction', '_functions_indicator', '_functions_interfaces_grid_function_1d', diff --git a/python/dune/xt/functions/gradient.cc b/python/dune/xt/functions/gradient.cc new file mode 100644 index 0000000000000000000000000000000000000000..34413212c35a4111dac4e1cf5f1676b650397d99 --- /dev/null +++ b/python/dune/xt/functions/gradient.cc @@ -0,0 +1,129 @@ +// 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(...)