Changeset 802a9d for src/Python/export_numpy.cpp

Timestamp:

Dec 21, 2025, 9:50:27 PM (10 days ago)

Author:

Frederik Heber <frederik.heber@…>

Branches:

Candidate_v1.7.1, stable

Children:

2e7888

Parents:

e4b9d5

git-author:

Frederik Heber <frederik.heber@…> (11/18/25 19:34:57)

git-committer:

Frederik Heber <frederik.heber@…> (12/21/25 21:50:27)

Message:

Adds scripts to generate a bond table.

• one python script for the computation and a bash script to call it. With this split, we can easily parallelize the computations to make use of JobMarket's multiple workers.

File:

• src/Python/export_numpy.cpp (modified) (3 diffs)

src/Python/export_numpy.cpp

@@ -41 +41 @@
 
 #include "CodePatterns/Assert.hpp"
-
+#include "CodePatterns/Log.hpp"
+
+#include "Geometry/GeometryRegistry.hpp"
 #include "World.hpp"
 
@@ -78 +80 @@
 
   return positions;
+}
+
+np::ndarray get_geometryobject(const std::string &name)
+{
+  LOG(1, "Getting GeometryObject of name " << name);
+  const GeometryObject *object = GeometryRegistry::getInstance().getByName(name);
+  p::tuple shape = p::make_tuple(3);
+  np::dtype dtype = np::dtype::get_builtin<double>();
+  np::ndarray distance = np::zeros(shape, dtype);
+  if (object != NULL) {
+    for (unsigned int i=0;i<NDIM;++i)
+      distance[i] = object->getVector()[i];
+  }
+  return distance;
 }
 
@@ -173 +189 @@
 void export_numpy()
 {
+  p::def("get_geometryobject", get_geometryobject, p::args("name"));
   p::def("get_positions", get_positions);
   p::def("get_velocities", get_velocities);