Add missing export for laue ray predictor. Add missing include. Fix t…

…ypo.
toastisme · Mar 21, 2024 · 647a1a7 · 647a1a7
1 parent f460b66
commit 647a1a7
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 1 deletion.
diff --git a/src/dials/algorithms/spot_prediction/boost_python/ray_predictor.cc b/src/dials/algorithms/spot_prediction/boost_python/ray_predictor.cc
@@ -59,4 +59,11 @@ namespace dials { namespace algorithms { namespace boost_python {
  .def("__call__", &call_with_miller_index_array);
  }
 
+ void export_laue_ray_predictor() {
+ class_<LaueRayPredictor>("LaueRayPredictor", no_init)
+ .def(init<vec3<double>, mat3<double>, mat3<double> >(
+ (arg("unit_s0"), arg("fixed_rotation"), arg("setting_rotation"))))
+ .def("__call__", &LaueRayPredictor::operator(), (arg("miller_index"), arg("UB")));
+ }
+
 }}} // namespace dials::algorithms::boost_python
diff --git a/src/dials/algorithms/spot_prediction/boost_python/reflection_predictor.cc b/src/dials/algorithms/spot_prediction/boost_python/reflection_predictor.cc
@@ -173,8 +173,9 @@ namespace dials { namespace algorithms { namespace boost_python {
  const af::const_ref<std::size_t>&) = &Predictor::operator();
 
  class_<Predictor>("LaueReflectionPredictor", no_init)
- .def(init<const vec3<double>&,
+ .def(init<const PolychromaticBeam&,
  const Detector&,
+ const Goniometer&,
  mat3<double>,
  const cctbx::uctbx::unit_cell&,
  const cctbx::sgtbx::space_group_type&,

diff --git a/src/dials/algorithms/spot_prediction/reflection_predictor.h b/src/dials/algorithms/spot_prediction/reflection_predictor.h
@@ -28,6 +28,7 @@
 #include <dials/algorithms/spot_prediction/scan_varying_ray_predictor.h>
 #include <dials/algorithms/spot_prediction/stills_ray_predictor.h>
 #include <dials/algorithms/spot_prediction/ray_intersection.h>
+#include <cctbx/miller/index_generator.h>
 
 namespace dials { namespace algorithms {