1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
#include <lib/base/Logging.hpp>
#include <lib/smoothing/WeightedAverage2d.hpp>
CREATE_CPP_LOCAL_LOGGER("WeightedAverage2d.cpp");
namespace yade { // Cannot have #include directive inside.
/* Tell whether point is inside polygon
*
* See _utils.cpp: pointInsidePolygon for docs and license.
*/
bool pyGaussAverage::pointInsidePolygon(const Vector2r& pt, const vector<Vector2r>& vertices)
{
int i /*current node*/, j /*previous node*/;
bool inside = false;
int rows = (int)vertices.size();
const Real &testx = pt[0], testy = pt[1];
for (i = 0, j = rows - 1; i < rows; j = i++) {
const Real &vx_i = vertices[i][0], vy_i = vertices[i][1], vx_j = vertices[j][0], vy_j = vertices[j][1];
if (((vy_i > testy) != (vy_j > testy)) && (testx < (vx_j - vx_i) * (testy - vy_i) / (vy_j - vy_i) + vx_i)) inside = !inside;
}
return inside;
}
} // namespace yade
// BOOST_PYTHON_MODULE cannot be inside yade namespace, it has 'extern "C"' keyword, which strips it out of any namespaces.
BOOST_PYTHON_MODULE(WeightedAverage2d)
try {
using pyGaussAverage = ::yade::pyGaussAverage;
using ::yade::Real;
boost::python::scope().attr("__doc__") = "Smoothing (2d gauss-weighted average) for postprocessing scalars in 2d.";
boost::python::class_<pyGaussAverage>(
"GaussAverage",
boost::python::init<boost::python::tuple, boost::python::tuple, boost::python::tuple, Real, boost::python::optional<Real>>(
boost::python::args("min", "max", "nCells", "stDev", "relThreshold"),
"Create empty container for data, which can be added using add and later retrieved using avg."))
.def("add", &pyGaussAverage::addPt)
.def("avg", &pyGaussAverage::avg)
.def("avgPerUnitArea", &pyGaussAverage::avgPerUnitArea)
.def("cellNum", &pyGaussAverage::cellNum)
.def("cellSum", &pyGaussAverage::cellSum)
.def("cellAvg", &pyGaussAverage::cellAvg)
.add_property("stDev", &pyGaussAverage::stDev_get, &pyGaussAverage::stDev_set)
.add_property("relThreshold", &pyGaussAverage::relThreshold_get, &pyGaussAverage::relThreshold_set)
.add_property("clips", &pyGaussAverage::clips_get, &pyGaussAverage::clips_set)
.add_property("data", &pyGaussAverage::data_get)
.add_property("aabb", &pyGaussAverage::aabb_get)
.add_property("nCells", &pyGaussAverage::nCells_get)
.add_property("cellArea", &pyGaussAverage::cellArea)
.add_property("cellDim", &pyGaussAverage::cellDim);
} catch (...) {
LOG_FATAL("Importing this module caused an exception and this module is in an inconsistent state now.");
PyErr_Print();
PyErr_SetString(PyExc_SystemError, __FILE__);
boost::python::handle_exception();
throw;
}
|
