* Pointer that can be casted to the underlying CellAccessor object.
*/
void *cell_accessor;
+
+ friend class MappingQGenericWrapper;
};
--- /dev/null
+// ---------------------------------------------------------------------
+//
+// Copyright (C) 2016 - 2017 by the deal.II authors
+//
+// This file is part of the deal.II library.
+//
+// The deal.II library is free software; you can use it, redistribute
+// it, and/or modify it under the terms of the GNU Lesser General
+// Public License as published by the Free Software Foundation; either
+// version 2.1 of the License, or (at your option) any later version.
+// The full text of the license can be found in the file LICENSE.md at
+// the top level directory of deal.II.
+//
+// ---------------------------------------------------------------------
+
+#ifndef dealii_mapping_wrapper_h
+#define dealii_mapping_wrapper_h
+
+#include <deal.II/base/config.h>
+
+#include <deal.II/fe/mapping_q_generic.h>
+
+#include <boost/python.hpp>
+
+#include <cell_accessor_wrapper.h>
+#include <point_wrapper.h>
+
+DEAL_II_NAMESPACE_OPEN
+
+namespace python
+{
+ class MappingQGenericWrapper
+ {
+ public:
+ /**
+ * Copy constructor.
+ */
+ MappingQGenericWrapper(const MappingQGenericWrapper &other);
+
+ /**
+ * Constructor.
+ */
+ MappingQGenericWrapper(const int dim, const int spacedim, const int degree);
+
+ /**
+ * Destructor.
+ */
+ ~MappingQGenericWrapper();
+
+ /**
+ * Map the point p on the unit cell to the corresponding point on the real
+ * cell.
+ */
+ PointWrapper
+ transform_unit_to_real_cell(CellAccessorWrapper &cell, PointWrapper &point);
+
+ /**
+ * Map the point p on the real cell to the corresponding point on the unit
+ * cell.
+ */
+ PointWrapper
+ transform_real_to_unit_cell(CellAccessorWrapper &cell, PointWrapper &point);
+
+ private:
+ /**
+ * Dimension of the underlying Mapping object.
+ */
+ int dim;
+
+ /**
+ * Space dimension of the underlying Mapping object.
+ */
+ int spacedim;
+
+ /**
+ * Degree of the underlying mapping.
+ */
+ int degree;
+
+ /**
+ * Pointer that can be casted to the underlying Mapping object.
+ */
+ void *mapping_ptr;
+ };
+
+} // namespace python
+
+DEAL_II_NAMESPACE_CLOSE
+
+#endif
* Pointer to the underlying Point object.
*/
void *point;
+
+ friend class MappingQGenericWrapper;
};
export_cell_accessor.cc
export_point.cc
export_triangulation.cc
+ export_mapping.cc
cell_accessor_wrapper.cc
tria_accessor_wrapper.cc
point_wrapper.cc
triangulation_wrapper.cc
+ mapping_wrapper.cc
)
FOREACH(_build ${DEAL_II_BUILD_TYPES})
static_cast<const CellAccessor<dim, spacedim> *>(cell_accessor);
Point<spacedim> barycenter = cell->barycenter();
boost::python::list barycenter_list;
- for (int i = 0; i < dim; ++i)
+ for (int i = 0; i < spacedim; ++i)
barycenter_list.append(barycenter[i]);
return PointWrapper(barycenter_list);
--- /dev/null
+// ---------------------------------------------------------------------
+//
+// Copyright (C) 2016 - 2017 by the deal.II authors
+//
+// This file is part of the deal.II library.
+//
+// The deal.II library is free software; you can use it, redistribute
+// it, and/or modify it under the terms of the GNU Lesser General
+// Public License as published by the Free Software Foundation; either
+// version 2.1 of the License, or (at your option) any later version.
+// The full text of the license can be found in the file LICENSE.md at
+// the top level directory of deal.II.
+//
+// ---------------------------------------------------------------------
+
+#include <boost/python.hpp>
+
+#include <mapping_wrapper.h>
+
+DEAL_II_NAMESPACE_OPEN
+
+namespace python
+{
+ const char transform_unit_to_real_cell_docstring[] =
+ " Map the point p on the unit cell to the corresponding point \n"
+ " on the real cell. \n";
+
+ const char transform_real_to_unit_cell_docstring[] =
+ " Map the point p on the real cell to the corresponding point \n"
+ " on the unit cell. \n";
+
+
+
+ void
+ export_mapping()
+ {
+ boost::python::class_<MappingQGenericWrapper>(
+ "MappingQGeneric",
+ boost::python::init<const int, const int, const int>(
+ boost::python::args("dim", "spacedim", "degree")))
+ .def("transform_real_to_unit_cell",
+ &MappingQGenericWrapper::transform_real_to_unit_cell,
+ transform_real_to_unit_cell_docstring,
+ boost::python::args("self", "cell_wrapper", "point_wrapper"))
+ .def("transform_unit_to_real_cell",
+ &MappingQGenericWrapper::transform_unit_to_real_cell,
+ transform_unit_to_real_cell_docstring,
+ boost::python::args("self", "cell_wrapper", "point_wrapper"));
+ }
+} // namespace python
+
+DEAL_II_NAMESPACE_CLOSE
--- /dev/null
+// ---------------------------------------------------------------------
+//
+// Copyright (C) 2016 - 2017 by the deal.II authors
+//
+// This file is part of the deal.II library.
+//
+// The deal.II library is free software; you can use it, redistribute
+// it, and/or modify it under the terms of the GNU Lesser General
+// Public License as published by the Free Software Foundation; either
+// version 2.1 of the License, or (at your option) any later version.
+// The full text of the license can be found in the file LICENSE.md at
+// the top level directory of deal.II.
+//
+// ---------------------------------------------------------------------
+
+#include <cell_accessor_wrapper.h>
+#include <mapping_wrapper.h>
+#include <point_wrapper.h>
+#include <triangulation_wrapper.h>
+
+DEAL_II_NAMESPACE_OPEN
+
+namespace python
+{
+ namespace internal
+ {
+ template <int dim, int spacedim>
+ PointWrapper
+ transform_unit_to_real_cell(void *mapping_ptr,
+ void *cell_accessor_ptr,
+ void *point_ptr)
+ {
+ const MappingQGeneric<dim, spacedim> *mapping =
+ static_cast<const MappingQGeneric<dim, spacedim> *>(mapping_ptr);
+
+ const CellAccessor<dim, spacedim> *cell_accessor =
+ static_cast<const CellAccessor<dim, spacedim> *>(cell_accessor_ptr);
+
+ const Point<dim> *point = static_cast<const Point<dim> *>(point_ptr);
+
+ typename Triangulation<dim, spacedim>::active_cell_iterator cell(
+ &cell_accessor->get_triangulation(),
+ cell_accessor->level(),
+ cell_accessor->index());
+
+ Point<spacedim> p_real =
+ mapping->transform_unit_to_real_cell(cell, *point);
+
+ boost::python::list coord_list;
+ for (int i = 0; i < spacedim; ++i)
+ coord_list.append(p_real[i]);
+
+ return PointWrapper(coord_list);
+ }
+
+
+
+ template <int dim, int spacedim>
+ PointWrapper
+ transform_real_to_unit_cell(void *mapping_ptr,
+ void *cell_accessor_ptr,
+ void *point_ptr)
+ {
+ const MappingQGeneric<dim, spacedim> *mapping =
+ static_cast<const MappingQGeneric<dim, spacedim> *>(mapping_ptr);
+
+ const CellAccessor<dim, spacedim> *cell_accessor =
+ static_cast<const CellAccessor<dim, spacedim> *>(cell_accessor_ptr);
+
+ const Point<spacedim> *point =
+ static_cast<const Point<spacedim> *>(point_ptr);
+
+ typename Triangulation<dim, spacedim>::active_cell_iterator cell(
+ &cell_accessor->get_triangulation(),
+ cell_accessor->level(),
+ cell_accessor->index());
+
+ Point<dim> p_real = mapping->transform_real_to_unit_cell(cell, *point);
+
+ boost::python::list coord_list;
+ for (int i = 0; i < dim; ++i)
+ coord_list.append(p_real[i]);
+
+ return PointWrapper(coord_list);
+ }
+ } // namespace internal
+
+
+
+ MappingQGenericWrapper::MappingQGenericWrapper(const int dim,
+ const int spacedim,
+ const int degree)
+ : dim(dim)
+ , spacedim(spacedim)
+ , degree(degree)
+ {
+ if ((dim == 2) && (spacedim == 2))
+ {
+ mapping_ptr = new MappingQGeneric<2, 2>(degree);
+ }
+ else if ((dim == 2) && (spacedim == 3))
+ {
+ mapping_ptr = new MappingQGeneric<2, 3>(degree);
+ }
+ else if ((dim == 3) && (spacedim == 3))
+ {
+ mapping_ptr = new MappingQGeneric<3, 3>(degree);
+ }
+ else
+ AssertThrow(false, ExcMessage("Wrong dim-spacedim combination."));
+ }
+
+
+
+ MappingQGenericWrapper::MappingQGenericWrapper(
+ const MappingQGenericWrapper &other)
+ {
+ dim = other.dim;
+ spacedim = other.spacedim;
+ degree = other.degree;
+
+ AssertThrow(other.mapping_ptr != nullptr,
+ ExcMessage("Underlying mapping does not exist."));
+
+ if ((dim == 2) && (spacedim == 2))
+ {
+ mapping_ptr = new MappingQGeneric<2, 2>(other.degree);
+ }
+ else if ((dim == 2) && (spacedim == 3))
+ {
+ mapping_ptr = new MappingQGeneric<2, 3>(other.degree);
+ }
+ else if ((dim == 3) && (spacedim == 3))
+ {
+ mapping_ptr = new MappingQGeneric<3, 3>(other.degree);
+ }
+ else
+ AssertThrow(false, ExcMessage("Wrong dim-spacedim combination."));
+ }
+
+
+
+ MappingQGenericWrapper::~MappingQGenericWrapper()
+ {
+ if (dim != -1)
+ {
+ if ((dim == 2) && (spacedim == 2))
+ {
+ // We cannot call delete on a void pointer so cast the void pointer
+ // back first.
+ MappingQGeneric<2, 2> *tmp =
+ static_cast<MappingQGeneric<2, 2> *>(mapping_ptr);
+ delete tmp;
+ }
+ else if ((dim == 2) && (spacedim == 3))
+ {
+ MappingQGeneric<2, 3> *tmp =
+ static_cast<MappingQGeneric<2, 3> *>(mapping_ptr);
+ delete tmp;
+ }
+ else
+ {
+ MappingQGeneric<3, 3> *tmp =
+ static_cast<MappingQGeneric<3, 3> *>(mapping_ptr);
+ delete tmp;
+ }
+
+ dim = -1;
+ spacedim = -1;
+ degree = -1;
+ mapping_ptr = nullptr;
+ }
+ }
+
+
+
+ PointWrapper
+ MappingQGenericWrapper::transform_unit_to_real_cell(CellAccessorWrapper &cell,
+ PointWrapper & p)
+ {
+ AssertThrow(
+ dim == p.get_dim(),
+ ExcMessage(
+ "Dimension of the point is not equal to the dimension of the mapping."));
+
+ if ((dim == 2) && (spacedim == 2))
+ return internal::transform_unit_to_real_cell<2, 2>(mapping_ptr,
+ cell.cell_accessor,
+ p.point);
+ else if ((dim == 2) && (spacedim == 3))
+ return internal::transform_unit_to_real_cell<2, 3>(mapping_ptr,
+ cell.cell_accessor,
+ p.point);
+ else
+ return internal::transform_unit_to_real_cell<3, 3>(mapping_ptr,
+ cell.cell_accessor,
+ p.point);
+ }
+
+
+
+ PointWrapper
+ MappingQGenericWrapper::transform_real_to_unit_cell(CellAccessorWrapper &cell,
+ PointWrapper & p)
+ {
+ AssertThrow(
+ spacedim == p.get_dim(),
+ ExcMessage(
+ "Dimension of the point is not equal to the space dimension of the mapping."));
+
+ if ((dim == 2) && (spacedim == 2))
+ return internal::transform_real_to_unit_cell<2, 2>(mapping_ptr,
+ cell.cell_accessor,
+ p.point);
+ else if ((dim == 2) && (spacedim == 3))
+ return internal::transform_real_to_unit_cell<2, 3>(mapping_ptr,
+ cell.cell_accessor,
+ p.point);
+ else
+ return internal::transform_real_to_unit_cell<3, 3>(mapping_ptr,
+ cell.cell_accessor,
+ p.point);
+ }
+
+} // namespace python
+
+DEAL_II_NAMESPACE_CLOSE
tria_accessor);
Point<spacedim> barycenter = accessor->barycenter();
boost::python::list barycenter_list;
- for (int i = 0; i < dim; ++i)
+ for (int i = 0; i < spacedim; ++i)
barycenter_list.append(barycenter[i]);
return PointWrapper(barycenter_list);
export_point();
void
export_triangulation();
+ void
+ export_mapping();
} // namespace python
DEAL_II_NAMESPACE_CLOSE
dealii::python::export_cell_accessor();
dealii::python::export_point();
dealii::python::export_triangulation();
+ dealii::python::export_mapping();
}
#else
dealii::python::export_cell_accessor();
dealii::python::export_point();
dealii::python::export_triangulation();
+ dealii::python::export_mapping();
}
#endif
--- /dev/null
+# ---------------------------------------------------------------------
+#
+# Copyright (C) 2016 by the deal.II authors
+#
+# This file is part of the deal.II library.
+#
+# The deal.II library is free software; you can use it, redistribute
+# it, and/or modify it under the terms of the GNU Lesser General
+# Public License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+# The full text of the license can be found in the file LICENSE.md at
+# the top level directory of deal.II.
+#
+# ---------------------------------------------------------------------
+
+import os
+import sys
+module_path = os.path.abspath('/home/agrayver/lib/dealii/build_gofem/lib/python3.7/site-packages')
+if module_path not in sys.path:
+ sys.path.append(module_path)
+
+import unittest
+from PyDealII.Debug import *
+
+class TestMappingWrapperCube(unittest.TestCase):
+
+ def setUp(self):
+ self.triangulation = Triangulation('2D')
+ self.triangulation.generate_hyper_cube()
+ self.triangulation.refine_global(1)
+
+ def test_mapping(self):
+ mapping = MappingQGeneric(dim = 2, spacedim = 2, degree = 1)
+ p_unit = Point([0.5, 0.5])
+
+ for cell in self.triangulation.active_cells():
+ p_real = mapping.transform_unit_to_real_cell(cell, p_unit)
+ p_unit_back = mapping.transform_real_to_unit_cell(cell, p_real)
+
+ self.assertTrue(p_unit.distance(p_unit_back) < 1e-10)
+
+class TestMappingWrapperSphere(unittest.TestCase):
+
+ def setUp(self):
+ self.triangulation = Triangulation('2D', '3D')
+ p_center = Point([0., 0., 0.])
+ self.triangulation.generate_hyper_sphere(p_center)
+
+ def test_mapping(self):
+ mapping = MappingQGeneric(dim = 2, spacedim = 3, degree = 4)
+ p_unit = Point([0.5, 0.5])
+
+ for cell in self.triangulation.active_cells():
+ p_real = mapping.transform_unit_to_real_cell(cell, p_unit)
+ p_unit_back = mapping.transform_real_to_unit_cell(cell, p_real)
+
+ self.assertTrue(p_unit.distance(p_unit_back) < 1e-10)
+
+if __name__ == '__main__':
+ unittest.main()
--- /dev/null
+Added: New python wrappers for the MappingQGeneric class
+<br>
+(Alexander Grayver, 2019/11/18)
projects / dealii.git / commitdiff